Add share capital

NeokingdomDAO · Dec 5, 2023 · ebd56cd · ebd56cd
1 parent 29d54af
commit ebd56cd
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/lib/internal/types.ts b/lib/internal/types.ts
@@ -1,7 +1,4 @@
 import { TransactionResponse } from "@ethersproject/providers";
-import { SignerWithAddress } from "@nomiclabs/hardhat-ethers/signers";
-import { Wallet } from "ethers";
-import { HardhatRuntimeEnvironment } from "hardhat/types";
 
 import {
   DAORoles,
@@ -45,7 +42,8 @@ export type ContractNames = keyof typeof FACTORIES;
 export type Contributor = {
   name?: string;
   address: string;
-  status?: "contributor" | "board" | "investor";
+  status: "contributor" | "board" | "investor";
+  shareBalance: string;
   balance?: string;
   vestingBalance?: string;
 };
@@ -64,6 +62,7 @@ export type DAOConfig = {
   reserveAddress: Address;
   usdcAddress: Address;
   diaOracleAddress: Address;
+  shareCapital: string;
   contributors: Contributor[];
 };
 

diff --git a/lib/sequence/setup.ts b/lib/sequence/setup.ts
@@ -29,11 +29,20 @@ export function generateSetupContext(config: DAOConfig) {
 }
 
 export const SETUP_SEQUENCE: Sequence<SetupContext> = [
-  // Give each address one share
+  (c) =>
+    c.shareholderRegistry.mint(
+      c.shareholderRegistry.address,
+      c.config.shareCapital
+    ),
+
   expandable((preprocessContext: SetupContext) =>
     preprocessContext.config.contributors.map(
       (contributor) => (c) =>
-        c.shareholderRegistry.mint(contributor.address, parseEther("1"))
+        c.shareholderRegistry.transferFrom(
+          c.shareholderRegistry.address,
+          contributor.address,
+          contributor.shareBalance
+        )
     )
   ),
 

diff --git a/test/utils/setup.ts b/test/utils/setup.ts
@@ -1,5 +1,6 @@
 import { SignerWithAddress } from "@nomiclabs/hardhat-ethers/signers";
 import { Wallet } from "ethers";
+import { parseEther } from "ethers/lib/utils";
 
 import { DEPLOY_SEQUENCE, generateDeployContext } from "../../lib";
 import { NeokingdomDAOMemory } from "../../lib/environment/memory";
@@ -13,6 +14,7 @@ const config: DAOConfig = {
   tokenSymbol: "TD",
   governanceTokenName: "Test DAO Governance",
   governanceTokenSymbol: "TDGOV",
+  shareCapital: parseEther("10000").toString(),
   shareTokenName: "Test DAO Share",
   shareTokenSymbol: "TDSHARE",
   reserveAddress: "0x0000000000000000000000000000000000000001",