init

internal/db/db.go

@@ -0,0 +1,114 @@
+package db
+
+import (
+	"fmt"
+	"time"
+
+	"charm.land/ssh"
+	"github.com/google/uuid"
+	"github.com/jmoiron/sqlx"
+	"golang.org/x/crypto/bcrypt"
+	gossh "golang.org/x/crypto/ssh"
+	"jp.wtf/go-fishing/internal/models"
+)
+
+type Database struct {
+	Conn *sqlx.DB
+}
+
+func (db *Database) GetAnglerByID(angler_id uuid.UUID) (models.Angler, error) {
+	var angler models.Angler
+	err := db.Conn.Get(&angler, "SELECT angler_id, angler_username, angler_email FROM anglers WHERE angler_id = $1", angler_id)
+
+	return angler, err
+}
+
+func (db *Database) GetAnglerByKey(ssh_key ssh.PublicKey) (models.Angler, error) {
+	var angler models.Angler
+	err := db.Conn.Get(&angler, "SELECT angler_id, angler_username, angler_email FROM ssh_keys INNER JOIN anglers USING (angler_id) WHERE key_data = $1", ssh_key.Marshal())
+
+	return angler, err
+}
+
+func (db *Database) GetRawAnglerByUsername(username string) (models.AnglerDB, error) {
+	var angler models.AnglerDB
+	err := db.Conn.Get(&angler, "SELECT angler_id, angler_username, angler_email, angler_password FROM anglers WHERE angler_username = $1", username)
+
+	return angler, err
+}
+
+func (db *Database) CreateNewAngler(username string, hashed_password string) (models.Angler, error) {
+	var new_angler models.Angler
+	err := db.Conn.Get(&new_angler, `
+		INSERT INTO public.anglers (angler_username, angler_password)
+		VALUES ($1, $2)
+		RETURNING angler_id, angler_username, angler_email
+	`, username, hashed_password)
+
+	return new_angler, err
+}
+
+func (db *Database) CreateNewAnglerFromKey(ssh_key ssh.PublicKey) (models.Angler, error) {
+	bytes, err := bcrypt.GenerateFromPassword([]byte(uuid.New().String()), bcrypt.DefaultCost)
+
+	if err != nil {
+		return models.Angler{}, err
+	}
+
+	fingerprint := gossh.FingerprintSHA256(ssh_key)
+
+	angler, err := db.CreateNewAngler(fingerprint, string(bytes))
+	if err != nil {
+		return models.Angler{}, err
+	}
+
+	if _, err := db.AttachKeyToAngler(angler.ID, ssh_key); err != nil {
+		return models.Angler{}, err
+	}
+
+	return angler, err
+}
+
+func (db *Database) AttachKeyToAngler(angler_id uuid.UUID, ssh_key ssh.PublicKey) (models.SSHKey, error) {
+	var new_key models.SSHKey
+	err := db.Conn.Get(&new_key, `
+		INSERT INTO public.ssh_keys (key_type, key_data, key_fingerprint, angler_id)
+		VALUES ($1, $2, $3, $4)
+		RETURNING key_id, key_type, key_data, key_fingerprint
+	`, ssh_key.Type(), ssh_key.Marshal(), gossh.FingerprintSHA256(ssh_key), angler_id)
+
+	return new_key, err
+}
+
+func (db *Database) IsCasting(angler_id uuid.UUID) (bool, error) {
+	var exists bool
+	err := db.Conn.Get(&exists, "SELECT EXISTS( SELECT 1 FROM public.casts WHERE angler_id = $1 AND cast_ended_at IS NULL )", angler_id)
+
+	if err != nil {
+		fmt.Print(err.Error())
+		return false, err
+	}
+
+	return exists, err
+}
+
+func (db *Database) GetMostRecentCast(angler_id uuid.UUID) (models.Cast, error) {
+	var cast models.Cast
+	err := db.Conn.Get(&cast, "SELECT cast_id, cast_started_at, cast_ended_at, catch_id FROM public.casts WHERE angler_id =! ORDER cast_started_at DESC LIMIT 1", angler_id)
+
+	return cast, err
+}
+
+func (db *Database) CreateNewCast(angler_id uuid.UUID) (models.Cast, error) {
+	var cast models.Cast
+	err := db.Conn.Get(&cast, "INSERT INTO public.casts (angler_id) VALUES ($1) RETURNING *", angler_id)
+
+	return cast, err
+}
+
+func (db *Database) EndCastByID(cast_id uuid.UUID) (models.Cast, error) {
+	var cast models.Cast
+	err := db.Conn.Get(&cast, "UPDATE public.casts SET cast_ended_at = $1 WHERE cast_id = $2 RETURNING *", time.Now(), cast_id)
+
+	return cast, err
+}