@openzeppelin/wizard-stellar 0.6.1 → 0.6.3

package/dist/vault.js

@@ -0,0 +1,334 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.functions = exports.defaults = void 0;
+exports.printVault = printVault;
+exports.withDefaults = withDefaults;
+exports.isAccessControlRequired = isAccessControlRequired;
+exports.buildVault = buildVault;
+const contract_1 = require("./contract");
+const set_access_control_1 = require("./set-access-control");
+const add_pausable_1 = require("./add-pausable");
+const add_upgradeable_1 = require("./add-upgradeable");
+const define_functions_1 = require("./utils/define-functions");
+const common_options_1 = require("./common-options");
+const set_info_1 = require("./set-info");
+const common_options_2 = require("./common-options");
+const print_1 = require("./print");
+const convert_strings_1 = require("./utils/convert-strings");
+const error_1 = require("./error");
+const wizard_common_1 = require("@openzeppelin/wizard-common");
+// Upper bound enforced by the library's `Vault::set_decimals_offset`
+// (`stellar_tokens::vault::MAX_DECIMALS_OFFSET`); a larger value panics on deploy.
+const MAX_DECIMALS_OFFSET = 10;
+exports.defaults = {
+    name: 'MyVault',
+    symbol: 'MTK',
+    decimalsOffset: '0',
+    pausable: false,
+    upgradeable: false,
+    access: common_options_2.contractDefaults.access,
+    info: common_options_2.contractDefaults.info,
+    explicitImplementations: common_options_2.contractDefaults.explicitImplementations,
+};
+function printVault(opts = exports.defaults) {
+    return (0, print_1.printContract)(buildVault(opts));
+}
+function withDefaults(opts) {
+    return {
+        ...opts,
+        ...(0, common_options_1.withCommonContractDefaults)(opts),
+        decimalsOffset: opts.decimalsOffset ?? exports.defaults.decimalsOffset,
+        pausable: opts.pausable ?? exports.defaults.pausable,
+        upgradeable: opts.upgradeable ?? exports.defaults.upgradeable,
+    };
+}
+function isAccessControlRequired(opts) {
+    return opts.pausable === true || opts.upgradeable === true;
+}
+function buildVault(opts) {
+    const c = new contract_1.ContractBuilder(opts.name);
+    const allOpts = withDefaults(opts);
+    const decimalsOffset = (0, convert_strings_1.toUint)(allOpts.decimalsOffset, 'decimalsOffset', 'u32');
+    if (decimalsOffset > MAX_DECIMALS_OFFSET) {
+        throw new error_1.OptionsError({
+            decimalsOffset: `Maximum decimals offset is ${MAX_DECIMALS_OFFSET}`,
+        });
+    }
+    addBase(c, (0, convert_strings_1.toByteArray)(allOpts.name), (0, convert_strings_1.toByteArray)(allOpts.symbol), decimalsOffset, allOpts.pausable, allOpts.explicitImplementations);
+    if (allOpts.pausable) {
+        (0, add_pausable_1.addPausable)(c, allOpts.access, allOpts.explicitImplementations);
+    }
+    if (allOpts.upgradeable) {
+        (0, add_upgradeable_1.addUpgradeable)(c, allOpts.access, allOpts.explicitImplementations);
+    }
+    (0, set_access_control_1.setAccessControl)(c, allOpts.access, allOpts.explicitImplementations);
+    // The AccessControl trait surface (e.g. `get_existing_roles`, role lookups)
+    // references `Symbol` and `Vec`, which the contracttrait macro and explicit
+    // implementations need in scope even when no function is role-gated.
+    if (allOpts.access === 'roles') {
+        c.addUseClause('soroban_sdk', 'Symbol');
+        c.addUseClause('soroban_sdk', 'Vec');
+    }
+    (0, set_info_1.setInfo)(c, allOpts.info);
+    return c;
+}
+function addBase(c, name, symbol, decimalsOffset, pausable, explicitImplementations) {
+    // The underlying asset is supplied at deployment, while the virtual decimals
+    // offset is a baked-in configuration value. The offset must be set before the
+    // metadata because the vault derives its decimals from the underlying asset's
+    // decimals plus the offset.
+    c.addConstructorArgument({ name: 'asset', type: 'Address' });
+    c.addConstructorCode('Vault::set_asset(e, asset);');
+    c.addConstructorCode(`Vault::set_decimals_offset(e, ${decimalsOffset});`);
+    c.addConstructorCode(`Base::set_metadata(e, Self::decimals(e), String::from_str(e, "${name}"), String::from_str(e, "${symbol}"));`);
+    // Set token functions
+    c.addUseClause('stellar_tokens::fungible', 'Base');
+    c.addUseClause('stellar_tokens::fungible', 'FungibleToken');
+    c.addUseClause('stellar_tokens::vault', 'FungibleVault');
+    c.addUseClause('stellar_tokens::vault', 'Vault');
+    // The base token functions we emit delegate to `Self::ContractType::...`,
+    // which resolve through the `ContractOverrides` trait that `Vault` implements,
+    // so it must be in scope. This happens with explicit implementations (all
+    // functions are spelled out) and when pausable adds transfer overrides.
+    if (explicitImplementations || pausable)
+        c.addUseClause('stellar_tokens::fungible', 'ContractOverrides');
+    c.addUseClause('soroban_sdk', 'contract');
+    c.addUseClause('soroban_sdk', 'contractimpl');
+    c.addUseClause('soroban_sdk', 'String');
+    c.addUseClause('soroban_sdk', 'Env');
+    c.addUseClause('soroban_sdk', 'Address');
+    c.addUseClause('soroban_sdk', 'MuxedAddress');
+    const fungibleTokenTrait = {
+        traitName: 'FungibleToken',
+        structName: c.name,
+        tags: explicitImplementations ? ['contractimpl'] : ['contractimpl(contracttrait)'],
+        assocType: 'type ContractType = Vault;',
+    };
+    c.addTraitImplBlock(fungibleTokenTrait);
+    if (explicitImplementations) {
+        c.addTraitForEachFunctions(fungibleTokenTrait, fungibleTokenTraitFunctions);
+    }
+    else {
+        // The vault overrides decimals to derive them from the underlying asset and
+        // the virtual decimals offset.
+        c.addTraitFunction(fungibleTokenTrait, exports.functions.decimals);
+    }
+    if (pausable) {
+        c.addUseClause('stellar_macros', 'when_not_paused');
+        c.addTraitFunction(fungibleTokenTrait, exports.functions.transfer);
+        c.addFunctionTag(exports.functions.transfer, 'when_not_paused', fungibleTokenTrait);
+        c.addTraitFunction(fungibleTokenTrait, exports.functions.transfer_from);
+        c.addFunctionTag(exports.functions.transfer_from, 'when_not_paused', fungibleTokenTrait);
+    }
+    const fungibleVaultTrait = {
+        traitName: 'FungibleVault',
+        structName: c.name,
+        tags: explicitImplementations ? ['contractimpl'] : ['contractimpl(contracttrait)'],
+        section: 'Extensions',
+    };
+    if (explicitImplementations) {
+        c.addTraitForEachFunctions(fungibleVaultTrait, fungibleVaultTraitFunctions);
+    }
+    else {
+        c.addTraitImplBlock(fungibleVaultTrait);
+    }
+}
+exports.functions = (0, define_functions_1.defineFunctions)({
+    // FungibleToken Trait
+    total_supply: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'i128',
+        code: ['Self::ContractType::total_supply(e)'],
+    },
+    balance: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'account', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::balance(e, &account)'],
+    },
+    allowance: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'owner', type: 'Address' }, { name: 'spender', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::allowance(e, &owner, &spender)'],
+    },
+    transfer: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'from', type: 'Address' },
+            { name: 'to', type: 'MuxedAddress' },
+            { name: 'amount', type: 'i128' },
+        ],
+        code: ['Self::ContractType::transfer(e, &from, &to, amount)'],
+    },
+    transfer_from: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'spender', type: 'Address' },
+            { name: 'from', type: 'Address' },
+            { name: 'to', type: 'Address' },
+            { name: 'amount', type: 'i128' },
+        ],
+        code: ['Self::ContractType::transfer_from(e, &spender, &from, &to, amount)'],
+    },
+    approve: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'owner', type: 'Address' },
+            { name: 'spender', type: 'Address' },
+            { name: 'amount', type: 'i128' },
+            { name: 'live_until_ledger', type: 'u32' },
+        ],
+        code: ['Self::ContractType::approve(e, &owner, &spender, amount, live_until_ledger)'],
+    },
+    // The vault derives decimals from the underlying asset's decimals plus the
+    // virtual offset, so it overrides the default metadata-based implementation.
+    decimals: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'u32',
+        code: ['Vault::decimals(e)'],
+    },
+    name: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'String',
+        code: ['Self::ContractType::name(e)'],
+    },
+    symbol: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'String',
+        code: ['Self::ContractType::symbol(e)'],
+    },
+    // FungibleVault Trait
+    query_asset: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'Address',
+        code: ['Self::ContractType::query_asset(e)'],
+    },
+    total_assets: {
+        args: [(0, common_options_1.getSelfArg)()],
+        returns: 'i128',
+        code: ['Self::ContractType::total_assets(e)'],
+    },
+    convert_to_shares: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'assets', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::convert_to_shares(e, assets)'],
+    },
+    convert_to_assets: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'shares', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::convert_to_assets(e, shares)'],
+    },
+    max_deposit: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'receiver', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::max_deposit(e, receiver)'],
+    },
+    preview_deposit: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'assets', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::preview_deposit(e, assets)'],
+    },
+    deposit: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'assets', type: 'i128' },
+            { name: 'receiver', type: 'Address' },
+            { name: 'from', type: 'Address' },
+            { name: 'operator', type: 'Address' },
+        ],
+        returns: 'i128',
+        code: ['Self::ContractType::deposit(e, assets, receiver, from, operator)'],
+    },
+    max_mint: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'receiver', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::max_mint(e, receiver)'],
+    },
+    preview_mint: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'shares', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::preview_mint(e, shares)'],
+    },
+    mint: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'shares', type: 'i128' },
+            { name: 'receiver', type: 'Address' },
+            { name: 'from', type: 'Address' },
+            { name: 'operator', type: 'Address' },
+        ],
+        returns: 'i128',
+        code: ['Self::ContractType::mint(e, shares, receiver, from, operator)'],
+    },
+    max_withdraw: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'owner', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::max_withdraw(e, owner)'],
+    },
+    preview_withdraw: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'assets', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::preview_withdraw(e, assets)'],
+    },
+    withdraw: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'assets', type: 'i128' },
+            { name: 'receiver', type: 'Address' },
+            { name: 'owner', type: 'Address' },
+            { name: 'operator', type: 'Address' },
+        ],
+        returns: 'i128',
+        code: ['Self::ContractType::withdraw(e, assets, receiver, owner, operator)'],
+    },
+    max_redeem: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'owner', type: 'Address' }],
+        returns: 'i128',
+        code: ['Self::ContractType::max_redeem(e, owner)'],
+    },
+    preview_redeem: {
+        args: [(0, common_options_1.getSelfArg)(), { name: 'shares', type: 'i128' }],
+        returns: 'i128',
+        code: ['Self::ContractType::preview_redeem(e, shares)'],
+    },
+    redeem: {
+        args: [
+            (0, common_options_1.getSelfArg)(),
+            { name: 'shares', type: 'i128' },
+            { name: 'receiver', type: 'Address' },
+            { name: 'owner', type: 'Address' },
+            { name: 'operator', type: 'Address' },
+        ],
+        returns: 'i128',
+        code: ['Self::ContractType::redeem(e, shares, receiver, owner, operator)'],
+    },
+});
+const fungibleTokenTraitFunctions = (0, wizard_common_1.pickKeys)(exports.functions, [
+    'total_supply',
+    'balance',
+    'allowance',
+    'transfer',
+    'transfer_from',
+    'approve',
+    'decimals',
+    'name',
+    'symbol',
+]);
+const fungibleVaultTraitFunctions = (0, wizard_common_1.pickKeys)(exports.functions, [
+    'query_asset',
+    'total_assets',
+    'convert_to_shares',
+    'convert_to_assets',
+    'max_deposit',
+    'preview_deposit',
+    'deposit',
+    'max_mint',
+    'preview_mint',
+    'mint',
+    'max_withdraw',
+    'preview_withdraw',
+    'withdraw',
+    'max_redeem',
+    'preview_redeem',
+    'redeem',
+]);
+//# sourceMappingURL=vault.js.map

package/dist/vault.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vault.js","sourceRoot":"","sources":["../src/vault.ts"],"names":[],"mappings":";;;AA8BA,gCAEC;AAUD,oCAQC;AAED,0DAEC;AAED,gCA0CC;AAlGD,yCAA6C;AAE7C,6DAAwD;AACxD,iDAA6C;AAC7C,uDAAmD;AACnD,+DAA2D;AAE3D,qDAA0E;AAC1E,yCAAqC;AACrC,qDAAsE;AACtE,mCAAwC;AACxC,6DAA8D;AAC9D,mCAAuC;AACvC,+DAAuD;AAEvD,qEAAqE;AACrE,mFAAmF;AACnF,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAElB,QAAA,QAAQ,GAA2B;IAC9C,IAAI,EAAE,SAAS;IACf,MAAM,EAAE,KAAK;IACb,cAAc,EAAE,GAAG;IACnB,QAAQ,EAAE,KAAK;IACf,WAAW,EAAE,KAAK;IAClB,MAAM,EAAE,iCAAc,CAAC,MAAM;IAC7B,IAAI,EAAE,iCAAc,CAAC,IAAI;IACzB,uBAAuB,EAAE,iCAAc,CAAC,uBAAuB;CACvD,CAAC;AAEX,SAAgB,UAAU,CAAC,OAAqB,gBAAQ;IACtD,OAAO,IAAA,qBAAa,EAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;AACzC,CAAC;AAUD,SAAgB,YAAY,CAAC,IAAkB;IAC7C,OAAO;QACL,GAAG,IAAI;QACP,GAAG,IAAA,2CAA0B,EAAC,IAAI,CAAC;QACnC,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,gBAAQ,CAAC,cAAc;QAC9D,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,gBAAQ,CAAC,QAAQ;QAC5C,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,gBAAQ,CAAC,WAAW;KACtD,CAAC;AACJ,CAAC;AAED,SAAgB,uBAAuB,CAAC,IAA2B;IACjE,OAAO,IAAI,CAAC,QAAQ,KAAK,IAAI,IAAI,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC;AAC7D,CAAC;AAED,SAAgB,UAAU,CAAC,IAAkB;IAC3C,MAAM,CAAC,GAAG,IAAI,0BAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEzC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAEnC,MAAM,cAAc,GAAG,IAAA,wBAAM,EAAC,OAAO,CAAC,cAAc,EAAE,gBAAgB,EAAE,KAAK,CAAC,CAAC;IAC/E,IAAI,cAAc,GAAG,mBAAmB,EAAE,CAAC;QACzC,MAAM,IAAI,oBAAY,CAAC;YACrB,cAAc,EAAE,8BAA8B,mBAAmB,EAAE;SACpE,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CACL,CAAC,EACD,IAAA,6BAAW,EAAC,OAAO,CAAC,IAAI,CAAC,EACzB,IAAA,6BAAW,EAAC,OAAO,CAAC,MAAM,CAAC,EAC3B,cAAc,EACd,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,uBAAuB,CAChC,CAAC;IAEF,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,IAAA,0BAAW,EAAC,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,IAAA,gCAAc,EAAC,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,uBAAuB,CAAC,CAAC;IACrE,CAAC;IAED,IAAA,qCAAgB,EAAC,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAErE,4EAA4E;IAC5E,4EAA4E;IAC5E,qEAAqE;IACrE,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;QAC/B,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QACxC,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,IAAA,kBAAO,EAAC,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzB,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,OAAO,CACd,CAAkB,EAClB,IAAY,EACZ,MAAc,EACd,cAAsB,EACtB,QAAiB,EACjB,uBAAgC;IAEhC,6EAA6E;IAC7E,8EAA8E;IAC9E,8EAA8E;IAC9E,4BAA4B;IAC5B,CAAC,CAAC,sBAAsB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC,kBAAkB,CAAC,6BAA6B,CAAC,CAAC;IACpD,CAAC,CAAC,kBAAkB,CAAC,iCAAiC,cAAc,IAAI,CAAC,CAAC;IAC1E,CAAC,CAAC,kBAAkB,CAClB,iEAAiE,IAAI,4BAA4B,MAAM,MAAM,CAC9G,CAAC;IAEF,sBAAsB;IACtB,CAAC,CAAC,YAAY,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC;IACnD,CAAC,CAAC,YAAY,CAAC,0BAA0B,EAAE,eAAe,CAAC,CAAC;IAC5D,CAAC,CAAC,YAAY,CAAC,uBAAuB,EAAE,eAAe,CAAC,CAAC;IACzD,CAAC,CAAC,YAAY,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IACjD,0EAA0E;IAC1E,+EAA+E;IAC/E,0EAA0E;IAC1E,wEAAwE;IACxE,IAAI,uBAAuB,IAAI,QAAQ;QAAE,CAAC,CAAC,YAAY,CAAC,0BAA0B,EAAE,mBAAmB,CAAC,CAAC;IACzG,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;IAC1C,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;IAC9C,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,YAAY,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;IAE9C,MAAM,kBAAkB,GAAG;QACzB,SAAS,EAAE,eAAe;QAC1B,UAAU,EAAE,CAAC,CAAC,IAAI;QAClB,IAAI,EAAE,uBAAuB,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B,CAAC;QAClF,SAAS,EAAE,4BAA4B;KACxC,CAAC;IAEF,CAAC,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;IAExC,IAAI,uBAAuB,EAAE,CAAC;QAC5B,CAAC,CAAC,wBAAwB,CAAC,kBAAkB,EAAE,2BAA2B,CAAC,CAAC;IAC9E,CAAC;SAAM,CAAC;QACN,4EAA4E;QAC5E,+BAA+B;QAC/B,CAAC,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,iBAAS,CAAC,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,YAAY,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;QAEpD,CAAC,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,iBAAS,CAAC,QAAQ,CAAC,CAAC;QAC3D,CAAC,CAAC,cAAc,CAAC,iBAAS,CAAC,QAAQ,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;QAE5E,CAAC,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,iBAAS,CAAC,aAAa,CAAC,CAAC;QAChE,CAAC,CAAC,cAAc,CAAC,iBAAS,CAAC,aAAa,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;IACnF,CAAC;IAED,MAAM,kBAAkB,GAAG;QACzB,SAAS,EAAE,eAAe;QAC1B,UAAU,EAAE,CAAC,CAAC,IAAI;QAClB,IAAI,EAAE,uBAAuB,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B,CAAC;QAClF,OAAO,EAAE,YAAY;KACtB,CAAC;IAEF,IAAI,uBAAuB,EAAE,CAAC;QAC5B,CAAC,CAAC,wBAAwB,CAAC,kBAAkB,EAAE,2BAA2B,CAAC,CAAC;IAC9E,CAAC;SAAM,CAAC;QACN,CAAC,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAEY,QAAA,SAAS,GAAG,IAAA,kCAAe,EAAC;IACvC,sBAAsB;IACtB,YAAY,EAAE;QACZ,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,qCAAqC,CAAC;KAC9C;IACD,OAAO,EAAE;QACP,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1D,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,0CAA0C,CAAC;KACnD;IACD,SAAS,EAAE;QACT,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC9F,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,oDAAoD,CAAC;KAC7D;IACD,QAAQ,EAAE;QACR,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE;YACjC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE;YACpC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;SACjC;QACD,IAAI,EAAE,CAAC,qDAAqD,CAAC;KAC9D;IACD,aAAa,EAAE;QACb,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE;YACpC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE;YACjC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE;YAC/B,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;SACjC;QACD,IAAI,EAAE,CAAC,oEAAoE,CAAC;KAC7E;IACD,OAAO,EAAE;QACP,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE;YAClC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE;YACpC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;YAChC,EAAE,IAAI,EAAE,mBAAmB,EAAE,IAAI,EAAE,KAAK,EAAE;SAC3C;QACD,IAAI,EAAE,CAAC,6EAA6E,CAAC;KACtF;IACD,2EAA2E;IAC3E,6EAA6E;IAC7E,QAAQ,EAAE;QACR,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,KAAK;QACd,IAAI,EAAE,CAAC,oBAAoB,CAAC;KAC7B;IACD,IAAI,EAAE;QACJ,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,QAAQ;QACjB,IAAI,EAAE,CAAC,6BAA6B,CAAC;KACtC;IACD,MAAM,EAAE;QACN,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,QAAQ;QACjB,IAAI,EAAE,CAAC,+BAA+B,CAAC;KACxC;IAED,sBAAsB;IACtB,WAAW,EAAE;QACX,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,SAAS;QAClB,IAAI,EAAE,CAAC,oCAAoC,CAAC;KAC7C;IACD,YAAY,EAAE;QACZ,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,CAAC;QACpB,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,qCAAqC,CAAC;KAC9C;IACD,iBAAiB,EAAE;QACjB,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,kDAAkD,CAAC;KAC3D;IACD,iBAAiB,EAAE;QACjB,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,kDAAkD,CAAC;KAC3D;IACD,WAAW,EAAE;QACX,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC3D,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,8CAA8C,CAAC;KACvD;IACD,eAAe,EAAE;QACf,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,gDAAgD,CAAC;KACzD;IACD,OAAO,EAAE;QACP,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;YAChC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;YACrC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE;YACjC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;SACtC;QACD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,kEAAkE,CAAC;KAC3E;IACD,QAAQ,EAAE;QACR,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC3D,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,2CAA2C,CAAC;KACpD;IACD,YAAY,EAAE;QACZ,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,6CAA6C,CAAC;KACtD;IACD,IAAI,EAAE;QACJ,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;YAChC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;YACrC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE;YACjC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;SACtC;QACD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,+DAA+D,CAAC;KACxE;IACD,YAAY,EAAE;QACZ,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACxD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,4CAA4C,CAAC;KACrD;IACD,gBAAgB,EAAE;QAChB,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,iDAAiD,CAAC;KAC1D;IACD,QAAQ,EAAE;QACR,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;YAChC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;YACrC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE;YAClC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;SACtC;QACD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,oEAAoE,CAAC;KAC7E;IACD,UAAU,EAAE;QACV,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACxD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,0CAA0C,CAAC;KACnD;IACD,cAAc,EAAE;QACd,IAAI,EAAE,CAAC,IAAA,2BAAU,GAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACtD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,+CAA+C,CAAC;KACxD;IACD,MAAM,EAAE;QACN,IAAI,EAAE;YACJ,IAAA,2BAAU,GAAE;YACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE;YAChC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;YACrC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE;YAClC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;SACtC;QACD,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,kEAAkE,CAAC;KAC3E;CACF,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,IAAA,wBAAQ,EAAC,iBAAS,EAAE;IACtD,cAAc;IACd,SAAS;IACT,WAAW;IACX,UAAU;IACV,eAAe;IACf,SAAS;IACT,UAAU;IACV,MAAM;IACN,QAAQ;CACT,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,IAAA,wBAAQ,EAAC,iBAAS,EAAE;IACtD,aAAa;IACb,cAAc;IACd,mBAAmB;IACnB,mBAAmB;IACnB,aAAa;IACb,iBAAiB;IACjB,SAAS;IACT,UAAU;IACV,cAAc;IACd,MAAM;IACN,cAAc;IACd,kBAAkB;IAClB,UAAU;IACV,YAAY;IACZ,gBAAgB;IAChB,QAAQ;CACT,CAAC,CAAC"}

package/dist/zip-rust.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"zip-rust.d.ts","sourceRoot":"","sources":["../src/zip-rust.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AA8B3C,eAAO,MAAM,wBAAwB,GAAI,GAAG,QAAQ,EAAE,MAAM,cAAc,UAczE,CAAC;AAIF,eAAO,MAAM,cAAc,GAAU,GAAG,QAAQ,EAAE,MAAM,cAAc,mBACb,CAAC"}
+{"version":3,"file":"zip-rust.d.ts","sourceRoot":"","sources":["../src/zip-rust.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AA+B3C,eAAO,MAAM,wBAAwB,GAAI,GAAG,QAAQ,EAAE,MAAM,cAAc,UAiBzE,CAAC;AAIF,eAAO,MAAM,cAAc,GAAU,GAAG,QAAQ,EAAE,MAAM,cAAc,mBACb,CAAC"}

package/dist/zip-rust.js

@@ -29,7 +29,7 @@ const createRustZipEnvironment = (c, opts) => {
     const contractName = (0, zip_shared_1.contractOptionsToContractName)(opts?.kind || 'contract');
     const requiredLibDeps = (0, zip_shared_1.getRequiredLibDependencies)(c);
     zip.file(`contracts/${contractName}/src/contract.rs`, (0, print_1.removeCreateLevelAttributes)((0, print_1.printContract)(c)));
-    zip.file(`contracts/${contractName}/src/test.rs`, (0, zip_shared_1.printRustNameTest)(c));
+    zip.file(`contracts/${contractName}/src/test.rs`, opts?.kind === 'Vault' ? (0, zip_shared_1.printVaultRustTest)(c) : (0, zip_shared_1.printRustNameTest)(c));
     zip.file(`contracts/${contractName}/src/lib.rs`, zip_shared_1.createRustLibFile);
     zip.file(`contracts/${contractName}/Cargo.toml`, (0, zip_shared_1.printContractCargo)(contractName, requiredLibDeps));
     zip.file('Cargo.toml', (0, zip_shared_1.workspaceCargo)(requiredLibDeps));

package/dist/zip-rust.js.map

@@ -1 +1 @@
-{"version":3,"file":"zip-rust.js","sourceRoot":"","sources":["../src/zip-rust.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAG1B,mCAAqE;AACrE,6CAOsB;AAEtB,MAAM,MAAM,GAAG;;;;;;;;;;;;;CAad,CAAC;AAEF,MAAM,SAAS,GAAG;;CAEjB,CAAC;AAEK,MAAM,wBAAwB,GAAG,CAAC,CAAW,EAAE,IAAoB,EAAE,EAAE;IAC5E,MAAM,GAAG,GAAG,IAAI,eAAK,EAAE,CAAC;IAExB,MAAM,YAAY,GAAG,IAAA,0CAA6B,EAAC,IAAI,EAAE,IAAI,IAAI,UAAU,CAAC,CAAC;IAC7E,MAAM,eAAe,GAAG,IAAA,uCAA0B,EAAC,CAAC,CAAC,CAAC;IAEtD,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,kBAAkB,EAAE,IAAA,mCAA2B,EAAC,IAAA,qBAAa,EAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrG,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,cAAc,EAAE,IAAA,8BAAiB,EAAC,CAAC,CAAC,CAAC,CAAC;IACxE,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,aAAa,EAAE,8BAAiB,CAAC,CAAC;IACpE,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,aAAa,EAAE,IAAA,+BAAkB,EAAC,YAAY,EAAE,eAAe,CAAC,CAAC,CAAC;IACpG,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,IAAA,2BAAc,EAAC,eAAe,CAAC,CAAC,CAAC;IACxD,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAElC,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAdW,QAAA,wBAAwB,4BAcnC;AAEF,MAAM,oBAAoB,GAAG,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;AAEpE,MAAM,cAAc,GAAG,KAAK,EAAE,CAAW,EAAE,IAAoB,EAAE,EAAE,CACxE,oBAAoB,CAAC,IAAA,gCAAwB,EAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;AAD7C,QAAA,cAAc,kBAC+B"}
+{"version":3,"file":"zip-rust.js","sourceRoot":"","sources":["../src/zip-rust.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAG1B,mCAAqE;AACrE,6CAQsB;AAEtB,MAAM,MAAM,GAAG;;;;;;;;;;;;;CAad,CAAC;AAEF,MAAM,SAAS,GAAG;;CAEjB,CAAC;AAEK,MAAM,wBAAwB,GAAG,CAAC,CAAW,EAAE,IAAoB,EAAE,EAAE;IAC5E,MAAM,GAAG,GAAG,IAAI,eAAK,EAAE,CAAC;IAExB,MAAM,YAAY,GAAG,IAAA,0CAA6B,EAAC,IAAI,EAAE,IAAI,IAAI,UAAU,CAAC,CAAC;IAC7E,MAAM,eAAe,GAAG,IAAA,uCAA0B,EAAC,CAAC,CAAC,CAAC;IAEtD,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,kBAAkB,EAAE,IAAA,mCAA2B,EAAC,IAAA,qBAAa,EAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrG,GAAG,CAAC,IAAI,CACN,aAAa,YAAY,cAAc,EACvC,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAA,+BAAkB,EAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAA,8BAAiB,EAAC,CAAC,CAAC,CACtE,CAAC;IACF,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,aAAa,EAAE,8BAAiB,CAAC,CAAC;IACpE,GAAG,CAAC,IAAI,CAAC,aAAa,YAAY,aAAa,EAAE,IAAA,+BAAkB,EAAC,YAAY,EAAE,eAAe,CAAC,CAAC,CAAC;IACpG,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,IAAA,2BAAc,EAAC,eAAe,CAAC,CAAC,CAAC;IACxD,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAElC,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAjBW,QAAA,wBAAwB,4BAiBnC;AAEF,MAAM,oBAAoB,GAAG,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;AAEpE,MAAM,cAAc,GAAG,KAAK,EAAE,CAAW,EAAE,IAAoB,EAAE,EAAE,CACxE,oBAAoB,CAAC,IAAA,gCAAwB,EAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;AAD7C,QAAA,cAAc,kBAC+B"}

package/dist/zip-shared.d.ts

@@ -3,6 +3,7 @@ declare function pascalToSnakeCase(string: string): string;
 export declare const contractOptionsToContractName: typeof pascalToSnakeCase;
 export declare function getAddressArgs(c: Pick<Contract, 'constructorArgs'>): string[];
 export declare const printRustNameTest: (c: Pick<Contract, "constructorArgs" | "name">) => string;
+export declare const printVaultRustTest: (c: Pick<Contract, "constructorArgs" | "name">) => string;
 export declare function getRequiredLibDependencies(c: Pick<Contract, 'useClauses'>): string[];
 export declare const addDependenciesWith: (dependencyValue: string, dependenciesToAdd: string[]) => string;
 export declare const printContractCargo: (scaffoldContractName: string, requiredLibDeps: readonly string[]) => string;

package/dist/zip-shared.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"zip-shared.d.ts","sourceRoot":"","sources":["../src/zip-shared.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAG3C,iBAAS,iBAAiB,CAAC,MAAM,EAAE,MAAM,UAKxC;AAED,eAAO,MAAM,6BAA6B,0BAAoB,CAAC;AAE/D,wBAAgB,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE,iBAAiB,CAAC,GAAG,MAAM,EAAE,CAI7E;AAED,eAAO,MAAM,iBAAiB,GAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,iBAAiB,GAAG,MAAM,CAAC,WAqB9E,CAAC;AAiBF,wBAAgB,0BAA0B,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,MAAM,EAAE,CAGpF;AAED,eAAO,MAAM,mBAAmB,GAAI,iBAAiB,MAAM,EAAE,mBAAmB,MAAM,EAAE,WACG,CAAC;AAE5F,eAAO,MAAM,kBAAkB,GAAI,sBAAsB,MAAM,EAAE,iBAAiB,SAAS,MAAM,EAAE,WAiBT,CAAC;AAE3F,eAAO,MAAM,iBAAiB,mEAK7B,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,iBAAiB,SAAS,MAAM,EAAE,WA0BhE,CAAC"}
+{"version":3,"file":"zip-shared.d.ts","sourceRoot":"","sources":["../src/zip-shared.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAG3C,iBAAS,iBAAiB,CAAC,MAAM,EAAE,MAAM,UAKxC;AAED,eAAO,MAAM,6BAA6B,0BAAoB,CAAC;AAE/D,wBAAgB,cAAc,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE,iBAAiB,CAAC,GAAG,MAAM,EAAE,CAI7E;AAED,eAAO,MAAM,iBAAiB,GAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,iBAAiB,GAAG,MAAM,CAAC,WAqB9E,CAAC;AAOF,eAAO,MAAM,kBAAkB,GAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,iBAAiB,GAAG,MAAM,CAAC,WAuD/E,CAAC;AAiBF,wBAAgB,0BAA0B,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,MAAM,EAAE,CAGpF;AAED,eAAO,MAAM,mBAAmB,GAAI,iBAAiB,MAAM,EAAE,mBAAmB,MAAM,EAAE,WACG,CAAC;AAE5F,eAAO,MAAM,kBAAkB,GAAI,sBAAsB,MAAM,EAAE,iBAAiB,SAAS,MAAM,EAAE,WAiBT,CAAC;AAE3F,eAAO,MAAM,iBAAiB,mEAK7B,CAAC;AAEF,eAAO,MAAM,cAAc,GAAI,iBAAiB,SAAS,MAAM,EAAE,WA0BhE,CAAC"}

package/dist/zip-shared.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.workspaceCargo = exports.createRustLibFile = exports.printContractCargo = exports.addDependenciesWith = exports.printRustNameTest = exports.contractOptionsToContractName = void 0;
+exports.workspaceCargo = exports.createRustLibFile = exports.printContractCargo = exports.addDependenciesWith = exports.printVaultRustTest = exports.printRustNameTest = exports.contractOptionsToContractName = void 0;
 exports.getAddressArgs = getAddressArgs;
 exports.getRequiredLibDependencies = getRequiredLibDependencies;
 const version_1 = require("./utils/version");
@@ -39,6 +39,65 @@ fn initial_state() {
 // Add more tests bellow
 `;
 exports.printRustNameTest = printRustNameTest;
+// The vault's constructor takes an underlying asset address and derives its own
+// decimals by calling into that asset contract. A generated address would have
+// no `decimals()` to call, so the test deploys a minimal mock fungible asset
+// first and registers the vault against it. A single-element tuple needs a
+// trailing comma in Rust.
+const printVaultRustTest = (c) => {
+    const registerArgs = (c.constructorArgs || []).map(arg => arg.name === 'asset' ? 'asset_address.clone()' : 'Address::generate(&env)');
+    const registerTuple = `(${registerArgs.join(', ')}${registerArgs.length === 1 ? ',' : ''})`;
+    return `#![cfg(test)]
+
+extern crate std;
+
+use soroban_sdk::{
+    contract, contractimpl, testutils::Address as _, Address, Env, MuxedAddress, String,
+};
+use stellar_tokens::fungible::{Base, FungibleToken};
+
+use crate::contract::{ ${c.name}, ${c.name}Client };
+
+// Mock asset contract: a minimal fungible token used as the vault's underlying asset.
+#[contract]
+pub struct MockAssetContract;
+
+#[contractimpl]
+impl MockAssetContract {
+    pub fn __constructor(e: &Env, initial_supply: i128, admin: Address) {
+        Base::set_metadata(
+            e,
+            18,
+            String::from_str(e, "Mock Asset Token"),
+            String::from_str(e, "MAT"),
+        );
+        Base::mint(e, &admin, initial_supply);
+    }
+}
+
+#[contractimpl(contracttrait)]
+impl FungibleToken for MockAssetContract {
+    type ContractType = Base;
+}
+
+#[test]
+fn initial_state() {
+    let env = Env::default();
+
+    let admin = Address::generate(&env);
+    let asset_address = env.register(MockAssetContract, (1_000_000_000i128, admin));
+
+    let contract_addr = env.register(${c.name}, ${registerTuple});
+    let client = ${c.name}Client::new(&env, &contract_addr);
+
+    assert_eq!(client.query_asset(), asset_address);
+    assert_eq!(client.name(), String::from_str(&env, "${c.name}"));
+}
+
+// Add more tests bellow
+`;
+};
+exports.printVaultRustTest = printVaultRustTest;
 const libDependencies = [
     'stellar-tokens',
     'stellar-access',

package/dist/zip-shared.js.map

@@ -1 +1 @@
-{"version":3,"file":"zip-shared.js","sourceRoot":"","sources":["../src/zip-shared.ts"],"names":[],"mappings":";;;AAYA,wCAIC;AAwCD,gEAGC;AA1DD,6CAA6E;AAE7E,SAAS,iBAAiB,CAAC,MAAc;IACvC,OAAO,MAAM;SACV,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;SAC1B,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;SACjB,WAAW,EAAE,CAAC;AACnB,CAAC;AAEY,QAAA,6BAA6B,GAAG,iBAAiB,CAAC;AAE/D,SAAgB,cAAc,CAAC,CAAoC;IACjE,OAAO,CAAC,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;SAC7B,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,SAAS,CAAC;SACnE,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;AAChD,CAAC;AAEM,MAAM,iBAAiB,GAAG,CAAC,CAA6C,EAAE,EAAE,CAAC;;;;qBAI/D,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE;;yBAEhE,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;;;;;;uCAMH,CAAC,CAAC,IAAI,MAAM,cAAc,CAAC,CAAC,CAAC;KAC7D,GAAG,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC;KACpC,IAAI,CAAC,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;mBACzC,CAAC,CAAC,IAAI;;wDAE+B,CAAC,CAAC,IAAI;;;;CAI7D,CAAC;AArBW,QAAA,iBAAiB,qBAqB5B;AAEF,MAAM,eAAe,GAAG;IACtB,gBAAgB;IAChB,gBAAgB;IAChB,wBAAwB;IACxB,oBAAoB;IACpB,gBAAgB;CACR,CAAC;AAEX,2EAA2E;AAC3E,6EAA6E;AAC7E,gFAAgF;AAChF,8EAA8E;AAC9E,2EAA2E;AAC3E,yEAAyE;AACzE,sEAAsE;AACtE,SAAgB,0BAA0B,CAAC,CAA+B;IACxE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IACxG,OAAO,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;AAC5D,CAAC;AAEM,MAAM,mBAAmB,GAAG,CAAC,eAAuB,EAAE,iBAA2B,EAAE,EAAE,CAC1F,GAAG,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG,UAAU,MAAM,eAAe,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;AAD/E,QAAA,mBAAmB,uBAC4D;AAErF,MAAM,kBAAkB,GAAG,CAAC,oBAA4B,EAAE,eAAkC,EAAE,EAAE,CAAC;UAC9F,oBAAoB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;;;;;;;;;;;;;;EAc9C,IAAA,2BAAmB,EAAC,sBAAsB,EAAE,CAAC,GAAG,eAAe,EAAE,aAAa,CAAC,CAAC;;EAEhF,IAAA,2BAAmB,EAAC,gDAAgD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;AAjB9E,QAAA,kBAAkB,sBAiB4D;AAE9E,QAAA,iBAAiB,GAAG;;;;;CAKhC,CAAC;AAEK,MAAM,cAAc,GAAG,CAAC,eAAkC,EAAE,EAAE,CAAC;;;;;;;;;;;EAWpE,IAAA,2BAAmB,EAAC,IAAI,kCAAwB,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,GAAG,IAAA,2BAAmB,EAAC,KAAK,0BAAgB,GAAG,EAAE,CAAC,GAAG,eAAe,CAAC,CAAC;;;;;;;;;;;;;;;CAe5I,CAAC;AA1BW,QAAA,cAAc,kBA0BzB"}
+{"version":3,"file":"zip-shared.js","sourceRoot":"","sources":["../src/zip-shared.ts"],"names":[],"mappings":";;;AAYA,wCAIC;AAsGD,gEAGC;AAxHD,6CAA6E;AAE7E,SAAS,iBAAiB,CAAC,MAAc;IACvC,OAAO,MAAM;SACV,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC;SAC1B,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;SACjB,WAAW,EAAE,CAAC;AACnB,CAAC;AAEY,QAAA,6BAA6B,GAAG,iBAAiB,CAAC;AAE/D,SAAgB,cAAc,CAAC,CAAoC;IACjE,OAAO,CAAC,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;SAC7B,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,SAAS,CAAC;SACnE,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;AAChD,CAAC;AAEM,MAAM,iBAAiB,GAAG,CAAC,CAA6C,EAAE,EAAE,CAAC;;;;qBAI/D,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE;;yBAEhE,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;;;;;;uCAMH,CAAC,CAAC,IAAI,MAAM,cAAc,CAAC,CAAC,CAAC;KAC7D,GAAG,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC;KACpC,IAAI,CAAC,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;mBACzC,CAAC,CAAC,IAAI;;wDAE+B,CAAC,CAAC,IAAI;;;;CAI7D,CAAC;AArBW,QAAA,iBAAiB,qBAqB5B;AAEF,gFAAgF;AAChF,+EAA+E;AAC/E,6EAA6E;AAC7E,2EAA2E;AAC3E,0BAA0B;AACnB,MAAM,kBAAkB,GAAG,CAAC,CAA6C,EAAE,EAAE;IAClF,MAAM,YAAY,GAAG,CAAC,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CACvD,GAAG,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,yBAAyB,CAC3E,CAAC;IACF,MAAM,aAAa,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC;IAE5F,OAAO;;;;;;;;;yBASgB,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uCA+BH,CAAC,CAAC,IAAI,KAAK,aAAa;mBAC5C,CAAC,CAAC,IAAI;;;wDAG+B,CAAC,CAAC,IAAI;;;;CAI7D,CAAC;AACF,CAAC,CAAC;AAvDW,QAAA,kBAAkB,sBAuD7B;AAEF,MAAM,eAAe,GAAG;IACtB,gBAAgB;IAChB,gBAAgB;IAChB,wBAAwB;IACxB,oBAAoB;IACpB,gBAAgB;CACR,CAAC;AAEX,2EAA2E;AAC3E,6EAA6E;AAC7E,gFAAgF;AAChF,8EAA8E;AAC9E,2EAA2E;AAC3E,yEAAyE;AACzE,sEAAsE;AACtE,SAAgB,0BAA0B,CAAC,CAA+B;IACxE,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IACxG,OAAO,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;AAC5D,CAAC;AAEM,MAAM,mBAAmB,GAAG,CAAC,eAAuB,EAAE,iBAA2B,EAAE,EAAE,CAC1F,GAAG,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG,UAAU,MAAM,eAAe,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;AAD/E,QAAA,mBAAmB,uBAC4D;AAErF,MAAM,kBAAkB,GAAG,CAAC,oBAA4B,EAAE,eAAkC,EAAE,EAAE,CAAC;UAC9F,oBAAoB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;;;;;;;;;;;;;;EAc9C,IAAA,2BAAmB,EAAC,sBAAsB,EAAE,CAAC,GAAG,eAAe,EAAE,aAAa,CAAC,CAAC;;EAEhF,IAAA,2BAAmB,EAAC,gDAAgD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;AAjB9E,QAAA,kBAAkB,sBAiB4D;AAE9E,QAAA,iBAAiB,GAAG;;;;;CAKhC,CAAC;AAEK,MAAM,cAAc,GAAG,CAAC,eAAkC,EAAE,EAAE,CAAC;;;;;;;;;;;EAWpE,IAAA,2BAAmB,EAAC,IAAI,kCAAwB,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,GAAG,IAAA,2BAAmB,EAAC,KAAK,0BAAgB,GAAG,EAAE,CAAC,GAAG,eAAe,CAAC,CAAC;;;;;;;;;;;;;;;CAe5I,CAAC;AA1BW,QAAA,cAAc,kBA0BzB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@openzeppelin/wizard-stellar",
-  "version": "0.6.1",
+  "version": "0.6.3",
   "description": "A boilerplate generator to get started with OpenZeppelin Stellar Soroban Contracts",
   "license": "AGPL-3.0-only",
   "repository": "https://github.com/OpenZeppelin/contracts-wizard",

package/src/api.ts

@@ -3,11 +3,17 @@ import type { FungibleOptions } from './fungible';
 import type { GovernorOptions } from './governor';
 import type { NonFungibleOptions } from './non-fungible';
 import type { StablecoinOptions } from './stablecoin';
+import type { VaultOptions } from './vault';
 import {
   printStablecoin,
   defaults as stablecoinDefaults,
   isAccessControlRequired as stablecoinIsAccessControlRequired,
 } from './stablecoin';
+import {
+  printVault,
+  defaults as vaultDefaults,
+  isAccessControlRequired as vaultIsAccessControlRequired,
+} from './vault';
 import {
   printFungible,
   defaults as fungibledefaults,
@@ -48,6 +54,7 @@ export type Fungible = WizardContractAPI<FungibleOptions> & AccessControlAPI<Fun
 export type Governor = WizardContractAPI<GovernorOptions> & AccessControlAPI<GovernorOptions>;
 export type NonFungible = WizardContractAPI<NonFungibleOptions> & AccessControlAPI<NonFungibleOptions>;
 export type Stablecoin = WizardContractAPI<StablecoinOptions> & AccessControlAPI<StablecoinOptions>;
+export type Vault = WizardContractAPI<VaultOptions> & AccessControlAPI<VaultOptions>;
 
 export const fungible: Fungible = {
   print: printFungible,
@@ -72,3 +79,9 @@ export const stablecoin: Stablecoin = {
   defaults: stablecoinDefaults,
   isAccessControlRequired: stablecoinIsAccessControlRequired,
 };
+
+export const vault: Vault = {
+  print: printVault,
+  defaults: vaultDefaults,
+  isAccessControlRequired: vaultIsAccessControlRequired,
+};

package/src/build-generic.ts

@@ -6,12 +6,15 @@ import type { NonFungibleOptions } from './non-fungible';
 import { buildNonFungible } from './non-fungible';
 import type { StablecoinOptions } from './stablecoin';
 import { buildStablecoin } from './stablecoin';
+import type { VaultOptions } from './vault';
+import { buildVault } from './vault';
 
 export interface KindedOptions {
   Fungible: { kind: 'Fungible' } & FungibleOptions;
   Governor: { kind: 'Governor' } & GovernorOptions;
   NonFungible: { kind: 'NonFungible' } & NonFungibleOptions;
   Stablecoin: { kind: 'Stablecoin' } & StablecoinOptions;
+  Vault: { kind: 'Vault' } & VaultOptions;
 }
 
 export type GenericOptions = KindedOptions[keyof KindedOptions];
@@ -30,6 +33,9 @@ export function buildGeneric(opts: GenericOptions) {
     case 'Stablecoin':
       return buildStablecoin(opts);
 
+    case 'Vault':
+      return buildVault(opts);
+
     default: {
       const _: never = opts;
       throw new Error('Unknown kind');

package/src/generate/sources.ts

@@ -6,6 +6,7 @@ import { generateFungibleOptions } from './fungible';
 import { generateGovernorOptions } from './governor';
 import { generateNonFungibleOptions } from './non-fungible';
 import { generateStablecoinOptions } from './stablecoin';
+import { generateVaultOptions } from './vault';
 import type { GenericOptions, KindedOptions } from '../build-generic';
 import { buildGeneric } from '../build-generic';
 import { printContract } from '../print';
@@ -38,6 +39,12 @@ export function* generateOptions(kind?: Kind): Generator<GenericOptions> {
       yield { kind: 'Stablecoin', ...kindOpts };
     }
   }
+
+  if (!kind || kind === 'Vault') {
+    for (const kindOpts of generateVaultOptions()) {
+      yield { kind: 'Vault', ...kindOpts };
+    }
+  }
 }
 
 interface GeneratedContract {

package/src/generate/vault.ts

@@ -0,0 +1,21 @@
+import type { VaultOptions } from '../vault';
+import { accessOptions } from '../set-access-control';
+import { infoOptions } from '../set-info';
+import { generateAlternatives } from './alternatives';
+
+const booleans = [true, false];
+
+const blueprint = {
+  name: ['MyVault'],
+  symbol: ['MTK'],
+  decimalsOffset: ['0', '2'],
+  pausable: booleans,
+  upgradeable: booleans,
+  access: accessOptions,
+  info: infoOptions,
+  explicitImplementations: booleans,
+};
+
+export function* generateVaultOptions(): Generator<Required<VaultOptions>> {
+  yield* generateAlternatives(blueprint);
+}

package/src/index.ts

@@ -25,8 +25,10 @@ export { fungible } from './api';
 export { governor } from './api';
 export { nonFungible } from './api';
 export { stablecoin } from './api';
+export { vault } from './api';
 
 export type { FungibleOptions } from './fungible';
 export type { GovernorOptions } from './governor';
 export type { NonFungibleOptions } from './non-fungible';
 export type { StablecoinOptions } from './stablecoin';
+export type { VaultOptions } from './vault';

package/src/kind.ts

@@ -18,6 +18,7 @@ function isKind<T>(value: Kind | T): value is Kind {
     case 'Governor':
     case 'NonFungible':
     case 'Stablecoin':
+    case 'Vault':
       return true;
 
     default: {

package/src/set-info.ts

@@ -1,4 +1,17 @@
 import type { ContractBuilder } from './contract';
+import { OptionsError } from './error';
+
+// These values are printed into a // comment line in the generated Rust. LF
+// ends that comment, letting following text become source. CR and U+2028/U+2029
+// cannot break out in Rust, but are rejected as defense in depth (review tools
+// render U+2028/U+2029 as line breaks).
+const LINE_TERMINATOR = /[\n\r\u2028\u2029]/u;
+
+function checkSingleLine(value: string, field: string): void {
+  if (LINE_TERMINATOR.test(value)) {
+    throw new OptionsError({ [field]: 'Must not contain line breaks' });
+  }
+}
 
 export const infoOptions = [{}, { license: 'WTFPL' }] as const;
 
@@ -10,7 +23,13 @@ export type Info = {
 };
 
 export function setInfo(c: ContractBuilder, { securityContact, license }: Info): void {
-  if (securityContact) c.addContractMetadata({ key: 'security_contact', value: securityContact });
+  if (securityContact) {
+    checkSingleLine(securityContact, 'securityContact');
+    c.addContractMetadata({ key: 'security_contact', value: securityContact });
+  }
 
-  if (license) c.license = license;
+  if (license) {
+    checkSingleLine(license, 'license');
+    c.license = license;
+  }
 }