Changeset 0.22.8 (#150)

2026-03-04 16:06:12 +00:00
parent 389e47b0f9
commit 7e26a2a261
114 changed files with 3700 additions and 7572 deletions
--- a/server/store/sqlite/file.go
+++ b/server/store/sqlite/file.go
@@ -0,0 +1,256 @@
+package sqlite
+
+import (
+	"context"
+	"database/sql"
+	"encoding/json"
+	"time"
+
+	"git.gobha.me/xcaliber/chat-switchboard/models"
+	"git.gobha.me/xcaliber/chat-switchboard/store"
+)
+
+// FileStore implements store.FileStore against the `files` table.
+// 
+type FileStore struct{}
+
+func NewFileStore() *FileStore { return &FileStore{} }
+
+const fileCols = `id, channel_id, user_id, message_id, project_id, origin,
+	filename, content_type, size_bytes, storage_key, display_hint,
+	extracted_text, metadata, created_at, updated_at`
+
+func scanFile(row interface{ Scan(dest ...interface{}) error }) (*models.File, error) {
+	var f models.File
+	var messageID sql.NullString
+	var projectID sql.NullString
+	var extractedText sql.NullString
+	var metadataJSON []byte
+
+	err := row.Scan(
+		&f.ID, &f.ChannelID, &f.UserID, &messageID, &projectID, &f.Origin,
+		&f.Filename, &f.ContentType, &f.SizeBytes,
+		&f.StorageKey, &f.DisplayHint,
+		&extractedText, &metadataJSON, st(&f.CreatedAt), st(&f.UpdatedAt),
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	f.MessageID = NullableStringPtr(messageID)
+	f.ProjectID = NullableStringPtr(projectID)
+	if extractedText.Valid {
+		f.ExtractedText = &extractedText.String
+	}
+	if len(metadataJSON) > 0 {
+		json.Unmarshal(metadataJSON, &f.Metadata)
+	}
+	return &f, nil
+}
+
+func (s *FileStore) Create(ctx context.Context, f *models.File) error {
+	f.ID = store.NewID()
+	now := time.Now().UTC()
+	f.CreatedAt = now
+	f.UpdatedAt = now
+	if f.Origin == "" {
+		f.Origin = models.FileOriginUserUpload
+	}
+	if f.DisplayHint == "" {
+		f.DisplayHint = models.FileHintDownload
+	}
+	_, err := DB.ExecContext(ctx, `
+		INSERT INTO files (id, channel_id, user_id, message_id, project_id, origin,
+			filename, content_type, size_bytes, storage_key, display_hint,
+			extracted_text, metadata, created_at, updated_at)
+		VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)`,
+		f.ID, f.ChannelID, f.UserID, models.NullString(f.MessageID),
+		models.NullString(f.ProjectID), f.Origin,
+		f.Filename, f.ContentType, f.SizeBytes,
+		f.StorageKey, f.DisplayHint, models.NullString(f.ExtractedText),
+		ToJSON(f.Metadata), now.Format(timeFmt), now.Format(timeFmt),
+	)
+	return err
+}
+
+func (s *FileStore) GetByID(ctx context.Context, id string) (*models.File, error) {
+	row := DB.QueryRowContext(ctx, `SELECT `+fileCols+` FROM files WHERE id = ?`, id)
+	return scanFile(row)
+}
+
+func (s *FileStore) GetByChannel(ctx context.Context, channelID string, origin string) ([]models.File, error) {
+	var rows *sql.Rows
+	var err error
+	if origin != "" {
+		rows, err = DB.QueryContext(ctx,
+			`SELECT `+fileCols+` FROM files WHERE channel_id = ? AND origin = ? ORDER BY created_at`,
+			channelID, origin)
+	} else {
+		rows, err = DB.QueryContext(ctx,
+			`SELECT `+fileCols+` FROM files WHERE channel_id = ? ORDER BY created_at`,
+			channelID)
+	}
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var out []models.File
+	for rows.Next() {
+		f, err := scanFile(rows)
+		if err != nil {
+			return nil, err
+		}
+		out = append(out, *f)
+	}
+	return out, rows.Err()
+}
+
+func (s *FileStore) GetByMessage(ctx context.Context, messageID string) ([]models.File, error) {
+	rows, err := DB.QueryContext(ctx,
+		`SELECT `+fileCols+` FROM files WHERE message_id = ? ORDER BY created_at`, messageID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var out []models.File
+	for rows.Next() {
+		f, err := scanFile(rows)
+		if err != nil {
+			return nil, err
+		}
+		out = append(out, *f)
+	}
+	return out, rows.Err()
+}
+
+func (s *FileStore) GetByProject(ctx context.Context, projectID string) ([]models.File, error) {
+	rows, err := DB.QueryContext(ctx,
+		`SELECT `+fileCols+` FROM files WHERE project_id = ? ORDER BY created_at`, projectID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var out []models.File
+	for rows.Next() {
+		f, err := scanFile(rows)
+		if err != nil {
+			return nil, err
+		}
+		out = append(out, *f)
+	}
+	return out, rows.Err()
+}
+
+func (s *FileStore) GetByUser(ctx context.Context, userID string, page, perPage int) ([]models.File, int, error) {
+	var total int
+	err := DB.QueryRowContext(ctx,
+		`SELECT COUNT(*) FROM files WHERE user_id = ?`, userID).Scan(&total)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	offset := (page - 1) * perPage
+	rows, err := DB.QueryContext(ctx,
+		`SELECT `+fileCols+` FROM files WHERE user_id = ?
+		 ORDER BY created_at DESC LIMIT ? OFFSET ?`, userID, perPage, offset)
+	if err != nil {
+		return nil, 0, err
+	}
+	defer rows.Close()
+
+	var out []models.File
+	for rows.Next() {
+		f, err := scanFile(rows)
+		if err != nil {
+			return nil, total, err
+		}
+		out = append(out, *f)
+	}
+	return out, total, rows.Err()
+}
+
+func (s *FileStore) SetMessageID(ctx context.Context, fileID, messageID string) error {
+	_, err := DB.ExecContext(ctx,
+		`UPDATE files SET message_id = ?, updated_at = ? WHERE id = ?`,
+		messageID, time.Now().UTC().Format(timeFmt), fileID)
+	return err
+}
+
+func (s *FileStore) UpdateMetadata(ctx context.Context, id string, metadata map[string]interface{}) error {
+	metaJSON, err := json.Marshal(metadata)
+	if err != nil {
+		return err
+	}
+	_, err = DB.ExecContext(ctx,
+		`UPDATE files SET metadata = json_patch(metadata, ?), updated_at = ? WHERE id = ?`,
+		string(metaJSON), time.Now().UTC().Format(timeFmt), id)
+	return err
+}
+
+func (s *FileStore) SetExtractedText(ctx context.Context, id string, text string) error {
+	_, err := DB.ExecContext(ctx,
+		`UPDATE files SET extracted_text = ?, updated_at = ? WHERE id = ?`,
+		text, time.Now().UTC().Format(timeFmt), id)
+	return err
+}
+
+func (s *FileStore) Delete(ctx context.Context, id string) (*models.File, error) {
+	row := DB.QueryRowContext(ctx,
+		`DELETE FROM files WHERE id = ? RETURNING `+fileCols, id)
+	return scanFile(row)
+}
+
+func (s *FileStore) DeleteByChannel(ctx context.Context, channelID string) ([]string, error) {
+	rows, err := DB.QueryContext(ctx,
+		`DELETE FROM files WHERE channel_id = ? RETURNING storage_key`, channelID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var keys []string
+	for rows.Next() {
+		var key string
+		if err := rows.Scan(&key); err != nil {
+			return keys, err
+		}
+		keys = append(keys, key)
+	}
+	return keys, rows.Err()
+}
+
+func (s *FileStore) UserUsageBytes(ctx context.Context, userID string) (int64, error) {
+	var total sql.NullInt64
+	err := DB.QueryRowContext(ctx,
+		`SELECT COALESCE(SUM(size_bytes), 0) FROM files WHERE user_id = ?`,
+		userID).Scan(&total)
+	if err != nil {
+		return 0, err
+	}
+	return total.Int64, nil
+}
+
+func (s *FileStore) ListOrphans(ctx context.Context, olderThan time.Duration) ([]models.File, error) {
+	cutoff := time.Now().Add(-olderThan)
+	rows, err := DB.QueryContext(ctx,
+		`SELECT `+fileCols+` FROM files
+		 WHERE message_id IS NULL AND origin = 'user_upload' AND created_at < ?
+		 ORDER BY created_at`, cutoff)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var out []models.File
+	for rows.Next() {
+		f, err := scanFile(rows)
+		if err != nil {
+			return nil, err
+		}
+		out = append(out, *f)
+	}
+	return out, rows.Err()
+}