Add db support for workspaces

This commit is contained in:
2024-10-14 21:08:37 +02:00
parent b36c5b30c6
commit 2d2b596f2c
8 changed files with 212 additions and 38 deletions

View File

@@ -18,6 +18,53 @@ var migrations = []Migration{
settings JSON NOT NULL settings JSON NOT NULL
)`, )`,
}, },
{
Version: 2,
SQL: `
-- Create users table
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT NOT NULL UNIQUE,
email TEXT NOT NULL UNIQUE,
password_hash TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
-- Create workspaces table
CREATE TABLE IF NOT EXISTS workspaces (
id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id INTEGER NOT NULL,
name TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (user_id) REFERENCES users (id)
);
-- Create workspace_settings table
CREATE TABLE IF NOT EXISTS workspace_settings (
workspace_id INTEGER PRIMARY KEY,
settings JSON NOT NULL,
FOREIGN KEY (workspace_id) REFERENCES workspaces (id)
);
-- Migrate existing settings to a default user and workspace
INSERT INTO users (username, email, password_hash)
VALUES ('default_user', 'default@example.com', 'placeholder_hash');
INSERT INTO workspaces (user_id, name, root_path)
SELECT 1, 'Default Workspace'
WHERE NOT EXISTS (SELECT 1 FROM workspaces);
INSERT INTO workspace_settings (workspace_id, settings)
SELECT w.id, s.settings
FROM workspaces w
CROSS JOIN settings s
WHERE w.name = 'Default Workspace'
AND NOT EXISTS (SELECT 1 FROM workspace_settings);
-- Drop the old settings table
DROP TABLE IF EXISTS settings;
`,
},
} }
func (db *DB) Migrate() error { func (db *DB) Migrate() error {

View File

@@ -7,39 +7,37 @@ import (
"novamd/internal/models" "novamd/internal/models"
) )
func (db *DB) GetSettings(userID int) (models.Settings, error) { func (db *DB) GetWorkspaceSettings(workspaceID int) (*models.WorkspaceSettings, error) {
var settings models.Settings var settings models.WorkspaceSettings
var settingsJSON []byte var settingsJSON []byte
err := db.QueryRow("SELECT user_id, settings FROM settings WHERE user_id = ?", userID).Scan(&settings.UserID, &settingsJSON) err := db.QueryRow("SELECT workspace_id, settings FROM workspace_settings WHERE workspace_id = ?", workspaceID).
Scan(&settings.WorkspaceID, &settingsJSON)
if err != nil { if err != nil {
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
// If no settings found, return default settings // If no settings found, return default settings
settings.UserID = userID settings.WorkspaceID = workspaceID
settings.Settings = models.UserSettings{} // This will be filled with defaults later settings.Settings = models.UserSettings{} // This will be filled with defaults later
return settings, nil return &settings, nil
} }
return settings, err return nil, err
} }
err = json.Unmarshal(settingsJSON, &settings.Settings) err = json.Unmarshal(settingsJSON, &settings.Settings)
if err != nil { if err != nil {
return settings, err return nil, err
} }
return settings, nil return &settings, nil
} }
func (db *DB) SaveSettings(settings models.Settings) error { func (db *DB) SaveWorkspaceSettings(settings *models.WorkspaceSettings) error {
if err := settings.Validate(); err != nil {
return err
}
settingsJSON, err := json.Marshal(settings.Settings) settingsJSON, err := json.Marshal(settings.Settings)
if err != nil { if err != nil {
return err return err
} }
_, err = db.Exec("INSERT OR REPLACE INTO settings (user_id, settings) VALUES (?, json(?))", settings.UserID, string(settingsJSON)) _, err = db.Exec("INSERT OR REPLACE INTO workspace_settings (workspace_id, settings) VALUES (?, ?)",
settings.WorkspaceID, settingsJSON)
return err return err
} }

View File

@@ -0,0 +1,31 @@
package db
import (
"novamd/internal/models"
)
func (db *DB) CreateUser(user *models.User) error {
_, err := db.Exec("INSERT INTO users (username, email, password_hash) VALUES (?, ?, ?)",
user.Username, user.Email, user.PasswordHash)
return err
}
func (db *DB) GetUserByID(id int) (*models.User, error) {
user := &models.User{}
err := db.QueryRow("SELECT id, username, email, created_at FROM users WHERE id = ?", id).
Scan(&user.ID, &user.Username, &user.Email, &user.CreatedAt)
if err != nil {
return nil, err
}
return user, nil
}
func (db *DB) GetUserByUsername(username string) (*models.User, error) {
user := &models.User{}
err := db.QueryRow("SELECT id, username, email, password_hash, created_at FROM users WHERE username = ?", username).
Scan(&user.ID, &user.Username, &user.Email, &user.PasswordHash, &user.CreatedAt)
if err != nil {
return nil, err
}
return user, nil
}

View File

@@ -0,0 +1,48 @@
package db
import (
"novamd/internal/models"
)
func (db *DB) CreateWorkspace(workspace *models.Workspace) error {
result, err := db.Exec("INSERT INTO workspaces (user_id, name) VALUES (?, ?, ?)",
workspace.UserID, workspace.Name)
if err != nil {
return err
}
id, err := result.LastInsertId()
if err != nil {
return err
}
workspace.ID = int(id)
return nil
}
func (db *DB) GetWorkspaceByID(id int) (*models.Workspace, error) {
workspace := &models.Workspace{}
err := db.QueryRow("SELECT id, user_id, name, root_path, created_at FROM workspaces WHERE id = ?", id).
Scan(&workspace.ID, &workspace.UserID, &workspace.Name, &workspace.CreatedAt)
if err != nil {
return nil, err
}
return workspace, nil
}
func (db *DB) GetWorkspacesByUserID(userID int) ([]*models.Workspace, error) {
rows, err := db.Query("SELECT id, user_id, name, root_path, created_at FROM workspaces WHERE user_id = ?", userID)
if err != nil {
return nil, err
}
defer rows.Close()
var workspaces []*models.Workspace
for rows.Next() {
workspace := &models.Workspace{}
err := rows.Scan(&workspace.ID, &workspace.UserID, &workspace.Name, &workspace.CreatedAt)
if err != nil {
return nil, err
}
workspaces = append(workspaces, workspace)
}
return workspaces, nil
}

View File

@@ -0,0 +1,29 @@
package filesystem
import (
"fmt"
"os"
"path/filepath"
"novamd/internal/models"
)
// GetWorkspacePath returns the file system path for a given workspace
func GetWorkspacePath(workspace *models.Workspace) string {
baseDir := os.Getenv("NOVAMD_WORKDIR")
if baseDir == "" {
baseDir = "./data" // Default if not set
}
return filepath.Join(baseDir, fmt.Sprintf("%d", workspace.UserID), workspace.Name)
}
// GetFilePath returns the file system path for a given file within a workspace
func GetFilePath(workspace *models.Workspace, relativeFilePath string) string {
return filepath.Join(GetWorkspacePath(workspace), relativeFilePath)
}
// EnsureWorkspaceDirectory creates the workspace directory if it doesn't exist
func EnsureWorkspaceDirectory(workspace *models.Workspace) error {
dir := GetWorkspacePath(workspace)
return os.MkdirAll(dir, 0755)
}

View File

@@ -17,42 +17,30 @@ type UserSettings struct {
GitCommitMsgTemplate string `json:"gitCommitMsgTemplate"` GitCommitMsgTemplate string `json:"gitCommitMsgTemplate"`
} }
type Settings struct { type WorkspaceSettings struct {
UserID int `json:"userId" validate:"required,min=1"` WorkspaceID int `json:"workspaceId" validate:"required,min=1"`
Settings UserSettings `json:"settings" validate:"required"` Settings UserSettings `json:"settings" validate:"required"`
}
var defaultUserSettings = UserSettings{
Theme: "light",
AutoSave: false,
GitEnabled: false,
GitCommitMsgTemplate: "Update ${filename}",
} }
var validate = validator.New() var validate = validator.New()
func (s *Settings) Validate() error { func (s *UserSettings) Validate() error {
return validate.Struct(s) return validate.Struct(s)
} }
func (s *Settings) SetDefaults() { func (ws *WorkspaceSettings) Validate() error {
if s.Settings.Theme == "" { return validate.Struct(ws)
s.Settings.Theme = defaultUserSettings.Theme
}
if s.Settings.GitCommitMsgTemplate == "" {
s.Settings.GitCommitMsgTemplate = defaultUserSettings.GitCommitMsgTemplate
}
} }
func (s *Settings) UnmarshalJSON(data []byte) error { func (ws *WorkspaceSettings) UnmarshalJSON(data []byte) error {
type Alias Settings type Alias WorkspaceSettings
aux := &struct { aux := &struct {
*Alias *Alias
}{ }{
Alias: (*Alias)(s), Alias: (*Alias)(ws),
} }
if err := json.Unmarshal(data, &aux); err != nil { if err := json.Unmarshal(data, &aux); err != nil {
return err return err
} }
return s.Validate() return ws.Validate()
} }

View File

@@ -0,0 +1,17 @@
package models
import (
"time"
)
type User struct {
ID int `json:"id" validate:"required,min=1"`
Username string `json:"username" validate:"required"`
Email string `json:"email" validate:"required,email"`
PasswordHash string `json:"-"`
CreatedAt time.Time `json:"createdAt"`
}
func (u *User) Validate() error {
return validate.Struct(u)
}

View File

@@ -0,0 +1,16 @@
package models
import (
"time"
)
type Workspace struct {
ID int `json:"id" validate:"required,min=1"`
UserID int `json:"userId" validate:"required,min=1"`
Name string `json:"name" validate:"required"`
CreatedAt time.Time `json:"createdAt"`
}
func (w *Workspace) Validate() error {
return validate.Struct(w)
}