From a3b13d21b3d61325f504ccd13658e4126c4d545d Mon Sep 17 00:00:00 2001
From: vilmibm <nathanielksmith@gmail.com>
Date: Tue, 24 Oct 2023 06:22:21 +0000
Subject: [PATCH] creating auth codes in db

---
 cmd/help/main.go       | 18 +++++++++++++---
 invites/invites.go     | 48 +++---------------------------------------
 sql/create_town_db.sql |  8 +++++++
 towndb/towndb.go       | 15 +++++++++++++
 4 files changed, 41 insertions(+), 48 deletions(-)

diff --git a/cmd/help/main.go b/cmd/help/main.go
index 05c4b1d..60087ef 100644
--- a/cmd/help/main.go
+++ b/cmd/help/main.go
@@ -7,6 +7,7 @@ import (
 	"strconv"
 	"strings"
 
+	"git.tilde.town/tildetown/town/codes"
 	"git.tilde.town/tildetown/town/towndb"
 	"github.com/charmbracelet/lipgloss"
 	_ "github.com/mattn/go-sqlite3"
@@ -162,11 +163,22 @@ func collectEmail(db *sql.DB, cs colorScheme, tty *tty.TTY) error {
 		return nil
 	}
 
-	fmt.Println("found a user so gonna email")
+	code := codes.NewCode(email)
+
+	fmt.Println(code)
+
+	ac := &towndb.AuthCode{
+		Code:  code,
+		Email: email,
+	}
+
+	err = ac.Insert(db)
+	if err != nil {
+		// TODO log
+		return err
+	}
 
-	// TODO generate reset code
 	// TODO send email
-	// TODO report success
 	return nil
 }
 
diff --git a/invites/invites.go b/invites/invites.go
index d342374..80a057a 100644
--- a/invites/invites.go
+++ b/invites/invites.go
@@ -1,20 +1,16 @@
 package invites
 
 import (
-	"crypto/rand"
 	"database/sql"
-	"encoding/base64"
 	"errors"
-	"math/big"
-	"strings"
 	"time"
 
+	"git.tilde.town/tildetown/town/codes"
 	_ "github.com/mattn/go-sqlite3"
 )
 
 const (
-	dsn     = "/town/var/invites/invites.db?mode=rw"
-	codeLen = 32
+	dsn = "/town/var/invites/invites.db?mode=rw"
 )
 
 type Invite struct {
@@ -33,7 +29,7 @@ func (i *Invite) Insert(db *sql.DB) error {
 		return err
 	}
 
-	i.Code = generateCode(i.Email)
+	i.Code = codes.NewCode(i.Email)
 
 	_, err = stmt.Exec(i.Code, i.Email)
 	if err != nil {
@@ -53,44 +49,6 @@ func ConnectDB() (*sql.DB, error) {
 	return db, nil
 }
 
-func generateCode(email string) string {
-
-	charset := "abcdefghijklmnopqrztuvwxyz"
-	charset += strings.ToUpper(charset)
-	charset += "0123456789"
-	charset += "`~!@#$%^&*()-=_+[]{}|;:,./<>?"
-
-	code := []byte{}
-
-	max := big.NewInt(int64(len(charset)))
-	for len(code) < codeLen {
-		ix, err := rand.Int(rand.Reader, max)
-		if err != nil {
-			// TODO this is bad but I'm just kind of hoping it doesn't happen...often
-			panic(err)
-		}
-		code = append(code, charset[ix.Int64()])
-	}
-
-	code = append(code, ' ')
-
-	eb := []byte(email)
-	for x := 0; x < len(eb); x++ {
-		code = append(code, eb[x])
-	}
-
-	return base64.StdEncoding.EncodeToString(code)
-}
-
-func Decode(code string) ([]string, error) {
-	decoded, err := base64.StdEncoding.DecodeString(code)
-	if err != nil {
-		return nil, err
-	}
-
-	return strings.Split(string(decoded), " "), nil
-}
-
 func Get(db *sql.DB, code string) (*Invite, error) {
 	inv := &Invite{
 		Code: code,
diff --git a/sql/create_town_db.sql b/sql/create_town_db.sql
index 169674a..3da22d2 100644
--- a/sql/create_town_db.sql
+++ b/sql/create_town_db.sql
@@ -32,3 +32,11 @@ CREATE TABLE IF NOT EXISTS notes (
 
   FOREIGN KEY (author) REFERENCES users(author)
 );
+
+CREATE TABLE IF NOT EXISTS auth_codes (
+  id      INTEGER PRIMARY KEY,
+  created TEXT DEFAULT (strftime('%Y-%m-%dT%H:%M', 'now', 'localtime')),
+  code    TEXT,
+  email   TEXT,
+  used    INTEGER DEFAULT 0
+);
diff --git a/towndb/towndb.go b/towndb/towndb.go
index b83169d..c13418a 100644
--- a/towndb/towndb.go
+++ b/towndb/towndb.go
@@ -173,3 +173,18 @@ func ConnectDB() (*sql.DB, error) {
 
 	return db, nil
 }
+
+type AuthCode struct {
+	ID      int64
+	Code    string
+	Email   string
+	Used    bool
+	Created time.Time
+}
+
+func (c *AuthCode) Insert(db *sql.DB) error {
+	// TODO
+	return nil
+}
+
+// TODO other auth code as needed