feat: add support for creating users and chirps

.gitignore

@@ -1 +1,2 @@
 bin/
+*.env*

go.mod

@@ -1,3 +1,9 @@
 module github.com/StevanFreeborn/chirpy
 
 go 1.26.3
+
+require (
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
+	github.com/lib/pq v1.12.3 // indirect
+)

go.sum

@@ -0,0 +1,6 @@
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/lib/pq v1.12.3 h1:tTWxr2YLKwIvK90ZXEw8GP7UFHtcbTtty8zsI+YjrfQ=
+github.com/lib/pq v1.12.3/go.mod h1:/p+8NSbOcwzAEI7wiMXFlgydTwcgTr3OSKMsD2BitpA=

internal/database/chirps.sql.go

@@ -0,0 +1,42 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: chirps.sql
+
+package database
+
+import (
+	"context"
+
+	"github.com/google/uuid"
+)
+
+const createChirp = `-- name: CreateChirp :one
+INSERT INTO chirps (id, created_at, updated_at, body, user_id)
+VALUES (
+  gen_random_uuid(),
+  NOW(),
+  NOW(),
+  $1,
+  $2
+)
+RETURNING id, created_at, updated_at, body, user_id
+`
+
+type CreateChirpParams struct {
+	Body   string
+	UserID uuid.UUID
+}
+
+func (q *Queries) CreateChirp(ctx context.Context, arg CreateChirpParams) (Chirp, error) {
+	row := q.db.QueryRowContext(ctx, createChirp, arg.Body, arg.UserID)
+	var i Chirp
+	err := row.Scan(
+		&i.ID,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+		&i.Body,
+		&i.UserID,
+	)
+	return i, err
+}

internal/database/db.go

@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package database
+
+import (
+	"context"
+	"database/sql"
+)
+
+type DBTX interface {
+	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
+	PrepareContext(context.Context, string) (*sql.Stmt, error)
+	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
+	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx *sql.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

internal/database/models.go

@@ -0,0 +1,26 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package database
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type Chirp struct {
+	ID        uuid.UUID
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	Body      string
+	UserID    uuid.UUID
+}
+
+type User struct {
+	ID        uuid.UUID
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	Email     string
+}

internal/database/users.sql.go

@@ -0,0 +1,61 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: users.sql
+
+package database
+
+import (
+	"context"
+
+	"github.com/google/uuid"
+)
+
+const createUser = `-- name: CreateUser :one
+INSERT INTO users (id, created_at, updated_at, email)
+VALUES (
+  gen_random_uuid(),
+  NOW(),
+  NOW(),
+  $1
+)
+RETURNING id, created_at, updated_at, email
+`
+
+func (q *Queries) CreateUser(ctx context.Context, email string) (User, error) {
+	row := q.db.QueryRowContext(ctx, createUser, email)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+		&i.Email,
+	)
+	return i, err
+}
+
+const deleteAllUsers = `-- name: DeleteAllUsers :exec
+DELETE FROM users
+`
+
+func (q *Queries) DeleteAllUsers(ctx context.Context) error {
+	_, err := q.db.ExecContext(ctx, deleteAllUsers)
+	return err
+}
+
+const getUserById = `-- name: GetUserById :one
+SELECT id, created_at, updated_at, email FROM users
+WHERE id = $1
+`
+
+func (q *Queries) GetUserById(ctx context.Context, id uuid.UUID) (User, error) {
+	row := q.db.QueryRowContext(ctx, getUserById, id)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+		&i.Email,
+	)
+	return i, err
+}

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -14,22 +15,17 @@ import (
 	"sync/atomic"
 	"syscall"
 	"time"
+
+	"github.com/StevanFreeborn/chirpy/internal/database"
+	"github.com/google/uuid"
+	"github.com/joho/godotenv"
+	_ "github.com/lib/pq"
 )
 
 type server struct {
 	fileServerHits atomic.Int32
-}
+	database       *database.Queries
-
+	platform       string
-type chirpRequest struct {
-	Body string `json:"body"`
-}
-
-type validationError struct {
-	Err string `json:"error"`
-}
-
-type chirpResponse struct {
-	CleanedBody string `json:"cleaned_body"`
 }
 
 func writeJsonResponse(w http.ResponseWriter, response any) {
@@ -56,8 +52,15 @@ func (s *server) HandleHealthChecks(w http.ResponseWriter, r *http.Request) {
 }
 
 func (s *server) HandleReset(w http.ResponseWriter, r *http.Request) {
+	if s.platform != "dev" {
+		w.WriteHeader(http.StatusForbidden)
+		return
+	}
+
 	s.fileServerHits.Store(0)
 
+	s.database.DeleteAllUsers(r.Context())
+
 	w.Header().Add("Content-Type", "text/plain; charset=utf-8")
 	w.WriteHeader(http.StatusOK)
 	w.Write([]byte("OK"))
@@ -81,38 +84,75 @@ func (s *server) HandleMetrics(w http.ResponseWriter, r *http.Request) {
 	fmt.Fprintf(w, template, hits)
 }
 
-func (s *server) HandleChirpValidation(w http.ResponseWriter, r *http.Request) {
+type apiError struct {
+	Err string `json:"error"`
+}
+
+type createChirpRequest struct {
+	Body   string `json:"body"`
+	UserId string `json:"user_id"`
+}
+
+type createChirpResponse struct {
+	Id        string `json:"id"`
+	CreatedAt string `json:"created_at"`
+	UpdatedAt string `json:"updated_at"`
+	Body      string `json:"body"`
+	UserId    string `json:"user_id"`
+}
+
+func (s *server) HandleCreateChirp(w http.ResponseWriter, r *http.Request) {
 	blacklist := []string{
 		"kerfuffle",
 		"sharbert",
 		"fornax",
 	}
 
-	var chirpRequest chirpRequest
+	var createChirpRequest createChirpRequest
 
 	defer r.Body.Close()
 
 	w.Header().Add("Content-Type", "application/json")
 
-	err := json.NewDecoder(r.Body).Decode(&chirpRequest)
+	err := json.NewDecoder(r.Body).Decode(&createChirpRequest)
 
 	if err != nil {
 		w.WriteHeader(http.StatusBadRequest)
-		writeJsonResponse(w, validationError{
+		writeJsonResponse(w, apiError{
 			Err: "Unable to deserialize JSON",
 		})
 		return
 	}
 
-	if len(chirpRequest.Body) > 140 {
+	if len(createChirpRequest.Body) > 140 {
 		w.WriteHeader(http.StatusBadRequest)
-		writeJsonResponse(w, validationError{
+		writeJsonResponse(w, apiError{
 			Err: "Chirp is too long",
 		})
 		return
 	}
 
-	words := strings.Split(chirpRequest.Body, " ")
+	requestUserId, err := uuid.Parse(createChirpRequest.UserId)
+
+	invalidUserIdError := apiError{
+		Err: "Invalid user_id. user_id must be valid UUID",
+	}
+
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		writeJsonResponse(w, invalidUserIdError)
+		return
+	}
+
+	existingUser, err := s.database.GetUserById(r.Context(), requestUserId)
+
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		writeJsonResponse(w, invalidUserIdError)
+		return
+	}
+
+	words := strings.Split(createChirpRequest.Body, " ")
 	sanitized := []string{}
 
 	for _, word := range words {
@@ -126,12 +166,105 @@ func (s *server) HandleChirpValidation(w http.ResponseWriter, r *http.Request) {
 
 	cleanedBody := strings.Join(sanitized, " ")
 
-	w.WriteHeader(http.StatusOK)
+	createChirpParams := database.CreateChirpParams{
-	writeJsonResponse(w, chirpResponse{CleanedBody: cleanedBody})
+		Body:   cleanedBody,
+		UserID: existingUser.ID,
+	}
+
+	createdChirp, err := s.database.CreateChirp(r.Context(), createChirpParams)
+
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		writeJsonResponse(w, apiError{
+			Err: "Failed to create chirp. 🤷🏻‍♂️",
+		})
+	}
+
+	w.WriteHeader(http.StatusCreated)
+	writeJsonResponse(w, createChirpResponse{
+		Id:        createdChirp.ID.String(),
+		CreatedAt: createdChirp.CreatedAt.Format(time.RFC3339),
+		UpdatedAt: createdChirp.UpdatedAt.Format(time.RFC3339),
+		Body:      createdChirp.Body,
+		UserId:    createdChirp.UserID.String(),
+	})
+}
+
+type createUserRequest struct {
+	Email string `json:"email"`
+}
+
+type createUserResponse struct {
+	Id        string `json:"id"`
+	CreatedAt string `json:"created_at"`
+	UpdatedAt string `json:"updated_at"`
+	Email     string `json:"email"`
+}
+
+func (s *server) HandleCreateUser(w http.ResponseWriter, r *http.Request) {
+	var createUserRequest createUserRequest
+
+	defer r.Body.Close()
+
+	err := json.NewDecoder(r.Body).Decode(&createUserRequest)
+
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		writeJsonResponse(w, apiError{
+			Err: "Unable to deserialize JSON",
+		})
+	}
+
+	trimmedEmail := strings.TrimSpace(createUserRequest.Email)
+
+	if strings.TrimSpace(trimmedEmail) == "" {
+		w.WriteHeader(http.StatusBadRequest)
+		writeJsonResponse(w, apiError{
+			Err: "email is required. must be valid email address.",
+		})
+	}
+
+	createdUser, err := s.database.CreateUser(r.Context(), trimmedEmail)
+
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		writeJsonResponse(w, apiError{
+			Err: "Uh oh we were unable to create a new user",
+		})
+	}
+
+	w.WriteHeader(http.StatusCreated)
+	writeJsonResponse(w, createUserResponse{
+		Id:        createdUser.ID.String(),
+		CreatedAt: createdUser.CreatedAt.Format(time.RFC3339),
+		UpdatedAt: createdUser.UpdatedAt.Format(time.RFC3339),
+		Email:     createdUser.Email,
+	})
 }
 
 func main() {
-	server := &server{}
+	err := godotenv.Load()
+
+	if err != nil {
+		log.Fatalf("Failed to load environment variables")
+		os.Exit(1)
+	}
+
+	dbURL := os.Getenv("DB_URL")
+	platform := os.Getenv("PLATFORM")
+
+	db, err := sql.Open("postgres", dbURL)
+
+	if err != nil {
+		log.Fatalf("Failed to connect to database: %v", err)
+		os.Exit(1)
+	}
+
+	server := &server{
+		database: database.New(db),
+		platform: platform,
+	}
+
 	mux := http.NewServeMux()
 
 	appRoot := "/app/"
@@ -140,9 +273,12 @@ func main() {
 	mux.HandleFunc("GET /admin/metrics", server.HandleMetrics)
 	mux.HandleFunc("POST /admin/reset", server.HandleReset)
 
-	mux.HandleFunc("POST /api/validate_chirp", server.HandleChirpValidation)
 	mux.HandleFunc("GET /api/healthz", server.HandleHealthChecks)
 
+	mux.HandleFunc("POST /api/users", server.HandleCreateUser)
+
+	mux.HandleFunc("POST /api/chirps", server.HandleCreateChirp)
+
 	port := ":8080"
 
 	httpServer := http.Server{

sql/queries/chirps.sql

@@ -0,0 +1,10 @@
+-- name: CreateChirp :one
+INSERT INTO chirps (id, created_at, updated_at, body, user_id)
+VALUES (
+  gen_random_uuid(),
+  NOW(),
+  NOW(),
+  $1,
+  $2
+)
+RETURNING *;

sql/queries/users.sql

@@ -0,0 +1,17 @@
+-- name: CreateUser :one
+INSERT INTO users (id, created_at, updated_at, email)
+VALUES (
+  gen_random_uuid(),
+  NOW(),
+  NOW(),
+  $1
+)
+RETURNING *;
+
+
+-- name: DeleteAllUsers :exec
+DELETE FROM users;
+
+-- name: GetUserById :one
+SELECT id, created_at, updated_at, email FROM users
+WHERE id = $1;

sql/schema/001_users.sql

@@ -0,0 +1,10 @@
+-- +goose Up
+CREATE TABLE users (
+  id UUID PRIMARY KEY,
+  created_at TIMESTAMP NOT NULL,
+  updated_at TIMESTAMP NOT NULL,
+  email TEXT UNIQUE NOT NULL
+);
+
+-- +goose Down
+DROP TABLE users;

sql/schema/002_chirps.sql

@@ -0,0 +1,12 @@
+-- +goose Up
+CREATE TABLE chirps (
+  id UUID PRIMARY KEY,
+  created_at TIMESTAMP NOT NULL,
+  updated_at TIMESTAMP NOT NULL,
+  body TEXT NOT NULL,
+  user_id UUID NOT NULL,
+  FOREIGN KEY (user_id) REFERENCES users (id) ON DELETE CASCADE
+);
+
+-- +goose Down
+DROP TABLE chirps;

sqlc.yml

@@ -0,0 +1,8 @@
+version: "2"
+sql:
+  - schema: "sql/schema"
+    queries: "sql/queries"
+    engine: "postgresql"
+    gen:
+      go:
+        out: "internal/database"