scrypt-testgen 1.0.0

package/dist/parser/ast-utils.js

@@ -0,0 +1,114 @@
+"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.getDecoratorText = getDecoratorText;
+exports.getTypeText = getTypeText;
+exports.getDefaultValueForType = getDefaultValueForType;
+exports.isSmartContractClass = isSmartContractClass;
+exports.findDecorator = findDecorator;
+const ts = __importStar(require("typescript"));
+function getDecoratorText(decorator) {
+    const decoratorText = decorator.getText();
+    if (decoratorText.includes('@prop')) {
+        return '@prop';
+    }
+    else if (decoratorText.includes('@method')) {
+        return '@method';
+    }
+    return decoratorText;
+}
+function getTypeText(type, checker) {
+    if (!type)
+        return 'any';
+    const typeText = type.getText();
+    // Handle sCrypt specific types
+    if (typeText.includes('bigint'))
+        return 'bigint';
+    if (typeText.includes('ByteString'))
+        return 'ByteString';
+    if (typeText.includes('PubKey'))
+        return 'PubKey';
+    if (typeText.includes('Sig'))
+        return 'Sig';
+    if (typeText.includes('boolean'))
+        return 'boolean';
+    if (typeText.includes('number'))
+        return 'number';
+    return typeText;
+}
+function getDefaultValueForType(type) {
+    switch (type) {
+        case 'bigint':
+            return '1n';
+        case 'boolean':
+            return 'true';
+        case 'ByteString':
+            return `toByteString('00', false)`;
+        case 'PubKey':
+            return `myPublicKey`;
+        case 'Sig':
+            return `mySignature`;
+        case 'number':
+            return '0';
+        case 'string':
+            return "''";
+        default:
+            if (type.includes('[]'))
+                return '[]';
+            return 'undefined';
+    }
+}
+function isSmartContractClass(node) {
+    if (!ts.isClassDeclaration(node))
+        return false;
+    // Check if extends SmartContract
+    if (node.heritageClauses) {
+        for (const heritage of node.heritageClauses) {
+            for (const type of heritage.types) {
+                if (type.getText().includes('SmartContract')) {
+                    return true;
+                }
+            }
+        }
+    }
+    return false;
+}
+function findDecorator(node, decoratorName) {
+    if (!ts.canHaveDecorators(node))
+        return undefined;
+    const decorators = ts.getDecorators(node);
+    return decorators?.find((decorator) => decorator.getText().includes(decoratorName));
+}
+//# sourceMappingURL=ast-utils.js.map

package/dist/parser/ast-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-utils.js","sourceRoot":"","sources":["../../src/parser/ast-utils.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEA,4CAQC;AAED,kCAcC;AAED,wDAoBC;AAED,oDAeC;AAED,sCAOC;AA1ED,+CAAiC;AAEjC,SAAgB,gBAAgB,CAAC,SAAuB;IACtD,MAAM,aAAa,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;IAC1C,IAAI,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACpC,OAAO,OAAO,CAAC;IACjB,CAAC;SAAM,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAgB,WAAW,CAAC,IAA6B,EAAE,OAAuB;IAChF,IAAI,CAAC,IAAI;QAAE,OAAO,KAAK,CAAC;IAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAEhC,+BAA+B;IAC/B,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IACjD,IAAI,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;QAAE,OAAO,YAAY,CAAC;IACzD,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IACjD,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC3C,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IACnD,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAEjD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,sBAAsB,CAAC,IAAY;IACjD,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC;QACd,KAAK,SAAS;YACZ,OAAO,MAAM,CAAC;QAChB,KAAK,YAAY;YACf,OAAO,2BAA2B,CAAC;QACrC,KAAK,QAAQ;YACX,OAAO,aAAa,CAAC;QACvB,KAAK,KAAK;YACR,OAAO,aAAa,CAAC;QACvB,KAAK,QAAQ;YACX,OAAO,GAAG,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC;QACd;YACE,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YACrC,OAAO,WAAW,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAgB,oBAAoB,CAAC,IAAa;IAChD,IAAI,CAAC,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAE/C,iCAAiC;IACjC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;QACzB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAClC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;oBAC7C,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAgB,aAAa,CAAC,IAAa,EAAE,aAAqB;IAChE,IAAI,CAAC,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAElD,MAAM,UAAU,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC1C,OAAO,UAAU,EAAE,IAAI,CAAC,CAAC,SAAuB,EAAE,EAAE,CAClD,SAAS,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAC5C,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,18 @@
+import { ContractModel } from '../model/contract-model';
+export declare class ContractParser {
+    private program;
+    private checker;
+    private sourceFile;
+    constructor(filePath: string);
+    parse(): ContractModel;
+    private findContractClass;
+    private parseConstructor;
+    private parseProperties;
+    private parseMethods;
+    private parseMethodParameters;
+    private classifyMethod;
+    private doesMethodMutateState;
+    private parseImports;
+    private getExtendedClass;
+}
+//# sourceMappingURL=contract-parser.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"contract-parser.d.ts","sourceRoot":"","sources":["../../src/parser/contract-parser.ts"],"names":[],"mappings":"AAEA,OAAO,EAML,aAAa,EACd,MAAM,yBAAyB,CAAC;AASjC,qBAAa,cAAc;IACzB,OAAO,CAAC,OAAO,CAAa;IAC5B,OAAO,CAAC,OAAO,CAAiB;IAChC,OAAO,CAAC,UAAU,CAAgB;gBAEtB,QAAQ,EAAE,MAAM;IAgB5B,KAAK,IAAI,aAAa;IAmBtB,OAAO,CAAC,iBAAiB;IAiBzB,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,eAAe;IA4BvB,OAAO,CAAC,YAAY;IAgDpB,OAAO,CAAC,qBAAqB;IAkB7B,OAAO,CAAC,cAAc;IAqBtB,OAAO,CAAC,qBAAqB;IAM7B,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,gBAAgB;CAczB"}

package/dist/parser/contract-parser.js

@@ -0,0 +1,224 @@
+"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.ContractParser = void 0;
+const ts = __importStar(require("typescript"));
+const contract_model_1 = require("../model/contract-model");
+const ast_utils_1 = require("./ast-utils");
+class ContractParser {
+    constructor(filePath) {
+        this.program = ts.createProgram([filePath], {
+            target: ts.ScriptTarget.ES2020,
+            module: ts.ModuleKind.CommonJS,
+            experimentalDecorators: true,
+            emitDecoratorMetadata: true
+        });
+        this.checker = this.program.getTypeChecker();
+        this.sourceFile = this.program.getSourceFile(filePath);
+        if (!this.sourceFile) {
+            throw new Error(`Cannot read source file: ${filePath}`);
+        }
+    }
+    parse() {
+        const contractClass = this.findContractClass();
+        if (!contractClass) {
+            throw new Error('No SmartContract class found in file');
+        }
+        const className = contractClass.name?.text || 'UnknownContract';
+        return {
+            name: className,
+            filePath: this.sourceFile.fileName,
+            constructorArgs: this.parseConstructor(contractClass),
+            props: this.parseProperties(contractClass),
+            methods: this.parseMethods(contractClass),
+            imports: this.parseImports(),
+            extendsClass: this.getExtendedClass(contractClass)
+        };
+    }
+    findContractClass() {
+        let contractClass;
+        const visit = (node) => {
+            if ((0, ast_utils_1.isSmartContractClass)(node)) {
+                if (contractClass) {
+                    throw new Error('Multiple SmartContract classes found. Only one per file is supported.');
+                }
+                contractClass = node;
+            }
+            ts.forEachChild(node, visit);
+        };
+        visit(this.sourceFile);
+        return contractClass;
+    }
+    parseConstructor(classNode) {
+        const constructor = classNode.members.find(member => ts.isConstructorDeclaration(member));
+        if (!constructor || !constructor.parameters) {
+            return [];
+        }
+        return constructor.parameters.map(param => {
+            const paramName = param.name.getText();
+            const type = (0, ast_utils_1.getTypeText)(param.type, this.checker);
+            return {
+                name: paramName,
+                type: type,
+                defaultValue: param.initializer ? param.initializer.getText() : undefined
+            };
+        });
+    }
+    parseProperties(classNode) {
+        const properties = [];
+        for (const member of classNode.members) {
+            if (ts.isPropertyDeclaration(member)) {
+                const decorators = ts.getDecorators(member);
+                if (decorators && decorators.length > 0) {
+                    const propDecorator = (0, ast_utils_1.findDecorator)(member, '@prop');
+                    if (propDecorator) {
+                        const propName = member.name.getText();
+                        const type = (0, ast_utils_1.getTypeText)(member.type, this.checker);
+                        const decoratorText = decorators[0].getText();
+                        const isMutable = decoratorText.includes('true');
+                        properties.push({
+                            name: propName,
+                            type: type,
+                            isMutable: isMutable,
+                            decoratorText: decoratorText
+                        });
+                    }
+                }
+            }
+        }
+        return properties;
+    }
+    parseMethods(classNode) {
+        const methods = [];
+        for (const member of classNode.members) {
+            if (ts.isMethodDeclaration(member)) {
+                const decorators = ts.getDecorators(member);
+                if (decorators && decorators.length > 0) {
+                    const methodDecorator = (0, ast_utils_1.findDecorator)(member, '@method');
+                    if (methodDecorator) {
+                        const methodName = member.name.getText();
+                        // Skip private methods
+                        if (!member.modifiers?.some(m => m.kind === ts.SyntaxKind.PublicKeyword)) {
+                            continue;
+                        }
+                        const parameters = this.parseMethodParameters(member);
+                        const returnType = (0, ast_utils_1.getTypeText)(member.type, this.checker);
+                        const methodText = member.getText();
+                        const usesHashOutputs = methodText.includes('this.ctx.hashOutputs');
+                        const hasAssert = methodText.includes('assert(');
+                        // Classify method
+                        const category = this.classifyMethod(member, methodText);
+                        // Determine if method mutates state
+                        const mutatesState = this.doesMethodMutateState(member, methodText);
+                        methods.push({
+                            name: methodName,
+                            parameters: parameters,
+                            returnType: returnType,
+                            category: category,
+                            isPublic: true,
+                            mutatesState: mutatesState,
+                            usesHashOutputs: usesHashOutputs,
+                            hasAssert: hasAssert,
+                            decoratorText: (0, ast_utils_1.getDecoratorText)(methodDecorator)
+                        });
+                    }
+                }
+            }
+        }
+        return methods;
+    }
+    parseMethodParameters(method) {
+        if (!method.parameters) {
+            return [];
+        }
+        return method.parameters.map(param => {
+            const paramName = param.name.getText();
+            const type = (0, ast_utils_1.getTypeText)(param.type, this.checker);
+            const isOptional = !!param.questionToken || !!param.initializer;
+            return {
+                name: paramName,
+                type: type,
+                isOptional: isOptional
+            };
+        });
+    }
+    classifyMethod(method, methodText) {
+        const returnType = method.type?.getText() || 'void';
+        // Methods that return boolean are often validation methods
+        if (returnType === 'boolean') {
+            return contract_model_1.MethodCategory.PURE_VALIDATION;
+        }
+        // Methods that modify @prop(true) properties
+        if (this.doesMethodMutateState(method, methodText)) {
+            return contract_model_1.MethodCategory.STATE_TRANSITION;
+        }
+        // Methods that check spending conditions
+        if (methodText.includes('this.ctx') || methodText.includes('hashOutputs')) {
+            return contract_model_1.MethodCategory.SPENDING_CONSTRAINT;
+        }
+        return contract_model_1.MethodCategory.PURE_VALIDATION;
+    }
+    doesMethodMutateState(method, methodText) {
+        // Check for assignment to this.props (especially mutable ones)
+        const assignmentRegex = /this\.\w+\s*=/;
+        return assignmentRegex.test(methodText);
+    }
+    parseImports() {
+        const imports = [];
+        const visit = (node) => {
+            if (ts.isImportDeclaration(node)) {
+                imports.push(node.getText());
+            }
+            ts.forEachChild(node, visit);
+        };
+        visit(this.sourceFile);
+        return imports;
+    }
+    getExtendedClass(classNode) {
+        if (!classNode.heritageClauses)
+            return '';
+        for (const heritage of classNode.heritageClauses) {
+            for (const type of heritage.types) {
+                const typeText = type.getText();
+                if (typeText.includes('SmartContract')) {
+                    return typeText;
+                }
+            }
+        }
+        return '';
+    }
+}
+exports.ContractParser = ContractParser;
+//# sourceMappingURL=contract-parser.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"contract-parser.js","sourceRoot":"","sources":["../../src/parser/contract-parser.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AAEjC,4DAOiC;AACjC,2CAMqB;AAErB,MAAa,cAAc;IAKzB,YAAY,QAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,EAAE;YAC1C,MAAM,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM;YAC9B,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ;YAC9B,sBAAsB,EAAE,IAAI;YAC5B,qBAAqB,EAAE,IAAI;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAE,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK;QACH,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC/C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,IAAI,IAAI,iBAAiB,CAAC;QAEhE,OAAO;YACL,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ;YAClC,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC;YACrD,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC;YAC1C,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;YACzC,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;YAC5B,YAAY,EAAE,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC;SACnD,CAAC;IACJ,CAAC;IAEO,iBAAiB;QACvB,IAAI,aAA8C,CAAC;QAEnD,MAAM,KAAK,GAAG,CAAC,IAAa,EAAE,EAAE;YAC9B,IAAI,IAAA,gCAAoB,EAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,IAAI,aAAa,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAC;gBAC3F,CAAC;gBACD,aAAa,GAAG,IAA2B,CAAC;YAC9C,CAAC;YACD,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC;QAEF,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvB,OAAO,aAAa,CAAC;IACvB,CAAC;IAEO,gBAAgB,CAAC,SAA8B;QACrD,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CACxC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC,CACjB,CAAC;QAE/B,IAAI,CAAC,WAAW,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC5C,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YACxC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,IAAA,uBAAW,EAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAEnD,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;aAC1E,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,SAA8B;QACpD,MAAM,UAAU,GAAmB,EAAE,CAAC;QAEtC,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACvC,IAAI,EAAE,CAAC,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;gBACrC,MAAM,UAAU,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,MAAM,aAAa,GAAG,IAAA,yBAAa,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;oBACrD,IAAI,aAAa,EAAE,CAAC;wBAClB,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACvC,MAAM,IAAI,GAAG,IAAA,uBAAW,EAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;wBACpD,MAAM,aAAa,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;wBAC9C,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;wBAEjD,UAAU,CAAC,IAAI,CAAC;4BACd,IAAI,EAAE,QAAQ;4BACd,IAAI,EAAE,IAAI;4BACV,SAAS,EAAE,SAAS;4BACpB,aAAa,EAAE,aAAa;yBAC7B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,YAAY,CAAC,SAA8B;QACjD,MAAM,OAAO,GAAqB,EAAE,CAAC;QAErC,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACvC,IAAI,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnC,MAAM,UAAU,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,MAAM,eAAe,GAAG,IAAA,yBAAa,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;oBACzD,IAAI,eAAe,EAAE,CAAC;wBACpB,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;wBAEzC,uBAAuB;wBACvB,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;4BACzE,SAAS;wBACX,CAAC;wBAED,MAAM,UAAU,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;wBACtD,MAAM,UAAU,GAAG,IAAA,uBAAW,EAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;wBAE1D,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;wBACpC,MAAM,eAAe,GAAG,UAAU,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;wBACpE,MAAM,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;wBAEjD,kBAAkB;wBAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;wBAEzD,oCAAoC;wBACpC,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;wBAEpE,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,UAAU;4BAChB,UAAU,EAAE,UAAU;4BACtB,UAAU,EAAE,UAAU;4BACtB,QAAQ,EAAE,QAAQ;4BAClB,QAAQ,EAAE,IAAI;4BACd,YAAY,EAAE,YAAY;4BAC1B,eAAe,EAAE,eAAe;4BAChC,SAAS,EAAE,SAAS;4BACpB,aAAa,EAAE,IAAA,4BAAgB,EAAC,eAAe,CAAC;yBACjD,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,qBAAqB,CAAC,MAA4B;QACxD,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YACnC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,IAAA,uBAAW,EAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC;YAEhE,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,IAAI;gBACV,UAAU,EAAE,UAAU;aACvB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,MAA4B,EAAE,UAAkB;QACrE,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,MAAM,CAAC;QAEpD,2DAA2D;QAC3D,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO,+BAAc,CAAC,eAAe,CAAC;QACxC,CAAC;QAED,6CAA6C;QAC7C,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,CAAC;YACnD,OAAO,+BAAc,CAAC,gBAAgB,CAAC;QACzC,CAAC;QAED,yCAAyC;QACzC,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YAC1E,OAAO,+BAAc,CAAC,mBAAmB,CAAC;QAC5C,CAAC;QAED,OAAO,+BAAc,CAAC,eAAe,CAAC;IACxC,CAAC;IAEO,qBAAqB,CAAC,MAA4B,EAAE,UAAkB;QAC5E,+DAA+D;QAC/D,MAAM,eAAe,GAAG,eAAe,CAAC;QACxC,OAAO,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,YAAY;QAClB,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,MAAM,KAAK,GAAG,CAAC,IAAa,EAAE,EAAE;YAC9B,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/B,CAAC;YACD,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC,CAAC;QAEF,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvB,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CAAC,SAA8B;QACrD,IAAI,CAAC,SAAS,CAAC,eAAe;YAAE,OAAO,EAAE,CAAC;QAE1C,KAAK,MAAM,QAAQ,IAAI,SAAS,CAAC,eAAe,EAAE,CAAC;YACjD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;oBACvC,OAAO,QAAQ,CAAC;gBAClB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAnOD,wCAmOC"}

package/dist/parser/index.d.ts

@@ -0,0 +1,3 @@
+export { ContractParser } from './contract-parser';
+export type { ContractModel } from '../model/contract-model';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/parser/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,YAAY,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC"}

package/dist/parser/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContractParser = void 0;
+var contract_parser_1 = require("./contract-parser");
+Object.defineProperty(exports, "ContractParser", { enumerable: true, get: function () { return contract_parser_1.ContractParser; } });
+//# sourceMappingURL=index.js.map

package/dist/parser/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/parser/index.ts"],"names":[],"mappings":";;;AAAA,qDAAmD;AAA1C,iHAAA,cAAc,OAAA"}

package/dist/utils/file-utils.d.ts

@@ -0,0 +1,4 @@
+export declare function ensureDirectoryExists(filePath: string): void;
+export declare function readContractFile(filePath: string): string;
+export declare function writeTestFile(filePath: string, content: string): void;
+//# sourceMappingURL=file-utils.d.ts.map

package/dist/utils/file-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.d.ts","sourceRoot":"","sources":["../../src/utils/file-utils.ts"],"names":[],"mappings":"AAGA,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAK5D;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAMzD;AAED,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAGrE"}

package/dist/utils/file-utils.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ensureDirectoryExists = ensureDirectoryExists;
+exports.readContractFile = readContractFile;
+exports.writeTestFile = writeTestFile;
+const fs_1 = require("fs");
+const path_1 = require("path");
+function ensureDirectoryExists(filePath) {
+    const dir = (0, path_1.dirname)(filePath);
+    if (!(0, fs_1.existsSync)(dir)) {
+        (0, fs_1.mkdirSync)(dir, { recursive: true });
+    }
+}
+function readContractFile(filePath) {
+    try {
+        return (0, fs_1.readFileSync)(filePath, 'utf-8');
+    }
+    catch (error) {
+        throw new Error(`Cannot read contract file: ${filePath}. ${error}`);
+    }
+}
+function writeTestFile(filePath, content) {
+    ensureDirectoryExists(filePath);
+    (0, fs_1.writeFileSync)(filePath, content, 'utf-8');
+}
+//# sourceMappingURL=file-utils.js.map

package/dist/utils/file-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.js","sourceRoot":"","sources":["../../src/utils/file-utils.ts"],"names":[],"mappings":";;AAGA,sDAKC;AAED,4CAMC;AAED,sCAGC;AArBD,2BAAwE;AACxE,+BAA+B;AAE/B,SAAgB,qBAAqB,CAAC,QAAgB;IACpD,MAAM,GAAG,GAAG,IAAA,cAAO,EAAC,QAAQ,CAAC,CAAC;IAC9B,IAAI,CAAC,IAAA,eAAU,EAAC,GAAG,CAAC,EAAE,CAAC;QACrB,IAAA,cAAS,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,SAAgB,gBAAgB,CAAC,QAAgB;IAC/C,IAAI,CAAC;QACH,OAAO,IAAA,iBAAY,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,KAAK,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;AACH,CAAC;AAED,SAAgB,aAAa,CAAC,QAAgB,EAAE,OAAe;IAC7D,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAChC,IAAA,kBAAa,EAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAC5C,CAAC"}

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "scrypt-testgen",
+  "version": "1.0.0",
+  "description": "Automatic test generator for sCrypt-ts smart contracts",
+  "main": "dist/cli.js",
+  "bin": {
+    "scrypt-testgen": "./bin/scrypt-testgen.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "start": "ts-node src/cli.ts",
+    "dev": "ts-node src/cli.ts",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": ["scrypt", "bitcoin", "smart-contract", "test", "mocha"],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@types/node": "^20.0.0",
+    "chalk": "^4.1.2",
+    "commander": "^11.0.0",
+    "typescript": "^5.0.0"
+  },
+  "devDependencies": {
+    "@types/chai": "^4.3.0",
+    "@types/mocha": "^10.0.0",
+    "ts-node": "^10.9.0",
+    "prettier": "^3.0.0"
+  }
+}

package/src/cli.ts

@@ -0,0 +1,51 @@
+#!/usr/bin/env node
+
+import { Command } from 'commander';
+import { generateTests } from './generator';
+import { resolve, dirname, basename, join } from 'path';
+import { mkdirSync, existsSync } from 'fs';
+import chalk from 'chalk';
+
+const program = new 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 = resolve(process.cwd(), contractPath);
+      
+      if (!existsSync(resolvedPath)) {
+        console.error(chalk.red(`Error: Contract file not found: ${resolvedPath}`));
+        process.exit(1);
+      }
+
+      const outputPath = options.output 
+        ? resolve(process.cwd(), options.output)
+        : join(dirname(resolvedPath), '..', 'test', basename(resolvedPath).replace('.ts', '.test.ts'));
+
+      // Ensure test directory exists
+      mkdirSync(dirname(outputPath), { recursive: true });
+
+      await generateTests(resolvedPath, outputPath, {
+        minimal: options.minimal,
+        coverage: options.coverage,
+        overwrite: options.overwrite
+      });
+
+      console.log(chalk.green(`✓ Tests generated: ${outputPath}`));
+    } catch (error) {
+      console.error(chalk.red(`Error: ${error}`));
+      process.exit(1);
+    }
+  });
+
+program.parse();

package/src/generator/index.ts

@@ -0,0 +1,45 @@
+import { ContractParser } from '../parser';
+import { TestGenerator } from './test-generator';
+import { existsSync, writeFileSync, readFileSync } from 'fs';
+import * as path from 'path';
+
+export async function generateTests(
+  contractPath: string,
+  outputPath: string,
+  options: {
+    minimal: boolean;
+    coverage: boolean;
+    overwrite: boolean;
+  }
+): Promise<void> {
+  
+  // Check if output file exists
+  if (existsSync(outputPath) && !options.overwrite) {
+    throw new Error(`Test file already exists: ${outputPath}. Use --overwrite to replace.`);
+  }
+
+  // Parse the contract
+  const parser = new ContractParser(contractPath);
+  const model = parser.parse();
+
+  // Generate tests
+  const generator = new 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
+  writeFileSync(outputPath, formattedCode, 'utf-8');
+}
+
+function formatCode(code: string): string {
+  // Simple formatting - in production, integrate with Prettier
+  return code
+    .replace(/\n\s*\n\s*\n/g, '\n\n')
+    .replace(/\s+$/gm, '')
+    .trim() + '\n';
+}