fix: merge internal functions templates

src/context.ts

@@ -46,8 +46,8 @@ export function internalFunctionContext(node: FunctionDefinition): InternalFunct
   }
 
   // Check if the function is view or pure
-  const isView = node.stateMutability === FunctionStateMutability.View;
   const isPure = node.stateMutability === FunctionStateMutability.Pure;
+  const isView = node.stateMutability === FunctionStateMutability.View || isPure;
 
   // Save the internal function information
   return {

src/templates/partials/internal-function.hbs

@@ -6,8 +6,8 @@ function mock_call_{{functionName}}({{parameters}}) public {
   );
 }
 
-function {{functionName}}({{inputs}}) internal override {{#if outputs}}returns ({{outputs}}){{/if}} {
-  (bool _success, bytes memory _data) = address(this).call(abi.encodeWithSignature('{{signature}}'{{#if inputs}}, {{#each inputNames}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}{{/if}}));
+function {{functionName}}{{#if isPure}}Helper{{/if}}({{inputs}}) internal {{#if isView}}view{{/if}} {{#unless isPure}}override{{/unless}} {{#if outputs}}returns ({{outputs}}){{/if}} {
+  (bool _success, bytes memory _data) = address(this).{{#if isView}}static{{/if}}call(abi.encodeWithSignature('{{signature}}'{{#if inputs}}, {{#each inputNames}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}{{/if}}));
 
   if (_success) return abi.decode(_data, ({{#each outputTypes}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}));
 
@@ -23,3 +23,21 @@ function call_{{functionName}}({{inputs}}) public {{#if outputs}}returns ({{outp
 function expectCall_{{functionName}}({{#if inputs}}{{inputs}}{{/if}}) public {
   vm.expectCall(address(this), abi.encodeWithSignature('{{signature}}'{{#if inputs}}, {{#each inputNames}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}{{/if}}));
 }
+
+{{#if isPure}}
+
+  function _{{functionName}}CastToPure(function({{#each inputTypes}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}) internal view {{#if outputs}}returns ({{explicitOutputTypes}}){{/if}} fnIn) 
+    internal 
+    pure 
+    returns (function({{#each inputTypes}}{{this}}{{#unless @last}}, {{/unless}}{{/each}}) internal pure {{#if outputs}}returns ({{explicitOutputTypes}}){{/if}} fnOut)
+  {
+    assembly {
+      fnOut := fnIn
+    }
+  }
+
+  function {{functionName}}({{inputs}}) internal pure override {{#if outputs}}returns ({{outputs}}){{/if}} {
+    return _{{functionName}}CastToPure({{functionName}}Helper)({{#each inputNames}}{{this}}{{#unless @last}}, {{/unless}}{{/each}});
+  }
+
+{{/if}}

src/utils.ts

@@ -74,7 +74,7 @@
  * @param templatePath The path of the template (if it's nested)
  * @returns The compiled template
  */
 export function compileTemplate(templateName: string, templatePath?: string[]): HandlebarsTemplateDelegate<any> {
   const templateContent = readTemplate(templateName, templatePath);
   return Handlebars.compile(templateContent, { noEscape: true });
 }
@@ -83,7 +83,7 @@
  * Gets the base contract template
  * @returns The contract template
  */
 export function getContractTemplate(): HandlebarsTemplateDelegate<any> {
   return compileTemplate('contract-template');
 }
 
@@ -91,7 +91,7 @@
  * Gets the smock helper template
  * @returns The helper template
  */
 export function getSmockHelperTemplate(): HandlebarsTemplateDelegate<any> {
   return compileTemplate('helper-template');
 }
 
@@ -192,7 +192,6 @@
     constructor: constructorContext,
     'external-or-public-function': externalOrPublicFunctionContext,
     'internal-function': internalFunctionContext,
-    'internal-view-function': internalFunctionContext,
     import: importContext,
   };
 
@@ -222,11 +221,7 @@
     } else if (node.visibility === 'external' || node.visibility === 'public') {
       return 'external-or-public-function';
     } else if (node.visibility === 'internal' && node.virtual) {
-      if (node.stateMutability === 'view' || node.stateMutability === 'pure') {
-        return 'internal-view-function';
-      } else {
-        return 'internal-function';
-      }
+      return 'internal-function';
     }
   } else if (node instanceof ImportDirective) {
     return 'import';
@@ -269,7 +264,7 @@
   const regex = /remappings[\s|\n]*=[\s\n]*\[(?<remappings>[^\]]+)]/;
   const matches = foundryConfigContent.match(regex);
   if (matches) {
     return matches
       .groups!.remappings.split(',')
       .map((line) => line.trim())
       .map((line) => line.replace(/["']/g, ''))

test/unit/context/internalFunctionContext.spec.ts

@@ -198,9 +198,10 @@ describe('internalFunctionContext', () => {
   it('determines whether the function is view or not', () => {
     for (const stateMutability in FunctionStateMutability) {
       const isView = FunctionStateMutability[stateMutability] === FunctionStateMutability.View;
+      const isPure = FunctionStateMutability[stateMutability] === FunctionStateMutability.Pure;
       const node = mockFunctionDefinition({ ...defaultAttributes, stateMutability: FunctionStateMutability[stateMutability] });
       const context = internalFunctionContext(node);
-      expect(context.isView).to.be.equal(isView);
+      expect(context.isView).to.be.equal(isView || isPure);
     }
   });