Fix disabling phash distance checks

stashapp · Jan 14, 2025 · 5ec5613 · 5ec5613
1 parent c9fc7ec
commit 5ec5613
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/pkg/sqlx/querybuilder_scene.go b/pkg/sqlx/querybuilder_scene.go
@@ -186,10 +186,12 @@ func (qb *sceneQueryBuilder) FindByFullFingerprints(fingerprints []*models.Finge
 		JOIN scene_fingerprints SFP ON SFP.fingerprint_id = FP.id
 	`
 
+	distance := config.GetPHashDistance()
+
 	var phashes []int64
 	var hashes []string
 	for _, fp := range fingerprints {
-		if fp.Algorithm == models.FingerprintAlgorithmPhash {
+		if fp.Algorithm == models.FingerprintAlgorithmPhash && distance > 0 {
 			// Postgres only supports signed integers, so we parse
 			// as uint64 and cast to int64 to ensure values are the same.
 			value, err := strconv.ParseUint(fp.Hash, 16, 64)
@@ -215,7 +217,7 @@ func (qb *sceneQueryBuilder) FindByFullFingerprints(fingerprints []*models.Finge
 	arg := map[string]interface{}{
 		"phashes":  phashes,
 		"hashes":   hashes,
-		"distance": config.GetPHashDistance(),
+		"distance": distance,
 	}
 
 	query := `
@@ -275,10 +277,12 @@ func (qb *sceneQueryBuilder) FindIdsBySceneFingerprints(fingerprints []*models.F
 		GROUP BY scene_id, phash
 	`
 
+	distance := config.GetPHashDistance()
+
 	var phashes []int64
 	var hashes []string
 	for _, fp := range fingerprints {
-		if fp.Algorithm == models.FingerprintAlgorithmPhash {
+		if fp.Algorithm == models.FingerprintAlgorithmPhash && distance > 0 {
 			// Postgres only supports signed integers, so we parse
 			// as uint64 and cast to int64 to ensure values are the same.
 			value, err := strconv.ParseUint(fp.Hash, 16, 64)
@@ -304,7 +308,7 @@ func (qb *sceneQueryBuilder) FindIdsBySceneFingerprints(fingerprints []*models.F
 	arg := map[string]interface{}{
 		"phashes":  phashes,
 		"hashes":   hashes,
-		"distance": config.GetPHashDistance(),
+		"distance": distance,
 	}
 
 	query := strings.Join(clauses, " UNION ")