Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-18 04:44:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support setting file fingerprints (#4376)

Browse Source 
* Support setting file fingerprints
* Disallow modifying managed hashes
This commit is contained in:
WithoutPants
2023-12-22 14:07:10 +11:00
committed by GitHub
parent a1bd7cf817
commit afda6decf2
7 changed files with 156 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
pkg/sqlite/fingerprint.go
View File

@@ -68,6 +68,25 @@ func (qb *fingerprintQueryBuilder) insertJoins(ctx context.Context, fileID model
return nil
}
func (qb *fingerprintQueryBuilder) upsertJoins(ctx context.Context, fileID models.FileID, f []models.Fingerprint) error {
types := make([]string, len(f))
for i, ff := range f {
types[i] = ff.Type
}
if err := qb.destroyJoins(ctx, fileID, types); err != nil {
return err
}
for _, ff := range f {
if err := qb.insert(ctx, fileID, ff); err != nil {
return err
}
}
return nil
}
func (qb *fingerprintQueryBuilder) replaceJoins(ctx context.Context, fileID models.FileID, f []models.Fingerprint) error {
if err := qb.destroy(ctx, []int{int(fileID)}); err != nil {
return err
@@ -76,6 +95,21 @@ func (qb *fingerprintQueryBuilder) replaceJoins(ctx context.Context, fileID mode
return qb.insertJoins(ctx, fileID, f)
}
func (qb *fingerprintQueryBuilder) destroyJoins(ctx context.Context, fileID models.FileID, types []string) error {
table := qb.table()
q := dialect.Delete(table).Where(
table.Col(fileIDColumn).Eq(fileID),
table.Col("type").In(types),
)
_, err := exec(ctx, q)
if err != nil {
return fmt.Errorf("deleting from %s: %w", table.GetTable(), err)
}
return nil
}
func (qb *fingerprintQueryBuilder) table() exp.IdentifierExpression {
return qb.tableMgr.table
}