coder · kylecarbs · Jan 20, 2022 · Jan 17, 2022 · Jan 17, 2022 · Jan 19, 2022
diff --git a/coderd/cmd/root.go b/coderd/cmd/root.go
@@ -5,12 +5,13 @@ import (
 	"net/http"
 	"os"
 
+	"github.com/spf13/cobra"
+	"golang.org/x/xerrors"
+
 	"cdr.dev/slog"
 	"cdr.dev/slog/sloggers/sloghuman"
 	"github.com/coder/coder/coderd"
-	"github.com/coder/coder/database"
-	"github.com/spf13/cobra"
-	"golang.org/x/xerrors"
+	"github.com/coder/coder/database/databasefake"
 )
 
 func Root() *cobra.Command {
@@ -22,7 +23,7 @@ func Root() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			handler := coderd.New(&coderd.Options{
 				Logger:   slog.Make(sloghuman.Sink(os.Stderr)),
-				Database: database.NewInMemory(),
+				Database: databasefake.New(),
 			})
 
 			listener, err := net.Listen("tcp", address)

diff --git a/coderd/cmd/root_test.go b/coderd/cmd/root_test.go
@@ -4,8 +4,9 @@ import (
 	"context"
 	"testing"
 
-	"github.com/coder/coder/coderd/cmd"
 	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/coderd/cmd"
 )
 
 func TestRoot(t *testing.T) {

diff --git a/coderd/coderd.go b/coderd/coderd.go
@@ -3,10 +3,12 @@ package coderd
 import (
 	"net/http"
 
+	"github.com/go-chi/chi"
+
 	"cdr.dev/slog"
 	"github.com/coder/coder/database"
-	"github.com/go-chi/chi"
-	"github.com/go-chi/render"
+	"github.com/coder/coder/httpapi"
+	"github.com/coder/coder/httpmw"
 )
 
 // Options are requires parameters for Coder to start.
@@ -17,15 +19,25 @@ type Options struct {
 
 // New constructs the Coder API into an HTTP handler.
 func New(options *Options) http.Handler {
+	users := &users{
+		Database: options.Database,
+	}
+
 	r := chi.NewRouter()
 	r.Route("/api/v2", func(r chi.Router) {
 		r.Get("/", func(w http.ResponseWriter, r *http.Request) {
-			render.JSON(w, r, struct {
-				Message string `json:"message"`
-			}{
+			httpapi.Write(w, http.StatusOK, httpapi.Response{
 				Message: "👋",
 			})
 		})
+		r.Group(func(r chi.Router) {
+			// Require an API key and authenticated user to call this route.
+			r.Use(
+				httpmw.ExtractAPIKey(options.Database, nil),
+				httpmw.ExtractUser(options.Database),
+			)
+			r.Get("/user", users.getAuthenticatedUser)
+		})
 	})
 	return r
 }
diff --git a/coderd/coderdtest/coderdtest.go b/coderd/coderdtest/coderdtest.go
@@ -0,0 +1,31 @@
+package coderdtest
+
+import (
+	"net/http/httptest"
+	"testing"
+
+	"cdr.dev/slog/sloggers/slogtest"
+	"github.com/coder/coder/coderd"
+	"github.com/coder/coder/database"
+	"github.com/coder/coder/database/databasefake"
+)
+
+type Server struct {
+	Database database.Store
+	URL      string
+}
+
+func New(t *testing.T) Server {
+	// This can be hotswapped for a live database instance.
+	db := databasefake.New()
+	handler := coderd.New(&coderd.Options{
+		Logger:   slogtest.Make(t, nil),
+		Database: db,
+	})
+	srv := httptest.NewServer(handler)
+	t.Cleanup(srv.Close)
+	return Server{
+		Database: db,
+		URL:      srv.URL,
+	}
+}
diff --git a/coderd/coderdtest/coderdtest_test.go b/coderd/coderdtest/coderdtest_test.go
@@ -0,0 +1,16 @@
+package coderdtest_test
+
+import (
+	"testing"
+
+	"github.com/coder/coder/coderd/coderdtest"
+	"go.uber.org/goleak"
+)
+
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m)
+}
+
+func TestNew(t *testing.T) {
+	_ = coderdtest.New(t)
+}
diff --git a/coderd/users.go b/coderd/users.go
@@ -0,0 +1,33 @@
+package coderd
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/go-chi/render"
+
+	"github.com/coder/coder/database"
+	"github.com/coder/coder/httpmw"
+)
+
+type User struct {
+	ID        string    `json:"id"`
+	Email     string    `json:"email"`
+	CreatedAt time.Time `json:"created_at"`
+	Username  string    `json:"username"`
+}
+
+type users struct {
+	Database database.Store
+}
+
+func (users *users) getAuthenticatedUser(rw http.ResponseWriter, r *http.Request) {
+	user := httpmw.User(r)
+
+	render.JSON(rw, r, User{
+		ID:        user.ID,
+		Email:     user.Email,
+		CreatedAt: user.CreatedAt,
+		Username:  user.Username,
+	})
+}
diff --git a/coderd/users_test.go b/coderd/users_test.go
@@ -0,0 +1,9 @@
+package coderd_test
+
+import "testing"
+
+func TestUsers(t *testing.T) {
+	t.Run("wow", func(t *testing.T) {
+
+	})
+}
diff --git a/database/databasefake/databasefake.go b/database/databasefake/databasefake.go
@@ -0,0 +1,97 @@
+package databasefake
+
+import (
+	"context"
+	"database/sql"
+
+	"github.com/coder/coder/database"
+)
+
+// New returns an in-memory fake of the database.
+func New() database.Store {
+	return &fakeQuerier{
+		apiKeys: make([]database.APIKey, 0),
+		users:   make([]database.User, 0),
+	}
+}
+
+type fakeQuerier struct {
+	apiKeys []database.APIKey
+	users   []database.User
+}
+
+// InTx doesn't rollback data properly for in-memory yet.
+func (q *fakeQuerier) InTx(ctx context.Context, fn func(database.Store) error) error {
+	return fn(q)
+}
+
+func (q *fakeQuerier) GetAPIKeyByID(ctx context.Context, id string) (database.APIKey, error) {
+	for _, apiKey := range q.apiKeys {
+		if apiKey.ID == id {
+			return apiKey, nil
+		}
+	}
+	return database.APIKey{}, sql.ErrNoRows
+}
+
+func (q *fakeQuerier) GetUserByID(ctx context.Context, id string) (database.User, error) {
+	for _, user := range q.users {
+		if user.ID == id {
+			return user, nil
+		}
+	}
+	return database.User{}, sql.ErrNoRows
+}
+
+func (q *fakeQuerier) InsertAPIKey(ctx context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {
+	key := database.APIKey{
+		ID:               arg.ID,
+		HashedSecret:     arg.HashedSecret,
+		UserID:           arg.UserID,
+		Application:      arg.Application,
+		Name:             arg.Name,
+		LastUsed:         arg.LastUsed,
+		ExpiresAt:        arg.ExpiresAt,
+		CreatedAt:        arg.CreatedAt,
+		UpdatedAt:        arg.UpdatedAt,
+		LoginType:        arg.LoginType,
+		OIDCAccessToken:  arg.OIDCAccessToken,
+		OIDCRefreshToken: arg.OIDCRefreshToken,
+		OIDCIDToken:      arg.OIDCIDToken,
+		OIDCExpiry:       arg.OIDCExpiry,
+		DevurlToken:      arg.DevurlToken,
+	}
+	q.apiKeys = append(q.apiKeys, key)
+	return key, nil
+}
+
+func (q *fakeQuerier) InsertUser(ctx context.Context, arg database.InsertUserParams) (database.User, error) {
+	user := database.User{
+		ID:             arg.ID,
+		Email:          arg.Email,
+		Name:           arg.Name,
+		LoginType:      arg.LoginType,
+		HashedPassword: arg.HashedPassword,
+		CreatedAt:      arg.CreatedAt,
+		UpdatedAt:      arg.UpdatedAt,
+		Username:       arg.Username,
+	}
+	q.users = append(q.users, user)
+	return user, nil
+}
+
+func (q *fakeQuerier) UpdateAPIKeyByID(ctx context.Context, arg database.UpdateAPIKeyByIDParams) error {
+	for index, apiKey := range q.apiKeys {
+		if apiKey.ID != arg.ID {
+			continue
+		}
+		apiKey.LastUsed = arg.LastUsed
+		apiKey.ExpiresAt = arg.ExpiresAt
+		apiKey.OIDCAccessToken = arg.OIDCAccessToken
+		apiKey.OIDCRefreshToken = arg.OIDCRefreshToken
+		apiKey.OIDCExpiry = arg.OIDCExpiry
+		q.apiKeys[index] = apiKey
+		return nil
+	}
+	return sql.ErrNoRows
+}
diff --git a/database/db_memory.go b/database/db_memory.go
diff --git a/database/models.go b/database/models.go
diff --git a/database/querier.go b/database/querier.go
diff --git a/database/query.sql b/database/query.sql
@@ -1,2 +1,20 @@
--- name: ExampleQuery :exec
-SELECT 'example query';
+-- name: GetAPIKeyByID :one
+SELECT
+    *
+FROM
+    api_keys
+WHERE
+    id = $1
+LIMIT 1;
+
+-- name: GetUserByID :one
+SELECT * FROM users WHERE id = $1 LIMIT 1;
+
+-- name: InsertAPIKey :one
+INSERT INTO api_keys (id, hashed_secret, user_id, application, name, last_used, expires_at, created_at, updated_at, login_type, oidc_access_token, oidc_refresh_token, oidc_id_token, oidc_expiry, devurl_token) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15) RETURNING *;
+
+-- name: InsertUser :one
+INSERT INTO users (id, email, name, login_type, hashed_password, created_at, updated_at, username) VALUES ($1, $2, $3, $4, $5, $6, $7, $8) RETURNING *;
+
+-- name: UpdateAPIKeyByID :exec
+UPDATE api_keys SET last_used = $2, expires_at = $3, oidc_access_token = $4, oidc_refresh_token = $5, oidc_expiry = $6 WHERE id = $1;