feat: add governance flag

burnt-labs · Nov 15, 2024 · 9890087 · 9890087
1 parent 4f1e335
commit 9890087
Show file tree

Hide file tree

Showing 4 changed files with 6,263 additions and 5,101 deletions.
diff --git a/packages/abstraxion-core/src/AbstraxionAuth.ts b/packages/abstraxion-core/src/AbstraxionAuth.ts
@@ -17,7 +17,8 @@ export class AbstraxionAuth {
   bank?: SpendLimit[];
   callbackUrl?: string;
   treasury?: string;
-
+  governance?: boolean;
+
   // Signer
   private client?: GranteeSignerClient;
 
@@ -44,6 +45,7 @@ export class AbstraxionAuth {
    * @param {SpendLimit[]} [bank] - The spend limits for the user.
    * @param {string} callbackUrl - preferred callback url to override default
    * @param {string} treasury - treasury contract instance address
+   * @param {boolean} [governance] - Indicates whether governance is enabled.
    */
   configureAbstraxionInstance(
     rpc: string,
@@ -53,6 +55,7 @@ export class AbstraxionAuth {
     bank?: SpendLimit[],
     callbackUrl?: string,
     treasury?: string,
+    governance?: boolean,
   ) {
     this.rpcUrl = rpc;
     this.restUrl = restUrl;
@@ -61,6 +64,7 @@ export class AbstraxionAuth {
     this.bank = bank;
     this.callbackUrl = callbackUrl;
     this.treasury = treasury;
+    this.governance = governance;
   }
 
   /**
@@ -262,6 +266,10 @@ export class AbstraxionAuth {
         urlParams.set("stake", "true");
       }
 
+      if (this.governance) {
+        urlParams.set("governance", "true");
+      }
+
       if (this.grantContracts) {
         urlParams.set("contracts", JSON.stringify(this.grantContracts));
       }

diff --git a/packages/abstraxion/src/components/Abstraxion/index.tsx b/packages/abstraxion/src/components/Abstraxion/index.tsx
@@ -70,6 +70,7 @@ export interface AbstraxionConfig {
   callbackUrl?: string;
   treasury?: string;
   gasPrice?: string;
+  governance?: boolean;
 }
 
 export function AbstraxionProvider({
@@ -82,6 +83,7 @@ export function AbstraxionProvider({
   return (
     <AbstraxionContextProvider
       contracts={config.contracts}
+      governance={config.governance}
       rpcUrl={config.rpcUrl}
       restUrl={config.restUrl}
       stake={config.stake}

diff --git a/packages/abstraxion/src/components/AbstraxionContext/index.tsx b/packages/abstraxion/src/components/AbstraxionContext/index.tsx
@@ -35,6 +35,7 @@ export interface AbstraxionContextProps {
   stake?: boolean;
   bank?: SpendLimit[];
   treasury?: string;
+  governance?: boolean;
   gasPrice: GasPrice;
   logout: () => void;
 }
@@ -52,6 +53,7 @@ export function AbstraxionContextProvider({
   bank,
   callbackUrl,
   treasury,
+  governance,
   gasPrice,
 }: {
   children: ReactNode;
@@ -63,6 +65,7 @@ export function AbstraxionContextProvider({
   bank?: SpendLimit[];
   callbackUrl?: string;
   treasury?: string;
+  governance?: boolean;
   gasPrice?: string;
 }): JSX.Element {
   const [abstraxionError, setAbstraxionError] = useState("");
@@ -91,8 +94,9 @@ export function AbstraxionContextProvider({
       bank,
       callbackUrl,
       treasury,
+      governance, 
     );
-  }, [rpcUrl, restUrl, contracts, stake, bank, callbackUrl, treasury]);
+  }, [rpcUrl, restUrl, contracts, stake, bank, callbackUrl, treasury, governance]);
 
   useEffect(() => {
     configureInstance();
@@ -170,6 +174,7 @@ export function AbstraxionContextProvider({
         stake,
         bank,
         treasury,
+        governance,
         logout,
         gasPrice: gasPrice ? GasPrice.fromString(gasPrice) : gasPriceDefault,
       }}