scrypt-testgen 1.0.0

package/.prettierrc

@@ -0,0 +1,7 @@
+{
+  "semi": true,
+  "trailingComma": "es5",
+  "singleQuote": true,
+  "printWidth": 100,
+  "tabWidth": 2
+}

package/README.md

@@ -0,0 +1,17 @@
+# scrypt-testgen
+
+Automatic test generator for sCrypt-ts smart contracts. Generates comprehensive Mocha test files by analyzing contract structure using TypeScript Compiler API.
+
+## Features
+
+- 📝 Parses sCrypt-ts contracts using TypeScript AST
+- 🧪 Generates ready-to-run Mocha test files
+- ✅ Includes happy-path tests for all public methods
+- 🚫 Generates revert tests for assertion failures
+- 🎯 Creates edge-case tests for comprehensive coverage
+- 🎨 Produces clean, readable, and idiomatic test code
+
+## Installation
+
+```bash
+npm install -g scrypt-testgen

package/bin/scrypt-testgen.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+require('../dist/cli.js');

package/dist/cli.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=cli.d.ts.map

package/dist/cli.d.ts.map

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

package/dist/cli.js

@@ -0,0 +1,48 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const generator_1 = require("./generator");
+const path_1 = require("path");
+const fs_1 = require("fs");
+const chalk_1 = __importDefault(require("chalk"));
+const program = new commander_1.Command();
+program
+    .name('scrypt-testgen')
+    .description('Generate Mocha tests for sCrypt-ts smart contracts')
+    .version('1.0.0');
+program
+    .argument('<contract-path>', 'Path to the sCrypt contract TypeScript file')
+    .option('-m, --minimal', 'Generate only happy-path tests', false)
+    .option('-c, --coverage', 'Include revert and edge-case tests', false)
+    .option('-o, --overwrite', 'Replace existing test file', false)
+    .option('-o, --output <path>', 'Custom output path for test file')
+    .action(async (contractPath, options) => {
+    try {
+        const resolvedPath = (0, path_1.resolve)(process.cwd(), contractPath);
+        if (!(0, fs_1.existsSync)(resolvedPath)) {
+            console.error(chalk_1.default.red(`Error: Contract file not found: ${resolvedPath}`));
+            process.exit(1);
+        }
+        const outputPath = options.output
+            ? (0, path_1.resolve)(process.cwd(), options.output)
+            : (0, path_1.join)((0, path_1.dirname)(resolvedPath), '..', 'test', (0, path_1.basename)(resolvedPath).replace('.ts', '.test.ts'));
+        // Ensure test directory exists
+        (0, fs_1.mkdirSync)((0, path_1.dirname)(outputPath), { recursive: true });
+        await (0, generator_1.generateTests)(resolvedPath, outputPath, {
+            minimal: options.minimal,
+            coverage: options.coverage,
+            overwrite: options.overwrite
+        });
+        console.log(chalk_1.default.green(`✓ Tests generated: ${outputPath}`));
+    }
+    catch (error) {
+        console.error(chalk_1.default.red(`Error: ${error}`));
+        process.exit(1);
+    }
+});
+program.parse();
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";;;;;;AAEA,yCAAoC;AACpC,2CAA4C;AAC5C,+BAAwD;AACxD,2BAA2C;AAC3C,kDAA0B;AAE1B,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,gBAAgB,CAAC;KACtB,WAAW,CAAC,oDAAoD,CAAC;KACjE,OAAO,CAAC,OAAO,CAAC,CAAC;AAEpB,OAAO;KACJ,QAAQ,CAAC,iBAAiB,EAAE,6CAA6C,CAAC;KAC1E,MAAM,CAAC,eAAe,EAAE,gCAAgC,EAAE,KAAK,CAAC;KAChE,MAAM,CAAC,gBAAgB,EAAE,oCAAoC,EAAE,KAAK,CAAC;KACrE,MAAM,CAAC,iBAAiB,EAAE,4BAA4B,EAAE,KAAK,CAAC;KAC9D,MAAM,CAAC,qBAAqB,EAAE,kCAAkC,CAAC;KACjE,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE;IACtC,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,IAAA,cAAO,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,YAAY,CAAC,CAAC;QAE1D,IAAI,CAAC,IAAA,eAAU,EAAC,YAAY,CAAC,EAAE,CAAC;YAC9B,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC,CAAC;YAC5E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM;YAC/B,CAAC,CAAC,IAAA,cAAO,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,OAAO,CAAC,MAAM,CAAC;YACxC,CAAC,CAAC,IAAA,WAAI,EAAC,IAAA,cAAO,EAAC,YAAY,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAA,eAAQ,EAAC,YAAY,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;QAEjG,+BAA+B;QAC/B,IAAA,cAAS,EAAC,IAAA,cAAO,EAAC,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAEpD,MAAM,IAAA,yBAAa,EAAC,YAAY,EAAE,UAAU,EAAE;YAC5C,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,sBAAsB,UAAU,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC;QAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/generator/index.d.ts

@@ -0,0 +1,6 @@
+export declare function generateTests(contractPath: string, outputPath: string, options: {
+    minimal: boolean;
+    coverage: boolean;
+    overwrite: boolean;
+}): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/dist/generator/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/generator/index.ts"],"names":[],"mappings":"AAKA,wBAAsB,aAAa,CACjC,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE;IACP,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;IAClB,SAAS,EAAE,OAAO,CAAC;CACpB,GACA,OAAO,CAAC,IAAI,CAAC,CAuBf"}

package/dist/generator/index.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateTests = generateTests;
+const parser_1 = require("../parser");
+const test_generator_1 = require("./test-generator");
+const fs_1 = require("fs");
+async function generateTests(contractPath, outputPath, options) {
+    // Check if output file exists
+    if ((0, fs_1.existsSync)(outputPath) && !options.overwrite) {
+        throw new Error(`Test file already exists: ${outputPath}. Use --overwrite to replace.`);
+    }
+    // Parse the contract
+    const parser = new parser_1.ContractParser(contractPath);
+    const model = parser.parse();
+    // Generate tests
+    const generator = new test_generator_1.TestGenerator();
+    const testCode = generator.generateTestFile(model, {
+        minimal: options.minimal,
+        coverage: options.coverage
+    });
+    // Format the output (simplified - in production, use prettier API)
+    const formattedCode = formatCode(testCode);
+    // Write to file
+    (0, fs_1.writeFileSync)(outputPath, formattedCode, 'utf-8');
+}
+function formatCode(code) {
+    // Simple formatting - in production, integrate with Prettier
+    return code
+        .replace(/\n\s*\n\s*\n/g, '\n\n')
+        .replace(/\s+$/gm, '')
+        .trim() + '\n';
+}
+//# sourceMappingURL=index.js.map

package/dist/generator/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/generator/index.ts"],"names":[],"mappings":";;AAKA,sCA+BC;AApCD,sCAA2C;AAC3C,qDAAiD;AACjD,2BAA6D;AAGtD,KAAK,UAAU,aAAa,CACjC,YAAoB,EACpB,UAAkB,EAClB,OAIC;IAGD,8BAA8B;IAC9B,IAAI,IAAA,eAAU,EAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,+BAA+B,CAAC,CAAC;IAC1F,CAAC;IAED,qBAAqB;IACrB,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC,YAAY,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;IAE7B,iBAAiB;IACjB,MAAM,SAAS,GAAG,IAAI,8BAAa,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,SAAS,CAAC,gBAAgB,CAAC,KAAK,EAAE;QACjD,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;KAC3B,CAAC,CAAC;IAEH,mEAAmE;IACrE,MAAM,aAAa,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAEzC,gBAAgB;IAChB,IAAA,kBAAa,EAAC,UAAU,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,6DAA6D;IAC7D,OAAO,IAAI;SACR,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC;SAChC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;SACrB,IAAI,EAAE,GAAG,IAAI,CAAC;AACnB,CAAC"}

package/dist/generator/test-generator.d.ts

@@ -0,0 +1,17 @@
+import { ContractModel } from '../model/contract-model';
+export declare class TestGenerator {
+    private valueGenerator;
+    constructor();
+    generateTestFile(model: ContractModel, options: {
+        minimal: boolean;
+        coverage: boolean;
+    }): string;
+    private generateImports;
+    private getRelativeTestPath;
+    private generateTestSuite;
+    private generateMethodTests;
+    private generateHappyPathTest;
+    private generateRevertTest;
+    private generateEdgeCaseTests;
+}
+//# sourceMappingURL=test-generator.d.ts.map

package/dist/generator/test-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-generator.d.ts","sourceRoot":"","sources":["../../src/generator/test-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAkB,MAAM,yBAAyB,CAAC;AAGxE,qBAAa,aAAa;IACxB,OAAO,CAAC,cAAc,CAAiB;;IAMvC,gBAAgB,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE;QAC9C,OAAO,EAAE,OAAO,CAAC;QACjB,QAAQ,EAAE,OAAO,CAAC;KACnB,GAAG,MAAM;IASV,OAAO,CAAC,eAAe;IAyBvB,OAAO,CAAC,mBAAmB;IAa3B,OAAO,CAAC,iBAAiB;IA0BzB,OAAO,CAAC,mBAAmB;IAuB3B,OAAO,CAAC,qBAAqB;IAsC7B,OAAO,CAAC,kBAAkB;IAqC1B,OAAO,CAAC,qBAAqB;CA+B9B"}

package/dist/generator/test-generator.js

@@ -0,0 +1,177 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestGenerator = void 0;
+const value_generator_1 = require("./value-generator");
+class TestGenerator {
+    constructor() {
+        this.valueGenerator = new value_generator_1.ValueGenerator();
+    }
+    generateTestFile(model, options) {
+        this.valueGenerator.reset();
+        const imports = this.generateImports(model);
+        const testSuite = this.generateTestSuite(model, options);
+        return `${imports}\n\n${testSuite}`;
+    }
+    generateImports(model) {
+        const contractName = model.name;
+        const contractPath = this.getRelativeTestPath(model.filePath);
+        return `import { expect } from 'chai';
+import { ${contractName} } from '${contractPath.replace('.ts', '')}';
+import { 
+  bsv, 
+  TestWallet, 
+  DefaultProvider, 
+  sha256,
+  toByteString,
+  PubKey,
+  Sig,
+  findSig,
+  MethodCallOptions,
+  ContractTransaction,
+  getDefaultSigner
+} from 'scrypt-ts';
+
+describe('${contractName}', () => {
+  let signer: TestWallet;
+  let instance: ${contractName};`;
+    }
+    getRelativeTestPath(contractPath) {
+        // Convert absolute path to relative from test directory
+        const pathParts = contractPath.split('/');
+        const srcIndex = pathParts.lastIndexOf('src');
+        if (srcIndex !== -1) {
+            const relativeParts = pathParts.slice(srcIndex);
+            return `../${relativeParts.join('/')}`;
+        }
+        return contractPath;
+    }
+    generateTestSuite(model, options) {
+        let output = `
+  before(async () => {
+    await ${model.name}.loadArtifact();
+    
+    signer = await getDefaultSigner();
+  });
+
+  beforeEach(async () => {
+    instance = new ${model.name}(${this.valueGenerator.generateConstructorArgs(model.constructorArgs)});
+    await instance.connect(signer);
+  });
+`;
+        // Generate tests for each method
+        for (const method of model.methods) {
+            output += this.generateMethodTests(method, model, options);
+        }
+        output += `});`;
+        return output;
+    }
+    generateMethodTests(method, model, options) {
+        let output = `\n\n  describe('#${method.name}', () => {`;
+        // Happy path test
+        output += this.generateHappyPathTest(method, model);
+        // Revert test (unless minimal mode)
+        if (!options.minimal) {
+            output += this.generateRevertTest(method, model);
+        }
+        // Edge case tests (if coverage mode)
+        if (options.coverage) {
+            output += this.generateEdgeCaseTests(method, model);
+        }
+        output += `\n  });`;
+        return output;
+    }
+    generateHappyPathTest(method, model) {
+        const args = this.valueGenerator.generateMethodArgs(method.parameters, method.name);
+        const methodCall = `instance.methods.${method.name}(${args})`;
+        let test = `
+    it('should succeed with valid parameters', async () => {
+   const deployTx = await instance.deploy(1)
+      const callTx = await ${methodCall};
+      
+      console.log('Deploy TX:', deployTx.id);
+      console.log('Call TX:', callTx.tx.id);
+    });
+`;
+        // Add assertions for state transitions
+        if (method.mutatesState) {
+            test = `
+    it('should succeed with valid parameters', async () => {
+    const deployTx = await instance.deploy(1)
+      const oldInstance = instance;
+      const callTx = await ${methodCall};
+      
+      console.log('Deploy TX:', deployTx.id);
+        console.log('Call TX:', callTx.tx.id);
+      
+      // Verify state transition if applicable
+      const result = await callTx.tx.id;
+      if (result.nexts && result.nexts.length > 0) {
+        const nextInstance = result.nexts[0].instance;
+        // Add state assertions here based on method logic
+      }
+    });
+`;
+        }
+        return test;
+    }
+    generateRevertTest(method, model) {
+        const args = this.valueGenerator.generateInvalidMethodArgs(method.parameters, method.name);
+        const methodCall = `instance.methods.${method.name}(${args})`;
+        let test = `
+    it('should revert with invalid parameters', async () => {
+      try {
+      const deployTx = await instance.deploy(1)
+        const callTx = await ${methodCall};
+        console.error('Expected method to revert, but it succeeded:', callTx.tx.id);
+        
+      } catch (error) {
+        // Expected to throw an error
+        expect(error).to.be.an('error');
+      }
+    });
+`;
+        // Special handling for assert statements
+        if (method.hasAssert) {
+            test = `
+    it('should revert when assertion fails', async () => {
+      // Use parameters that will cause assert() to fail
+      const deployTx = await instance.deploy(1)
+      const callTx = await ${methodCall};
+
+      console.log('Deploy TX:', deployTx.id);
+      console.log('Call TX:', callTx.tx.id);
+      
+    
+    });
+`;
+        }
+        return test;
+    }
+    generateEdgeCaseTests(method, model) {
+        let tests = '';
+        // Generate edge case for each parameter
+        method.parameters.forEach((param, index) => {
+            const edgeCaseValue = this.valueGenerator.getEdgeCaseValue(param.type);
+            const modifiedParams = method.parameters.map((p, i) => i === index ? { ...p, type: param.type } : p);
+            const args = this.valueGenerator.generateMethodArgs(modifiedParams, method.name).replace(this.valueGenerator.generateMethodArgs(method.parameters, method.name), '').trim() || edgeCaseValue;
+            const methodCall = `instance.methods.${method.name}(${args})`;
+            tests += `
+    it('should handle edge case for ${param.name} (${edgeCaseValue})', async () => {
+      try {
+      const deployTx = await instance.deploy(1)
+        const callTx = await ${methodCall};
+        // Edge case might succeed or fail depending on contract logic
+        console.log('Deploy TX:', deployTx.id);
+        console.log('Call TX:', callTx.tx.id);
+      } catch (error) {
+        // Handle expected errors gracefully
+        expect(error).to.be.an('error');
+      }
+    });
+`;
+        });
+        return tests;
+    }
+}
+exports.TestGenerator = TestGenerator;
+//# sourceMappingURL=test-generator.js.map

package/dist/generator/test-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-generator.js","sourceRoot":"","sources":["../../src/generator/test-generator.ts"],"names":[],"mappings":";;;AACA,uDAAmD;AAEnD,MAAa,aAAa;IAGxB;QACE,IAAI,CAAC,cAAc,GAAG,IAAI,gCAAc,EAAE,CAAC;IAC7C,CAAC;IAED,gBAAgB,CAAC,KAAoB,EAAE,OAGtC;QACC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAE5B,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEzD,OAAO,GAAG,OAAO,OAAO,SAAS,EAAE,CAAC;IACtC,CAAC;IAEO,eAAe,CAAC,KAAoB;QAC1C,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC;QAChC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAE9D,OAAO;WACA,YAAY,YAAY,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;;;;;;;;;;;;;;;YAetD,YAAY;;kBAEN,YAAY,GAAG,CAAC;IAChC,CAAC;IAEO,mBAAmB,CAAC,YAAoB;QAC9C,wDAAwD;QACxD,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;YACpB,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAChD,OAAO,MAAM,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACzC,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,iBAAiB,CAAC,KAAoB,EAAE,OAG/C;QACC,IAAI,MAAM,GAAG;;YAEL,KAAK,CAAC,IAAI;;;;;;qBAMD,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,uBAAuB,CAAC,KAAK,CAAC,eAAe,CAAC;;;CAGpG,CAAC;QAEE,iCAAiC;QACjC,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACnC,MAAM,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,IAAI,KAAK,CAAC;QAChB,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,mBAAmB,CAAC,MAAmC,EAAE,KAAoB,EAAE,OAGtF;QACC,IAAI,MAAM,GAAG,oBAAoB,MAAM,CAAC,IAAI,YAAY,CAAC;QAEzD,kBAAkB;QAClB,MAAM,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAEpD,oCAAoC;QACpC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM,IAAI,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,qCAAqC;QACrC,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,IAAI,SAAS,CAAC;QACpB,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,qBAAqB,CAAC,MAAmC,EAAE,KAAoB;QACrF,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACpF,MAAM,UAAU,GAAG,oBAAoB,MAAM,CAAC,IAAI,IAAI,IAAI,GAAG,CAAC;QAE9D,IAAI,IAAI,GAAG;;;6BAGc,UAAU;;;;;CAKtC,CAAC;QAEE,uCAAuC;QACvC,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,IAAI,GAAG;;;;6BAIgB,UAAU;;;;;;;;;;;;CAYtC,CAAC;QACE,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,kBAAkB,CAAC,MAAmC,EAAE,KAAoB;QAClF,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,yBAAyB,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3F,MAAM,UAAU,GAAG,oBAAoB,MAAM,CAAC,IAAI,IAAI,IAAI,GAAG,CAAC;QAE9D,IAAI,IAAI,GAAG;;;;+BAIgB,UAAU;;;;;;;;CAQxC,CAAC;QAEE,yCAAyC;QACzC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,IAAI,GAAG;;;;6BAIgB,UAAU;;;;;;;CAOtC,CAAC;QACE,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,qBAAqB,CAAC,MAAmC,EAAE,KAAoB;QACrF,IAAI,KAAK,GAAG,EAAE,CAAC;QAEf,wCAAwC;QACxC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvE,MAAM,cAAc,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACpD,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAC7C,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,aAAa,CAAC;YAE7L,MAAM,UAAU,GAAG,oBAAoB,MAAM,CAAC,IAAI,IAAI,IAAI,GAAG,CAAC;YAE9D,KAAK,IAAI;sCACuB,KAAK,CAAC,IAAI,KAAK,aAAa;;;+BAGnC,UAAU;;;;;;;;;CASxC,CAAC;QACE,CAAC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AApND,sCAoNC"}

package/dist/generator/value-generator.d.ts

@@ -0,0 +1,19 @@
+import { ConstructorArg, MethodParameter } from '../model/contract-model';
+import { ContractModel } from '../model/contract-model';
+export declare class ValueGenerator {
+    private usedVariableNames;
+    private testKeyPairs;
+    constructor();
+    generateConstructorArgs(args: ConstructorArg[]): string;
+    generateMethodArgs(parameters: MethodParameter[], methodName: string): string;
+    generateInvalidMethodArgs(parameters: MethodParameter[], methodName: string): string;
+    private generateValueForType;
+    private generateInvalidValueForType;
+    getTestKey(name?: string): {
+        pubKey: string;
+        privKey: string;
+    };
+    getEdgeCaseValue(type: string, model?: ContractModel): string;
+    reset(): void;
+}
+//# sourceMappingURL=value-generator.d.ts.map

package/dist/generator/value-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"value-generator.d.ts","sourceRoot":"","sources":["../../src/generator/value-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,qBAAa,cAAc;IACzB,OAAO,CAAC,iBAAiB,CAAqB;IAC9C,OAAO,CAAC,YAAY,CAA0D;;IAkB9E,uBAAuB,CAAC,IAAI,EAAE,cAAc,EAAE,GAAG,MAAM;IAWvD,kBAAkB,CAAC,UAAU,EAAE,eAAe,EAAE,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM;IAM7E,yBAAyB,CAAC,UAAU,EAAE,eAAe,EAAE,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM;IAMpF,OAAO,CAAC,oBAAoB;IAuC5B,OAAO,CAAC,2BAA2B;IAqBnC,UAAU,CAAC,IAAI,GAAE,MAAkB,GAAG;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE;IAQzE,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,aAAa,GAAG,MAAM;IAoB7D,KAAK,IAAI,IAAI;CAGd"}

package/dist/generator/value-generator.js

@@ -0,0 +1,130 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ValueGenerator = void 0;
+class ValueGenerator {
+    constructor() {
+        this.usedVariableNames = new Set();
+        this.testKeyPairs = new Map();
+        // Initialize with some test key pairs
+        this.testKeyPairs.set('default', {
+            pubKey: 'myPublicKey',
+            privKey: 'myPrivateKey'
+        });
+        this.testKeyPairs.set('alice', {
+            pubKey: 'alicePublicKey',
+            privKey: 'alicePrivateKey'
+        });
+        this.testKeyPairs.set('bob', {
+            pubKey: 'bobPublicKey',
+            privKey: 'bobPrivateKey'
+        });
+    }
+    generateConstructorArgs(args) {
+        if (args.length === 0)
+            return '';
+        return args.map(arg => {
+            if (arg.defaultValue) {
+                return arg.defaultValue;
+            }
+            return this.generateValueForType(arg.type, arg.name);
+        }).join(', ');
+    }
+    generateMethodArgs(parameters, methodName) {
+        return parameters.map(param => {
+            return this.generateValueForType(param.type, `${methodName}_${param.name}`);
+        }).join(', ');
+    }
+    generateInvalidMethodArgs(parameters, methodName) {
+        return parameters.map(param => {
+            return this.generateInvalidValueForType(param.type, `${methodName}_${param.name}`);
+        }).join(', ');
+    }
+    generateValueForType(type, variableName, model) {
+        // Ensure unique variable names
+        let uniqueName = variableName;
+        let counter = 1;
+        while (this.usedVariableNames.has(uniqueName)) {
+            uniqueName = `${variableName}${counter}`;
+            counter++;
+        }
+        this.usedVariableNames.add(uniqueName);
+        switch (type) {
+            case 'bigint':
+                return '1n';
+            case 'boolean':
+                return 'true';
+            case 'ByteString':
+                return `toByteString('00', false)`;
+            case 'PubKey':
+                return this.getTestKey('default').pubKey;
+            case 'Sig':
+                return ` (sigResps) => findSig(sigResps, myPublicKey)),
+                {
+                    pubKeyOrAddrToSign: myPublicKey,
+                } as MethodCallOptions<${model?.name}>`;
+            case 'number':
+                return '42';
+            case 'string':
+                return `'test'`;
+            default:
+                if (type.includes('bigint[]')) {
+                    return '[1n, 2n, 3n]';
+                }
+                if (type.includes('[]')) {
+                    return `[]`;
+                }
+                return 'undefined';
+        }
+    }
+    generateInvalidValueForType(type, variableName) {
+        switch (type) {
+            case 'bigint':
+                return '0n'; // Often invalid for amounts
+            case 'boolean':
+                return 'false';
+            case 'ByteString':
+                return `toByteString('', false)`; // Empty byte string
+            case 'PubKey':
+                return `PubKey(toByteString('00', false))`; // Invalid pubkey
+            case 'Sig':
+                return `Sig(toByteString('00', false))`; // Invalid signature
+            case 'number':
+                return '-1';
+            case 'string':
+                return `''`;
+            default:
+                return this.generateValueForType(type, variableName);
+        }
+    }
+    getTestKey(name = 'default') {
+        const keyPair = this.testKeyPairs.get(name);
+        if (!keyPair) {
+            throw new Error(`Test key pair '${name}' not found`);
+        }
+        return keyPair;
+    }
+    getEdgeCaseValue(type, model) {
+        switch (type) {
+            case 'bigint':
+                return '0n';
+            case 'boolean':
+                return 'false';
+            case 'ByteString':
+                return `toByteString('ff', false)`;
+            case 'PubKey':
+                return `PubKey(toByteString('00'.repeat(33), false))`;
+            case 'Sig':
+                return ` (sigResps) => findSig(sigResps, myPublickey)),
+                {
+                    pubKeyOrAddrToSign: myPublicKey,
+                } as MethodCallOptions<${model?.name}>`;
+            default:
+                return this.generateValueForType(type, 'edgeCase');
+        }
+    }
+    reset() {
+        this.usedVariableNames.clear();
+    }
+}
+exports.ValueGenerator = ValueGenerator;
+//# sourceMappingURL=value-generator.js.map

package/dist/generator/value-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"value-generator.js","sourceRoot":"","sources":["../../src/generator/value-generator.ts"],"names":[],"mappings":";;;AAGA,MAAa,cAAc;IAIzB;QAHQ,sBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,iBAAY,GAAG,IAAI,GAAG,EAA+C,CAAC;QAG5E,sCAAsC;QACtC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE;YAC/B,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,cAAc;SACxB,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE;YAC7B,MAAM,EAAE,gBAAgB;YACxB,OAAO,EAAE,iBAAiB;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE;YAC3B,MAAM,EAAE,cAAc;YACtB,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;IACL,CAAC;IAED,uBAAuB,CAAC,IAAsB;QAC5C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YACpB,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;gBACrB,OAAO,GAAG,CAAC,YAAY,CAAC;YAC1B,CAAC;YACD,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,kBAAkB,CAAC,UAA6B,EAAE,UAAkB;QAClE,OAAO,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YAC5B,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,UAAU,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,yBAAyB,CAAC,UAA6B,EAAE,UAAkB;QACzE,OAAO,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YAC5B,OAAO,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,UAAU,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAEO,oBAAoB,CAAC,IAAY,EAAE,YAAoB,EAAE,KAAqB;QACpF,+BAA+B;QAC/B,IAAI,UAAU,GAAG,YAAY,CAAC;QAC9B,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9C,UAAU,GAAG,GAAG,YAAY,GAAG,OAAO,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAEvC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd,KAAK,SAAS;gBACZ,OAAO,MAAM,CAAC;YAChB,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAAC;YACrC,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;YAC3C,KAAK,KAAK;gBACR,OAAO;;;yCAG0B,KAAK,EAAE,IAAI,GAAG,CAAC;YAClD,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB;gBACE,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBAC9B,OAAO,cAAc,CAAC;gBACxB,CAAC;gBACD,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACxB,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,WAAW,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,2BAA2B,CAAC,IAAY,EAAE,YAAoB;QACpE,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC,CAAC,4BAA4B;YAC3C,KAAK,SAAS;gBACZ,OAAO,OAAO,CAAC;YACjB,KAAK,YAAY;gBACf,OAAO,yBAAyB,CAAC,CAAC,oBAAoB;YACxD,KAAK,QAAQ;gBACX,OAAO,mCAAmC,CAAC,CAAC,iBAAiB;YAC/D,KAAK,KAAK;gBACR,OAAO,gCAAgC,CAAC,CAAC,oBAAoB;YAC/D,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd;gBACE,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,UAAU,CAAC,OAAe,SAAS;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,aAAa,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,gBAAgB,CAAC,IAAY,EAAE,KAAqB;QAClD,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC;YACd,KAAK,SAAS;gBACZ,OAAO,OAAO,CAAC;YACjB,KAAK,YAAY;gBACf,OAAO,2BAA2B,CAAC;YACrC,KAAK,QAAQ;gBACX,OAAO,8CAA8C,CAAC;YACxD,KAAK,KAAK;gBACR,OAAO;;;yCAG0B,KAAK,EAAE,IAAI,GAAG,CAAC;YAClD;gBACE,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;CACF;AAtID,wCAsIC"}

package/dist/model/contract-model.d.ts

@@ -0,0 +1,42 @@
+export interface ConstructorArg {
+    name: string;
+    type: string;
+    defaultValue?: string;
+}
+export interface ContractProp {
+    name: string;
+    type: string;
+    isMutable: boolean;
+    decoratorText: string;
+}
+export interface MethodParameter {
+    name: string;
+    type: string;
+    isOptional: boolean;
+}
+export declare enum MethodCategory {
+    PURE_VALIDATION = "pure-validation",
+    STATE_TRANSITION = "state-transition",
+    SPENDING_CONSTRAINT = "spending-constraint"
+}
+export interface ContractMethod {
+    name: string;
+    parameters: MethodParameter[];
+    returnType: string;
+    category: MethodCategory;
+    isPublic: boolean;
+    mutatesState: boolean;
+    usesHashOutputs: boolean;
+    hasAssert: boolean;
+    decoratorText: string;
+}
+export interface ContractModel {
+    name: string;
+    filePath: string;
+    constructorArgs: ConstructorArg[];
+    props: ContractProp[];
+    methods: ContractMethod[];
+    imports: string[];
+    extendsClass: string;
+}
+//# sourceMappingURL=contract-model.d.ts.map

package/dist/model/contract-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contract-model.d.ts","sourceRoot":"","sources":["../../src/model/contract-model.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,OAAO,CAAC;CACrB;AAED,oBAAY,cAAc;IACxB,eAAe,oBAAoB;IACnC,gBAAgB,qBAAqB;IACrC,mBAAmB,wBAAwB;CAC5C;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,cAAc,CAAC;IACzB,QAAQ,EAAE,OAAO,CAAC;IAClB,YAAY,EAAE,OAAO,CAAC;IACtB,eAAe,EAAE,OAAO,CAAC;IACzB,SAAS,EAAE,OAAO,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC,KAAK,EAAE,YAAY,EAAE,CAAC;IACtB,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;CACtB"}

package/dist/model/contract-model.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MethodCategory = void 0;
+var MethodCategory;
+(function (MethodCategory) {
+    MethodCategory["PURE_VALIDATION"] = "pure-validation";
+    MethodCategory["STATE_TRANSITION"] = "state-transition";
+    MethodCategory["SPENDING_CONSTRAINT"] = "spending-constraint";
+})(MethodCategory || (exports.MethodCategory = MethodCategory = {}));
+//# sourceMappingURL=contract-model.js.map

package/dist/model/contract-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contract-model.js","sourceRoot":"","sources":["../../src/model/contract-model.ts"],"names":[],"mappings":";;;AAmBA,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,qDAAmC,CAAA;IACnC,uDAAqC,CAAA;IACrC,6DAA2C,CAAA;AAC7C,CAAC,EAJW,cAAc,8BAAd,cAAc,QAIzB"}

package/dist/parser/ast-utils.d.ts

@@ -0,0 +1,7 @@
+import * as ts from 'typescript';
+export declare function getDecoratorText(decorator: ts.Decorator): string;
+export declare function getTypeText(type: ts.TypeNode | undefined, checker: ts.TypeChecker): string;
+export declare function getDefaultValueForType(type: string): string;
+export declare function isSmartContractClass(node: ts.Node): boolean;
+export declare function findDecorator(node: ts.Node, decoratorName: string): ts.Decorator | undefined;
+//# sourceMappingURL=ast-utils.d.ts.map

package/dist/parser/ast-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-utils.d.ts","sourceRoot":"","sources":["../../src/parser/ast-utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAEjC,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,GAAG,MAAM,CAQhE;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,GAAG,SAAS,EAAE,OAAO,EAAE,EAAE,CAAC,WAAW,GAAG,MAAM,CAc1F;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAoB3D;AAED,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,GAAG,OAAO,CAe3D;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,GAAG,EAAE,CAAC,SAAS,GAAG,SAAS,CAO5F"}