add model identities and restore httpx router

2026-07-19 00:06:21 +08:00
parent f7cd69c70f
commit 4e46fee809
20 changed files with 468 additions and 123 deletions
--- a/backend/internal/models/identity.go
+++ b/backend/internal/models/identity.go
@@ -0,0 +1,202 @@
+package models
+
+import (
+	"database/sql"
+
+	"github.com/google/uuid"
+	"gorm.io/gorm"
+)
+
+func (m *SenlinAgentUser) BeforeCreate(tx *gorm.DB) error {
+	return ensureIdentity(&m.Identity)
+}
+
+func (m *SenlinAgentProject) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentUser{}, m.OwnerID, &m.OwnerIdentity)
+}
+
+func (m *SenlinAgentInboxItem) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity)
+}
+
+func (m *SenlinAgentTask) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity); err != nil {
+		return err
+	}
+	if err := resolveOptionalIdentity(tx, &SenlinAgentUser{}, m.AssigneeID, &m.AssigneeIdentity); err != nil {
+		return err
+	}
+	return resolveOptionalIdentity(tx, &SenlinAgentInboxItem{}, m.SourceInboxItemID, &m.SourceInboxItemIdentity)
+}
+
+func (m *SenlinAgentNote) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity); err != nil {
+		return err
+	}
+	return resolveOptionalIdentity(tx, &SenlinAgentInboxItem{}, m.SourceInboxItemID, &m.SourceInboxItemIdentity)
+}
+
+func (m *SenlinAgentSource) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity); err != nil {
+		return err
+	}
+	return resolveOptionalIdentity(tx, &SenlinAgentInboxItem{}, m.SourceInboxItemID, &m.SourceInboxItemIdentity)
+}
+
+func (m *SenlinAgentAISession) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity)
+}
+
+func (m *SenlinAgentTag) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity)
+}
+
+func (m *SenlinAgentProjectEvent) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentProject{}, m.ProjectID, &m.ProjectIdentity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentUser{}, m.ActorID, &m.ActorIdentity); err != nil {
+		return err
+	}
+	return resolveEntityIdentity(tx, m.EntityType, m.EntityID, &m.EntityIdentity)
+}
+
+func (m *SenlinAgentAIKey) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentUser{}, m.UserID, &m.UserIdentity)
+}
+
+func (m *SenlinAgentAICallLog) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	return resolveIdentity(tx, &SenlinAgentUser{}, m.UserID, &m.UserIdentity)
+}
+
+func (m *SenlinAgentTaskShare) BeforeCreate(tx *gorm.DB) error {
+	if err := ensureIdentity(&m.Identity); err != nil {
+		return err
+	}
+	if err := resolveIdentity(tx, &SenlinAgentTask{}, m.TaskID, &m.TaskIdentity); err != nil {
+		return err
+	}
+	return resolveEntityIdentity(tx, m.ObjectType, m.ObjectID, &m.ObjectIdentity)
+}
+
+func ensureIdentity(identity *string) error {
+	if *identity != "" {
+		return nil
+	}
+	next, err := uuidV7()
+	if err != nil {
+		return err
+	}
+	*identity = next
+	return nil
+}
+
+func resolveIdentity(tx *gorm.DB, model any, id uint, identity *string) error {
+	if id == 0 || *identity != "" {
+		return nil
+	}
+	resolved, err := lookupIdentity(tx, model, id)
+	if err != nil {
+		return err
+	}
+	*identity = resolved
+	return nil
+}
+
+func resolveOptionalIdentity(tx *gorm.DB, model any, id *uint, identity **string) error {
+	if id == nil || *id == 0 || (identity != nil && *identity != nil && **identity != "") {
+		return nil
+	}
+	resolved, err := lookupIdentity(tx, model, *id)
+	if err != nil {
+		return err
+	}
+	*identity = &resolved
+	return nil
+}
+
+func resolveEntityIdentity(tx *gorm.DB, entityType string, entityID uint, identity *string) error {
+	if entityID == 0 || *identity != "" {
+		return nil
+	}
+	switch entityType {
+	case "project":
+		return resolveIdentity(tx, &SenlinAgentProject{}, entityID, identity)
+	case "inbox", "inbox_item":
+		return resolveIdentity(tx, &SenlinAgentInboxItem{}, entityID, identity)
+	case "task":
+		return resolveIdentity(tx, &SenlinAgentTask{}, entityID, identity)
+	case "note":
+		return resolveIdentity(tx, &SenlinAgentNote{}, entityID, identity)
+	case "source":
+		return resolveIdentity(tx, &SenlinAgentSource{}, entityID, identity)
+	case "ai_session":
+		return resolveIdentity(tx, &SenlinAgentAISession{}, entityID, identity)
+	case "tag":
+		return resolveIdentity(tx, &SenlinAgentTag{}, entityID, identity)
+	default:
+		return nil
+	}
+}
+
+func lookupIdentity(tx *gorm.DB, model any, id uint) (string, error) {
+	var identity string
+	err := tx.Model(model).Select("identity").Where("id = ?", id).Row().Scan(&identity)
+	if err == sql.ErrNoRows {
+		return "", nil
+	}
+	return identity, err
+}
+
+func uuidV7() (string, error) {
+	next, err := uuid.NewV7()
+	if err != nil {
+		return "", err
+	}
+	return next.String(), nil
+}