@openzeppelin/wizard-stellar 0.5.0 → 0.6.1

package/dist/zip-shared.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.workspaceCargo = exports.createRustLibFile = exports.printContractCargo = exports.addDependenciesWith = exports.allDependencies = exports.libDependencies = exports.printRustNameTest = exports.contractOptionsToContractName = void 0;
+exports.workspaceCargo = exports.createRustLibFile = exports.printContractCargo = exports.addDependenciesWith = exports.printRustNameTest = exports.contractOptionsToContractName = void 0;
 exports.getAddressArgs = getAddressArgs;
+exports.getRequiredLibDependencies = getRequiredLibDependencies;
 const version_1 = require("./utils/version");
 function pascalToSnakeCase(string) {
     return string
@@ -38,16 +39,27 @@ fn initial_state() {
 // Add more tests bellow
 `;
 exports.printRustNameTest = printRustNameTest;
-exports.libDependencies = [
+const libDependencies = [
     'stellar-tokens',
     'stellar-access',
     'stellar-contract-utils',
+    'stellar-governance',
     'stellar-macros',
 ];
-exports.allDependencies = [...exports.libDependencies, 'soroban-sdk'];
+// Derives required lib crate dependencies from the contract's use clauses.
+// Each use clause has a containerPath like "stellar_tokens::fungible::Base".
+// We extract the first segment (the crate name), convert underscores to hyphens
+// to match Cargo naming (e.g., stellar_tokens -> stellar-tokens), then filter
+// libDependencies to include only crates the contract actually references.
+// Filtering against the known list preserves stable ordering and ignores
+// non-lib crates like soroban_sdk (which is always added separately).
+function getRequiredLibDependencies(c) {
+    const usedCrates = new Set(c.useClauses.map(uc => uc.containerPath.split('::')[0].replace(/_/g, '-')));
+    return libDependencies.filter(dep => usedCrates.has(dep));
+}
 const addDependenciesWith = (dependencyValue, dependenciesToAdd) => `${dependenciesToAdd.map(dependency => `${dependency} = ${dependencyValue}\n`).join('')}`;
 exports.addDependenciesWith = addDependenciesWith;
-const printContractCargo = (scaffoldContractName) => `[package]
+const printContractCargo = (scaffoldContractName, requiredLibDeps) => `[package]
 name = "${scaffoldContractName.replace(/_/, '-')}-contract"
 edition.workspace = true
 license.workspace = true
@@ -62,7 +74,7 @@ crate-type = ["cdylib"]
 doctest = false
 
 [dependencies]
-${(0, exports.addDependenciesWith)('{ workspace = true }', [...exports.allDependencies])}
+${(0, exports.addDependenciesWith)('{ workspace = true }', [...requiredLibDeps, 'soroban-sdk'])}
 [dev-dependencies]
 ${(0, exports.addDependenciesWith)('{ workspace = true, features = ["testutils"] }', ['soroban-sdk'])}`;
 exports.printContractCargo = printContractCargo;
@@ -72,7 +84,7 @@ exports.createRustLibFile = `#![no_std]
 mod contract;
 mod test;
 `;
-exports.workspaceCargo = `[workspace]
+const workspaceCargo = (requiredLibDeps) => `[workspace]
 resolver = "2"
 members = ["contracts/*"]
 
@@ -83,7 +95,7 @@ license = "Apache-2.0"
 version = "0.0.1"
 
 [workspace.dependencies]
-${(0, exports.addDependenciesWith)(`"${version_1.compatibleSorobanVersion}"`, ['soroban-sdk'])}${(0, exports.addDependenciesWith)(`"=${version_1.contractsVersion}"`, [...exports.libDependencies])}
+${(0, exports.addDependenciesWith)(`"${version_1.compatibleSorobanVersion}"`, ['soroban-sdk'])}${(0, exports.addDependenciesWith)(`"=${version_1.contractsVersion}"`, [...requiredLibDeps])}
 
 [profile.release]
 opt-level = "z"
@@ -99,4 +111,5 @@ lto = true
 inherits = "release"
 debug-assertions = true
 `;
+exports.workspaceCargo = workspaceCargo;
 //# sourceMappingURL=zip-shared.js.map

package/dist/zip-shared.js.map

@@ -1 +1 @@
-{"version":3,"file":"zip-shared.js","sourceRoot":"","sources":["../src/zip-shared.ts"],"names":[],"mappings":";;;AAYA,wCAIC;AAfD,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;AAEW,QAAA,eAAe,GAAG;IAC7B,gBAAgB;IAChB,gBAAgB;IAChB,wBAAwB;IACxB,gBAAgB;CACR,CAAC;AAEE,QAAA,eAAe,GAAG,CAAC,GAAG,uBAAe,EAAE,aAAa,CAAU,CAAC;AAErE,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,EAAE,CAAC;UAC1D,oBAAoB,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;;;;;;;;;;;;;;EAc9C,IAAA,2BAAmB,EAAC,sBAAsB,EAAE,CAAC,GAAG,uBAAe,CAAC,CAAC;;EAEjE,IAAA,2BAAmB,EAAC,gDAAgD,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;AAjB9E,QAAA,kBAAkB,sBAiB4D;AAE9E,QAAA,iBAAiB,GAAG;;;;;CAKhC,CAAC;AAEW,QAAA,cAAc,GAAG;;;;;;;;;;;EAW5B,IAAA,2BAAmB,EAAC,IAAI,kCAAwB,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,GAAG,IAAA,2BAAmB,EAAC,KAAK,0BAAgB,GAAG,EAAE,CAAC,GAAG,uBAAe,CAAC,CAAC;;;;;;;;;;;;;;;CAe5I,CAAC"}
+{"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"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@openzeppelin/wizard-stellar",
-  "version": "0.5.0",
+  "version": "0.6.1",
   "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/add-upgradeable.ts

@@ -6,31 +6,30 @@ import { defineFunctions } from './utils/define-functions';
 
 export function addUpgradeable(c: ContractBuilder, access: Access, explicitImplementations: boolean) {
   const functions = defineFunctions({
-    _require_auth: {
-      args: [getSelfArg(), { name: 'operator', type: '&Address' }],
-      code: ['operator.require_auth();'],
+    upgrade: {
+      args: [getSelfArg(), { name: 'new_wasm_hash', type: 'BytesN<32>' }, { name: 'operator', type: 'Address' }],
+      code: ['upgradeable::upgrade(e, &new_wasm_hash)'],
     },
-    _require_auth_unused_operator: {
-      name: '_require_auth',
-      args: [getSelfArg(), { name: '_operator', type: '&Address' }],
-      code: [],
+    upgrade_unused_operator: {
+      name: 'upgrade',
+      args: [getSelfArg(), { name: 'new_wasm_hash', type: 'BytesN<32>' }, { name: '_operator', type: 'Address' }],
+      code: ['upgradeable::upgrade(e, &new_wasm_hash)'],
     },
   });
 
-  c.addUseClause('stellar_contract_utils::upgradeable', 'UpgradeableInternal');
-  c.addUseClause('stellar_macros', 'Upgradeable');
-
-  c.addDerives('Upgradeable');
+  c.addUseClause('soroban_sdk', 'Address');
+  c.addUseClause('soroban_sdk', 'BytesN');
+  c.addUseClause('stellar_contract_utils::upgradeable', 'self', { alias: 'upgradeable' });
+  c.addUseClause('stellar_contract_utils::upgradeable', 'Upgradeable');
 
   const upgradeableTrait = {
-    traitName: 'UpgradeableInternal',
+    traitName: 'Upgradeable',
     structName: c.name,
-    tags: [],
+    tags: ['contractimpl'],
     section: 'Utils',
   };
 
-  const upgradeFn: BaseFunction =
-    access === 'ownable' ? functions._require_auth_unused_operator : functions._require_auth;
+  const upgradeFn: BaseFunction = access === 'ownable' ? functions.upgrade_unused_operator : functions.upgrade;
 
   c.addTraitFunction(upgradeableTrait, upgradeFn);
 
@@ -40,7 +39,7 @@ export function addUpgradeable(c: ContractBuilder, access: Access, explicitImple
     upgradeFn,
     access,
     {
-      useMacro: false,
+      useMacro: true,
       role: 'upgrader',
       caller: 'operator',
     },

package/src/api.ts

@@ -1,5 +1,6 @@
-import type { CommonContractOptions } from './common-options';
+import type { CommonContractOptions, CommonOptions } from './common-options';
 import type { FungibleOptions } from './fungible';
+import type { GovernorOptions } from './governor';
 import type { NonFungibleOptions } from './non-fungible';
 import type { StablecoinOptions } from './stablecoin';
 import {
@@ -12,13 +13,18 @@ import {
   defaults as fungibledefaults,
   isAccessControlRequired as fungibleIsAccessControlRequired,
 } from './fungible';
+import {
+  printGovernor,
+  defaults as governorDefaults,
+  isAccessControlRequired as governorIsAccessControlRequired,
+} from './governor';
 import {
   printNonFungible,
   defaults as nonFungibledefaults,
   isAccessControlRequired as nonFungibleIsAccessControlRequired,
 } from './non-fungible';
 
-export interface WizardContractAPI<Options extends CommonContractOptions> {
+export interface WizardContractAPI<Options extends CommonOptions> {
   /**
    * Returns a string representation of a contract generated using the provided options. If opts is not provided, uses `defaults`.
    */
@@ -39,6 +45,7 @@ export interface AccessControlAPI<Options extends CommonContractOptions> {
 }
 
 export type Fungible = WizardContractAPI<FungibleOptions> & AccessControlAPI<FungibleOptions>;
+export type Governor = WizardContractAPI<GovernorOptions> & AccessControlAPI<GovernorOptions>;
 export type NonFungible = WizardContractAPI<NonFungibleOptions> & AccessControlAPI<NonFungibleOptions>;
 export type Stablecoin = WizardContractAPI<StablecoinOptions> & AccessControlAPI<StablecoinOptions>;
 
@@ -48,6 +55,12 @@ export const fungible: Fungible = {
   isAccessControlRequired: fungibleIsAccessControlRequired,
 };
 
+export const governor: Governor = {
+  print: printGovernor,
+  defaults: governorDefaults,
+  isAccessControlRequired: governorIsAccessControlRequired,
+};
+
 export const nonFungible: NonFungible = {
   print: printNonFungible,
   defaults: nonFungibledefaults,

package/src/build-generic.ts

@@ -1,5 +1,7 @@
 import type { FungibleOptions } from './fungible';
 import { buildFungible } from './fungible';
+import type { GovernorOptions } from './governor';
+import { buildGovernor } from './governor';
 import type { NonFungibleOptions } from './non-fungible';
 import { buildNonFungible } from './non-fungible';
 import type { StablecoinOptions } from './stablecoin';
@@ -7,6 +9,7 @@ import { buildStablecoin } from './stablecoin';
 
 export interface KindedOptions {
   Fungible: { kind: 'Fungible' } & FungibleOptions;
+  Governor: { kind: 'Governor' } & GovernorOptions;
   NonFungible: { kind: 'NonFungible' } & NonFungibleOptions;
   Stablecoin: { kind: 'Stablecoin' } & StablecoinOptions;
 }
@@ -18,6 +21,9 @@ export function buildGeneric(opts: GenericOptions) {
     case 'Fungible':
       return buildFungible(opts);
 
+    case 'Governor':
+      return buildGovernor(opts);
+
     case 'NonFungible':
       return buildNonFungible(opts);
 

package/src/contract.ts

@@ -10,6 +10,7 @@ export interface Contract {
   constructorArgs: Argument[];
   implementedTraits: TraitImplBlock[];
   freeFunctions: ContractFunction[];
+  topLevelDeclarations: string[][];
   variables: Variable[];
   errors: Error[];
   ownable: boolean;
@@ -81,6 +82,7 @@ export class ContractBuilder implements Contract {
 
   readonly constructorArgs: Argument[] = [];
   readonly constructorCode: string[] = [];
+  readonly topLevelDeclarations: string[][] = [];
 
   private implementedTraitsMap: Map<string, TraitImplBlock> = new Map();
   private freeFunctionsMap: Map<string, ContractFunction> = new Map();
@@ -250,6 +252,10 @@ export class ContractBuilder implements Contract {
     existingFn.code = [...code];
   }
 
+  addTopLevelDeclaration(lines: string[]): void {
+    this.topLevelDeclarations.push(lines);
+  }
+
   addConstructorArgument(arg: Argument): void {
     for (const existingArg of this.constructorArgs) {
       if (existingArg.name == arg.name) {

package/src/fungible.ts

@@ -13,10 +13,14 @@ import { printContract } from './print';
 import { toByteArray, toUint } from './utils/convert-strings';
 import { pickKeys } from '@openzeppelin/wizard-common';
 
+const DEFAULT_DECIMALS = 7;
+
 export const defaults: Required<FungibleOptions> = {
   name: 'MyToken',
   symbol: 'MTK',
+  decimals: DEFAULT_DECIMALS.toString(),
   burnable: false,
+  votes: false,
   pausable: false,
   upgradeable: false,
   premint: '0',
@@ -33,7 +37,9 @@ export function printFungible(opts: FungibleOptions = defaults): string {
 export interface FungibleOptions extends CommonContractOptions {
   name: string;
   symbol: string;
+  decimals?: string;
   burnable?: boolean;
+  votes?: boolean;
   pausable?: boolean;
   upgradeable?: boolean;
   premint?: string;
@@ -44,7 +50,9 @@ export function withDefaults(opts: FungibleOptions): Required<FungibleOptions> {
   return {
     ...opts,
     ...withCommonContractDefaults(opts),
+    decimals: opts.decimals ?? defaults.decimals,
     burnable: opts.burnable ?? defaults.burnable,
+    votes: opts.votes ?? defaults.votes,
     pausable: opts.pausable ?? defaults.pausable,
     upgradeable: opts.upgradeable ?? defaults.upgradeable,
     premint: opts.premint || defaults.premint,
@@ -61,10 +69,20 @@ export function buildFungible(opts: FungibleOptions): ContractBuilder {
 
   const allOpts = withDefaults(opts);
 
-  addBase(c, toByteArray(allOpts.name), toByteArray(allOpts.symbol), allOpts.pausable, allOpts.explicitImplementations);
+  const decimals = toUint(allOpts.decimals, 'decimals', 'u32');
+
+  addBase(
+    c,
+    toByteArray(allOpts.name),
+    toByteArray(allOpts.symbol),
+    decimals,
+    allOpts.votes,
+    allOpts.pausable,
+    allOpts.explicitImplementations,
+  );
 
   if (allOpts.premint) {
-    addPremint(c, allOpts.premint);
+    addPremint(c, allOpts.premint, decimals, allOpts.votes);
   }
 
   if (allOpts.pausable) {
@@ -76,11 +94,11 @@ export function buildFungible(opts: FungibleOptions): ContractBuilder {
   }
 
   if (allOpts.burnable) {
-    addBurnable(c, allOpts.pausable, allOpts.explicitImplementations);
+    addBurnable(c, allOpts.votes, allOpts.pausable, allOpts.explicitImplementations);
   }
 
   if (allOpts.mintable) {
-    addMintable(c, allOpts.access, allOpts.pausable, allOpts.explicitImplementations);
+    addMintable(c, allOpts.access, allOpts.votes, allOpts.pausable, allOpts.explicitImplementations);
   }
 
   setAccessControl(c, allOpts.access, allOpts.explicitImplementations);
@@ -93,15 +111,24 @@ function addBase(
   c: ContractBuilder,
   name: string,
   symbol: string,
+  decimals: bigint,
+  votes: boolean,
   pausable: boolean,
   explicitImplementations: boolean,
 ) {
   // Set metadata
-  c.addConstructorCode(`Base::set_metadata(e, 7, String::from_str(e, "${name}"), String::from_str(e, "${symbol}"));`);
+  c.addConstructorCode(
+    `Base::set_metadata(e, ${decimals}, 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');
+  if (votes) {
+    c.addUseClause('stellar_tokens::fungible', 'ContractOverrides');
+    c.addUseClause('stellar_governance::votes', 'Votes');
+    c.addUseClause('stellar_tokens::fungible', 'votes::FungibleVotes');
+  }
   c.addUseClause('soroban_sdk', 'contract');
   c.addUseClause('soroban_sdk', 'contractimpl');
   c.addUseClause('soroban_sdk', 'String');
@@ -115,7 +142,7 @@ function addBase(
     traitName: 'FungibleToken',
     structName: c.name,
     tags: explicitImplementations ? ['contractimpl'] : ['contractimpl(contracttrait)'],
-    assocType: 'type ContractType = Base;',
+    assocType: `type ContractType = ${votes ? 'FungibleVotes' : 'Base'};`,
   };
 
   c.addTraitImplBlock(fungibleTokenTrait);
@@ -131,29 +158,47 @@ function addBase(
     c.addTraitFunction(fungibleTokenTrait, functions.transfer_from);
     c.addFunctionTag(functions.transfer_from, 'when_not_paused', fungibleTokenTrait);
   }
+
+  if (votes) {
+    c.addTraitImplBlock({
+      traitName: 'Votes',
+      structName: c.name,
+      tags: explicitImplementations ? ['contractimpl'] : ['contractimpl(contracttrait)'],
+      section: 'Extensions',
+    });
+  }
 }
 
-function addMintable(c: ContractBuilder, access: Access, pausable: boolean, explicitImplementations: boolean) {
+function addMintable(
+  c: ContractBuilder,
+  access: Access,
+  votes: boolean,
+  pausable: boolean,
+  explicitImplementations: boolean,
+) {
+  const mintFn = votes ? votesFunctions.mint : functions.mint;
+  const mintWithCallerFn = votes ? votesFunctions.mint_with_caller : functions.mint_with_caller;
+
   switch (access) {
     case false:
       break;
     case 'ownable': {
-      c.addFreeFunction(functions.mint);
+      c.addFreeFunction(mintFn);
 
-      requireAccessControl(c, undefined, functions.mint, access, undefined, explicitImplementations);
+      requireAccessControl(c, undefined, mintFn, access, undefined, explicitImplementations);
 
       if (pausable) {
-        c.addFunctionTag(functions.mint, 'when_not_paused');
+        c.addFunctionTag(mintFn, 'when_not_paused');
       }
       break;
     }
     case 'roles': {
-      c.addFreeFunction(functions.mint_with_caller);
+      c.addFreeFunction(mintWithCallerFn);
 
       requireAccessControl(
         c,
         undefined,
-        functions.mint_with_caller,
+        mintWithCallerFn,
         access,
         {
           useMacro: true,
@@ -164,7 +209,7 @@ function addMintable(c: ContractBuilder, access: Access, pausable: boolean, expl
       );
 
       if (pausable) {
-        c.addFunctionTag(functions.mint_with_caller, 'when_not_paused');
+        c.addFunctionTag(mintWithCallerFn, 'when_not_paused');
       }
       break;
     }
@@ -175,7 +220,7 @@ function addMintable(c: ContractBuilder, access: Access, pausable: boolean, expl
   }
 }
 
-function addBurnable(c: ContractBuilder, pausable: boolean, explicitImplementations: boolean) {
+function addBurnable(c: ContractBuilder, votes: boolean, pausable: boolean, explicitImplementations: boolean) {
   c.addUseClause('stellar_tokens::fungible', 'burnable::FungibleBurnable');
 
   const fungibleBurnableTrait = {
@@ -185,23 +230,29 @@ function addBurnable(c: ContractBuilder, pausable: boolean, explicitImplementati
     section: 'Extensions',
   };
 
-  if (pausable) {
-    c.addUseClause('stellar_macros', 'when_not_paused');
+  const burnFn = votes ? votesFunctions.burn : functions.burn;
+  const burnFromFn = votes ? votesFunctions.burn_from : functions.burn_from;
+
+  if (pausable || votes || explicitImplementations) {
+    if (pausable) {
+      c.addUseClause('stellar_macros', 'when_not_paused');
+    }
 
-    c.addTraitFunction(fungibleBurnableTrait, functions.burn);
-    c.addFunctionTag(functions.burn, 'when_not_paused', fungibleBurnableTrait);
+    c.addTraitFunction(fungibleBurnableTrait, burnFn);
+    c.addTraitFunction(fungibleBurnableTrait, burnFromFn);
 
-    c.addTraitFunction(fungibleBurnableTrait, functions.burn_from);
-    c.addFunctionTag(functions.burn_from, 'when_not_paused', fungibleBurnableTrait);
-  } else if (explicitImplementations) c.addTraitForEachFunctions(fungibleBurnableTrait, fungibleBurnableFunctions);
-  else {
+    if (pausable) {
+      c.addFunctionTag(burnFn, 'when_not_paused', fungibleBurnableTrait);
+      c.addFunctionTag(burnFromFn, 'when_not_paused', fungibleBurnableTrait);
+    }
+  } else {
     c.addTraitImplBlock(fungibleBurnableTrait);
   }
 }
 
 export const premintPattern = /^(\d*\.?\d*)$/;
 
-function addPremint(c: ContractBuilder, amount: string) {
+function addPremint(c: ContractBuilder, amount: string, decimals: bigint, votes: boolean) {
   if (amount !== undefined && amount !== '0') {
     if (!premintPattern.test(amount)) {
       throw new OptionsError({
@@ -210,10 +261,10 @@ function addPremint(c: ContractBuilder, amount: string) {
     }
 
     // TODO: handle signed int?
-    const premintAbsolute = toUint(getInitialSupply(amount, 7), 'premint', 'u128');
+    const premintAbsolute = toUint(getInitialSupply(amount, Number(decimals)), 'premint', 'u128');
 
     c.addConstructorArgument({ name: 'recipient', type: 'Address' });
-    c.addConstructorCode(`Base::mint(e, &recipient, ${premintAbsolute});`);
+    c.addConstructorCode(`${votes ? 'FungibleVotes' : 'Base'}::mint(e, &recipient, ${premintAbsolute});`);
   }
 }
 
@@ -350,6 +401,36 @@ export const functions = defineFunctions({
   },
 });
 
+const votesFunctions = defineFunctions({
+  burn: {
+    args: [getSelfArg(), { name: 'from', type: 'Address' }, { name: 'amount', type: 'i128' }],
+    code: ['FungibleVotes::burn(e, &from, amount)'],
+  },
+  burn_from: {
+    args: [
+      getSelfArg(),
+      { name: 'spender', type: 'Address' },
+      { name: 'from', type: 'Address' },
+      { name: 'amount', type: 'i128' },
+    ],
+    code: ['FungibleVotes::burn_from(e, &spender, &from, amount)'],
+  },
+  mint: {
+    args: [getSelfArg(), { name: 'account', type: 'Address' }, { name: 'amount', type: 'i128' }],
+    code: ['FungibleVotes::mint(e, &account, amount);'],
+  },
+  mint_with_caller: {
+    name: 'mint',
+    args: [
+      getSelfArg(),
+      { name: 'account', type: 'Address' },
+      { name: 'amount', type: 'i128' },
+      { name: 'caller', type: 'Address' },
+    ],
+    code: ['FungibleVotes::mint(e, &account, amount);'],
+  },
+});
+
 const fungibleTokenTraitFunctions = pickKeys(functions, [
   'total_supply',
   'balance',
@@ -361,5 +442,3 @@ const fungibleTokenTraitFunctions = pickKeys(functions, [
   'name',
   'symbol',
 ]);
-
-const fungibleBurnableFunctions = pickKeys(functions, ['burn', 'burn_from']);

package/src/generate/fungible.ts

@@ -8,7 +8,9 @@ const booleans = [true, false];
 const blueprint = {
   name: ['MyToken'],
   symbol: ['MTK'],
+  decimals: ['7', '18'],
   burnable: booleans,
+  votes: booleans,
   pausable: booleans,
   upgradeable: booleans,
   mintable: booleans,

package/src/generate/governor.ts

@@ -0,0 +1,24 @@
+import type { GovernorOptions } from '../governor';
+import { accessOptions } from '../set-access-control';
+import { infoOptions } from '../set-info';
+import { generateAlternatives } from './alternatives';
+
+const booleans = [true, false];
+
+const blueprint = {
+  name: ['MyGovernor'],
+  version: ['1.0.0'],
+  votingDelay: ['10'],
+  votingPeriod: ['100'],
+  proposalThreshold: ['100'],
+  quorum: ['500'],
+  upgradeable: booleans,
+  timelock: booleans,
+  access: accessOptions,
+  info: infoOptions,
+  explicitImplementations: booleans,
+};
+
+export function* generateGovernorOptions(): Generator<Required<GovernorOptions>> {
+  yield* generateAlternatives(blueprint);
+}

package/src/generate/non-fungible.ts

@@ -10,6 +10,7 @@ const blueprint = {
   symbol: ['MTK'],
   tokenUri: ['https://www.mytoken.com'],
   burnable: booleans,
+  votes: booleans,
   pausable: booleans,
   upgradeable: booleans,
   enumerable: booleans,

package/src/generate/sources.ts

@@ -3,6 +3,7 @@ import path from 'path';
 import crypto from 'crypto';
 
 import { generateFungibleOptions } from './fungible';
+import { generateGovernorOptions } from './governor';
 import { generateNonFungibleOptions } from './non-fungible';
 import { generateStablecoinOptions } from './stablecoin';
 import type { GenericOptions, KindedOptions } from '../build-generic';
@@ -20,6 +21,12 @@ export function* generateOptions(kind?: Kind): Generator<GenericOptions> {
     }
   }
 
+  if (!kind || kind === 'Governor') {
+    for (const kindOpts of generateGovernorOptions()) {
+      yield { kind: 'Governor', ...kindOpts };
+    }
+  }
+
   if (!kind || kind === 'NonFungible') {
     for (const kindOpts of generateNonFungibleOptions()) {
       yield { kind: 'NonFungible', ...kindOpts };

package/src/generate/stablecoin.ts

@@ -8,7 +8,9 @@ const booleans = [true, false];
 const blueprint = {
   name: ['MyStablecoin'],
   symbol: ['MST'],
+  decimals: ['7', '18'],
   burnable: booleans,
+  votes: booleans,
   pausable: booleans,
   upgradeable: booleans,
   mintable: booleans,