recon-generate 0.0.13 → 0.0.15

package/dist/link.js

@@ -74,22 +74,53 @@ const parseLibrariesFromOutput = (output, verbose = false) => {
 };
 const runCryticCompile = (cwd, verbose = false) => {
     return new Promise((resolve, reject) => {
-        const cmd = 'crytic-compile . --foundry-compile-all --print-libraries';
+        const cmd = 'crytic-compile';
+        const args = ['.', '--foundry-compile-all', '--print-libraries'];
         if (verbose) {
-            console.log(`[VERBOSE] Running command: ${cmd}`);
+            console.log(`[VERBOSE] Running command: ${cmd} ${args.join(' ')}`);
             console.log(`[VERBOSE] Working directory: ${cwd}`);
         }
-        (0, child_process_1.exec)(cmd, { cwd, env: { ...process.env, PATH: (0, utils_1.getEnvPath)() } }, (error, stdout, stderr) => {
-            if (verbose) {
-                console.log(`[VERBOSE] stdout:\n${stdout}`);
-                console.log(`[VERBOSE] stderr:\n${stderr}`);
+        const child = (0, child_process_1.spawn)(cmd, args, { cwd, env: { ...process.env, PATH: (0, utils_1.getEnvPath)() } });
+        const usesPattern = /^\s+uses:\s+\[(.*?)\]/;
+        let stdoutRemainder = '';
+        let stderrBuffer = '';
+        const relevantLines = [];
+        const handleStdoutChunk = (chunk) => {
+            var _a;
+            const text = stdoutRemainder + chunk.toString('utf8');
+            const lines = text.split(/\r?\n/);
+            stdoutRemainder = (_a = lines.pop()) !== null && _a !== void 0 ? _a : '';
+            for (const line of lines) {
+                const sanitized = (0, utils_1.stripAnsiCodes)(line);
+                if (usesPattern.test(sanitized)) {
+                    relevantLines.push(sanitized);
+                }
             }
-            if (error) {
-                const errorMsg = `crytic-compile failed with exit code ${error.code}\nstderr: ${stderr}\nstdout: ${stdout}\nerror: ${error.message}`;
+        };
+        child.stdout.on('data', handleStdoutChunk);
+        child.stderr.on('data', chunk => {
+            stderrBuffer += chunk.toString('utf8');
+        });
+        child.on('error', err => {
+            reject(new Error(`crytic-compile failed to start: ${err.message}`));
+        });
+        child.on('close', code => {
+            if (stdoutRemainder) {
+                const sanitized = (0, utils_1.stripAnsiCodes)(stdoutRemainder);
+                if (usesPattern.test(sanitized)) {
+                    relevantLines.push(sanitized);
+                }
+            }
+            if (code !== 0) {
+                const truncatedStderr = stderrBuffer.slice(-4000); // avoid flooding the buffer in errors
+                const errorMsg = `crytic-compile failed with exit code ${code}\nstderr (tail): ${truncatedStderr}`;
                 reject(new Error(errorMsg));
                 return;
             }
-            resolve(stdout || '');
+            if (verbose) {
+                console.log(`[VERBOSE] Collected ${relevantLines.length} lines containing 'uses:'`);
+            }
+            resolve(relevantLines.join('\n'));
         });
     });
 };

package/dist/templates/targets/managers-targets.js

@@ -39,10 +39,10 @@ abstract contract ManagersTargets is
     }
 
     /// @dev Deploy a new token and add it to the list of assets, then set it as the current asset
-    function add_new_asset(uint8 decimals) public returns (address) {
-        address newAsset = _newAsset(decimals);
-        return newAsset;
-    }
+    // function add_new_asset(uint8 decimals) public returns (address) {
+    //     address newAsset = _newAsset(decimals);
+    //     return newAsset;
+    // }
 
     /// === GHOST UPDATING HANDLERS ===///
     /// We updateGhosts cause you never know (e.g. donations)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "recon-generate",
-  "version": "0.0.13",
+  "version": "0.0.15",
   "description": "CLI to scaffold Recon fuzzing suite inside Foundry projects",
   "main": "dist/index.js",
   "bin": {

package/dist/templates/wake/conftest.d.ts

@@ -1 +0,0 @@
-export declare const conftestTemplate: HandlebarsTemplateDelegate<any>;

package/dist/templates/wake/conftest.js

@@ -1,18 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.conftestTemplate = void 0;
-const handlebars_1 = __importDefault(require("handlebars"));
-const handlebars_helpers_1 = require("./handlebars_helpers");
-(0, handlebars_helpers_1.registerWakeHelpers)(handlebars_1.default);
-exports.conftestTemplate = handlebars_1.default.compile(`import pytest
-from wake.testing import chain
-
-@pytest.fixture(scope="module", autouse=True)
-def chain_connection():
-    """Connect to a local chain for all tests in this module."""
-    with chain.connect():
-        yield
-`, { noEscape: true });

package/dist/templates/wake/flows.d.ts

@@ -1 +0,0 @@
-export declare const flowsTemplate: HandlebarsTemplateDelegate<any>;

package/dist/templates/wake/flows.js

@@ -1,18 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.flowsTemplate = void 0;
-const handlebars_1 = __importDefault(require("handlebars"));
-const helpers_1 = require("./helpers");
-(0, helpers_1.registerWakeHelpers)(handlebars_1.default);
-exports.flowsTemplate = handlebars_1.default.compile(`
-from wake.testing import *
-from wake.testing.fuzzing import *
-
-class Flows:
-{{#each flows}}
-{{flowDefinition this}}
-{{/each}}
-`, { noEscape: true });

package/dist/templates/wake/fuzz-test.d.ts

@@ -1 +0,0 @@
-export declare const fuzzTestTemplate: HandlebarsTemplateDelegate<any>;

package/dist/templates/wake/fuzz-test.js

@@ -1,62 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.fuzzTestTemplate = void 0;
-const handlebars_1 = __importDefault(require("handlebars"));
-const helpers_1 = require("./helpers");
-(0, helpers_1.registerWakeHelpers)(handlebars_1.default);
-exports.fuzzTestTemplate = handlebars_1.default.compile(`"""
-{{pascal suiteName}} Fuzz Test
-"""
-from wake.testing import *
-from wake.testing.fuzzing import *
-import random
-
-# Import pytypes
-{{#each contracts}}
-from {{module}} import {{name}}
-{{/each}}
-
-from .flows import Flows
-from .invariants import Invariants
-
-class {{pascal suiteName}}FuzzTest(FuzzTest, Flows, Invariants):
-    # Contract instances
-    {{#each contracts}}
-    {{camel name}}: {{name}}
-    {{/each}}
-    
-    # Actors
-    actors: list[Account]
-    admin: Account
-
-    def pre_sequence(self) -> None:
-        # Setup actors
-        self.admin = default_chain.accounts[0]
-        self.actors = [Account.new() for _ in range(3)]
-
-        # Deploy contracts
-        {{#each contracts}}
-        # Deploy {{name}}
-        # Args: {{#each deploy_args}}{{name}}: {{type}}, {{/each}}
-        self.{{camel name}} = {{name}}.deploy(
-            {{#each deploy_args}}{{name}}={{pythonDefault type}}, {{/each}}
-            from_=self.admin
-        )
-        {{/each}}
-
-
-    def _get_actor(self) -> Account:
-        return self.actors[0] if self.actors else self.admin
-
-    @flow()
-    def flow_switch_actor(self, entropy: uint8) -> None:
-        if self.actors:
-            idx = entropy % len(self.actors)
-            self.actors = self.actors[idx:] + self.actors[:idx]
-
-def test_{{snake suiteName}}_fuzz():
-    {{pascal suiteName}}FuzzTest().run(sequences_count=10, flows_count=100)
-`, { noEscape: true });

package/dist/templates/wake/helpers.d.ts

@@ -1,2 +0,0 @@
-import Handlebars from "handlebars";
-export declare function registerWakeHelpers(handlebars: typeof Handlebars): void;

package/dist/templates/wake/helpers.js

@@ -1,81 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.registerWakeHelpers = registerWakeHelpers;
-const case_1 = require("case");
-const types_1 = require("../../types"); // Keeping types for Actor/Mode
-function registerWakeHelpers(handlebars) {
-    handlebars.registerHelper('snake', function (str) {
-        return (0, case_1.snake)(str);
-    });
-    handlebars.registerHelper('camel', function (str) {
-        return (0, case_1.camel)(str);
-    });
-    handlebars.registerHelper('pascal', function (str) {
-        return (0, case_1.pascal)(str);
-    });
-    /**
-     * Python default value based on type string.
-     */
-    function pythonDefault(type) {
-        if (type.includes("Address"))
-            return "Address(0)";
-        if (type.includes("int"))
-            return "0";
-        if (type === "bool")
-            return "False";
-        if (type.includes("bytes"))
-            return 'b""';
-        if (type.includes("list") || type.includes("List"))
-            return "[]";
-        return "None";
-    }
-    handlebars.registerHelper('pythonDefault', pythonDefault);
-    /**
-     * Generate a @flow decorated method definition for Wake FuzzTest.
-     * Expects: { contractName, method: { name, args: [{name, type}] }, actor, mode }
-     */
-    handlebars.registerHelper('flowDefinition', function ({ contractName, method, actor, mode }) {
-        const contractVar = (0, case_1.camel)(contractName);
-        const funcName = method.name;
-        // Build parameter list
-        const params = method.args.map((arg) => {
-            let typeStr = arg.type;
-            // Fix imported types if needed? Usually they are unqualified if imported, or fully qualified.
-            // Pytypes usually gives "ComplexVault.Status" or "Address".
-            return `${arg.name}: ${typeStr}`;
-        });
-        // Add self as first parameter
-        const paramStr = params.length > 0 ? `self, ${params.join(', ')}` : 'self';
-        // Determine sender based on actor type
-        let senderLine;
-        if (actor === types_1.Actor.ADMIN) {
-            senderLine = '        sender = default_chain.accounts[0]  # Admin';
-        }
-        else {
-            senderLine = '        sender = self._get_actor()';
-        }
-        // Build argument list for the contract call
-        const callArgs = method.args.map((arg) => arg.name);
-        const callArgsStr = callArgs.length > 0 ? `${callArgs.join(', ')}, ` : '';
-        // Handle different modes (using existing logic, though pytypes inspection doesn't perfectly enable this yet)
-        let methodBody = `${senderLine}
-        # self.${contractVar}.${funcName}(${callArgsStr}from_=sender)`;
-        // We wrap in try-except by default per "identical suite" robustness usually found in these tools
-        // But user said "user only need to fix the setup and add invariants" implies flows should work or fail?
-        // Let's assume standard basic call.
-        methodBody = `${senderLine}
-        self.${contractVar}.${funcName}(${callArgsStr}from_=sender)`;
-        return `
-    @flow()
-    def flow_${(0, case_1.snake)(contractName)}_${(0, case_1.snake)(funcName)}(${paramStr}) -> None:
-        """Call ${contractName}.${funcName}"""
-${methodBody}
-`;
-    });
-    handlebars.registerHelper('hasItems', function (arr, options) {
-        if (arr && arr.length > 0) {
-            return options.fn(this);
-        }
-        return options.inverse(this);
-    });
-}

package/dist/templates/wake/invariants.d.ts

@@ -1 +0,0 @@
-export declare const invariantsTemplate: HandlebarsTemplateDelegate<any>;

package/dist/templates/wake/invariants.js

@@ -1,21 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.invariantsTemplate = void 0;
-const handlebars_1 = __importDefault(require("handlebars"));
-const helpers_1 = require("./helpers");
-(0, helpers_1.registerWakeHelpers)(handlebars_1.default);
-exports.invariantsTemplate = handlebars_1.default.compile(`
-from wake.testing import *
-from wake.testing.fuzzing import *
-
-class Invariants:
-    @invariant()
-    def invariant_placeholder(self) -> None:
-        """
-        Placeholder invariant - replace with your actual property checks.
-        """
-        pass
-`, { noEscape: true });

package/dist/templates/wake/test_fuzz.d.ts

@@ -1 +0,0 @@
-export declare const fuzzTestTemplate: HandlebarsTemplateDelegate<any>;

package/dist/templates/wake/test_fuzz.js

@@ -1,27 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.fuzzTestTemplate = void 0;
-const handlebars_1 = __importDefault(require("handlebars"));
-const handlebars_helpers_1 = require("./handlebars_helpers");
-(0, handlebars_helpers_1.registerWakeHelpers)(handlebars_1.default);
-exports.fuzzTestTemplate = handlebars_1.default.compile(`"""
-{{pascal suiteName}} Fuzz Test
-"""
-from wake.testing import *
-from wake.testing.fuzzing import *
-
-from .setup import Setup
-from .target_functions import TargetFunctions
-from .properties import Properties
-from .before_after import BeforeAfter
-
-class {{pascal suiteName}}FuzzTest(FuzzTest, Setup, TargetFunctions, Properties, BeforeAfter):
-    def pre_sequence(self) -> None:
-        self.setup()
-
-def test_{{snake suiteName}}_fuzz():
-    {{pascal suiteName}}FuzzTest().run(sequences_count=10, flows_count=100)
-`);