@goshenkata/dryscan-core 1.0.10 → 1.0.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goshenkata/dryscan-core",
-  "version": "1.0.10",
+  "version": "1.0.11",
   "description": "Core library for DryScan - semantic code duplication analyzer",
   "type": "module",
   "main": "./dist/index.js",
@@ -35,7 +35,6 @@
     "access": "public"
   },
   "devDependencies": {
-    "@types/better-sqlite3": "^7.6.13",
     "@types/debug": "^4.1.12",
     "@types/node": "^25.0.3",
     "chai": "^6.2.2",
@@ -49,7 +48,6 @@
     "@langchain/core": "^1.1.8",
     "@langchain/google-genai": "^2.1.3",
     "@langchain/ollama": "^1.1.0",
-    "better-sqlite3": "^12.5.0",
     "debug": "^4.4.3",
     "glob-gitignore": "^1.0.15",
     "ignore": "^7.0.5",
@@ -58,9 +56,9 @@
     "minimatch": "^10.1.1",
     "reflect-metadata": "^0.2.2",
     "short-uuid": "^6.0.3",
+    "sqlite3": "^5.1.7",
     "tree-sitter": "^0.25.0",
     "tree-sitter-java": "^0.23.5",
-    "tree-sitter-python": "^0.25.0",
     "typeorm": "^0.3.28",
     "upath": "^2.0.1"
   }

package/src/db/DryScanDatabase.ts

@@ -19,7 +19,7 @@ export class DryScanDatabase {
     await fs.mkdir(upath.dirname(dbPath), { recursive: true });
 
     this.dataSource = new DataSource({
-      type: "better-sqlite3",
+      type: "sqlite",
       database: dbPath,
       entities: [IndexUnitEntity, FileEntity],
       synchronize: true,

package/src/services/DuplicateService.ts

@@ -149,19 +149,32 @@ export class DuplicateService {
     }
 
     if (left.unitType === IndexUnitType.FUNCTION) {
-      const parentClassSimilarity = this.parentSimilarity(left, right, IndexUnitType.CLASS);
       const weights = indexConfig.weights.function;
-      return (weights.self * selfSimilarity) + (weights.parentClass * parentClassSimilarity);
+      const hasParentClass = !!this.findParentOfType(left, IndexUnitType.CLASS) && !!this.findParentOfType(right, IndexUnitType.CLASS);
+      const parentClassSimilarity = hasParentClass ? this.parentSimilarity(left, right, IndexUnitType.CLASS) : 0;
+
+      // Re-normalize weights when parent context is missing, so standalone units aren't penalized.
+      const totalWeight = weights.self + (hasParentClass ? weights.parentClass : 0);
+      return ((weights.self * selfSimilarity) + (hasParentClass ? (weights.parentClass * parentClassSimilarity) : 0)) / totalWeight;
     }
 
     const weights = indexConfig.weights.block;
-    const parentFuncSim = this.parentSimilarity(left, right, IndexUnitType.FUNCTION);
-    const parentClassSim = this.parentSimilarity(left, right, IndexUnitType.CLASS);
+    const hasParentFunction = !!this.findParentOfType(left, IndexUnitType.FUNCTION) && !!this.findParentOfType(right, IndexUnitType.FUNCTION);
+    const hasParentClass = !!this.findParentOfType(left, IndexUnitType.CLASS) && !!this.findParentOfType(right, IndexUnitType.CLASS);
+    const parentFuncSim = hasParentFunction ? this.parentSimilarity(left, right, IndexUnitType.FUNCTION) : 0;
+    const parentClassSim = hasParentClass ? this.parentSimilarity(left, right, IndexUnitType.CLASS) : 0;
+
+    // Re-normalize weights when some parent context is missing.
+    const totalWeight =
+      weights.self +
+      (hasParentFunction ? weights.parentFunction : 0) +
+      (hasParentClass ? weights.parentClass : 0);
+
     return (
-      weights.self * selfSimilarity +
-      weights.parentFunction * parentFuncSim +
-      weights.parentClass * parentClassSim
-    );
+      (weights.self * selfSimilarity) +
+      (hasParentFunction ? (weights.parentFunction * parentFuncSim) : 0) +
+      (hasParentClass ? (weights.parentClass * parentClassSim) : 0)
+    ) / totalWeight;
   }
 
   private parentSimilarity(left: IndexUnit, right: IndexUnit, targetType: IndexUnitType): number {