feat: add supply option to ERC1155

Author: ericnordelo

.changeset/cairo-alpha-erc1155-supply.md

@@ -0,0 +1,5 @@
+---
+"@openzeppelin/wizard-common": patch
+---
+
+Cairo: ERC1155 supply tracking description for AI prompts.

packages/common/src/ai/descriptions/cairo.ts

@@ -66,6 +66,7 @@ export const cairoERC1155Descriptions = {
   baseUri:
     'The location of the metadata for the token. Clients will replace any instance of {id} in this string with the tokenId.',
   updatableUri: 'Whether privileged accounts will be able to set a new URI for all token types.',
+  supply: 'Whether to keep track of total supply of tokens.',
 };
 
 export const cairoGovernorDescriptions = {

packages/core/cairo_alpha/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 3.1.0 (Unreleased)
+
+- Add ERC1155 supply tracking extension () <!-- TODO: Add linked PR>
+
 ## 3.0.0 (2026-01-10)
 
 - Add support for `with_components` macro. ([#703](https://github.com/OpenZeppelin/contracts-wizard/pull/703))

packages/core/cairo_alpha/ava.config.js

@@ -1,6 +1,7 @@
 module.exports = {
   extensions: ['ts'],
   require: ['ts-node/register'],
+  files: ['src/**/*.test.ts'],
   watchmode: {
     ignoreChanges: ['contracts', 'artifacts', 'cache'],
   },

packages/core/cairo_alpha/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@openzeppelin/wizard-cairo-alpha",
   "private": true,
-  "version": "3.0.0",
+  "version": "4.0.0-alpha.0",
   "description": "A boilerplate generator to get started with the latest alpha version of OpenZeppelin Contracts for Cairo",
   "license": "AGPL-3.0-only",
   "repository": "https://github.com/OpenZeppelin/contracts-wizard",

packages/core/cairo_alpha/src/erc1155.ts

@@ -23,6 +23,7 @@ export const defaults: Required<ERC1155Options> = {
   burnable: false,
   pausable: false,
   mintable: false,
+  supply: false,
   updatableUri: true,
   royaltyInfo: royaltyInfoDefaults,
   access: commonDefaults.access,
@@ -41,6 +42,7 @@ export interface ERC1155Options extends CommonContractOptions {
   burnable?: boolean;
   pausable?: boolean;
   mintable?: boolean;
+  supply?: boolean;
   updatableUri?: boolean;
   royaltyInfo?: RoyaltyInfoOptions;
 }
@@ -52,6 +54,7 @@ function withDefaults(opts: ERC1155Options): Required<ERC1155Options> {
     burnable: opts.burnable ?? defaults.burnable,
     pausable: opts.pausable ?? defaults.pausable,
     mintable: opts.mintable ?? defaults.mintable,
+    supply: opts.supply ?? defaults.supply,
     updatableUri: opts.updatableUri ?? defaults.updatableUri,
     royaltyInfo: opts.royaltyInfo ?? defaults.royaltyInfo,
   };
@@ -86,6 +89,10 @@ export function buildERC1155(opts: ERC1155Options): Contract {
     addMintable(c, allOpts.access);
   }
 
+  if (allOpts.supply) {
+    addSupply(c);
+  }
+
   if (allOpts.updatableUri) {
     addSetBaseUri(c, allOpts.access);
   }
@@ -101,7 +108,7 @@ export function buildERC1155(opts: ERC1155Options): Contract {
 }
 
 function addHooks(c: ContractBuilder, allOpts: Required<ERC1155Options>) {
-  const usesCustomHooks = allOpts.pausable;
+  const usesCustomHooks = allOpts.pausable || allOpts.supply;
   if (usesCustomHooks) {
     const hooksTrait = {
       name: 'ERC1155HooksImpl',
@@ -112,20 +119,43 @@ function addHooks(c: ContractBuilder, allOpts: Required<ERC1155Options>) {
     c.addImplementedTrait(hooksTrait);
     c.addUseClause('starknet', 'ContractAddress');
 
-    c.addFunction(hooksTrait, {
-      name: 'before_update',
-      args: [
-        {
-          name: 'ref self',
-          type: `ERC1155Component::ComponentState<ContractState>`,
-        },
-        { name: 'from', type: 'ContractAddress' },
-        { name: 'to', type: 'ContractAddress' },
-        { name: 'token_ids', type: 'Span<u256>' },
-        { name: 'values', type: 'Span<u256>' },
-      ],
-      code: ['let contract_state = self.get_contract()', 'contract_state.pausable.assert_not_paused()'],
-    });
+    if (allOpts.pausable) {
+      c.addFunction(hooksTrait, {
+        name: 'before_update',
+        args: [
+          {
+            name: 'ref self',
+            type: `ERC1155Component::ComponentState<ContractState>`,
+          },
+          { name: 'from', type: 'ContractAddress' },
+          { name: 'to', type: 'ContractAddress' },
+          { name: 'token_ids', type: 'Span<u256>' },
+          { name: 'values', type: 'Span<u256>' },
+        ],
+        code: ['let contract_state = self.get_contract()', 'contract_state.pausable.assert_not_paused()'],
+      });
+    }
+
+    if (allOpts.supply) {
+      const afterUpdateFn = c.addFunction(hooksTrait, {
+        name: 'after_update',
+        args: [
+          {
+            name: 'ref self',
+            type: `ERC1155Component::ComponentState<ContractState>`,
+          },
+          { name: 'from', type: 'ContractAddress' },
+          { name: 'to', type: 'ContractAddress' },
+          { name: 'token_ids', type: 'Span<u256>' },
+          { name: 'values', type: 'Span<u256>' },
+        ],
+        code: [],
+      });
+      afterUpdateFn.code.push(
+        'let mut contract_state = self.get_contract_mut();',
+        'contract_state.erc1155_supply.after_update(from, to, token_ids, values);',
+      );
+    }
   } else {
     c.addUseClause('openzeppelin_token::erc1155', 'ERC1155HooksEmptyImpl');
   }
@@ -153,6 +183,10 @@ function addBurnable(c: ContractBuilder) {
   c.addFunction(externalTrait, functions.batchBurn);
 }
 
+function addSupply(c: ContractBuilder) {
+  c.addComponent(components.ERC1155SupplyComponent, [], true);
+}
+
 function addMintable(c: ContractBuilder, access: Access) {
   c.addUseClause('starknet', 'ContractAddress');
   requireAccessControl(c, externalTrait, functions.mint, access, 'MINTER', 'minter');
@@ -188,6 +222,29 @@ const components = defineComponents({
       },
     ],
   },
+  ERC1155SupplyComponent: {
+    path: 'openzeppelin_token::erc1155::extensions',
+    substorage: {
+      name: 'erc1155_supply',
+      type: 'ERC1155SupplyComponent::Storage',
+    },
+    event: {
+      name: 'ERC1155SupplyEvent',
+      type: 'ERC1155SupplyComponent::Event',
+    },
+    impls: [
+      {
+        name: 'ERC1155SupplyImpl',
+        embed: true,
+        value: 'ERC1155SupplyComponent::ERC1155SupplyImpl<ContractState>',
+      },
+      {
+        name: 'ERC1155SupplyInternalImpl',
+        embed: false,
+        value: 'ERC1155SupplyComponent::InternalImpl<ContractState>',
+      },
+    ],
+  },
 });
 
 const functions = defineFunctions({

packages/core/cairo_alpha/src/generate/erc1155.ts

@@ -24,6 +24,7 @@ function prepareBlueprint(opts: GeneratorOptions) {
     burnable: booleans,
     pausable: booleans,
     mintable: booleans,
+    supply: booleans,
     updatableUri: booleans,
     upgradeable: upgradeableOptions,
     royaltyInfo: resolveRoyaltyOptionsSubset(opts.royaltyInfo),