From 754e6df986c15e06b2a0de718fad308832ba6452 Mon Sep 17 00:00:00 2001
From: lbqds <lbqds@outlook.com>
Date: Fri, 21 Feb 2025 10:11:51 +0800
Subject: [PATCH] Add getAllContracts method

---
 artifacts/ts/contracts.ts   | 4 ++++
 packages/cli/src/codegen.ts | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/artifacts/ts/contracts.ts b/artifacts/ts/contracts.ts
index a868a39c6..be2cd2703 100644
--- a/artifacts/ts/contracts.ts
+++ b/artifacts/ts/contracts.ts
@@ -6,6 +6,10 @@ import { Contract, ContractFactory } from "@alephium/web3";
 
 let contracts: ContractFactory<any>[] | undefined = undefined;
 
+export function getAllContracts(): ContractFactory<any>[] {
+  return contracts ?? [];
+}
+
 export function registerContract(factory: ContractFactory<any>) {
   if (contracts === undefined) {
     contracts = [factory];
diff --git a/packages/cli/src/codegen.ts b/packages/cli/src/codegen.ts
index a0a18a830..398cbc9a1 100644
--- a/packages/cli/src/codegen.ts
+++ b/packages/cli/src/codegen.ts
@@ -643,6 +643,10 @@ function genContractByCodeHash(outDir: string) {
 
     let contracts: ContractFactory<any>[] | undefined = undefined
 
+    export function getAllContracts(): ContractFactory<any>[] {
+      return contracts ?? []
+    }
+
     export function registerContract(factory: ContractFactory<any>) {
       if (contracts === undefined) {
         contracts = [factory]