database.go

package main

import (
	"crypto/rand"
	"database/sql"
	"encoding/base64"
	"encoding/json"

	_ "github.com/mattn/go-sqlite3"
)

type User struct {
	ID                      int64
	Username                string
	Email                   string
	Password                string
	RecoveryToken           sql.NullString
	RecoveryTokenExpiration sql.NullTime
}

type Project struct {
	ID     string
	UserID string
	Plan   ProjectPlan
}

func InitDB() *sql.DB {

	db, err := sql.Open("sqlite3", "db.db")
	_, err = db.Exec(`
		CREATE TABLE IF NOT EXISTS projects (
			id TEXT PRIMARY KEY UNIQUE,
			user_id TEXT,
			data TEXT NOT NULL
		);
	`)
	if err != nil {
		panic(err)
	}

	return db
}

func GenerateRandomString(length int) (string, error) {
	// Generate random bytes
	bytes := make([]byte, (length+3)/4*3) // Ensure enough bytes are generated for Base64 encoding
	_, err := rand.Read(bytes)
	if err != nil {
		return "", err
	}

	// Convert bytes to a Base64 string and trim to desired length
	randomString := base64.RawURLEncoding.EncodeToString(bytes)
	return randomString[:length], nil
}

func SaveProject(db *sql.DB, user_id string, unique_id string, project *ProjectPlan) error {

	data, err := json.Marshal(project)
	if err != nil {
		return err
	}

	_, err = db.Exec("INSERT INTO projects (id, user_id, data) VALUES (?, ?, ?)", unique_id, user_id, string(data))
	if err != nil {
		return err
	}

	return nil
}

func LoadProject(db *sql.DB, id string) (*ProjectPlan, error) {
	var data string
	err := db.QueryRow("SELECT data FROM projects WHERE id = ?", id).Scan(&data)
	if err != nil {
		return nil, err
	}

	objData := &ProjectPlan{}
	err = json.Unmarshal([]byte(data), &objData)
	if err != nil {
		return nil, err
	}

	return objData, nil
}