update TruffleProject#fileBelongs

NomicFoundation · Apr 13, 2023 · c30b9bb · c30b9bb
1 parent 2ac6291
commit c30b9bb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/client/package.json b/client/package.json
@@ -42,7 +42,6 @@
     "workspaceContains:**/hardhat.config.{ts,js}",
     "workspaceContains:**/foundry.toml",
     "workspaceContains:**/truffle-config.js"
-
   ],
   "scripts": {
     "build": "tsc -b",

diff --git a/server/src/frameworks/Truffle/TruffleProject.ts b/server/src/frameworks/Truffle/TruffleProject.ts
@@ -13,7 +13,7 @@ import { isRelativeImport } from "../../utils";
 import { directoryContains } from "../../utils/directoryContains";
 import { CompilationDetails } from "../base/CompilationDetails";
 import { BuildInputError } from "../base/Errors";
-import { Project } from "../base/Project";
+import { FileBelongsResult, Project } from "../base/Project";
 import { Remapping } from "../base/Remapping";
 import { getDependenciesAndPragmas } from "../shared/crawlDependencies";
 
@@ -96,15 +96,21 @@ export class TruffleProject extends Project {
     }
   }
 
-  public async fileBelongs(file: string): Promise<boolean> {
+  public async fileBelongs(file: string): Promise<FileBelongsResult> {
+    let belongs: boolean;
     if (this.status === Status.INITIALIZED_SUCCESS) {
-      return [this.sourcesPath, this.testsPath].some((dir) =>
+      belongs = [this.sourcesPath, this.testsPath].some((dir) =>
         directoryContains(dir, path.dirname(file))
       );
     } else {
       // Claim ownership under all contracts on the base folder if initialization failed
-      return directoryContains(this.basePath, file);
+      belongs = directoryContains(this.basePath, file);
     }
+
+    return {
+      belongs,
+      isLocal: true,
+    };
   }
 
   public async resolveImportPath(