mirror of
https://github.com/stashapp/stash.git
synced 2025-12-14 12:25:23 +01:00
f69bd8a94f
* Move main to cmd * Move api to internal * Move logger and manager to internal * Move shell hiding code to separate package * Decouple job from desktop and utils * Decouple session from config * Move static into internal * Decouple config from dlna * Move desktop to internal * Move dlna to internal * Decouple remaining packages from config * Move config into internal * Move jsonschema and paths to models * Make ffmpeg functions private * Move file utility methods into fsutil package * Move symwalk into fsutil * Move single-use util functions into client package * Move slice functions to separate packages * Add env var to suppress windowsgui arg * Move hash functions into separate package * Move identify to internal * Move autotag to internal * Touch UI when generating backend
73 lines
2.3 KiB
Go
73 lines
2.3 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/stashapp/stash/internal/manager/config"
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
)
|
|
|
|
func setInitialMD5Config(ctx context.Context, txnManager models.TransactionManager) {
|
|
// if there are no scene files in the database, then default the
|
|
// VideoFileNamingAlgorithm config setting to oshash and calculateMD5 to
|
|
// false, otherwise set them to true for backwards compatibility purposes
|
|
var count int
|
|
if err := txnManager.WithReadTxn(ctx, func(r models.ReaderRepository) error {
|
|
var err error
|
|
count, err = r.Scene().Count()
|
|
return err
|
|
}); err != nil {
|
|
logger.Errorf("Error while counting scenes: %s", err.Error())
|
|
return
|
|
}
|
|
|
|
usingMD5 := count != 0
|
|
defaultAlgorithm := models.HashAlgorithmOshash
|
|
if usingMD5 {
|
|
defaultAlgorithm = models.HashAlgorithmMd5
|
|
}
|
|
|
|
config := config.GetInstance()
|
|
config.SetChecksumDefaultValues(defaultAlgorithm, usingMD5)
|
|
if err := config.Write(); err != nil {
|
|
logger.Errorf("Error while writing configuration file: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
// ValidateVideoFileNamingAlgorithm validates changing the
|
|
// VideoFileNamingAlgorithm configuration flag.
|
|
//
|
|
// If setting VideoFileNamingAlgorithm to MD5, then this function will ensure
|
|
// that all checksum values are set on all scenes.
|
|
//
|
|
// Likewise, if VideoFileNamingAlgorithm is set to oshash, then this function
|
|
// will ensure that all oshash values are set on all scenes.
|
|
func ValidateVideoFileNamingAlgorithm(txnManager models.TransactionManager, newValue models.HashAlgorithm) error {
|
|
// if algorithm is being set to MD5, then all checksums must be present
|
|
return txnManager.WithReadTxn(context.TODO(), func(r models.ReaderRepository) error {
|
|
qb := r.Scene()
|
|
if newValue == models.HashAlgorithmMd5 {
|
|
missingMD5, err := qb.CountMissingChecksum()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if missingMD5 > 0 {
|
|
return errors.New("some checksums are missing on scenes. Run Scan with calculateMD5 set to true")
|
|
}
|
|
} else if newValue == models.HashAlgorithmOshash {
|
|
missingOSHash, err := qb.CountMissingOSHash()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if missingOSHash > 0 {
|
|
return errors.New("some oshash values are missing on scenes. Run Scan to populate")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|