更新项目架构与存储适配器，添加用户认证功能

本次提交包含以下主要更改： 1. 更新 `.gitignore` 文件，添加对 `node_modules` 和环境变量文件的忽略。 2. 修改 `.gitmodules` 文件，替换为新的子模块 `cloudflare-worker`。 3. 新增 `ARCHITECTURE.md` 和 `PROJECT_REFACTOR_PLAN.md` 文档，详细描述项目架构和改造计划。 4. 实现用户认证功能，添加 GitHub OAuth 处理逻辑，支持 JWT 生成与解析。 5. 引入新的存储接口 `CanvasStore`，并实现相应的存储逻辑，支持用户画布的增删改查。 6. 更新 `main.go` 文件，整合新的认证与存储逻辑，优化路由设置。这些更改旨在提升项目的可扩展性与用户体验，支持多用户环境下的画布管理与存储。
2026-06-03 22:02:57 +00:00 · 2025-07-05 23:13:17 +08:00
parent 61abbc612b
commit 94953a5eac
26 changed files with 2078 additions and 293 deletions
@@ -0,0 +1,157 @@
+package sqlite
+
+import (
+	"bytes"
+	"context"
+	"database/sql"
+	"excalidraw-complete/core"
+	"fmt"
+	"log"
+	"time"
+
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/oklog/ulid/v2"
+	"github.com/sirupsen/logrus"
+)
+
+type sqliteStore struct {
+	db *sql.DB
+}
+
+// NewStore creates a new SQLite-based store.
+func NewStore(dataSourceName string) *sqliteStore {
+	db, err := sql.Open("sqlite3", dataSourceName)
+	if err != nil {
+		log.Fatalf("failed to open sqlite database: %v", err)
+	}
+
+	// Initialize table for anonymous documents
+	docTableStmt := `CREATE TABLE IF NOT EXISTS documents (id TEXT PRIMARY KEY, data BLOB);`
+	if _, err = db.Exec(docTableStmt); err != nil {
+		log.Fatalf("failed to create documents table: %v", err)
+	}
+
+	// Initialize table for user-owned canvases
+	canvasTableStmt := `
+	CREATE TABLE IF NOT EXISTS canvases (
+		id TEXT NOT NULL,
+		user_id TEXT NOT NULL,
+		name TEXT,
+		data BLOB,
+		created_at DATETIME,
+		updated_at DATETIME,
+		PRIMARY KEY (user_id, id)
+	);`
+	if _, err = db.Exec(canvasTableStmt); err != nil {
+		log.Fatalf("failed to create canvases table: %v", err)
+	}
+
+	return &sqliteStore{db}
+}
+
+// DocumentStore implementation
+func (s *sqliteStore) FindID(ctx context.Context, id string) (*core.Document, error) {
+	log := logrus.WithField("document_id", id)
+	log.Debug("Retrieving document by ID")
+	var data []byte
+	err := s.db.QueryRowContext(ctx, "SELECT data FROM documents WHERE id = ?", id).Scan(&data)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			log.WithField("error", "document not found").Warn("Document with specified ID not found")
+			return nil, fmt.Errorf("document with id %s not found", id)
+		}
+		log.WithError(err).Error("Failed to retrieve document")
+		return nil, err
+	}
+	document := core.Document{
+		Data: *bytes.NewBuffer(data),
+	}
+	log.Info("Document retrieved successfully")
+	return &document, nil
+}
+
+func (s *sqliteStore) Create(ctx context.Context, document *core.Document) (string, error) {
+	id := ulid.Make().String()
+	data := document.Data.Bytes()
+	log := logrus.WithFields(logrus.Fields{
+		"document_id": id,
+		"data_length": len(data),
+	})
+
+	_, err := s.db.ExecContext(ctx, "INSERT INTO documents (id, data) VALUES (?, ?)", id, data)
+	if err != nil {
+		log.WithError(err).Error("Failed to create document")
+		return "", err
+	}
+	log.Info("Document created successfully")
+	return id, nil
+}
+
+// CanvasStore implementation
+func (s *sqliteStore) List(ctx context.Context, userID string) ([]*core.Canvas, error) {
+	rows, err := s.db.QueryContext(ctx, "SELECT id, name, updated_at FROM canvases WHERE user_id = ?", userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var canvases []*core.Canvas
+	for rows.Next() {
+		var canvas core.Canvas
+		canvas.UserID = userID
+		if err := rows.Scan(&canvas.ID, &canvas.Name, &canvas.UpdatedAt); err != nil {
+			return nil, err
+		}
+		canvases = append(canvases, &canvas)
+	}
+	return canvases, nil
+}
+
+func (s *sqliteStore) Get(ctx context.Context, userID, id string) (*core.Canvas, error) {
+	var canvas core.Canvas
+	canvas.UserID = userID
+	canvas.ID = id
+	err := s.db.QueryRowContext(ctx, "SELECT name, data, created_at, updated_at FROM canvases WHERE user_id = ? AND id = ?", userID, id).Scan(&canvas.Name, &canvas.Data, &canvas.CreatedAt, &canvas.UpdatedAt)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			return nil, fmt.Errorf("canvas not found")
+		}
+		return nil, err
+	}
+	return &canvas, nil
+}
+
+func (s *sqliteStore) Save(ctx context.Context, canvas *core.Canvas) error {
+	tx, err := s.db.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback() // Rollback on any error
+
+	var exists bool
+	err = tx.QueryRowContext(ctx, "SELECT 1 FROM canvases WHERE user_id = ? AND id = ?", canvas.UserID, canvas.ID).Scan(&exists)
+
+	now := time.Now()
+	if err != nil && err != sql.ErrNoRows {
+		return err
+	}
+
+	if exists {
+		// Update
+		_, err = tx.ExecContext(ctx, "UPDATE canvases SET name = ?, data = ?, updated_at = ? WHERE user_id = ? AND id = ?", canvas.Name, canvas.Data, now, canvas.UserID, canvas.ID)
+	} else {
+		// Insert
+		_, err = tx.ExecContext(ctx, "INSERT INTO canvases (id, user_id, name, data, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)", canvas.ID, canvas.UserID, canvas.Name, canvas.Data, now, now)
+	}
+
+	if err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (s *sqliteStore) Delete(ctx context.Context, userID, id string) error {
+	_, err := s.db.ExecContext(ctx, "DELETE FROM canvases WHERE user_id = ? AND id = ?", userID, id)
+	return err
+}