299 lines
12 KiB
Go
299 lines
12 KiB
Go
package repository
|
||
|
||
import (
|
||
"strings"
|
||
|
||
"infinite-canvas/backend/internal/model"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// ResourceReferenceDocument 是资源删除校验使用的只读业务文档快照。
|
||
// repository 只按用户范围读取记录；JSON 中的资源引用合同由 service 统一解释。
|
||
type ResourceReferenceDocument struct {
|
||
Kind string
|
||
ID string
|
||
Title string
|
||
PrimaryJSON string
|
||
SecondaryJSON string
|
||
}
|
||
|
||
type ResourceDirectReference struct {
|
||
Kind string
|
||
ID string
|
||
Title string
|
||
ResourceID string
|
||
}
|
||
|
||
type ResourceReferenceSnapshot struct {
|
||
Documents []ResourceReferenceDocument
|
||
Direct []ResourceDirectReference
|
||
}
|
||
|
||
func (r *Repository) AssetResourceRecords(assetID string) ([]model.AssetVersion, []model.AssetRepresentation, error) {
|
||
var versions []model.AssetVersion
|
||
if err := r.db.Where("asset_id = ?", assetID).Find(&versions).Error; err != nil {
|
||
return nil, nil, err
|
||
}
|
||
if len(versions) == 0 {
|
||
return versions, nil, nil
|
||
}
|
||
versionIDs := make([]string, 0, len(versions))
|
||
for _, version := range versions {
|
||
versionIDs = append(versionIDs, version.ID)
|
||
}
|
||
var representations []model.AssetRepresentation
|
||
if err := r.db.Where("asset_version_id IN ?", versionIDs).Find(&representations).Error; err != nil {
|
||
return nil, nil, err
|
||
}
|
||
return versions, representations, nil
|
||
}
|
||
|
||
func (r *Repository) ResourcesForUserIDs(userID string, resourceIDs []string) ([]model.Resource, error) {
|
||
if len(resourceIDs) == 0 {
|
||
return []model.Resource{}, nil
|
||
}
|
||
var resources []model.Resource
|
||
err := r.db.Where("user_id = ? AND id IN ?", userID, resourceIDs).Find(&resources).Error
|
||
return resources, err
|
||
}
|
||
|
||
// ResourceStorageReferenceCount 返回未包含在 excludedResourceIDs 中、但指向同一物理对象的资源记录数。
|
||
// 这用于兼容历史数据中多个 Resource 行复用一个对象路径的情况。
|
||
func (r *Repository) ResourceStorageReferenceCount(resource *model.Resource, excludedResourceIDs []string) (int64, error) {
|
||
if resource == nil {
|
||
return 0, nil
|
||
}
|
||
query := r.db.Model(&model.Resource{}).
|
||
Where("endpoint = ? AND bucket = ? AND object_key = ?", resource.Endpoint, resource.Bucket, resource.ObjectKey)
|
||
if strings.TrimSpace(resource.Provider) == "" || strings.EqualFold(strings.TrimSpace(resource.Provider), "local") {
|
||
query = query.Where("provider IN ?", []string{"", "local"})
|
||
} else {
|
||
query = query.Where("provider = ?", resource.Provider)
|
||
}
|
||
if len(excludedResourceIDs) > 0 {
|
||
query = query.Where("id NOT IN ?", excludedResourceIDs)
|
||
}
|
||
var count int64
|
||
err := query.Count(&count).Error
|
||
return count, err
|
||
}
|
||
|
||
func (r *Repository) ResourceReferenceSnapshot(userID string, excludingAssetID string, resourceIDs []string) (ResourceReferenceSnapshot, error) {
|
||
snapshot := ResourceReferenceSnapshot{Documents: []ResourceReferenceDocument{}, Direct: []ResourceDirectReference{}}
|
||
if len(resourceIDs) == 0 {
|
||
return snapshot, nil
|
||
}
|
||
|
||
var assets []model.Asset
|
||
assetQuery := r.db.Where("user_id = ? AND id <> ?", userID, excludingAssetID)
|
||
if err := assetQuery.Find(&assets).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, asset := range assets {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "素材", ID: asset.ID, Title: asset.Title, PrimaryJSON: asset.PayloadJSON})
|
||
}
|
||
|
||
var canvases []model.CanvasProject
|
||
if err := r.db.Where("user_id = ?", userID).Find(&canvases).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, canvas := range canvases {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "画布", ID: canvas.ID, Title: canvas.Title, PrimaryJSON: canvas.PayloadJSON})
|
||
}
|
||
|
||
var projects []model.Project
|
||
if err := r.db.Where("user_id = ?", userID).Find(&projects).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, project := range projects {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "项目", ID: project.ID, Title: project.Name, PrimaryJSON: project.StyleProfileJSON})
|
||
}
|
||
|
||
var styles []model.StyleProfile
|
||
if err := r.db.Where("user_id = ?", userID).Find(&styles).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, style := range styles {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "风格", ID: style.ID, Title: style.Name, PrimaryJSON: style.CoverURL, SecondaryJSON: style.ProfileJSON})
|
||
}
|
||
|
||
type joinedDocument struct {
|
||
ID string
|
||
Title string
|
||
PrimaryJSON string
|
||
SecondaryJSON string
|
||
}
|
||
var versions []joinedDocument
|
||
versionQuery := r.db.Table("asset_versions").
|
||
Select("asset_versions.id, assets.title, asset_versions.definition_json AS primary_json").
|
||
Joins("JOIN assets ON assets.id = asset_versions.asset_id").
|
||
Where("assets.user_id = ? AND assets.id <> ?", userID, excludingAssetID)
|
||
if err := versionQuery.Scan(&versions).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, version := range versions {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "素材", ID: version.ID, Title: version.Title, PrimaryJSON: version.PrimaryJSON})
|
||
}
|
||
|
||
var candidates []joinedDocument
|
||
candidateQuery := r.db.Table("project_asset_candidates").
|
||
Select("project_asset_candidates.id, projects.name AS title, project_asset_candidates.details_json AS primary_json").
|
||
Joins("JOIN projects ON projects.id = project_asset_candidates.project_id").
|
||
Where("projects.user_id = ?", userID)
|
||
if err := candidateQuery.Scan(&candidates).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, candidate := range candidates {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "项目", ID: candidate.ID, Title: candidate.Title, PrimaryJSON: candidate.PrimaryJSON})
|
||
}
|
||
|
||
var steps []joinedDocument
|
||
stepQuery := r.db.Table("workflow_step_instances").
|
||
Select("workflow_step_instances.id, workflow_step_instances.name AS title, workflow_step_instances.input_json AS primary_json, workflow_step_instances.output_json AS secondary_json").
|
||
Joins("JOIN workflow_instances ON workflow_instances.id = workflow_step_instances.workflow_instance_id").
|
||
Joins("JOIN projects ON projects.id = workflow_instances.project_id").
|
||
Where("projects.user_id = ?", userID)
|
||
if err := stepQuery.Scan(&steps).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, step := range steps {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "工作流", ID: step.ID, Title: step.Title, PrimaryJSON: step.PrimaryJSON, SecondaryJSON: step.SecondaryJSON})
|
||
}
|
||
|
||
var shotArtifacts []joinedDocument
|
||
artifactDocumentQuery := r.db.Table("shot_artifacts").
|
||
Select("shot_artifacts.id, shots.title, shot_artifacts.metadata_json AS primary_json").
|
||
Joins("JOIN shots ON shots.id = shot_artifacts.shot_id").
|
||
Joins("JOIN projects ON projects.id = shots.project_id").
|
||
Where("projects.user_id = ?", userID)
|
||
if err := artifactDocumentQuery.Scan(&shotArtifacts).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, artifact := range shotArtifacts {
|
||
snapshot.Documents = append(snapshot.Documents, ResourceReferenceDocument{Kind: "镜头产物", ID: artifact.ID, Title: artifact.Title, PrimaryJSON: artifact.PrimaryJSON})
|
||
}
|
||
|
||
type joinedRepresentation struct {
|
||
ID string
|
||
Title string
|
||
ResourceID string
|
||
}
|
||
var representations []joinedRepresentation
|
||
if err := r.db.Table("asset_representations").
|
||
Select("asset_representations.id, assets.title, asset_representations.resource_id").
|
||
Joins("JOIN asset_versions ON asset_versions.id = asset_representations.asset_version_id").
|
||
Joins("JOIN assets ON assets.id = asset_versions.asset_id").
|
||
Where("assets.user_id = ? AND assets.id <> ? AND asset_representations.resource_id IN ?", userID, excludingAssetID, resourceIDs).
|
||
Scan(&representations).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, representation := range representations {
|
||
snapshot.Direct = append(snapshot.Direct, ResourceDirectReference{Kind: "素材", ID: representation.ID, Title: representation.Title, ResourceID: representation.ResourceID})
|
||
}
|
||
|
||
var voices []model.VoiceProfile
|
||
if err := r.db.Where("user_id = ? AND sample_resource_id IN ?", userID, resourceIDs).Find(&voices).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, voice := range voices {
|
||
snapshot.Direct = append(snapshot.Direct, ResourceDirectReference{Kind: "声音", ID: voice.ID, Title: voice.Name, ResourceID: voice.SampleResourceID})
|
||
}
|
||
|
||
type joinedShotArtifact struct {
|
||
ID string
|
||
Title string
|
||
ResourceID string
|
||
}
|
||
var artifacts []joinedShotArtifact
|
||
if err := r.db.Table("shot_artifacts").
|
||
Select("shot_artifacts.id, shots.title, shot_artifacts.resource_id").
|
||
Joins("JOIN shots ON shots.id = shot_artifacts.shot_id").
|
||
Joins("JOIN projects ON projects.id = shots.project_id").
|
||
Where("projects.user_id = ? AND shot_artifacts.resource_id IN ?", userID, resourceIDs).
|
||
Scan(&artifacts).Error; err != nil {
|
||
return snapshot, err
|
||
}
|
||
for _, artifact := range artifacts {
|
||
snapshot.Direct = append(snapshot.Direct, ResourceDirectReference{Kind: "镜头产物", ID: artifact.ID, Title: artifact.Title, ResourceID: artifact.ResourceID})
|
||
}
|
||
return snapshot, nil
|
||
}
|
||
|
||
func (r *Repository) AssetBusinessReferences(userID string, assetID string) ([]ResourceDirectReference, error) {
|
||
type projectReference struct {
|
||
ID string
|
||
Title string
|
||
}
|
||
var projects []projectReference
|
||
if err := r.db.Table("project_asset_links").
|
||
Distinct("projects.id, projects.name AS title").
|
||
Joins("JOIN projects ON projects.id = project_asset_links.project_id").
|
||
Where("projects.user_id = ? AND project_asset_links.asset_id = ?", userID, assetID).
|
||
Scan(&projects).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
var shotProjects []projectReference
|
||
if err := r.db.Table("shot_asset_references").
|
||
Distinct("projects.id, projects.name AS title").
|
||
Joins("JOIN asset_versions ON asset_versions.id = shot_asset_references.asset_version_id").
|
||
Joins("JOIN shots ON shots.id = shot_asset_references.shot_id").
|
||
Joins("JOIN projects ON projects.id = shots.project_id").
|
||
Where("projects.user_id = ? AND asset_versions.asset_id = ?", userID, assetID).
|
||
Scan(&shotProjects).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
var candidateProjects []projectReference
|
||
if err := r.db.Table("project_asset_candidates").
|
||
Distinct("projects.id, projects.name AS title").
|
||
Joins("JOIN projects ON projects.id = project_asset_candidates.project_id").
|
||
Where("projects.user_id = ? AND project_asset_candidates.resolved_asset_id = ?", userID, assetID).
|
||
Scan(&candidateProjects).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
result := make([]ResourceDirectReference, 0, len(projects)+len(shotProjects)+len(candidateProjects))
|
||
for _, project := range append(append(projects, shotProjects...), candidateProjects...) {
|
||
result = append(result, ResourceDirectReference{Kind: "项目", ID: project.ID, Title: project.Title})
|
||
}
|
||
return result, nil
|
||
}
|
||
|
||
func (r *Repository) DeleteAssetAndResources(userID string, assetID string, resourceIDs []string, deletionJobs []model.ResourceDeletionJob) error {
|
||
return r.db.Transaction(func(tx *gorm.DB) error {
|
||
versionIDs := tx.Model(&model.AssetVersion{}).Select("id").Where("asset_id = ?", assetID)
|
||
if err := tx.Where("asset_version_id IN (?)", versionIDs).Delete(&model.ShotAssetReference{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Where("asset_version_id IN (?)", versionIDs).Delete(&model.CharacterVoiceBinding{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Where("asset_version_id IN (?)", versionIDs).Delete(&model.AssetRepresentation{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Where("asset_id = ?", assetID).Delete(&model.ProjectAssetLink{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Where("resolved_asset_id = ?", assetID).Delete(&model.ProjectAssetCandidate{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Where("asset_id = ?", assetID).Delete(&model.AssetVersion{}).Error; err != nil {
|
||
return err
|
||
}
|
||
if err := tx.Delete(&model.Asset{}, "id = ? AND user_id = ?", assetID, userID).Error; err != nil {
|
||
return err
|
||
}
|
||
if len(deletionJobs) > 0 {
|
||
if err := tx.Create(&deletionJobs).Error; err != nil {
|
||
return err
|
||
}
|
||
}
|
||
if len(resourceIDs) == 0 {
|
||
return nil
|
||
}
|
||
if err := tx.Where("resource_id IN ?", resourceIDs).Delete(&model.ArkPrivateAssetBinding{}).Error; err != nil {
|
||
return err
|
||
}
|
||
return tx.Where("user_id = ? AND id IN ?", userID, resourceIDs).Delete(&model.Resource{}).Error
|
||
})
|
||
}
|