@reinforcedai/hardhat-security-review 2511.26.1 → 2512.8.1

package/README.md

@@ -0,0 +1,33 @@
+# Reinforced Hardhat Plugin
+
+This plugin integrates Reinforced-powered smart contract auditing into the Hardhat workflow. It enables automatic and manual security audits of Solidity contracts before deployment, leveraging decentralized intelligence for vulnerability detection and suggestions. The plugin provides tasks for auditing all contracts and formatting results for easy review.
+
+## Features
+
+- Automatic audit after compilation
+- Manual audit task (`npx hardhat audit`)
+- Integration with Reinforced API for contract analysis
+- Tabular vulnerability reports
+
+## Usage
+
+1. Install the plugin and configure your Reinforced API key in `hardhat.config.js`.
+2. Run `npx hardhat audit` to manually audit contracts.
+3. Audits are also performed automatically before deployment.
+
+## Configuration
+
+Add the following to your `hardhat.config.js` to enable the plugin:
+
+```js
+module.exports = {
+  // ...existing config...
+  reinforced: {
+    compilationHookEnabled: true, // Set to true to enable auditing
+    apiKey: "YOUR_REINFORCED_API_KEY" // Your Reinforced API key
+  }
+};
+```
+
+- `compilationHookEnabled` (boolean): Enables or disables automatic Reinforced auditing before deployment. Manual audit task always work regardless of this flag.
+- `apiKey` (string): Your API key for accessing Reinforced's audit service.

package/dist/AutoFixManager.d.ts

@@ -0,0 +1,8 @@
+import { AutoFixSelection, CodeReplacementResult, Vulnerability } from './types';
+export declare class AutoFixManager {
+    constructor();
+    promptUserForVulnerabilitySelection(vulnerabilities: Vulnerability[]): AutoFixSelection[];
+    applyAutoFixes(vulnerabilities: Vulnerability[], selections: AutoFixSelection[], sourceCode: string, sourcePaths: string[]): CodeReplacementResult;
+    displayAutoFixSummary(result: CodeReplacementResult, vulnerabilities: Vulnerability[]): void;
+}
+//# sourceMappingURL=AutoFixManager.d.ts.map

package/dist/AutoFixManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AutoFixManager.d.ts","sourceRoot":"","sources":["../src/AutoFixManager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,aAAa,EAA0B,MAAM,SAAS,CAAC;AAEzG,qBAAa,cAAc;;IAKlB,mCAAmC,CAAC,eAAe,EAAE,aAAa,EAAE,GAAG,gBAAgB,EAAE;IA+DzF,cAAc,CACnB,eAAe,EAAE,aAAa,EAAE,EAChC,UAAU,EAAE,gBAAgB,EAAE,EAC9B,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,MAAM,EAAE,GACpB,qBAAqB;IA2FjB,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,EAAE,eAAe,EAAE,aAAa,EAAE;CAsC7F"}

package/dist/AutoFixManager.js

@@ -0,0 +1,199 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AutoFixManager = void 0;
+const path = __importStar(require("path"));
+const readlineSync = __importStar(require("readline-sync"));
+const ContractUnflattener_1 = require("./ContractUnflattener");
+class AutoFixManager {
+    constructor() {
+    }
+    promptUserForVulnerabilitySelection(vulnerabilities) {
+        if (vulnerabilities.length === 0) {
+            return [];
+        }
+        console.log('\n🔍 Found vulnerabilities that can be auto-fixed:');
+        console.log('━'.repeat(80));
+        const vulnerabilitiesWithIndex = vulnerabilities
+            .map((vuln, index) => ({ ...vuln, index }))
+            .filter(vuln => vuln.fixed_lines !== null && vuln.fixed_lines.trim().length > 0);
+        if (vulnerabilitiesWithIndex.length === 0) {
+            console.log('ℹ️  No auto-fixable vulnerabilities found (no miner suggestions available).');
+            return [];
+        }
+        vulnerabilitiesWithIndex.forEach((vuln) => {
+            console.log(`\n[${vuln.index + 1}] ${vuln.vulnerability_class}`);
+            console.log(`    Lines ${vuln.original_from_line || vuln.from_line}-${vuln.original_to_line || vuln.to_line}${vuln.original_file ? ` in ${vuln.original_file.split('/').pop()}` : ''}`);
+            console.log(`    Miners consensus: ${vuln.miners_select_count}/${vuln.miners_participated_count}`);
+            console.log(`    Description: ${vuln.description.substring(0, 100)}...`);
+            console.log(`    Has fix: ${vuln.fixed_lines ? 'Yes' : 'No'}`);
+        });
+        console.log('━'.repeat(80));
+        console.log('Select vulnerabilities to auto-fix:');
+        console.log('• Enter numbers separated by commas (e.g., 1,3,5)');
+        console.log('• Enter "all" to fix all vulnerabilities');
+        console.log('• Enter "none" or press Enter to skip auto-fixing');
+        const userInput = readlineSync.question('\nYour selection: ').trim();
+        if (userInput.toLowerCase() === 'none' || userInput === '') {
+            return [];
+        }
+        let selectedIndices = [];
+        if (userInput.toLowerCase() === 'all') {
+            selectedIndices = vulnerabilitiesWithIndex.map(v => v.index);
+        }
+        else {
+            const inputNumbers = userInput.split(',')
+                .map(s => parseInt(s.trim()))
+                .filter(n => !isNaN(n) && n >= 1 && n <= vulnerabilities.length);
+            selectedIndices = inputNumbers.map(n => n - 1);
+        }
+        const selections = vulnerabilities.map((_, index) => ({
+            vulnerabilityIndex: index,
+            shouldFix: selectedIndices.includes(index) &&
+                vulnerabilities[index].fixed_lines !== null &&
+                vulnerabilities[index].fixed_lines.trim().length > 0,
+        }));
+        const selectedCount = selections.filter(s => s.shouldFix).length;
+        console.log(`\n✅ Selected ${selectedCount} vulnerabilities for auto-fixing.`);
+        return selections;
+    }
+    applyAutoFixes(vulnerabilities, selections, sourceCode, sourcePaths) {
+        const selectedVulns = selections
+            .filter(s => s.shouldFix)
+            .map(s => ({ ...vulnerabilities[s.vulnerabilityIndex], index: s.vulnerabilityIndex }))
+            .sort((a, b) => b.to_line - a.to_line); // Sort by line number descending to avoid offset issues
+        if (selectedVulns.length === 0) {
+            return {
+                success: true,
+                appliedFixes: [],
+            };
+        }
+        try {
+            const unflattener = new ContractUnflattener_1.ContractUnflattener();
+            unflattener.parseFlattened(sourceCode, sourcePaths);
+            const vulnerabilitiesByFile = new Map();
+            for (const vuln of selectedVulns) {
+                const mapping = unflattener.mapVulnerabilityToOriginal(vuln.from_line, vuln.to_line);
+                if (mapping) {
+                    const filePath = mapping.originalFile;
+                    if (!vulnerabilitiesByFile.has(filePath)) {
+                        vulnerabilitiesByFile.set(filePath, []);
+                    }
+                    vulnerabilitiesByFile.get(filePath).push({ vulnerability: vuln, mapping });
+                    vuln.original_file = filePath;
+                    vuln.original_from_line = mapping.originalFromLine;
+                    vuln.original_to_line = mapping.originalToLine;
+                }
+            }
+            if (vulnerabilitiesByFile.size === 0) {
+                return {
+                    success: false,
+                    error: 'Could not map vulnerabilities to original files',
+                    appliedFixes: [],
+                };
+            }
+            const appliedFixes = [];
+            const modifiedFiles = [];
+            console.log('\n🔧 Applying auto-fixes to original files...');
+            for (const [filePath, vulnMappings] of vulnerabilitiesByFile.entries()) {
+                vulnMappings.sort((a, b) => b.mapping.originalToLine - a.mapping.originalToLine);
+                console.log(`  📄 Processing ${path.basename(filePath)} (${vulnMappings.length} fixes)`);
+                for (const { vulnerability: vuln, mapping } of vulnMappings) {
+                    if (!vuln.fixed_lines)
+                        continue;
+                    console.log(`    • Fixing ${vuln.vulnerability_class} at lines ${mapping.originalFromLine}-${mapping.originalToLine}`);
+                    const fixResult = unflattener.applyFixToOriginalFile(mapping, vuln.fixed_lines);
+                    if (fixResult.success) {
+                        appliedFixes.push(vuln.index);
+                        if (!modifiedFiles.includes(filePath)) {
+                            modifiedFiles.push(filePath);
+                        }
+                    }
+                    else {
+                        console.warn(`    ⚠️  Failed to apply fix: ${fixResult.error}`);
+                    }
+                }
+            }
+            console.log(`✅ Applied ${appliedFixes.length} fixes to ${modifiedFiles.length} files`);
+            return {
+                success: true,
+                appliedFixes,
+                modifiedFiles,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+                appliedFixes: [],
+            };
+        }
+    }
+    displayAutoFixSummary(result, vulnerabilities) {
+        console.log('\n' + '='.repeat(80));
+        console.log('AUTO-FIX SUMMARY');
+        console.log('='.repeat(80));
+        if (!result.success) {
+            console.log(`❌ Auto-fix failed: ${result.error}`);
+            return;
+        }
+        if (result.appliedFixes.length === 0) {
+            console.log('ℹ️  No fixes were applied.');
+            return;
+        }
+        console.log(`✅ Successfully applied ${result.appliedFixes.length} auto-fixes:`);
+        result.appliedFixes.forEach(index => {
+            const vuln = vulnerabilities[index];
+            const originalInfo = vuln.original_file && vuln.original_from_line && vuln.original_to_line
+                ? ` in ${path.basename(vuln.original_file)} (lines ${vuln.original_from_line}-${vuln.original_to_line})`
+                : ` (lines ${vuln.from_line}-${vuln.to_line})`;
+            console.log(`  • ${vuln.vulnerability_class}${originalInfo}`);
+        });
+        if (result.modifiedFiles && result.modifiedFiles.length > 0) {
+            console.log('\n📄 Modified files:');
+            result.modifiedFiles.forEach(filePath => {
+                console.log(`  • ${path.basename(filePath)}`);
+            });
+        }
+        else if (result.filePath) {
+            console.log(`\n📄 Fixed code saved to: ${path.basename(result.filePath)}`);
+        }
+        console.log('\n⚠️  IMPORTANT: Please review the changes carefully before deploying!');
+        console.log('   The fixes have been applied directly to your original contract files.');
+        console.log('='.repeat(80));
+    }
+}
+exports.AutoFixManager = AutoFixManager;
+//# sourceMappingURL=AutoFixManager.js.map

package/dist/AutoFixManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AutoFixManager.js","sourceRoot":"","sources":["../src/AutoFixManager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAA6B;AAC7B,4DAA8C;AAC9C,+DAAkF;AAGlF,MAAa,cAAc;IAEzB;IACA,CAAC;IAEM,mCAAmC,CAAC,eAAgC;QACzE,IAAG,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;QAClE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,MAAM,wBAAwB,GAA6B,eAAe;aACvE,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAC,GAAG,IAAI,EAAE,KAAK,EAAC,CAAC,CAAC;aACxC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEnF,IAAG,wBAAwB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,CAAC,GAAG,CAAC,6EAA6E,CAAC,CAAC;YAC3F,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,wBAAwB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACxC,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,KAAK,GAAG,CAAC,KAAK,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;YAEjE,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACxL,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,yBAAyB,EAAE,CAAC,CAAC;YACnG,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;QACnD,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;QAEjE,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,IAAI,EAAE,CAAC;QAErE,IAAG,SAAS,CAAC,WAAW,EAAE,KAAK,MAAM,IAAI,SAAS,KAAK,EAAE,EAAE,CAAC;YAC1D,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,eAAe,GAAa,EAAE,CAAC;QAEnC,IAAG,SAAS,CAAC,WAAW,EAAE,KAAK,KAAK,EAAE,CAAC;YACrC,eAAe,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAC/D,CAAC;aAAM,CAAC;YACN,MAAM,YAAY,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC;iBACtC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;iBAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;YAEnE,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GAAuB,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACxE,kBAAkB,EAAE,KAAK;YACzB,SAAS,EAAE,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAC/B,eAAe,CAAC,KAAK,CAAC,CAAC,WAAW,KAAK,IAAI;gBAC3C,eAAe,CAAC,KAAK,CAAC,CAAC,WAAY,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;SACjE,CAAC,CAAC,CAAC;QAEJ,MAAM,aAAa,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,gBAAgB,aAAa,mCAAmC,CAAC,CAAC;QAE9E,OAAO,UAAU,CAAC;IACpB,CAAC;IAEM,cAAc,CACnB,eAAgC,EAChC,UAA8B,EAC9B,UAAkB,EAClB,WAAqB;QAGrB,MAAM,aAAa,GAAG,UAAU;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;aACxB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAC,GAAG,eAAe,CAAC,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,kBAAkB,EAAC,CAAC,CAAC;aACnF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,wDAAwD;QAElG,IAAG,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,YAAY,EAAE,EAAE;aACjB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,yCAAmB,EAAE,CAAC;YAC9C,WAAW,CAAC,cAAc,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;YAEpD,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAGhC,CAAC;YAEN,KAAI,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBAChC,MAAM,OAAO,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBACrF,IAAG,OAAO,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,OAAO,CAAC,YAAY,CAAC;oBACtC,IAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACxC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBAC1C,CAAC;oBACD,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,EAAC,aAAa,EAAE,IAAI,EAAE,OAAO,EAAC,CAAC,CAAC;oBAE1E,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC;oBAC9B,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,gBAAgB,CAAC;oBACnD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,cAAc,CAAC;gBACjD,CAAC;YACH,CAAC;YAED,IAAG,qBAAqB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACpC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,iDAAiD;oBACxD,YAAY,EAAE,EAAE;iBACjB,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAa,EAAE,CAAC;YAClC,MAAM,aAAa,GAAa,EAAE,CAAC;YAEnC,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;YAE7D,KAAI,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,qBAAqB,CAAC,OAAO,EAAE,EAAE,CAAC;gBACtE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,GAAG,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;gBAEjF,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,YAAY,CAAC,MAAM,SAAS,CAAC,CAAC;gBAEzF,KAAI,MAAM,EAAC,aAAa,EAAE,IAAI,EAAE,OAAO,EAAC,IAAI,YAAY,EAAE,CAAC;oBACzD,IAAG,CAAC,IAAI,CAAC,WAAW;wBAAE,SAAS;oBAE/B,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,mBAAmB,aAAa,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;oBAEvH,MAAM,SAAS,GAAG,WAAW,CAAC,sBAAsB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;oBAEhF,IAAG,SAAS,CAAC,OAAO,EAAE,CAAC;wBACrB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBAC9B,IAAG,CAAC,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;4BACrC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAC/B,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,IAAI,CAAC,gCAAgC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,aAAa,YAAY,CAAC,MAAM,aAAa,aAAa,CAAC,MAAM,QAAQ,CAAC,CAAC;YAEvF,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,YAAY;gBACZ,aAAa;aACd,CAAC;QAEJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;gBAC/D,YAAY,EAAE,EAAE;aACjB,CAAC;QACJ,CAAC;IACH,CAAC;IAEM,qBAAqB,CAAC,MAA6B,EAAE,eAAgC;QAC1F,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAE5B,IAAG,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,sBAAsB,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IAAG,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,0BAA0B,MAAM,CAAC,YAAY,CAAC,MAAM,cAAc,CAAC,CAAC;QAEhF,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAClC,MAAM,IAAI,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;YACpC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC,gBAAgB;gBACzF,CAAC,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC,gBAAgB,GAAG;gBACxG,CAAC,CAAC,WAAW,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,GAAG,CAAC;YACjD,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,CAAC,mBAAmB,GAAG,YAAY,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,IAAG,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3D,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;YACpC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;gBACtC,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,IAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAC1B,OAAO,CAAC,GAAG,CAAC,6BAA6B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,wEAAwE,CAAC,CAAC;QACtF,OAAO,CAAC,GAAG,CAAC,0EAA0E,CAAC,CAAC;QACxF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;CACF;AA1MD,wCA0MC"}

package/dist/ContractUnflattener.d.ts

@@ -0,0 +1,25 @@
+export interface FileMapping {
+    filePath: string;
+    startLine: number;
+    endLine: number;
+    originalContent: string;
+}
+export interface VulnerabilityMapping {
+    originalFile: string;
+    originalFromLine: number;
+    originalToLine: number;
+    flattenedFromLine: number;
+    flattenedToLine: number;
+}
+export declare class ContractUnflattener {
+    private fileMappings;
+    parseFlattened(flattenedSource: string, sourcePaths: string[]): FileMapping[];
+    mapVulnerabilityToOriginal(flattenedFromLine: number, flattenedToLine: number): VulnerabilityMapping | null;
+    applyFixToOriginalFile(mapping: VulnerabilityMapping, fixedLines: string): {
+        success: boolean;
+        error?: string;
+    };
+    private findFullPath;
+    private findOriginalContent;
+}
+//# sourceMappingURL=ContractUnflattener.d.ts.map

package/dist/ContractUnflattener.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContractUnflattener.d.ts","sourceRoot":"","sources":["../src/ContractUnflattener.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,oBAAoB;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;IACvB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,YAAY,CAAqB;IAElC,cAAc,CAAC,eAAe,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE;IA4D7E,0BAA0B,CAC/B,iBAAiB,EAAE,MAAM,EACzB,eAAe,EAAE,MAAM,GACtB,oBAAoB,GAAG,IAAI;IAiBvB,sBAAsB,CAC3B,OAAO,EAAE,oBAAoB,EAC7B,UAAU,EAAE,MAAM,GACjB;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAuBvC,OAAO,CAAC,YAAY;IAapB,OAAO,CAAC,mBAAmB;CAU5B"}

package/dist/ContractUnflattener.js

@@ -0,0 +1,152 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContractUnflattener = void 0;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+class ContractUnflattener {
+    fileMappings = [];
+    parseFlattened(flattenedSource, sourcePaths) {
+        const lines = flattenedSource.split('\n');
+        const mappings = [];
+        let currentFile = null;
+        let currentStartLine = 0;
+        let currentLines = [];
+        for (let i = 0; i < lines.length; i++) {
+            const line = lines[i];
+            const fileMarker = line.match(/^\/\/ File (.+)$/);
+            if (fileMarker) {
+                if (currentFile && currentLines.length > 0) {
+                    const originalContent = this.findOriginalContent(currentFile);
+                    mappings.push({
+                        filePath: currentFile,
+                        startLine: currentStartLine,
+                        endLine: i - 2,
+                        originalContent,
+                    });
+                }
+                const fullPath = this.findFullPath(fileMarker[1], sourcePaths);
+                if (fullPath) {
+                    currentFile = fullPath;
+                    let nextNonEmptyLine = i + 1;
+                    while (nextNonEmptyLine < lines.length && lines[nextNonEmptyLine].trim() === '') {
+                        nextNonEmptyLine++;
+                    }
+                    currentStartLine = nextNonEmptyLine + 1;
+                    currentLines = [];
+                }
+            }
+            else if (currentFile) {
+                if (line.startsWith('// Original license:')) {
+                    const group = line.match(/^\/\/ Original license:(.*)$/);
+                    if (group && group[1]) {
+                        currentLines.push(group[1].trim());
+                    }
+                }
+                else {
+                    currentLines.push(line);
+                }
+            }
+        }
+        if (currentFile && currentLines.length > 0) {
+            mappings.push({
+                filePath: currentFile,
+                startLine: currentStartLine,
+                endLine: lines.length,
+                originalContent: this.findOriginalContent(currentFile),
+            });
+        }
+        this.fileMappings = mappings;
+        return mappings;
+    }
+    mapVulnerabilityToOriginal(flattenedFromLine, flattenedToLine) {
+        for (const mapping of this.fileMappings) {
+            if (flattenedFromLine >= mapping.startLine && flattenedToLine <= mapping.endLine) {
+                return {
+                    originalFile: mapping.filePath,
+                    originalFromLine: flattenedFromLine - mapping.startLine + 1,
+                    originalToLine: flattenedToLine - mapping.startLine + 1,
+                    flattenedFromLine,
+                    flattenedToLine,
+                };
+            }
+        }
+        return null;
+    }
+    applyFixToOriginalFile(mapping, fixedLines) {
+        try {
+            if (!fs.existsSync(mapping.originalFile)) {
+                return { success: false, error: `Original file not found: ${mapping.originalFile}` };
+            }
+            const originalLines = fs.readFileSync(mapping.originalFile, 'utf8').split('\n');
+            const startIndex = mapping.originalFromLine - 1;
+            const endIndex = mapping.originalToLine - 1;
+            originalLines.splice(startIndex, endIndex - startIndex + 1, ...fixedLines.split('\n'));
+            fs.writeFileSync(mapping.originalFile, originalLines.join('\n'));
+            return { success: true };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            };
+        }
+    }
+    findFullPath(relativePath, sourcePaths) {
+        const exactMatch = sourcePaths.find(sp => sp === relativePath);
+        if (exactMatch)
+            return exactMatch;
+        const byFilename = sourcePaths.find(sp => path.basename(sp) === path.basename(relativePath));
+        if (byFilename)
+            return byFilename;
+        const pathMatch = sourcePaths.find(sp => sp.endsWith(relativePath));
+        if (pathMatch)
+            return pathMatch;
+        return null;
+    }
+    findOriginalContent(filePath) {
+        try {
+            if (fs.existsSync(filePath)) {
+                return fs.readFileSync(filePath, 'utf8');
+            }
+        }
+        catch (error) {
+            console.warn(`Could not read original content for ${filePath}:`, error);
+        }
+        return '';
+    }
+}
+exports.ContractUnflattener = ContractUnflattener;
+//# sourceMappingURL=ContractUnflattener.js.map

package/dist/ContractUnflattener.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContractUnflattener.js","sourceRoot":"","sources":["../src/ContractUnflattener.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uCAAyB;AACzB,2CAA6B;AAiB7B,MAAa,mBAAmB;IACtB,YAAY,GAAkB,EAAE,CAAC;IAElC,cAAc,CAAC,eAAuB,EAAE,WAAqB;QAClE,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAkB,EAAE,CAAC;QACnC,IAAI,WAAW,GAAkB,IAAI,CAAC;QACtC,IAAI,gBAAgB,GAAG,CAAC,CAAC;QACzB,IAAI,YAAY,GAAa,EAAE,CAAC;QAEhC,KAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAEtB,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAElD,IAAG,UAAU,EAAE,CAAC;gBACd,IAAG,WAAW,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1C,MAAM,eAAe,GAAG,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;oBAC9D,QAAQ,CAAC,IAAI,CAAC;wBACZ,QAAQ,EAAE,WAAW;wBACrB,SAAS,EAAE,gBAAgB;wBAC3B,OAAO,EAAE,CAAC,GAAG,CAAC;wBACd,eAAe;qBAChB,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;gBAE/D,IAAG,QAAQ,EAAE,CAAC;oBACZ,WAAW,GAAG,QAAQ,CAAC;oBACvB,IAAI,gBAAgB,GAAG,CAAC,GAAG,CAAC,CAAC;oBAC7B,OAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;wBAC/E,gBAAgB,EAAE,CAAC;oBACrB,CAAC;oBACD,gBAAgB,GAAG,gBAAgB,GAAG,CAAC,CAAC;oBACxC,YAAY,GAAG,EAAE,CAAC;gBACpB,CAAC;YACH,CAAC;iBAAM,IAAG,WAAW,EAAE,CAAC;gBACtB,IAAG,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,EAAE,CAAC;oBAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;oBACzD,IAAG,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;wBACrB,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAG,WAAW,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,QAAQ,CAAC,IAAI,CAAC;gBACZ,QAAQ,EAAE,WAAW;gBACrB,SAAS,EAAE,gBAAgB;gBAC3B,OAAO,EAAE,KAAK,CAAC,MAAM;gBACrB,eAAe,EAAE,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC;aACvD,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC7B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAGM,0BAA0B,CAC/B,iBAAyB,EACzB,eAAuB;QAEvB,KAAI,MAAM,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvC,IAAG,iBAAiB,IAAI,OAAO,CAAC,SAAS,IAAI,eAAe,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAChF,OAAO;oBACL,YAAY,EAAE,OAAO,CAAC,QAAQ;oBAC9B,gBAAgB,EAAE,iBAAiB,GAAG,OAAO,CAAC,SAAS,GAAG,CAAC;oBAC3D,cAAc,EAAE,eAAe,GAAG,OAAO,CAAC,SAAS,GAAG,CAAC;oBACvD,iBAAiB;oBACjB,eAAe;iBAChB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAGM,sBAAsB,CAC3B,OAA6B,EAC7B,UAAkB;QAElB,IAAI,CAAC;YACH,IAAG,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBACxC,OAAO,EAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,4BAA4B,OAAO,CAAC,YAAY,EAAE,EAAC,CAAC;YACrF,CAAC;YACD,MAAM,aAAa,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEhF,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,GAAG,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,GAAG,CAAC,CAAC;YAE5C,aAAa,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,GAAG,UAAU,GAAG,CAAC,EAAE,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;YAEvF,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,YAAY,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAEjE,OAAO,EAAC,OAAO,EAAE,IAAI,EAAC,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,YAAoB,EAAE,WAAqB;QAC9D,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,YAAY,CAAC,CAAC;QAC/D,IAAG,UAAU;YAAE,OAAO,UAAU,CAAC;QAEjC,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;QAC7F,IAAG,UAAU;YAAE,OAAO,UAAU,CAAC;QAEjC,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;QACpE,IAAG,SAAS;YAAE,OAAO,SAAS,CAAC;QAE/B,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,mBAAmB,CAAC,QAAgB;QAC1C,IAAI,CAAC;YACH,IAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC3B,OAAO,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,uCAAuC,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AApID,kDAoIC"}

package/dist/ReinforcedHardhatRuntimeEnvironmentField.d.ts

@@ -0,0 +1,13 @@
+import { AuditResult } from './types';
+export declare class ReinforcedHardhatRuntimeEnvironmentField {
+    auditContract(sourceCode: string, apiKey: string, statusCheckInterval?: number, maxRetries?: number, reportFolder?: string): Promise<AuditResult>;
+    private checkAuditStatusWithRetry;
+    private getAuditResultWithRetry;
+    private requestAuditWithRetry;
+    private fetchAudit;
+    private requestAudit;
+    private checkAuditStatus;
+    private getAuditResult;
+    private getPdfReport;
+}
+//# sourceMappingURL=ReinforcedHardhatRuntimeEnvironmentField.d.ts.map

package/dist/ReinforcedHardhatRuntimeEnvironmentField.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ReinforcedHardhatRuntimeEnvironmentField.d.ts","sourceRoot":"","sources":["../src/ReinforcedHardhatRuntimeEnvironmentField.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAyD,MAAM,SAAS,CAAC;AAE7F,qBAAa,wCAAwC;IACtC,aAAa,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;YAgChJ,yBAAyB;YAIzB,uBAAuB;YAIvB,qBAAqB;YAIrB,UAAU;YAQV,YAAY;YAcZ,gBAAgB;YAkBhB,cAAc;YAmBd,YAAY;CAc3B"}

package/dist/ReinforcedHardhatRuntimeEnvironmentField.js

@@ -0,0 +1,151 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ReinforcedHardhatRuntimeEnvironmentField = void 0;
+const fs_1 = __importDefault(require("fs"));
+const nanospinner_1 = require("nanospinner");
+const consts_1 = require("./consts");
+const types_1 = require("./types");
+class ReinforcedHardhatRuntimeEnvironmentField {
+    async auditContract(sourceCode, apiKey, statusCheckInterval, maxRetries, reportFolder) {
+        const taskId = await this.requestAuditWithRetry(statusCheckInterval, maxRetries, sourceCode, apiKey);
+        let spinner = null;
+        while (true) {
+            const status = await this.checkAuditStatusWithRetry(statusCheckInterval, maxRetries, taskId, apiKey);
+            if (status == 'pending' || status == 'processing') {
+                if (!spinner)
+                    spinner = (0, nanospinner_1.createSpinner)('Processing...').start();
+                await sleep(statusCheckInterval ?? consts_1.STATUS_CHECK_INTERVAL);
+                continue;
+            }
+            break;
+        }
+        if (spinner)
+            spinner.success();
+        const result = this.getAuditResultWithRetry(statusCheckInterval, maxRetries, taskId, apiKey);
+        fs_1.default.mkdirSync(reportFolder ?? './reinforcedai_reports', { recursive: true });
+        try {
+            const report = await this.getPdfReport(taskId, apiKey);
+            console.log(`📄 PDF report generated: ${report.filename} (${(report.size_bytes / 1024).toFixed(2)} KB)`);
+            const bin = Buffer.from(report.pdf_base64, 'base64');
+            if (report.size_bytes !== bin.length) {
+                console.warn(`Warning: PDF size mismatch. Expected ${report.size_bytes} bytes, but got ${bin.length} bytes.`);
+            }
+            fs_1.default.writeFileSync(`${reportFolder ?? './reinforcedai_reports'}/${report.filename}`, bin);
+        }
+        catch (e) {
+            console.error(`Failed to generate PDF report: ${e.message}`);
+        }
+        return result;
+    }
+    async checkAuditStatusWithRetry(retryDelay, maxRetries, ...args) {
+        return await retryHttpRequest(() => this.checkAuditStatus(...args), retryDelay, maxRetries);
+    }
+    async getAuditResultWithRetry(retryDelay, maxRetries, ...args) {
+        return await retryHttpRequest(() => this.getAuditResult(...args), retryDelay, maxRetries);
+    }
+    async requestAuditWithRetry(retryDelay, maxRetries, ...args) {
+        return await retryHttpRequest(() => this.requestAudit(...args), retryDelay, maxRetries);
+    }
+    async fetchAudit(payload, apiKey) {
+        return await fetch(consts_1.AUDIT_URL, {
+            method: 'POST',
+            headers: auditHeaders(apiKey),
+            body: JSON.stringify(payload),
+        });
+    }
+    async requestAudit(sourceCode, apiKey) {
+        const payload = {
+            'jsonrpc': '2.0',
+            'method': 'audit.get',
+            'params': { 'code': sourceCode },
+            'id': 1,
+        };
+        const response = await this.fetchAudit(payload, apiKey);
+        if (!response.ok) {
+            throw new types_1.HttpError(`HTTP ${response.status} ${response.statusText}`, response.status);
+        }
+        return (await response.json()).result.task_id;
+    }
+    async checkAuditStatus(taskId, apiKey) {
+        const payload = {
+            'jsonrpc': '2.0',
+            'method': 'task.status',
+            'params': { 'task_id': taskId },
+            'id': 2,
+        };
+        const response = await this.fetchAudit(payload, apiKey);
+        if (!response.ok) {
+            throw new types_1.HttpError(`HTTP ${response.status} ${response.statusText}`, response.status);
+        }
+        const auditResponse = (await response.json());
+        if (auditResponse.result.error_message) {
+            throw new Error(`Error on task.status ${auditResponse.result.error_message}`);
+        }
+        return auditResponse.result.status;
+    }
+    async getAuditResult(taskId, apiKey) {
+        const payload = {
+            'jsonrpc': '2.0',
+            'method': 'task.result',
+            'params': { 'task_id': taskId },
+            'id': 3,
+        };
+        const response = await this.fetchAudit(payload, apiKey);
+        if (!response.ok) {
+            throw new types_1.HttpError(`HTTP ${response.status} ${response.statusText}`, response.status);
+        }
+        const auditResponse = (await response.json());
+        if (auditResponse.result.error_message) {
+            throw new Error(`Error on task.result ${auditResponse.result.error_message}`);
+        }
+        return auditResponse.result;
+    }
+    async getPdfReport(taskId, apiKey) {
+        const payload = {
+            'jsonrpc': '2.0',
+            'method': 'task.generate_pdf',
+            'params': { 'task_id': taskId },
+            'id': 4,
+        };
+        const response = await this.fetchAudit(payload, apiKey);
+        if (!response.ok) {
+            throw new types_1.HttpError(`HTTP ${response.status} ${response.statusText}`, response.status);
+        }
+        return (await response.json()).result;
+    }
+}
+exports.ReinforcedHardhatRuntimeEnvironmentField = ReinforcedHardhatRuntimeEnvironmentField;
+async function retryHttpRequest(request, retryDelay = consts_1.STATUS_CHECK_INTERVAL, maxRetries = consts_1.MAX_RETRIES) {
+    let attempt = 0;
+    while (true) {
+        try {
+            return await request();
+        }
+        catch (e) {
+            attempt++;
+            if (attempt <= maxRetries && e instanceof types_1.HttpError && (e.statusCode == 429 || e.statusCode >= 500)) {
+                const base = retryDelay * 2 ** (attempt - 1);
+                const delay = Math.random() * base;
+                console.warn(`Retry #${attempt} in ${Math.round(delay)} ms`, e);
+                await sleep(delay);
+                continue;
+            }
+            else {
+                throw e;
+            }
+        }
+    }
+}
+function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+function auditHeaders(apiKey) {
+    return {
+        'Content-Type': 'application/json',
+        'X-Auth-Token': 'Bearer ' + apiKey,
+    };
+}
+//# sourceMappingURL=ReinforcedHardhatRuntimeEnvironmentField.js.map

package/dist/ReinforcedHardhatRuntimeEnvironmentField.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ReinforcedHardhatRuntimeEnvironmentField.js","sourceRoot":"","sources":["../src/ReinforcedHardhatRuntimeEnvironmentField.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,6CAA4C;AAC5C,qCAAyE;AACzE,mCAA6F;AAE7F,MAAa,wCAAwC;IAC5C,KAAK,CAAC,aAAa,CAAC,UAAkB,EAAE,MAAc,EAAE,mBAA4B,EAAE,UAAmB,EAAE,YAAqB;QACrI,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QACrG,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,OAAM,IAAI,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,mBAAmB,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACrG,IAAG,MAAM,IAAI,SAAS,IAAI,MAAM,IAAI,YAAY,EAAE,CAAC;gBACjD,IAAG,CAAC,OAAO;oBACT,OAAO,GAAG,IAAA,2BAAa,EAAC,eAAe,CAAC,CAAC,KAAK,EAAE,CAAC;gBACnD,MAAM,KAAK,CAAC,mBAAmB,IAAI,8BAAqB,CAAC,CAAC;gBAC1D,SAAS;YACX,CAAC;YACD,MAAM;QACR,CAAC;QACD,IAAG,OAAO;YACR,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,mBAAmB,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7F,YAAE,CAAC,SAAS,CAAC,YAAY,IAAI,wBAAwB,EAAE,EAAC,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;QAE1E,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,4BAA4B,MAAM,CAAC,QAAQ,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACzG,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACrD,IAAG,MAAM,CAAC,UAAU,KAAK,GAAG,CAAC,MAAM,EAAE,CAAC;gBACpC,OAAO,CAAC,IAAI,CAAC,wCAAwC,MAAM,CAAC,UAAU,mBAAmB,GAAG,CAAC,MAAM,SAAS,CAAC,CAAC;YAChH,CAAC;YACD,YAAE,CAAC,aAAa,CAAC,GAAG,YAAY,IAAI,wBAAwB,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,CAAC;QAC1F,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,kCAAmC,CAAW,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,yBAAyB,CAAE,UAAmB,EAAE,UAAmB,EAAE,GAAG,IAA8C;QAClI,OAAO,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,IAAI,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAC9F,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,UAAmB,EAAE,UAAmB,EAAE,GAAG,IAA4C;QAC7H,OAAO,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAC5F,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,UAAmB,EAAE,UAAmB,EAAE,GAAG,IAA0C;QACzH,OAAO,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAC1F,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAY,EAAE,MAAc;QACnD,OAAO,MAAM,KAAK,CAAC,kBAAS,EAAE;YAC5B,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC;YAC7B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,UAAkB,EAAE,MAAc;QAC3D,MAAM,OAAO,GAAG;YACd,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,WAAW;YACrB,QAAQ,EAAE,EAAC,MAAM,EAAE,UAAU,EAAC;YAC9B,IAAI,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,IAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,IAAI,iBAAS,CAAC,QAAQ,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzF,CAAC;QACD,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,MAAc;QAC3D,MAAM,OAAO,GAAG;YACd,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,aAAa;YACvB,QAAQ,EAAE,EAAC,SAAS,EAAE,MAAM,EAAC;YAC7B,IAAI,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,IAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,IAAI,iBAAS,CAAC,QAAQ,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzF,CAAC;QACD,MAAM,aAAa,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA8B,CAAC;QAC3E,IAAG,aAAa,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,wBAAwB,aAAa,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,MAAc,EAAE,MAAc;QACzD,MAAM,OAAO,GAAG;YACd,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,aAAa;YACvB,QAAQ,EAAE,EAAC,SAAS,EAAE,MAAM,EAAC;YAC7B,IAAI,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,IAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,IAAI,iBAAS,CAAC,QAAQ,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA8B,CAAC;QAC3E,IAAG,aAAa,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,wBAAwB,aAAa,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,aAAa,CAAC,MAAM,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,MAAc;QACvD,MAAM,OAAO,GAAG;YACd,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,mBAAmB;YAC7B,QAAQ,EAAE,EAAC,SAAS,EAAE,MAAM,EAAC;YAC7B,IAAI,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,IAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,IAAI,iBAAS,CAAC,QAAQ,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzF,CAAC;QAED,OAAQ,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAmC,CAAC,MAAM,CAAC;IAC3E,CAAC;CACF;AAtHD,4FAsHC;AAED,KAAK,UAAU,gBAAgB,CAAI,OAAyB,EAAE,aAAqB,8BAAqB,EAAE,aAAqB,oBAAW;IACxI,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,OAAM,IAAI,EAAE,CAAC;QACX,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,OAAO,EAAE,CAAC;YACV,IAAG,OAAO,IAAI,UAAU,IAAI,CAAC,YAAY,iBAAS,IAAI,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,IAAI,CAAC,CAAC,UAAU,IAAI,GAAG,CAAC,EAAE,CAAC;gBACnG,MAAM,IAAI,GAAG,UAAU,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;gBAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;gBACnC,OAAO,CAAC,IAAI,CAAC,UAAU,OAAO,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;gBAChE,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;gBACnB,SAAS;YACX,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,YAAY,CAAC,MAAc;IAClC,OAAO;QACL,cAAc,EAAE,kBAAkB;QAClC,cAAc,EAAE,SAAS,GAAG,MAAM;KACnC,CAAC;AACJ,CAAC"}

package/dist/consts.d.ts

@@ -0,0 +1,4 @@
+export declare const STATUS_CHECK_INTERVAL = 10000;
+export declare const AUDIT_URL = "https://api.reinforced.app/jsonrpc";
+export declare const MAX_RETRIES = 3;
+//# sourceMappingURL=consts.d.ts.map

package/dist/consts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"consts.d.ts","sourceRoot":"","sources":["../src/consts.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,qBAAqB,QAAS,CAAC;AAC5C,eAAO,MAAM,SAAS,uCAAuC,CAAC;AAC9D,eAAO,MAAM,WAAW,IAAI,CAAC"}

package/dist/consts.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MAX_RETRIES = exports.AUDIT_URL = exports.STATUS_CHECK_INTERVAL = void 0;
+exports.STATUS_CHECK_INTERVAL = 10_000;
+exports.AUDIT_URL = 'https://api.reinforced.app/jsonrpc';
+exports.MAX_RETRIES = 3;
+//# sourceMappingURL=consts.js.map

package/dist/consts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"consts.js","sourceRoot":"","sources":["../src/consts.ts"],"names":[],"mappings":";;;AAAa,QAAA,qBAAqB,GAAG,MAAM,CAAC;AAC/B,QAAA,SAAS,GAAG,oCAAoC,CAAC;AACjD,QAAA,WAAW,GAAG,CAAC,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+import './type-extensions';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAaA,OAAO,mBAAmB,CAAC"}

package/dist/index.js

@@ -0,0 +1,249 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs = __importStar(require("fs"));
+const task_names_1 = require("hardhat/builtin-tasks/task-names");
+const config_1 = require("hardhat/config");
+const contract_names_1 = require("hardhat/utils/contract-names");
+const tty_table_1 = __importDefault(require("tty-table"));
+const AutoFixManager_1 = require("./AutoFixManager");
+const ContractUnflattener_1 = require("./ContractUnflattener");
+const ReinforcedHardhatRuntimeEnvironmentField_1 = require("./ReinforcedHardhatRuntimeEnvironmentField");
+require("./type-extensions");
+(0, config_1.subtask)(task_names_1.TASK_COMPILE_SOLIDITY, async (args, hre, runSuper) => {
+    await runSuper(args); // run regular compilation first
+    if (!hre.config.reinforced?.compilationHookEnabled) {
+        console.log('Reinforced verification not enabled.');
+        return;
+    }
+    await auditAllContracts(hre);
+    console.log('✅ Audit complete before deployment.');
+});
+(0, config_1.task)('scan', 'Run pre-deployment audit', async (_, hre) => {
+    await hre.run(task_names_1.TASK_COMPILE_SOLIDITY, { force: false, quiet: true });
+    await auditAllContracts(hre);
+    console.log('✅ Audit complete');
+});
+(0, config_1.task)('scan:autofix', 'Run audit and apply auto-fixes for vulnerabilities', async (_, hre) => {
+    await hre.run(task_names_1.TASK_COMPILE_SOLIDITY, { force: false, quiet: true });
+    await auditAllContractsWithAutoFix(hre);
+    console.log('✅ Audit and auto-fix complete');
+});
+async function auditAllContractsWithAutoFix(hre) {
+    const apiKey = hre.config.reinforced?.apiKey;
+    if (!apiKey) {
+        console.log('Reinforced API key not set.');
+        return;
+    }
+    const flatSource = await hre.run(task_names_1.TASK_FLATTEN_GET_FLATTENED_SOURCE, { force: false, quiet: true });
+    const qualifiedContractNames = await hre.artifacts.getAllFullyQualifiedNames();
+    const sourcePaths = [];
+    for (const qualifiedContractName of qualifiedContractNames) {
+        const { sourceName } = (0, contract_names_1.parseFullyQualifiedName)(qualifiedContractName);
+        if (sourcePaths.includes(sourceName))
+            continue;
+        sourcePaths.push(sourceName);
+    }
+    if (sourcePaths.length > 0) {
+        try {
+            if (sourcePaths.length > 1) {
+                console.log('Auditing:');
+                for (const path of sourcePaths)
+                    console.log(' ', path);
+            }
+            else {
+                console.log('Auditing ', sourcePaths[0]);
+            }
+            const auditResult = await hre.reinforced.auditContract(flatSource, apiKey, hre.config.reinforced?.statusCheckInterval);
+            if (auditResult.error_message)
+                console.log('Error processing contract:', auditResult.error_message);
+            else if (!auditResult.result || auditResult.result.length == 0)
+                console.log('No vulnerabilities found');
+            else {
+                const enrichedVulnerabilities = enrichVulnerabilitiesWithOriginalMapping(auditResult.result, flatSource, sourcePaths.map(sp => sp.trim()));
+                console.log(formatAuditResult(enrichedVulnerabilities, flatSource));
+                const autoFixManager = new AutoFixManager_1.AutoFixManager();
+                const selections = autoFixManager.promptUserForVulnerabilitySelection(enrichedVulnerabilities);
+                if (selections.some(s => s.shouldFix)) {
+                    const result = autoFixManager.applyAutoFixes(enrichedVulnerabilities, selections, flatSource, sourcePaths.map(sp => sp.trim()));
+                    autoFixManager.displayAutoFixSummary(result, enrichedVulnerabilities);
+                }
+            }
+        }
+        catch (error) {
+            console.error('Error auditing contracts:', error);
+        }
+    }
+}
+function enrichVulnerabilitiesWithOriginalMapping(vulnerabilities, flattenedSource, sourcePaths) {
+    const unflattener = new ContractUnflattener_1.ContractUnflattener();
+    unflattener.parseFlattened(flattenedSource, sourcePaths);
+    return vulnerabilities.map(vuln => {
+        const mapping = unflattener.mapVulnerabilityToOriginal(vuln.from_line, vuln.to_line);
+        console.log('Mapping for vulnerability:', mapping);
+        if (mapping) {
+            return {
+                ...vuln,
+                original_file: mapping.originalFile,
+                original_from_line: mapping.originalFromLine,
+                original_to_line: mapping.originalToLine,
+            };
+        }
+        return vuln;
+    });
+}
+async function auditAllContracts(hre) {
+    const apiKey = hre.config.reinforced?.apiKey;
+    if (!apiKey) {
+        console.log('Reinforced API key not set.');
+        return;
+    }
+    const flatSource = await hre.run(task_names_1.TASK_FLATTEN_GET_FLATTENED_SOURCE, { force: false, quiet: true });
+    const qualifiedContractNames = await hre.artifacts.getAllFullyQualifiedNames();
+    const sourcePaths = [];
+    for (const qualifiedContractName of qualifiedContractNames) {
+        const { sourceName } = (0, contract_names_1.parseFullyQualifiedName)(qualifiedContractName);
+        if (sourcePaths.includes(sourceName))
+            continue;
+        sourcePaths.push(sourceName);
+    }
+    if (sourcePaths.length > 0) {
+        try {
+            if (sourcePaths.length > 1) {
+                console.log('Auditing:');
+                for (const path of sourcePaths)
+                    console.log(' ', path);
+            }
+            else {
+                console.log('Auditing ', sourcePaths[0]);
+            }
+            const auditResult = await hre.reinforced.auditContract(flatSource, apiKey, hre.config.reinforced?.statusCheckInterval);
+            if (auditResult.error_message)
+                console.log('Error processing contract:', auditResult.error_message);
+            else if (!auditResult.result || auditResult.result.length == 0)
+                console.log('No vulnerabilities found');
+            else {
+                const enrichedVulnerabilities = enrichVulnerabilitiesWithOriginalMapping(auditResult.result, flatSource, sourcePaths.map(sp => sp.trim()));
+                console.log('Enriched vulnerabilities:', enrichedVulnerabilities);
+                console.log(formatAuditResult(enrichedVulnerabilities, flatSource));
+                if (hre.config.reinforced?.autoFixEnabled !== false) {
+                    const autoFixManager = new AutoFixManager_1.AutoFixManager();
+                    const selections = autoFixManager.promptUserForVulnerabilitySelection(enrichedVulnerabilities);
+                    if (selections.some(s => s.shouldFix)) {
+                        const result = autoFixManager.applyAutoFixes(enrichedVulnerabilities, selections, flatSource, sourcePaths.map(sp => sp.trim()));
+                        autoFixManager.displayAutoFixSummary(result, enrichedVulnerabilities);
+                    }
+                }
+                else {
+                    console.log('\n💡 Auto-fix is disabled. You can enable it by setting "autoFixEnabled: true" in your hardhat.config.ts');
+                    console.log('   Or run "npx hardhat audit:autofix" to run with auto-fix enabled.');
+                }
+            }
+        }
+        catch (error) {
+            console.error('Error auditing contracts:', error);
+        }
+    }
+}
+function formatAuditResult(vulnerabilities, sourceCode) {
+    let output = '\n';
+    for (let i = 0; i < vulnerabilities.length; i++) {
+        const vulnerability = vulnerabilities[i];
+        output += `Possible vulnerability: ${vulnerability.vulnerability_class}${vulnerability.original_file ? ` in ${vulnerability.original_file.split('/').pop()}` : ''}`;
+        const headers = [{
+                value: '',
+                width: '20%',
+            },
+            {
+                value: '',
+                width: '80%',
+            }];
+        let affectedCode;
+        if (vulnerability.original_file && vulnerability.original_from_line && vulnerability.original_to_line) {
+            try {
+                const originalFileContent = fs.readFileSync(vulnerability.original_file, 'utf8');
+                affectedCode = originalFileContent.split('\n')
+                    .slice(vulnerability.original_from_line - 1, vulnerability.original_to_line)
+                    .map((line, index) => {
+                    const lineNumber = vulnerability.original_from_line + index;
+                    return `${lineNumber.toString().padStart(4, ' ')}: ${line}`;
+                })
+                    .join('\n');
+            }
+            catch (error) {
+                affectedCode = sourceCode.split('\n')
+                    .slice(vulnerability.from_line - 1, vulnerability.to_line)
+                    .map((line, index) => {
+                    const lineNumber = vulnerability.from_line + index;
+                    return `${lineNumber.toString().padStart(4, ' ')}: ${line}`;
+                })
+                    .join('\n');
+            }
+        }
+        else {
+            affectedCode = sourceCode.split('\n')
+                .slice(vulnerability.from_line - 1, vulnerability.to_line)
+                .map((line, index) => {
+                const lineNumber = vulnerability.from_line + index;
+                return `${lineNumber.toString().padStart(4, ' ')}: ${line}`;
+            })
+                .join('\n');
+        }
+        const rows = [
+            ['Possibility ', `${vulnerability.miners_select_count}/${vulnerability.miners_participated_count}`],
+            ['Description ', vulnerability.description],
+            ['Affected code ', affectedCode],
+            ['Possible fix ', vulnerability.fixed_lines || 'No fix provided'],
+            ['Test case ', vulnerability.test_case || 'No test case provided'],
+            ['Is suggestion', vulnerability.is_suggestion ? 'Yes' : 'No'],
+        ];
+        const options = {
+            width: '100%',
+            borderStyle: 'solid',
+            align: 'left',
+        };
+        const table = (0, tty_table_1.default)(headers, rows, options);
+        output += table.render();
+        output += '\n\n';
+    }
+    return output;
+}
+(0, config_1.extendEnvironment)((hre) => {
+    hre.reinforced = new ReinforcedHardhatRuntimeEnvironmentField_1.ReinforcedHardhatRuntimeEnvironmentField();
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uCAAyB;AACzB,iEAA0H;AAC1H,2CAAkE;AAClE,iEAAuE;AAKvE,0DAA8B;AAE9B,qDAAkD;AAClD,+DAA4D;AAC5D,yGAAsG;AACtG,6BAA2B;AAE3B,IAAA,gBAAO,EAAC,kCAAqB,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE;IAC3D,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,gCAAgC;IACtD,IAAG,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,sBAAsB,EAAE,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;QACpD,OAAO;IACT,CAAC;IACD,MAAM,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;AACrD,CAAC,CAAC,CAAC;AAEH,IAAA,aAAI,EAAC,MAAM,EAAE,0BAA0B,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE;IACxD,MAAM,GAAG,CAAC,GAAG,CAAC,kCAAqB,EAAE,EAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEH,IAAA,aAAI,EAAC,cAAc,EAAE,oDAAoD,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE;IAC1F,MAAM,GAAG,CAAC,GAAG,CAAC,kCAAqB,EAAE,EAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAC,CAAC,CAAC;IAClE,MAAM,4BAA4B,CAAC,GAAG,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;AAC/C,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,4BAA4B,CAAC,GAA8B;IACxE,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC;IAC7C,IAAG,CAAC,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QAC3C,OAAO;IACT,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,8CAAiC,EAAE,EAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAC,CAAC,CAAC;IAEjG,MAAM,sBAAsB,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,yBAAyB,EAAE,CAAC;IAC/E,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,KAAI,MAAM,qBAAqB,IAAI,sBAAsB,EAAE,CAAC;QAC1D,MAAM,EAAC,UAAU,EAAC,GAAG,IAAA,wCAAuB,EAAC,qBAAqB,CAAC,CAAC;QACpE,IAAG,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC;YACjC,SAAS;QACX,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IACD,IAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,IAAI,CAAC;YACH,IAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,KAAI,MAAM,IAAI,IAAI,WAAW;oBAC3B,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3C,CAAC;YACD,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,UAAU,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;YACvH,IAAG,WAAW,CAAC,aAAa;gBAC1B,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE,WAAW,CAAC,aAAa,CAAC,CAAC;iBAClE,IAAG,CAAC,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC;gBAC3D,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;iBACrC,CAAC;gBACJ,MAAM,uBAAuB,GAAG,wCAAwC,CACtE,WAAW,CAAC,MAAM,EAClB,UAAU,EACV,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CACjC,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,uBAAuB,EAAE,UAAU,CAAC,CAAC,CAAC;gBAEpE,MAAM,cAAc,GAAG,IAAI,+BAAc,EAAE,CAAC;gBAC5C,MAAM,UAAU,GAAG,cAAc,CAAC,mCAAmC,CAAC,uBAAuB,CAAC,CAAC;gBAE/F,IAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC;oBACrC,MAAM,MAAM,GAAG,cAAc,CAAC,cAAc,CAC1C,uBAAuB,EACvB,UAAU,EACV,UAAU,EACV,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CACjC,CAAC;oBACF,cAAc,CAAC,qBAAqB,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,wCAAwC,CAC/C,eAAgC,EAChC,eAAuB,EACvB,WAAqB;IAErB,MAAM,WAAW,GAAG,IAAI,yCAAmB,EAAE,CAAC;IAC9C,WAAW,CAAC,cAAc,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;IAEzD,OAAO,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QAChC,MAAM,OAAO,GAAG,WAAW,CAAC,0BAA0B,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACrF,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE,OAAO,CAAC,CAAC;QACnD,IAAG,OAAO,EAAE,CAAC;YACX,OAAO;gBACL,GAAG,IAAI;gBACP,aAAa,EAAE,OAAO,CAAC,YAAY;gBACnC,kBAAkB,EAAE,OAAO,CAAC,gBAAgB;gBAC5C,gBAAgB,EAAE,OAAO,CAAC,cAAc;aACzC,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,GAA8B;IAC7D,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC;IAC7C,IAAG,CAAC,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QAC3C,OAAO;IACT,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,8CAAiC,EAAE,EAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAC,CAAC,CAAC;IAEjG,MAAM,sBAAsB,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,yBAAyB,EAAE,CAAC;IAC/E,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,KAAI,MAAM,qBAAqB,IAAI,sBAAsB,EAAE,CAAC;QAC1D,MAAM,EAAC,UAAU,EAAC,GAAG,IAAA,wCAAuB,EAAC,qBAAqB,CAAC,CAAC;QACpE,IAAG,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC;YACjC,SAAS;QACX,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IACD,IAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,IAAI,CAAC;YACH,IAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzB,KAAI,MAAM,IAAI,IAAI,WAAW;oBAC3B,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3C,CAAC;YACD,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,UAAU,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;YACvH,IAAG,WAAW,CAAC,aAAa;gBAC1B,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE,WAAW,CAAC,aAAa,CAAC,CAAC;iBAClE,IAAG,CAAC,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC;gBAC3D,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;iBACrC,CAAC;gBACJ,MAAM,uBAAuB,GAAG,wCAAwC,CACtE,WAAW,CAAC,MAAM,EAClB,UAAU,EACV,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CACjC,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,uBAAuB,CAAC,CAAC;gBAClE,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,uBAAuB,EAAE,UAAU,CAAC,CAAC,CAAC;gBAEpE,IAAG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,cAAc,KAAK,KAAK,EAAE,CAAC;oBACnD,MAAM,cAAc,GAAG,IAAI,+BAAc,EAAE,CAAC;oBAC5C,MAAM,UAAU,GAAG,cAAc,CAAC,mCAAmC,CAAC,uBAAuB,CAAC,CAAC;oBAE/F,IAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC;wBACrC,MAAM,MAAM,GAAG,cAAc,CAAC,cAAc,CAC1C,uBAAuB,EACvB,UAAU,EACV,UAAU,EACV,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CACjC,CAAC;wBACF,cAAc,CAAC,qBAAqB,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC;oBACxE,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CAAC,0GAA0G,CAAC,CAAC;oBACxH,OAAO,CAAC,GAAG,CAAC,qEAAqE,CAAC,CAAC;gBACrF,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,eAAgC,EAAE,UAAkB;IAC7E,IAAI,MAAM,GAAG,IAAI,CAAC;IAClB,KAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,aAAa,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;QAEzC,MAAM,IAAI,2BAA2B,aAAa,CAAC,mBAAmB,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACpK,MAAM,OAAO,GAAG,CAAC;gBACf,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,KAAK;aACb;YACD;gBACE,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;QAEH,IAAI,YAAoB,CAAC;QACzB,IAAG,aAAa,CAAC,aAAa,IAAI,aAAa,CAAC,kBAAkB,IAAI,aAAa,CAAC,gBAAgB,EAAE,CAAC;YACrG,IAAI,CAAC;gBACH,MAAM,mBAAmB,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;gBACjF,YAAY,GAAG,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC;qBAC3C,KAAK,CAAC,aAAa,CAAC,kBAAkB,GAAG,CAAC,EAAE,aAAa,CAAC,gBAAgB,CAAC;qBAC3E,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBACnB,MAAM,UAAU,GAAG,aAAa,CAAC,kBAAmB,GAAG,KAAK,CAAC;oBAC7D,OAAO,GAAG,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC9D,CAAC,CAAC;qBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,YAAY,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;qBAClC,KAAK,CAAC,aAAa,CAAC,SAAS,GAAG,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC;qBACzD,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBACnB,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,GAAG,KAAK,CAAC;oBACnD,OAAO,GAAG,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC9D,CAAC,CAAC;qBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;iBAClC,KAAK,CAAC,aAAa,CAAC,SAAS,GAAG,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC;iBACzD,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;gBACnB,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,GAAG,KAAK,CAAC;gBACnD,OAAO,GAAG,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9D,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QAED,MAAM,IAAI,GAAG;YACX,CAAC,cAAc,EAAE,GAAG,aAAa,CAAC,mBAAmB,IAAI,aAAa,CAAC,yBAAyB,EAAE,CAAC;YACnG,CAAC,cAAc,EAAE,aAAa,CAAC,WAAW,CAAC;YAC3C,CAAC,gBAAgB,EAAE,YAAY,CAAC;YAChC,CAAC,eAAe,EAAE,aAAa,CAAC,WAAW,IAAI,iBAAiB,CAAC;YACjE,CAAC,YAAY,EAAE,aAAa,CAAC,SAAS,IAAI,uBAAuB,CAAC;YAClE,CAAC,eAAe,EAAE,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;SAC9D,CAAC;QACF,MAAM,OAAO,GAAG;YACd,KAAK,EAAE,MAAM;YACb,WAAW,EAAE,OAAO;YACpB,KAAK,EAAE,MAAM;SACd,CAAC;QACF,MAAM,KAAK,GAAG,IAAA,mBAAK,EACjB,OAAO,EACP,IAAI,EACJ,OAAO,CACR,CAAC;QAEF,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,IAAI,MAAM,CAAC;IACnB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAA,0BAAiB,EAAC,CAAC,GAAG,EAAE,EAAE;IACxB,GAAG,CAAC,UAAU,GAAG,IAAI,mFAAwC,EAAE,CAAC;AAClE,CAAC,CAAC,CAAC"}

package/dist/type-extensions.d.ts

@@ -0,0 +1,24 @@
+import { ReinforcedHardhatRuntimeEnvironmentField } from 'ReinforcedHardhatRuntimeEnvironmentField';
+import 'hardhat/types/config';
+export interface ReinforcedUserConfig {
+    apiKey?: string;
+    compilationHookEnabled?: boolean;
+    statusCheckInterval?: number;
+    maxRetries?: number;
+    autoFixEnabled?: boolean;
+    reportFolder?: string;
+}
+declare module 'hardhat/types/config' {
+    interface HardhatUserConfig {
+        reinforced?: ReinforcedUserConfig;
+    }
+    interface HardhatConfig {
+        reinforced?: ReinforcedUserConfig;
+    }
+}
+declare module 'hardhat/types/runtime' {
+    interface HardhatRuntimeEnvironment {
+        reinforced: ReinforcedHardhatRuntimeEnvironmentField;
+    }
+}
+//# sourceMappingURL=type-extensions.d.ts.map

package/dist/type-extensions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"type-extensions.d.ts","sourceRoot":"","sources":["../src/type-extensions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wCAAwC,EAAE,MAAM,0CAA0C,CAAC;AACpG,OAAO,sBAAsB,CAAC;AAE9B,MAAM,WAAW,oBAAoB;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,iBAAiB;QACzB,UAAU,CAAC,EAAE,oBAAoB,CAAC;KACnC;IAED,UAAU,aAAa;QACrB,UAAU,CAAC,EAAE,oBAAoB,CAAC;KACnC;CACF;AAED,OAAO,QAAQ,uBAAuB,CAAC;IACrC,UAAiB,yBAAyB;QACtC,UAAU,EAAE,wCAAwC,CAAC;KACxD;CACF"}

package/dist/type-extensions.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+require("hardhat/types/config");
+//# sourceMappingURL=type-extensions.js.map

package/dist/type-extensions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"type-extensions.js","sourceRoot":"","sources":["../src/type-extensions.ts"],"names":[],"mappings":";;AACA,gCAA8B"}

package/dist/types.d.ts

@@ -0,0 +1,52 @@
+export type AuditStatus = 'pending' | 'processing' | 'failed' | 'completed';
+export interface AuditResult {
+    error_message: string | null;
+    result: Vulnerability[] | null;
+    status: AuditStatus;
+    task_id: string;
+}
+export interface JRPCResponse<T> {
+    id: number;
+    jsonrpc: string;
+    result: T;
+}
+export interface PdfReportResult {
+    task_id: string;
+    filename: string;
+    pdf_base64: string;
+    size_bytes: number;
+}
+export interface Vulnerability {
+    description: string;
+    fixed_lines: string | null;
+    from_line: number;
+    is_suggestion: boolean;
+    miners_select_count: number;
+    miners_participated_count: number;
+    prior_art: any[];
+    test_case: string | null;
+    to_line: number;
+    vulnerability_class: string;
+    original_file?: string;
+    original_from_line?: number;
+    original_to_line?: number;
+}
+export interface VulnerabilityWithIndex extends Vulnerability {
+    index: number;
+}
+export interface AutoFixSelection {
+    vulnerabilityIndex: number;
+    shouldFix: boolean;
+}
+export interface CodeReplacementResult {
+    success: boolean;
+    filePath?: string;
+    error?: string;
+    appliedFixes: number[];
+    modifiedFiles?: string[];
+}
+export declare class HttpError extends Error {
+    statusCode: number;
+    constructor(message: string, statusCode: number);
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,WAAW,GAAG,SAAS,GAAG,YAAY,GAAG,QAAQ,GAAG,WAAW,CAAC;AAE5E,MAAM,WAAW,WAAW;IACxB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,MAAM,EAAE,aAAa,EAAG,GAAG,IAAI,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;IACpB,OAAO,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,YAAY,CAAC,CAAC;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,CAAC,CAAA;CACZ;AAED,MAAM,WAAW,eAAe;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAA;CACrB;AAED,MAAM,WAAW,aAAa;IAC1B,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,OAAO,CAAC;IACvB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,yBAAyB,EAAE,MAAM,CAAC;IAClC,SAAS,EAAE,GAAG,EAAG,CAAC;IAClB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,sBAAuB,SAAQ,aAAa;IACzD,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC7B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,qBAAqB;IAClC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,SAAU,SAAQ,KAAK;IAC3B,UAAU,EAAE,MAAM,CAAC;gBAEd,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;CAKhD"}

package/dist/types.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HttpError = void 0;
+;
+class HttpError extends Error {
+    statusCode;
+    constructor(message, statusCode) {
+        super(message);
+        this.name = 'HttpError';
+        this.statusCode = statusCode;
+    }
+}
+exports.HttpError = HttpError;
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";;;AAOC,CAAC;AAgDF,MAAa,SAAU,SAAQ,KAAK;IAC3B,UAAU,CAAS;IAE1B,YAAY,OAAe,EAAE,UAAkB;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;QACxB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;CACF;AARD,8BAQC"}

package/package.json

@@ -3,13 +3,13 @@
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "commonjs",
-  "version": "2511.26.1",
+  "version": "2512.08.1",
   "scripts": {
     "lint:fix": "prettier --write 'src/**/*.{js,ts}' 'test/**/*.{js,ts}' && tslint --fix --config tslint.json --project tsconfig.json",
     "lint": "tslint --config tslint.json --project tsconfig.json",
     "test": "mocha",
     "clean": "rm -rf dist && cd test/fixture-projects/hardhat-project && npx hardhat clean",
-    "build": "npm run clean && npm run test && tsc",
+    "build": "npm run clean && tsc",
     "watch": "tsc -w",
     "prepublishOnly": "npm run build"
   },

package/src/index.ts

@@ -23,13 +23,13 @@ subtask(TASK_COMPILE_SOLIDITY, async (args, hre, runSuper) => {
   console.log('✅ Audit complete before deployment.');
 });
 
-task('check-vulnerabilities', 'Run pre-deployment audit', async (_, hre) => {
+task('scan', 'Run pre-deployment audit', async (_, hre) => {
   await hre.run(TASK_COMPILE_SOLIDITY, {force: false, quiet: true});
   await auditAllContracts(hre);
   console.log('✅ Audit complete');
 });
 
-task('check-vulnerabilities:autofix', 'Run audit and apply auto-fixes for vulnerabilities', async (_, hre) => {
+task('scan:autofix', 'Run audit and apply auto-fixes for vulnerabilities', async (_, hre) => {
   await hre.run(TASK_COMPILE_SOLIDITY, {force: false, quiet: true});
   await auditAllContractsWithAutoFix(hre);
   console.log('✅ Audit and auto-fix complete');