recon-generate 0.0.26 → 0.0.28

package/dist/coverage.js

@@ -85,47 +85,6 @@ const runCmd = (cmd, cwd) => {
         });
     });
 };
-const loadLatestSourceUnits = async (foundryRoot) => {
-    var _a;
-    const outDir = path.join(foundryRoot, '.recon', 'out');
-    const buildInfoDir = path.join(outDir, 'build-info');
-    let files = [];
-    try {
-        const entries = await fs.readdir(buildInfoDir);
-        const jsonFiles = entries.filter((f) => f.endsWith('.json'));
-        files = await Promise.all(jsonFiles.map(async (f) => ({
-            name: f,
-            path: path.join(buildInfoDir, f),
-            mtime: (await fs.stat(path.join(buildInfoDir, f))).mtime,
-        })));
-    }
-    catch (e) {
-        throw new Error(`No build-info directory found at ${buildInfoDir}: ${e}`);
-    }
-    if (files.length === 0) {
-        throw new Error(`No build-info JSON files found in ${buildInfoDir}.`);
-    }
-    const latestFile = files.sort((a, b) => b.mtime.getTime() - a.mtime.getTime())[0].path;
-    const fileContent = await fs.readFile(latestFile, 'utf-8');
-    const buildInfo = JSON.parse(fileContent);
-    const buildOutput = (_a = buildInfo.output) !== null && _a !== void 0 ? _a : buildInfo;
-    if (!buildOutput) {
-        throw new Error(`Build-info file ${latestFile} is missing output data.`);
-    }
-    const filteredAstData = { ...buildOutput };
-    if (filteredAstData.sources) {
-        const validSources = {};
-        for (const [key, content] of Object.entries(filteredAstData.sources)) {
-            const ast = content.ast || content.legacyAST || content.AST;
-            if (ast && (ast.nodeType === 'SourceUnit' || ast.name === 'SourceUnit')) {
-                validSources[key] = content;
-            }
-        }
-        filteredAstData.sources = validSources;
-    }
-    const reader = new solc_typed_ast_1.ASTReader();
-    return reader.read(filteredAstData);
-};
 const shouldIncludeFunction = (fnDef) => {
     if (!fnDef.implemented)
         return false;
@@ -204,7 +163,7 @@ const runCoverage = async (foundryRoot, foundryConfigPath, cryticName, suiteName
     const outDir = path.join(foundryRoot, '.recon', 'out');
     const buildCmd = `forge build --contracts ${cryticName} --build-info --out .recon/out`.replace(/\s+/g, ' ').trim();
     await runCmd(buildCmd, foundryRoot);
-    const sourceUnits = await loadLatestSourceUnits(foundryRoot);
+    const { sourceUnits } = await (0, utils_1.loadLatestBuildInfo)(foundryRoot, '.recon/out');
     if (!sourceUnits || sourceUnits.length === 0) {
         throw new Error('No source units were produced from the Crytic build; cannot generate coverage.');
     }

package/dist/dictionary.d.ts

@@ -17,7 +17,7 @@ interface DictionaryOutput {
 /**
  * Run the dictionary command
  */
-export declare const runDictionary: (buildInfoPath: string, options?: {
+export declare const runDictionary: (foundryRoot: string, options?: {
     outputPath?: string;
     json?: boolean;
     srcPrefix?: string;

package/dist/dictionary.js

@@ -37,32 +37,6 @@ exports.runDictionary = void 0;
 const fs = __importStar(require("fs/promises"));
 const solc_typed_ast_1 = require("solc-typed-ast");
 const utils_1 = require("./utils");
-/**
- * Load build-info from a JSON file
- */
-const loadBuildInfo = async (buildInfoPath) => {
-    var _a;
-    const fileContent = await fs.readFile(buildInfoPath, 'utf-8');
-    const buildInfo = JSON.parse(fileContent);
-    const buildOutput = (_a = buildInfo.output) !== null && _a !== void 0 ? _a : buildInfo;
-    if (!buildOutput) {
-        throw new Error(`Build-info file ${buildInfoPath} is missing output data.`);
-    }
-    const filteredAstData = { ...buildOutput };
-    if (filteredAstData.sources) {
-        const validSources = {};
-        for (const [key, content] of Object.entries(filteredAstData.sources)) {
-            const ast = content.ast || content.legacyAST || content.AST;
-            if (ast && (ast.nodeType === 'SourceUnit' || ast.name === 'SourceUnit')) {
-                validSources[key] = content;
-            }
-        }
-        filteredAstData.sources = validSources;
-    }
-    const reader = new solc_typed_ast_1.ASTReader();
-    const sourceUnits = reader.read(filteredAstData);
-    return { sourceUnits };
-};
 /**
  * Check if a type string represents an address, contract, or interface type
  */
@@ -181,14 +155,14 @@ const findStateVarCasts = (contract, stateVarMap) => {
 /**
  * Run the dictionary command
  */
-const runDictionary = async (buildInfoPath, options = {}) => {
+const runDictionary = async (foundryRoot, options = {}) => {
     var _a;
     const log = options.json ? () => { } : console.log.bind(console);
     const srcPrefix = (_a = options.srcPrefix) !== null && _a !== void 0 ? _a : 'src/';
-    log(`[recon-generate] Loading build-info from ${buildInfoPath}`);
-    const { sourceUnits } = await loadBuildInfo(buildInfoPath);
+    log(`[recon-generate] Loading build-info from ${foundryRoot}`);
+    const { sourceUnits } = await (0, utils_1.loadLatestBuildInfo)(foundryRoot);
     if (!sourceUnits || sourceUnits.length === 0) {
-        throw new Error('No source units were produced from the build-info; cannot generate dictionary.');
+        throw new Error('No source units were produced from the build; cannot generate dictionary.');
     }
     log(`[recon-generate] Found ${sourceUnits.length} source units`);
     const output = {};

package/dist/generator.js

@@ -46,7 +46,6 @@ const templateManager_1 = require("./templateManager");
 const processor_1 = require("./processor");
 const types_1 = require("./types");
 const utils_1 = require("./utils");
-const solc_typed_ast_1 = require("solc-typed-ast");
 const inputParams_1 = require("./inputParams");
 class ReconGenerator {
     constructor(foundryRoot, options) {
@@ -546,7 +545,7 @@ class ReconGenerator {
         return config;
     }
     async run() {
-        var _a, _b, _c, _d, _e;
+        var _a, _b, _c;
         await this.ensureFoundryConfigExists();
         const outPath = this.outDir();
         const mockTargets = (_a = this.options.mocks) !== null && _a !== void 0 ? _a : new Set();
@@ -568,71 +567,24 @@ class ReconGenerator {
             await this.ensureBuild(); // first build to get ABIs
         }
         let sourceUnits = [];
-        const reader = new solc_typed_ast_1.ASTReader();
         try {
-            const buildInfoDir = path.join(this.outDir(), 'build-info');
-            this.logDebug('Scanning build-info directory', { buildInfoDir });
-            const allFiles = await fs.readdir(buildInfoDir);
-            const jsonFiles = allFiles.filter(f => f.endsWith('.json'));
-            const filesWithStats = await Promise.all(jsonFiles.map(async (f) => ({
-                name: f,
-                path: path.join(buildInfoDir, f),
-                mtime: (await fs.stat(path.join(buildInfoDir, f))).mtime
-            })));
-            const files = filesWithStats.sort((a, b) => b.mtime.getTime() - a.mtime.getTime());
-            if (files.length === 0) {
-                console.warn('❌ No build-info JSON files found');
-            }
-            const latestFile = (_b = files[0]) === null || _b === void 0 ? void 0 : _b.path;
-            if (latestFile) {
-                const fileContent = await fs.readFile(latestFile, 'utf-8');
-                const buildInfo = JSON.parse(fileContent);
-                const buildOutput = (_c = buildInfo.output) !== null && _c !== void 0 ? _c : buildInfo;
-                if (buildOutput) {
-                    const filteredAstData = { ...buildOutput };
-                    if (filteredAstData.sources) {
-                        const validSources = {};
-                        // Paths to skip - test and script directories
-                        const skipPatterns = [
-                            'test/',
-                            'tests/',
-                            'script/',
-                            'scripts/',
-                            'contracts/test/',
-                            'contracts/script/',
-                            'contracts/scripts/',
-                            'forge-std/'
-                        ];
-                        for (const [key, content] of Object.entries(filteredAstData.sources)) {
-                            // Skip test and script files
-                            const shouldSkip = skipPatterns.some(pattern => key.includes(pattern));
-                            if (shouldSkip) {
-                                this.logDebug('Skipping source (pattern match)', { key });
-                                continue;
-                            }
-                            const ast = content.ast || content.legacyAST || content.AST;
-                            if (ast && (ast.nodeType === "SourceUnit" || ast.name === "SourceUnit")) {
-                                validSources[key] = content;
-                            }
-                            else {
-                                this.logDebug('Skipping source (no source unit)', { key });
-                            }
-                        }
-                        filteredAstData.sources = validSources;
-                        this.logDebug('Filtered sources for AST read', { kept: Object.keys(validSources).length, keys: Object.keys(validSources) });
-                    }
-                    else {
-                        this.logDebug('No sources present in build output');
-                    }
-                    try {
-                        sourceUnits = reader.read(filteredAstData);
-                    }
-                    catch (e) {
-                    }
-                }
-            }
+            // Paths to skip - test and script directories
+            const skipPatterns = [
+                'test/',
+                'tests/',
+                'script/',
+                'scripts/',
+                'contracts/test/',
+                'contracts/script/',
+                'contracts/scripts/',
+                'forge-std/'
+            ];
+            const result = await (0, utils_1.loadLatestBuildInfo)(this.foundryRoot, '.recon/out', skipPatterns);
+            sourceUnits = result.sourceUnits;
+            this.logDebug('Loaded source units', { count: sourceUnits.length });
         }
         catch (e) {
+            this.logDebug('Failed to load build info', { error: String(e) });
         }
         // Generate mocks if requested, then rebuild to pick them up
         let tempMockSrc = null;
@@ -687,7 +639,7 @@ class ReconGenerator {
                         continue;
                     }
                     const fnSet = new Set();
-                    for (const fn of (_d = cfg.functions) !== null && _d !== void 0 ? _d : []) {
+                    for (const fn of (_b = cfg.functions) !== null && _b !== void 0 ? _b : []) {
                         if (fn === null || fn === void 0 ? void 0 : fn.signature) {
                             fnSet.add(String(fn.signature));
                         }
@@ -723,7 +675,7 @@ class ReconGenerator {
                 this.logDebug('Wrote parameter file', { paramPath });
             }
         }
-        const tm = new templateManager_1.TemplateManager(this.foundryRoot, this.options.suiteDir, this.options.suiteNameSnake, this.options.suiteNamePascal, (_e = this.options.dynamicDeploy) !== null && _e !== void 0 ? _e : new Set());
+        const tm = new templateManager_1.TemplateManager(this.foundryRoot, this.options.suiteDir, this.options.suiteNameSnake, this.options.suiteNamePascal, (_c = this.options.dynamicDeploy) !== null && _c !== void 0 ? _c : new Set());
         await tm.generateTemplates(collected.enabledContracts, collected.adminFunctions, collected.nonSeparatedFunctions, collected.separatedByContract, collected.allContractNames);
     }
     async listAvailable() {

package/dist/index.js

@@ -47,6 +47,8 @@ const link_1 = require("./link");
 const link2_1 = require("./link2");
 const sourcemap_1 = require("./sourcemap");
 const dictionary_1 = require("./dictionary");
+// eslint-disable-next-line @typescript-eslint/no-var-requires
+const packageJson = require('../package.json');
 function parseFilter(input) {
     if (!input)
         return undefined;
@@ -114,6 +116,7 @@ async function main() {
     const program = new commander_1.Command();
     program
         .name('recon-generate')
+        .version(packageJson.version, '-v, --version', 'Output the current version')
         .description('Scaffold Recon fuzzing suite inside a Foundry project')
         .option('-o, --output <path>', 'Custom output folder for tests (recon will be appended if missing)')
         .option('-r, --recon <path>', 'Path to recon.json configuration')
@@ -245,21 +248,22 @@ async function main() {
         });
     });
     program
-        .command('dictionary <buildInfo>')
+        .command('dictionary')
         .description('Extract storage variables with contract/interface types from build-info')
         .option('-o, --output <path>', 'Custom output path for the dictionary JSON file')
         .option('--json', 'Output JSON to terminal (also saves to file if -o is specified)')
         .option('--src-prefix <prefix>', 'Source directory prefix to filter (default: "src/")', 'src/')
-        .action(async function (buildInfoPath) {
+        .option('--foundry-config <path>', 'Path to foundry.toml (defaults to ./foundry.toml)')
+        .action(async function () {
         // @ts-ignore - Commander types are complex
         const opts = this.opts();
-        const resolvedBuildInfo = path.isAbsolute(buildInfoPath)
-            ? buildInfoPath
-            : path.resolve(process.cwd(), buildInfoPath);
+        const workspaceRoot = process.cwd();
+        const foundryConfig = (0, utils_1.getFoundryConfigPath)(workspaceRoot, opts.foundryConfig);
+        const foundryRoot = path.dirname(foundryConfig);
         const outputPath = opts.output
-            ? (path.isAbsolute(opts.output) ? opts.output : path.resolve(process.cwd(), opts.output))
+            ? (path.isAbsolute(opts.output) ? opts.output : path.join(foundryRoot, opts.output))
             : undefined;
-        await (0, dictionary_1.runDictionary)(resolvedBuildInfo, {
+        await (0, dictionary_1.runDictionary)(foundryRoot, {
             outputPath,
             json: !!opts.json,
             srcPrefix: opts.srcPrefix,

package/dist/info.d.ts

@@ -32,6 +32,7 @@ interface InfoOutput {
     with_fallback: string[];
     with_receive: string[];
     coverage_map: Record<string, Record<string, CoverageBlock>>;
+    exclude_from_fuzzing: string[];
 }
 export declare const runInfo: (foundryRoot: string, contractName: string, options?: {
     outputPath?: string;

package/dist/info.js

@@ -40,34 +40,10 @@ const solc_typed_ast_1 = require("solc-typed-ast");
 const utils_1 = require("./utils");
 const z3Solver_1 = require("./z3Solver");
 const call_tree_builder_1 = require("./analyzer/call-tree-builder");
-const loadLatestBuildInfo = async (foundryRoot) => {
-    var _a;
-    const outDir = path.join(foundryRoot, 'out');
-    const buildInfoDir = path.join(outDir, 'build-info');
-    let files = [];
-    try {
-        const entries = await fs.readdir(buildInfoDir);
-        const jsonFiles = entries.filter((f) => f.endsWith('.json'));
-        files = await Promise.all(jsonFiles.map(async (f) => ({
-            name: f,
-            path: path.join(buildInfoDir, f),
-            mtime: (await fs.stat(path.join(buildInfoDir, f))).mtime,
-        })));
-    }
-    catch (e) {
-        throw new Error(`No build-info directory found at ${buildInfoDir}: ${e}`);
-    }
-    if (files.length === 0) {
-        throw new Error(`No build-info JSON files found in ${buildInfoDir}.`);
-    }
-    const latestFile = files.sort((a, b) => b.mtime.getTime() - a.mtime.getTime())[0].path;
-    const fileContent = await fs.readFile(latestFile, 'utf-8');
-    const buildInfo = JSON.parse(fileContent);
-    const buildOutput = (_a = buildInfo.output) !== null && _a !== void 0 ? _a : buildInfo;
-    if (!buildOutput) {
-        throw new Error(`Build-info file ${latestFile} is missing output data.`);
-    }
-    // Extract ABIs from contracts
+/**
+ * Extract ABIs from build output
+ */
+const extractAbis = (buildOutput) => {
     const contractAbis = new Map();
     if (buildOutput.contracts) {
         for (const [_filePath, contracts] of Object.entries(buildOutput.contracts)) {
@@ -78,20 +54,7 @@ const loadLatestBuildInfo = async (foundryRoot) => {
             }
         }
     }
-    const filteredAstData = { ...buildOutput };
-    if (filteredAstData.sources) {
-        const validSources = {};
-        for (const [key, content] of Object.entries(filteredAstData.sources)) {
-            const ast = content.ast || content.legacyAST || content.AST;
-            if (ast && (ast.nodeType === 'SourceUnit' || ast.name === 'SourceUnit')) {
-                validSources[key] = content;
-            }
-        }
-        filteredAstData.sources = validSources;
-    }
-    const reader = new solc_typed_ast_1.ASTReader();
-    const sourceUnits = reader.read(filteredAstData);
-    return { sourceUnits, contractAbis };
+    return contractAbis;
 };
 const stripDataLocation = (raw) => {
     if (!raw)
@@ -907,6 +870,39 @@ const hasReceive = (contract) => {
     const allFunctions = (0, utils_1.getDefinitions)(contract, 'vFunctions', true);
     return allFunctions.some(fn => fn.kind === solc_typed_ast_1.FunctionKind.Receive);
 };
+/**
+ * Check if a function contains a fuzzFromHere() call
+ */
+const hasFuzzFromHereCall = (fnDef) => {
+    for (const call of fnDef.getChildrenByType(solc_typed_ast_1.FunctionCall)) {
+        const expr = call.vExpression;
+        if (expr instanceof solc_typed_ast_1.MemberAccess && expr.memberName === 'fuzzFromHere') {
+            return true;
+        }
+    }
+    return false;
+};
+/**
+ * Collect public/external functions that contain vm.fuzzFromHere() calls
+ * These functions should be excluded from fuzzing
+ */
+const collectExcludeFromFuzzing = (contract) => {
+    const exclude = [];
+    const allFunctions = (0, utils_1.getDefinitions)(contract, 'vFunctions', true).reverse();
+    for (const fnDef of allFunctions) {
+        // Only check public/external functions
+        if (fnDef.visibility !== solc_typed_ast_1.FunctionVisibility.External && fnDef.visibility !== solc_typed_ast_1.FunctionVisibility.Public) {
+            continue;
+        }
+        if (hasFuzzFromHereCall(fnDef)) {
+            // Use function name (not signature) as per the expected output format
+            if (!exclude.includes(fnDef.name)) {
+                exclude.push(fnDef.name);
+            }
+        }
+    }
+    return exclude;
+};
 /**
  * Load source file contents for source location info
  */
@@ -932,10 +928,11 @@ const runInfo = async (foundryRoot, contractName, options = {}) => {
     if (options.json) {
         (0, z3Solver_1.setZ3Silent)(true);
     }
-    const { sourceUnits, contractAbis } = await loadLatestBuildInfo(foundryRoot);
+    const { sourceUnits, buildOutput } = await (0, utils_1.loadLatestBuildInfo)(foundryRoot);
     if (!sourceUnits || sourceUnits.length === 0) {
         throw new Error('No source units were produced from the build; cannot generate info.');
     }
+    const contractAbis = extractAbis(buildOutput);
     // Load source file contents for assert info
     const sourceContents = await loadSourceContents(sourceUnits);
     const allContracts = getAllContracts(sourceUnits);
@@ -956,7 +953,10 @@ const runInfo = async (foundryRoot, contractName, options = {}) => {
         with_fallback: [],
         with_receive: [],
         coverage_map: {},
+        exclude_from_fuzzing: [],
     };
+    // Collect functions to exclude from fuzzing (from target contract)
+    output.exclude_from_fuzzing = collectExcludeFromFuzzing(targetContract);
     // Collect info for all related contracts
     for (const contract of relatedContracts) {
         // Skip abstract contracts (their functions are inherited by concrete contracts)

package/dist/pathsGenerator.js

@@ -61,52 +61,11 @@ const runCmd = (cmd, cwd) => {
         });
     });
 };
-const loadLatestSourceUnits = async (foundryRoot) => {
-    var _a;
-    const outDir = path.join(foundryRoot, '.recon', 'out');
-    const buildInfoDir = path.join(outDir, 'build-info');
-    let files = [];
-    try {
-        const entries = await fs.readdir(buildInfoDir);
-        const jsonFiles = entries.filter((f) => f.endsWith('.json'));
-        files = await Promise.all(jsonFiles.map(async (f) => ({
-            name: f,
-            path: path.join(buildInfoDir, f),
-            mtime: (await fs.stat(path.join(buildInfoDir, f))).mtime,
-        })));
-    }
-    catch (e) {
-        throw new Error(`No build-info directory found at ${buildInfoDir}: ${e}`);
-    }
-    if (files.length === 0) {
-        throw new Error(`No build-info JSON files found in ${buildInfoDir}.`);
-    }
-    const latestFile = files.sort((a, b) => b.mtime.getTime() - a.mtime.getTime())[0].path;
-    const fileContent = await fs.readFile(latestFile, 'utf-8');
-    const buildInfo = JSON.parse(fileContent);
-    const buildOutput = (_a = buildInfo.output) !== null && _a !== void 0 ? _a : buildInfo;
-    if (!buildOutput) {
-        throw new Error(`Build-info file ${latestFile} is missing output data.`);
-    }
-    const filteredAstData = { ...buildOutput };
-    if (filteredAstData.sources) {
-        const validSources = {};
-        for (const [key, content] of Object.entries(filteredAstData.sources)) {
-            const ast = content.ast || content.legacyAST || content.AST;
-            if (ast && (ast.nodeType === 'SourceUnit' || ast.name === 'SourceUnit')) {
-                validSources[key] = content;
-            }
-        }
-        filteredAstData.sources = validSources;
-    }
-    const reader = new solc_typed_ast_1.ASTReader();
-    return reader.read(filteredAstData);
-};
 // ==================== Main Entry ====================
 const runPaths = async (foundryRoot, cryticName, suiteNameSnake, options = {}) => {
     const buildCmd = `forge build --contracts ${cryticName} --build-info --out .recon/out`.replace(/\s+/g, ' ').trim();
     await runCmd(buildCmd, foundryRoot);
-    const sourceUnits = await loadLatestSourceUnits(foundryRoot);
+    const { sourceUnits } = await (0, utils_1.loadLatestBuildInfo)(foundryRoot, '.recon/out');
     if (!sourceUnits || sourceUnits.length === 0) {
         throw new Error('No source units were produced from the Crytic build; cannot generate paths.');
     }

package/dist/utils.d.ts

@@ -1,4 +1,4 @@
-import { ASTNode, ContractDefinition, Assignment, FunctionCall, FunctionDefinition, VariableDeclaration, EventDefinition, ErrorDefinition } from 'solc-typed-ast';
+import { ASTNode, ContractDefinition, Assignment, FunctionCall, FunctionDefinition, SourceUnit, VariableDeclaration, EventDefinition, ErrorDefinition } from 'solc-typed-ast';
 import { CallType } from './types';
 export declare function fileExists(p: string): Promise<boolean>;
 export declare function getFoundryConfigPath(workspaceRoot: string, override?: string): string;
@@ -39,3 +39,36 @@ export declare function forwardLinearization(root: ContractDefinition): Contract
  * contract. `current` is the contract whose code contains the `super` call.
  */
 export declare function resolveSuper(root: ContractDefinition, current: ContractDefinition, signature: string): FunctionDefinition | undefined;
+/**
+ * Result from loading build info
+ */
+export interface BuildInfoResult {
+    sourceUnits: SourceUnit[];
+    buildOutput: any;
+}
+/**
+ * Parse build output and return source units.
+ * This is the core parsing logic used by both loadLatestBuildInfo and loadBuildInfoFromFile.
+ *
+ * @param buildOutput - The raw build output object
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Array of SourceUnit objects
+ */
+export declare const parseBuildOutput: (buildOutput: any, skipPatterns?: string[]) => SourceUnit[];
+/**
+ * Load build info from a specific file path.
+ *
+ * @param buildInfoPath - Path to the build-info JSON file
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Object containing sourceUnits and raw buildOutput
+ */
+export declare const loadBuildInfoFromFile: (buildInfoPath: string, skipPatterns?: string[]) => Promise<BuildInfoResult>;
+/**
+ * Load the latest build info from a Foundry project.
+ *
+ * @param foundryRoot - The root directory of the Foundry project
+ * @param outputDir - The output directory relative to foundryRoot (default: 'out')
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Object containing sourceUnits and raw buildOutput
+ */
+export declare const loadLatestBuildInfo: (foundryRoot: string, outputDir?: string, skipPatterns?: string[]) => Promise<BuildInfoResult>;

package/dist/utils.js

@@ -36,7 +36,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.signatureEquals = exports.getFunctionName = exports.getIndex = exports.getSource = exports.getLines = void 0;
+exports.loadLatestBuildInfo = exports.loadBuildInfoFromFile = exports.parseBuildOutput = exports.signatureEquals = exports.getFunctionName = exports.getIndex = exports.getSource = exports.getLines = void 0;
 exports.fileExists = fileExists;
 exports.getFoundryConfigPath = getFoundryConfigPath;
 exports.findOutputDirectory = findOutputDirectory;
@@ -436,3 +436,84 @@ function resolveSuper(root, current, signature) {
     }
     return undefined;
 }
+/**
+ * Parse build output and return source units.
+ * This is the core parsing logic used by both loadLatestBuildInfo and loadBuildInfoFromFile.
+ *
+ * @param buildOutput - The raw build output object
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Array of SourceUnit objects
+ */
+const parseBuildOutput = (buildOutput, skipPatterns = []) => {
+    const filteredAstData = { ...buildOutput };
+    if (filteredAstData.sources) {
+        const validSources = {};
+        for (const [key, content] of Object.entries(filteredAstData.sources)) {
+            // Skip sources matching any of the skip patterns
+            if (skipPatterns.length > 0) {
+                const shouldSkip = skipPatterns.some(pattern => key.includes(pattern));
+                if (shouldSkip) {
+                    continue;
+                }
+            }
+            const ast = content.ast || content.legacyAST || content.AST;
+            if (ast && (ast.nodeType === 'SourceUnit' || ast.name === 'SourceUnit')) {
+                validSources[key] = content;
+            }
+        }
+        filteredAstData.sources = validSources;
+    }
+    const reader = new solc_typed_ast_1.ASTReader();
+    return reader.read(filteredAstData);
+};
+exports.parseBuildOutput = parseBuildOutput;
+/**
+ * Load build info from a specific file path.
+ *
+ * @param buildInfoPath - Path to the build-info JSON file
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Object containing sourceUnits and raw buildOutput
+ */
+const loadBuildInfoFromFile = async (buildInfoPath, skipPatterns = []) => {
+    var _a;
+    const fileContent = await fs.readFile(buildInfoPath, 'utf-8');
+    const buildInfo = JSON.parse(fileContent);
+    const buildOutput = (_a = buildInfo.output) !== null && _a !== void 0 ? _a : buildInfo;
+    if (!buildOutput) {
+        throw new Error(`Build-info file ${buildInfoPath} is missing output data.`);
+    }
+    const sourceUnits = (0, exports.parseBuildOutput)(buildOutput, skipPatterns);
+    return { sourceUnits, buildOutput };
+};
+exports.loadBuildInfoFromFile = loadBuildInfoFromFile;
+/**
+ * Load the latest build info from a Foundry project.
+ *
+ * @param foundryRoot - The root directory of the Foundry project
+ * @param outputDir - The output directory relative to foundryRoot (default: 'out')
+ * @param skipPatterns - Optional array of patterns to skip (e.g., ['test/', 'script/'])
+ * @returns Object containing sourceUnits and raw buildOutput
+ */
+const loadLatestBuildInfo = async (foundryRoot, outputDir = 'out', skipPatterns = []) => {
+    const outDir = path.join(foundryRoot, outputDir);
+    const buildInfoDir = path.join(outDir, 'build-info');
+    let files = [];
+    try {
+        const entries = await fs.readdir(buildInfoDir);
+        const jsonFiles = entries.filter((f) => f.endsWith('.json'));
+        files = await Promise.all(jsonFiles.map(async (f) => ({
+            name: f,
+            path: path.join(buildInfoDir, f),
+            mtime: (await fs.stat(path.join(buildInfoDir, f))).mtime,
+        })));
+    }
+    catch (e) {
+        throw new Error(`No build-info directory found at ${buildInfoDir}: ${e}`);
+    }
+    if (files.length === 0) {
+        throw new Error(`No build-info JSON files found in ${buildInfoDir}.`);
+    }
+    const latestFile = files.sort((a, b) => b.mtime.getTime() - a.mtime.getTime())[0].path;
+    return (0, exports.loadBuildInfoFromFile)(latestFile, skipPatterns);
+};
+exports.loadLatestBuildInfo = loadLatestBuildInfo;

package/package.json

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