feat: license mock contracts

solidity/contracts/ContractTest.sol

@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.8.0;
 
 import {IContractTest, MyContract} from '../interfaces/IContractTest.sol';

solidity/contracts/Lib.sol

@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.8.0;
 
 library Lib {

solidity/interfaces/IContractTest.sol

@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.8.0;
 
 contract MyContract {}

src/smock-foundry.ts

@@ -80,7 +80,10 @@ export const generateMockContracts = async (
       // Check if the abi and ast exist
       if (!ast) return;
 
-      // Get the absolute path from the ast
+      // Get the license from the ast
+      const license: string = ast.license;
+
+      // Get the absolute path
       const contractImport: string = ast.absolutePath;
       if (!contractImport) return;
 
@@ -97,10 +100,11 @@ export const generateMockContracts = async (
       if (!contractNode || contractNode.contractKind === 'library') return;
 
       const functions: StateVariablesOptions = getStateVariables(contractNode);
-      // All data which will be use for create the template
 
+      // All data which will be used for creating the template
       const data = {
         contractName: contractName,
+        license: license,
         contractImport: contractImport,
         exportedSymbols: exportedSymbols.join(', '),
         import: getImports(ast),

src/templates/mockContractTemplate.hbs

@@ -1,4 +1,4 @@
-/// SPDX-License-Identifier: UNLICENSED
+// SPDX-License-Identifier: {{#if license}}{{license}}{{else}}UNLICENSED{{/if}}
 pragma solidity ^0.8.0;
 
 import {Test} from 'forge-std/Test.sol';

src/types.ts

@@ -51,12 +51,12 @@ export interface FunctionDefinitionNode {
 }
 
 export interface Ast {
-  absolutePath: string;
+  nodeType: 'SourceUnit';
   id?: number;
-  nodeType: string;
   src: string;
   nodes: AstNode[];
   license: string;
+  absolutePath: string;
   exportedSymbols: { [key: string]: number[] };
 }
 

test/unit/get-imports.spec.ts

@@ -15,6 +15,7 @@ describe('getImports', () => {
       exportedSymbols: {},
     };
   });
+
   it('should return an empty array if there are no import directives', async () => {
     const importStatements = getImports(ast);
     expect(importStatements).to.be.an('array').that.is.empty;