Changeset 0.19.0.1 (#82)

2026-02-28 23:46:23 +00:00
parent 091ce2af6a
commit 748f49bedd
30 changed files with 3873 additions and 151 deletions
--- a/server/store/postgres/project.go
+++ b/server/store/postgres/project.go
@@ -0,0 +1,403 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+
+	"github.com/lib/pq"
+
+	"git.gobha.me/xcaliber/chat-switchboard/models"
+)
+
+// ── ProjectStore ───────────────────────────
+
+type ProjectStore struct{}
+
+func NewProjectStore() *ProjectStore { return &ProjectStore{} }
+
+// ── CRUD ────────────────────────────────────
+
+func (s *ProjectStore) Create(ctx context.Context, p *models.Project) error {
+	return DB.QueryRowContext(ctx, `
+		INSERT INTO projects (name, description, color, icon, scope, owner_id, team_id, settings)
+		VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
+		RETURNING id, created_at, updated_at`,
+		p.Name, p.Description, p.Color, p.Icon, p.Scope,
+		p.OwnerID, models.NullString(p.TeamID), ToJSON(p.Settings),
+	).Scan(&p.ID, &p.CreatedAt, &p.UpdatedAt)
+}
+
+func (s *ProjectStore) GetByID(ctx context.Context, id string) (*models.Project, error) {
+	var p models.Project
+	var teamID sql.NullString
+	err := DB.QueryRowContext(ctx, `
+		SELECT p.id, p.name, p.description, p.color, p.icon, p.scope,
+		       p.owner_id, p.team_id, p.is_archived, p.settings,
+		       p.created_at, p.updated_at,
+		       (SELECT COUNT(*) FROM project_channels WHERE project_id = p.id),
+		       (SELECT COUNT(*) FROM project_knowledge_bases WHERE project_id = p.id),
+		       (SELECT COUNT(*) FROM project_notes WHERE project_id = p.id)
+		FROM projects p
+		WHERE p.id = $1`, id).Scan(
+		&p.ID, &p.Name, &p.Description, &p.Color, &p.Icon, &p.Scope,
+		&p.OwnerID, &teamID, &p.IsArchived, &p.Settings,
+		&p.CreatedAt, &p.UpdatedAt,
+		&p.ChannelCount, &p.KBCount, &p.NoteCount,
+	)
+	if err != nil {
+		return nil, err
+	}
+	p.TeamID = NullableStringPtr(teamID)
+	return &p, nil
+}
+
+func (s *ProjectStore) Update(ctx context.Context, id string, patch models.ProjectPatch) error {
+	b := NewUpdate("projects")
+	if patch.Name != nil {
+		b.Set("name", *patch.Name)
+	}
+	if patch.Description != nil {
+		b.Set("description", *patch.Description)
+	}
+	if patch.Color != nil {
+		b.Set("color", *patch.Color)
+	}
+	if patch.Icon != nil {
+		b.Set("icon", *patch.Icon)
+	}
+	if patch.IsArchived != nil {
+		b.Set("is_archived", *patch.IsArchived)
+	}
+	if !b.HasSets() {
+		return nil
+	}
+	b.Where("id", id)
+	res, err := b.Exec(DB)
+	if err != nil {
+		return err
+	}
+	if n, _ := res.RowsAffected(); n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}
+
+func (s *ProjectStore) Delete(ctx context.Context, id string) error {
+	// CASCADE handles junction tables.
+	// Channels get project_id set to NULL via ON DELETE SET NULL.
+	res, err := DB.ExecContext(ctx, "DELETE FROM projects WHERE id = $1", id)
+	if err != nil {
+		return err
+	}
+	if n, _ := res.RowsAffected(); n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}
+
+// ── Listing ─────────────────────────────────
+
+func (s *ProjectStore) ListForUser(ctx context.Context, userID string, teamIDs []string, includeArchived bool) ([]models.Project, error) {
+	// User sees: their personal projects + team projects for their teams + global projects
+	q := `
+		SELECT p.id, p.name, p.description, p.color, p.icon, p.scope,
+		       p.owner_id, p.team_id, p.is_archived, p.settings,
+		       p.created_at, p.updated_at,
+		       (SELECT COUNT(*) FROM project_channels WHERE project_id = p.id),
+		       (SELECT COUNT(*) FROM project_knowledge_bases WHERE project_id = p.id),
+		       (SELECT COUNT(*) FROM project_notes WHERE project_id = p.id)
+		FROM projects p
+		WHERE (
+			(p.scope = 'personal' AND p.owner_id = $1)
+			OR p.scope = 'global'`
+
+	args := []interface{}{userID}
+
+	if len(teamIDs) > 0 {
+		q += fmt.Sprintf(`
+			OR (p.scope = 'team' AND p.team_id = ANY($%d))`, len(args)+1)
+		args = append(args, pq.Array(teamIDs))
+	}
+	q += `)`
+
+	if !includeArchived {
+		q += ` AND p.is_archived = false`
+	}
+	q += ` ORDER BY p.name`
+
+	return queryProjects(ctx, q, args...)
+}
+
+// ── Channel Association ─────────────────────
+
+func (s *ProjectStore) AddChannel(ctx context.Context, projectID, channelID string, position int) error {
+	// Atomic move: remove from old project (if any), add to new.
+	tx, err := DB.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	// Remove existing association (handles the "move" case)
+	tx.ExecContext(ctx, `DELETE FROM project_channels WHERE channel_id = $1`, channelID)
+
+	// Insert new association
+	_, err = tx.ExecContext(ctx, `
+		INSERT INTO project_channels (project_id, channel_id, position)
+		VALUES ($1, $2, $3)`,
+		projectID, channelID, position)
+	if err != nil {
+		return err
+	}
+
+	// Update denormalized column
+	_, err = tx.ExecContext(ctx, `UPDATE channels SET project_id = $1 WHERE id = $2`,
+		projectID, channelID)
+	if err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (s *ProjectStore) RemoveChannel(ctx context.Context, projectID, channelID string) error {
+	tx, err := DB.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	res, err := tx.ExecContext(ctx, `
+		DELETE FROM project_channels WHERE project_id = $1 AND channel_id = $2`,
+		projectID, channelID)
+	if err != nil {
+		return err
+	}
+	if n, _ := res.RowsAffected(); n == 0 {
+		return sql.ErrNoRows
+	}
+
+	// Clear denormalized column
+	_, err = tx.ExecContext(ctx, `UPDATE channels SET project_id = NULL WHERE id = $1`, channelID)
+	if err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (s *ProjectStore) ListChannels(ctx context.Context, projectID string) ([]models.ProjectChannel, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT project_id, channel_id, position, COALESCE(folder, ''), added_at
+		FROM project_channels
+		WHERE project_id = $1
+		ORDER BY position, added_at`, projectID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var result []models.ProjectChannel
+	for rows.Next() {
+		var pc models.ProjectChannel
+		if err := rows.Scan(&pc.ProjectID, &pc.ChannelID, &pc.Position, &pc.Folder, &pc.AddedAt); err != nil {
+			return nil, err
+		}
+		result = append(result, pc)
+	}
+	return result, rows.Err()
+}
+
+func (s *ProjectStore) ReorderChannels(ctx context.Context, projectID string, channelIDs []string) error {
+	tx, err := DB.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	for i, chID := range channelIDs {
+		_, err := tx.ExecContext(ctx, `
+			UPDATE project_channels SET position = $1
+			WHERE project_id = $2 AND channel_id = $3`,
+			i, projectID, chID)
+		if err != nil {
+			return err
+		}
+	}
+	return tx.Commit()
+}
+
+// ── KB Association ──────────────────────────
+
+func (s *ProjectStore) AddKB(ctx context.Context, projectID, kbID string, autoSearch bool) error {
+	_, err := DB.ExecContext(ctx, `
+		INSERT INTO project_knowledge_bases (project_id, kb_id, auto_search)
+		VALUES ($1, $2, $3)
+		ON CONFLICT (project_id, kb_id) DO UPDATE SET auto_search = EXCLUDED.auto_search`,
+		projectID, kbID, autoSearch)
+	return err
+}
+
+func (s *ProjectStore) RemoveKB(ctx context.Context, projectID, kbID string) error {
+	res, err := DB.ExecContext(ctx, `
+		DELETE FROM project_knowledge_bases WHERE project_id = $1 AND kb_id = $2`,
+		projectID, kbID)
+	if err != nil {
+		return err
+	}
+	if n, _ := res.RowsAffected(); n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}
+
+func (s *ProjectStore) ListKBs(ctx context.Context, projectID string) ([]models.ProjectKB, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT project_id, kb_id, auto_search, added_at
+		FROM project_knowledge_bases
+		WHERE project_id = $1
+		ORDER BY added_at`, projectID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var result []models.ProjectKB
+	for rows.Next() {
+		var pkb models.ProjectKB
+		if err := rows.Scan(&pkb.ProjectID, &pkb.KBID, &pkb.AutoSearch, &pkb.AddedAt); err != nil {
+			return nil, err
+		}
+		result = append(result, pkb)
+	}
+	return result, rows.Err()
+}
+
+func (s *ProjectStore) GetKBIDs(ctx context.Context, projectID string) ([]string, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT kb_id FROM project_knowledge_bases
+		WHERE project_id = $1`, projectID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var ids []string
+	for rows.Next() {
+		var id string
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		ids = append(ids, id)
+	}
+	return ids, rows.Err()
+}
+
+// ── Note Association ────────────────────────
+
+func (s *ProjectStore) AddNote(ctx context.Context, projectID, noteID string) error {
+	_, err := DB.ExecContext(ctx, `
+		INSERT INTO project_notes (project_id, note_id)
+		VALUES ($1, $2)
+		ON CONFLICT DO NOTHING`,
+		projectID, noteID)
+	return err
+}
+
+func (s *ProjectStore) RemoveNote(ctx context.Context, projectID, noteID string) error {
+	res, err := DB.ExecContext(ctx, `
+		DELETE FROM project_notes WHERE project_id = $1 AND note_id = $2`,
+		projectID, noteID)
+	if err != nil {
+		return err
+	}
+	if n, _ := res.RowsAffected(); n == 0 {
+		return sql.ErrNoRows
+	}
+	return nil
+}
+
+func (s *ProjectStore) ListNotes(ctx context.Context, projectID string) ([]models.ProjectNote, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT project_id, note_id, added_at
+		FROM project_notes
+		WHERE project_id = $1
+		ORDER BY added_at`, projectID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var result []models.ProjectNote
+	for rows.Next() {
+		var pn models.ProjectNote
+		if err := rows.Scan(&pn.ProjectID, &pn.NoteID, &pn.AddedAt); err != nil {
+			return nil, err
+		}
+		result = append(result, pn)
+	}
+	return result, rows.Err()
+}
+
+// ── Access Check ────────────────────────────
+
+func (s *ProjectStore) UserCanAccess(ctx context.Context, userID, projectID string, teamIDs []string) (bool, error) {
+	q := `
+		SELECT EXISTS(
+			SELECT 1 FROM projects p
+			WHERE p.id = $1 AND (
+				(p.scope = 'personal' AND p.owner_id = $2)
+				OR p.scope = 'global'`
+
+	args := []interface{}{projectID, userID}
+
+	if len(teamIDs) > 0 {
+		q += fmt.Sprintf(`
+				OR (p.scope = 'team' AND p.team_id = ANY($%d))`, len(args)+1)
+		args = append(args, pq.Array(teamIDs))
+	}
+	q += `))`
+
+	var ok bool
+	err := DB.QueryRowContext(ctx, q, args...).Scan(&ok)
+	return ok, err
+}
+
+func (s *ProjectStore) GetProjectIDForChannel(ctx context.Context, channelID string) (string, error) {
+	var projectID string
+	err := DB.QueryRowContext(ctx, `
+		SELECT project_id FROM project_channels WHERE channel_id = $1`,
+		channelID).Scan(&projectID)
+	if err == sql.ErrNoRows {
+		return "", nil // no project — not an error
+	}
+	return projectID, err
+}
+
+// ── Query Helper ────────────────────────────
+
+func queryProjects(ctx context.Context, q string, args ...interface{}) ([]models.Project, error) {
+	rows, err := DB.QueryContext(ctx, q, args...)
+	if err != nil {
+		return nil, fmt.Errorf("queryProjects: %w", err)
+	}
+	defer rows.Close()
+
+	var result []models.Project
+	for rows.Next() {
+		var p models.Project
+		var teamID sql.NullString
+		if err := rows.Scan(
+			&p.ID, &p.Name, &p.Description, &p.Color, &p.Icon, &p.Scope,
+			&p.OwnerID, &teamID, &p.IsArchived, &p.Settings,
+			&p.CreatedAt, &p.UpdatedAt,
+			&p.ChannelCount, &p.KBCount, &p.NoteCount,
+		); err != nil {
+			return nil, err
+		}
+		p.TeamID = NullableStringPtr(teamID)
+		result = append(result, p)
+	}
+	return result, rows.Err()
+}
--- a/server/store/postgres/stores.go
+++ b/server/store/postgres/stores.go
@@ -32,5 +32,6 @@ func NewStores(db *sql.DB) store.Stores {
 		Groups:         NewGroupStore(),
 		ResourceGrants: NewResourceGrantStore(),
 		Memories:       NewMemoryStore(),
+		Projects:       NewProjectStore(),
 	}
 }