feat(contract-name): defaults to file name without extension

Rubilmax · Mar 14, 2023 · f7d4931 · f7d4931
1 parent 63df748
commit f7d4931
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -563,13 +563,14 @@ const parseLayout = (content) => {
 };
 exports.parseLayout = parseLayout;
 const parseSource = (contract) => {
-    const [path, contractName] = contract.split(":");
+    let [path, contractName] = contract.split(":");
+    contractName ||= path.replace(/\.sol$/, "");
     const { children, tokens = [] } = parser.parse(fs_1.default.readFileSync(path, { encoding: "utf-8" }), {
         tolerant: true,
         tokens: true,
         loc: true,
     });
-    const def = children.find((child) => child.type === "ContractDefinition" && child.name === contractName);
+    let def = children.find((child) => child.type === "ContractDefinition" && child.name === contractName);
     if (!def)
         throw Error(`Contract definition not found: ${contractName}`);
     return { path, def, tokens };

diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/input.ts b/src/input.ts
@@ -50,17 +50,19 @@ export const parseLayout = (content: string): StorageLayoutReportExact => {
 };
 
 export const parseSource = (contract: string): ParsedSource => {
-  const [path, contractName] = contract.split(":");
+  let [path, contractName] = contract.split(":");
+  contractName ||= path.replace(/\.sol$/, "");
 
   const { children, tokens = [] } = parser.parse(fs.readFileSync(path, { encoding: "utf-8" }), {
     tolerant: true,
     tokens: true,
     loc: true,
   });
 
-  const def = children.find(
+  let def = children.find(
     (child) => child.type === "ContractDefinition" && child.name === contractName
   ) as ContractDefinition | undefined;
+
   if (!def) throw Error(`Contract definition not found: ${contractName}`);
 
   return { path, def, tokens };