docDB/doc.go

package docDB

import (
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"time"

	"apigo.cc/go/cast"
	"apigo.cc/go/file"
	"apigo.cc/go/id"
	"apigo.cc/go/tableDB"
)

// DocDB 实例
type DocDB struct {
	db           *tableDB.TableDBUnauthorized
	baseDir      string
	onCreatedDoc func(doc *Document)
	onUpdatedDoc func(doc *Document)
	onRemoved    func(id string)
}

// App 实例 (带用户权限)
type App struct {
	docDB  *DocDB
	userID string
	table  *tableDB.Table
}

// Document 结构体
type Document struct {
	ID            string         // 唯一标识符
	Path          string         // 层级路径
	Title         string         // 标题
	TextContent   string         // 纯文本
	BinaryContent io.Reader      `json:"-"` // 仅为内存瞬态，流式读取
	FilePath      string         // 原文件落盘绝对路径
	Thumbnail     string         // 缩略图路径
	Summary       string         // 摘要
	Metadata      map[string]any // 元数据
	ToC           []ToCNode      // 目录树
	Tags          string         // 标签
	Language      string         // 语言
	Type          string         // 类型
	StoreType     string         // 存储方式 (file/db)
	AsFile        bool           // 是否以文件存储
	Version       uint64         // 版本号
	IsSecret      bool           // 是否加密
	CreateTime    int64          // 创建时间
	UpdateTime    int64          // 更新时间
	Creator       string         // 创建人 ID
	Updater       string         // 更新人 ID
}

// ToCNode 目录树节点
type ToCNode struct {
	Title    string
	Level    int
	Children []ToCNode
}

// New 创建 DocDB 实例
func New(db *tableDB.TableDBUnauthorized, baseDir string) *DocDB {
	return &DocDB{
		db:      db,
		baseDir: baseDir,
	}
}

// Auth 授权并初始化表结构
func (d *DocDB) Auth(userID string) *App {
	appTableDB := d.db.Auth(userID)
	
	// 检查 _Doc 表是否存在，不存在则初始化
	res, _ := appTableDB.Table("_Table").Get("_Doc")
	if res == nil {
		// 使用高级接口 SetTable 和 SetField
		_ = appTableDB.SetTable(tableDB.TableSchema{ID: "_Doc", Name: "_Doc", Memo: "Document Repository"})
		_ = appTableDB.Table("_Doc").SetField(
			tableDB.FieldSchema{Name: "path", Type: "v256", IsIndex: true},
			tableDB.FieldSchema{Name: "title", Type: "v255"},
			tableDB.FieldSchema{Name: "textContent", Type: "t"},
			tableDB.FieldSchema{Name: "filePath", Type: "v256"},
			tableDB.FieldSchema{Name: "thumbnail", Type: "v256"},
			tableDB.FieldSchema{Name: "summary", Type: "t"},
			tableDB.FieldSchema{Name: "metadata", Type: "o"},
			tableDB.FieldSchema{Name: "toc", Type: "o"},
			tableDB.FieldSchema{Name: "tags", Type: "v1024"},
			tableDB.FieldSchema{Name: "language", Type: "v32"},
			tableDB.FieldSchema{Name: "type", Type: "v32"},
			tableDB.FieldSchema{Name: "storeType", Type: "v16"},
			tableDB.FieldSchema{Name: "asFile", Type: "b"},
			tableDB.FieldSchema{Name: "version", Type: "bi"},
			tableDB.FieldSchema{Name: "isSecret", Type: "b"},
			tableDB.FieldSchema{Name: "updateTime", Type: "bi"},
			tableDB.FieldSchema{Name: "updater", Type: "v64"},
			tableDB.FieldSchema{Name: "createTime", Type: "bi"},
			tableDB.FieldSchema{Name: "creator", Type: "v64"},
		)
	}

	return &App{
		docDB:  d,
		userID: userID,
		table:  appTableDB.Table("_Doc"),
	}
}

// SetDoc 更新内容 (强制提升版本号)
func (a *App) SetDoc(docObj *Document) error {
	if docObj.ID == "" {
		docObj.ID = id.MakeID(10)
	}

	now := time.Now().UnixMilli()
	existing, err := a.table.Get(docObj.ID)
	
	isUpdate := err == nil && existing != nil

	if isUpdate {
		docObj.Version = cast.Uint64(existing["version"]) + 1
		docObj.CreateTime = cast.Int64(existing["createTime"])
		docObj.Creator = cast.String(existing["creator"])
	} else {
		docObj.Version = 1
		docObj.CreateTime = now
		docObj.Creator = a.userID
	}

	docObj.UpdateTime = now
	docObj.Updater = a.userID

	// 处理文件落盘
	if docObj.BinaryContent != nil {
		if err := a.saveFile(docObj); err != nil {
			return err
		}
		docObj.AsFile = true
		docObj.StoreType = "file"
	}

	// 自动提取 ToC
	if docObj.ToC == nil && (docObj.Type == "markdown" || strings.HasSuffix(strings.ToLower(docObj.Path), ".md")) {
		docObj.ToC = ExtractToC(docObj.TextContent)
	}

	// 转换为 map 并清理瞬态/冲突字段
	record := a.toRecord(docObj)

	err = a.table.Set(record)
	if err == nil {
		if isUpdate {
			a.docDB.triggerUpdated(docObj)
		} else {
			a.docDB.triggerCreated(docObj)
		}
	}
	return err
}

// SetMeta 更新元数据 (不提升版本号)
func (a *App) SetMeta(id string, meta map[string]any) error {
	existing, err := a.table.Get(id)
	if err != nil || existing == nil {
		return fmt.Errorf("document not found: %s", id)
	}

	now := time.Now().UnixMilli()
	for k, v := range meta {
		existing[k] = v
	}
	existing["updateTime"] = now
	existing["updater"] = a.userID

	return a.table.Set(existing)
}

func (a *App) toRecord(docObj *Document) map[string]any {
	record := make(map[string]any)
	cast.Convert(&record, docObj)
	
	// tableDB 内部约定使用小写 id，这里强制转换以避免产生重复字段
	if v, ok := record["ID"]; ok {
		record["id"] = v
		delete(record, "ID")
	}
	
	// 移除瞬态字段
	delete(record, "binaryContent")
	
	return record
}

func (a *App) saveFile(docObj *Document) error {
	subDir1 := docObj.ID[0:2]
	subDir2 := docObj.ID[2:6]
	subDir3 := docObj.ID[6:10]

	ext := filepath.Ext(docObj.Path)
	fileName := fmt.Sprintf("v%d%s", docObj.Version, ext)
	filePath := filepath.Join(a.docDB.baseDir, "doc", subDir1, subDir2, subDir3, fileName)
	file.EnsureParentDir(filePath)

	out, err := os.Create(filePath)
	if err != nil {
		return err
	}
	defer out.Close()

	if _, err := io.Copy(out, docObj.BinaryContent); err != nil {
		return err
	}

	docObj.FilePath = filePath
	return nil
}

// Get 获取文档
func (a *App) Get(id string) (*Document, error) {
	res, err := a.table.Get(id)
	if err != nil || res == nil {
		return nil, err
	}
	var doc Document
	cast.Convert(&doc, res)
	return &doc, nil
}

// Remove 删除文档
func (a *App) Remove(id string) error {
	err := a.table.Remove(id)
	if err == nil {
		a.docDB.triggerRemoved(id)
	}
	return err
}

// 事件订阅
func (d *DocDB) OnCreatedDoc(fn func(doc *Document)) { d.onCreatedDoc = fn }
func (d *DocDB) OnUpdatedDoc(fn func(doc *Document)) { d.onUpdatedDoc = fn }
func (d *DocDB) OnRemoved(fn func(id string))       { d.onRemoved = fn }

func (d *DocDB) triggerCreated(doc *Document) {
	if d.onCreatedDoc != nil {
		d.onCreatedDoc(doc)
	}
}

func (d *DocDB) triggerUpdated(doc *Document) {
	if d.onUpdatedDoc != nil {
		d.onUpdatedDoc(doc)
	}
}

func (d *DocDB) triggerRemoved(id string) {
	if d.onRemoved != nil {
		d.onRemoved(id)
	}
}

// ExtractToC 保持提取目录逻辑
func ExtractToC(content string) []ToCNode {
	lines := strings.Split(content, "\n")
	var root []ToCNode
	stack := []*[]ToCNode{&root}

	for _, line := range lines {
		trimmed := strings.TrimSpace(line)
		if !strings.HasPrefix(trimmed, "#") {
			continue
		}

		level := 0
		for level < len(trimmed) && trimmed[level] == '#' {
			level++
		}
		if level > 0 && level < len(trimmed) && trimmed[level] == ' ' {
			title := strings.TrimSpace(trimmed[level:])

			for len(stack) > level {
				stack = stack[:len(stack)-1]
			}

			for len(stack) < level {
				parentSlice := stack[len(stack)-1]
				if len(*parentSlice) == 0 {
					*parentSlice = append(*parentSlice, ToCNode{
						Title: "...",
						Level: len(stack),
					})
				}
				lastNode := &((*parentSlice)[len(*parentSlice)-1])
				stack = append(stack, &(lastNode.Children))
			}

			parentSlice := stack[len(stack)-1]
			*parentSlice = append(*parentSlice, ToCNode{
				Title: title,
				Level: level,
			})
			lastNode := &((*parentSlice)[len(*parentSlice)-1])
			stack = append(stack, &(lastNode.Children))
		}
	}
	return root
}