#3

internal/config/state.go

@@ -1,5 +1,8 @@
 package config
 
+import "github.com/rdarius/boot-dev-blog-aggregator/internal/database"
+
 type State struct {
+	DB     *database.Queries
 	Config *Config
 }

internal/database/db.go

@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.28.0
+
+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/getUser.sql.go

@@ -0,0 +1,26 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.28.0
+// source: getUser.sql
+
+package database
+
+import (
+	"context"
+)
+
+const getUser = `-- name: GetUser :one
+SELECT id, created_at, updated_at, name FROM users WHERE name = $1
+`
+
+func (q *Queries) GetUser(ctx context.Context, name string) (User, error) {
+	row := q.db.QueryRowContext(ctx, getUser, name)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+		&i.Name,
+	)
+	return i, err
+}

internal/database/models.go

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

internal/database/users.sql.go

@@ -0,0 +1,48 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.28.0
+// source: users.sql
+
+package database
+
+import (
+	"context"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+const createUser = `-- name: CreateUser :one
+INSERT INTO users (id, created_at, updated_at, name)
+VALUES (
+           $1,
+           $2,
+           $3,
+           $4
+       )
+RETURNING id, created_at, updated_at, name
+`
+
+type CreateUserParams struct {
+	ID        uuid.UUID
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	Name      string
+}
+
+func (q *Queries) CreateUser(ctx context.Context, arg CreateUserParams) (User, error) {
+	row := q.db.QueryRowContext(ctx, createUser,
+		arg.ID,
+		arg.CreatedAt,
+		arg.UpdatedAt,
+		arg.Name,
+	)
+	var i User
+	err := row.Scan(
+		&i.ID,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+		&i.Name,
+	)
+	return i, err
+}

internal/handlers/loginHandler.go

@@ -1,17 +1,27 @@
 package handlers
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"github.com/rdarius/boot-dev-blog-aggregator/internal/config"
+	"os"
 )
 
-func HandlerLogin(s *config.State, cmd config.Command) error {
+func LoginHandler(s *config.State, cmd config.Command) error {
 	if len(cmd.Args) < 1 {
 		return errors.New("missing required argument USERNAME")
 	}
 
-	err := s.Config.SetUser(cmd.Args[0])
+	ctx := context.Background()
+
+	u, err := s.DB.GetUser(ctx, cmd.Args[0])
+	if err != nil {
+		fmt.Println("User does not exist")
+		os.Exit(1)
+	}
+
+	err = s.Config.SetUser(u.Name)
 	if err != nil {
 		return fmt.Errorf("failed to set current user: %w", err)
 	}

internal/handlers/registerHandler.go

@@ -0,0 +1,46 @@
+package handlers
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/google/uuid"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/config"
+	"github.com/rdarius/boot-dev-blog-aggregator/internal/database"
+	"os"
+	"time"
+)
+
+func RegisterHandler(s *config.State, cmd config.Command) error {
+	if len(cmd.Args) < 1 {
+		return errors.New("missing required argument USERNAME")
+	}
+
+	ctx := context.Background()
+
+	_, err := s.DB.GetUser(ctx, cmd.Args[0])
+	if err == nil {
+		fmt.Println("User already exists")
+		os.Exit(1)
+	}
+
+	u, err := s.DB.CreateUser(ctx, database.CreateUserParams{
+		ID:        uuid.New(),
+		CreatedAt: time.Now(),
+		UpdatedAt: time.Now(),
+		Name:      cmd.Args[0],
+	})
+	if err != nil {
+		fmt.Printf("Failed to create user %v\n", err)
+		os.Exit(1)
+	}
+
+	err = s.Config.SetUser(u.Name)
+	if err != nil {
+		return fmt.Errorf("failed to set current user: %w", err)
+	}
+
+	fmt.Printf("User Created %v\n", u)
+
+	return nil
+}