feat(governance): implement governance checks and upgrade wallet component

Author: QSchlegel

src/components/multisig/inspect-multisig-script.tsx

@@ -72,11 +72,11 @@ export default function InspectMultisigScript({
     );
   }
 
-  if (mWallet?.buildScript(3) !== undefined) {
+  if (mWallet?.buildScript(3) !== undefined && mWallet.isGovernanceEnabled()) {
     slides.push(
       <RowLabelInfo
         key="stake-3"
-        label="stake:"
+        label="drep:"
         value={<Code>{JSON.stringify(mWallet.buildScript(3), null, 2)}</Code>}
       />,
     );
@@ -98,7 +98,7 @@ export default function InspectMultisigScript({
         />
       )}
 {/* add pending rewards like balance */}
-      {mWallet.stakingEnabled() && <RowLabelInfo
+      {mWallet.isGovernanceEnabled() && <RowLabelInfo
         label="dRep ID"
         value={<Code>{mWallet.getDRepId()}</Code>}
         copyString={mWallet.getDRepId()}

src/components/pages/wallet/info/index.tsx

@@ -8,6 +8,7 @@ import { ArchiveWallet } from "./archive-wallet";
 import InspectMultisigScript from "@/components/multisig/inspect-multisig-script";
 import { UpgradeStakingWallet } from "./upgrade-staking-wallet";
 import { RegisterWallet } from "./register-wallet";
+import { UpgradeGovernanceWallet } from "./upgrade-governance-wallet";
 
 export default function WalletInfo() {
   const { appWallet } = useAppWallet();
@@ -25,6 +26,7 @@ export default function WalletInfo() {
           {multisigWallet && multisigWallet.stakingEnabled() && <InspectMultisigScript mWallet={multisigWallet} />}
           {multisigWallet && <RegisterWallet mWallet={multisigWallet} appWallet={appWallet} />}
           {multisigWallet && <UpgradeStakingWallet mWallet={multisigWallet} appWallet={appWallet} />}
+          {multisigWallet && <UpgradeGovernanceWallet mWallet={multisigWallet} />}
           <ArchiveWallet appWallet={appWallet} />
           <MigrateWallet appWallet={appWallet} />
         </div>

src/components/pages/wallet/info/upgrade-governance-wallet.tsx

@@ -0,0 +1,35 @@
+import {useMemo} from "react";
+
+import type { MultisigWallet } from "@/utils/multisigSDK";
+
+import CardUI from "@/components/ui/card-content";
+
+export function UpgradeGovernanceWallet({
+  mWallet,
+}: {
+  mWallet?: MultisigWallet;
+}) {
+
+
+
+  const upgraded = useMemo(() => {
+    return mWallet?.isGovernanceEnabled();
+  }, [mWallet]);
+
+  if (!mWallet || upgraded ) return null;
+
+  return (
+    <CardUI
+      title="Upgrade Wallet for Governance"
+      description="Add a drep key script to your multisig Wallet."
+      cardClassName="col-span-2"
+    >
+      {!mWallet.stakingEnabled() && (
+        <div>
+          Not all drep keys have been added. Click Edit Signers to add your
+          drep key!
+        </div>
+      )}
+    </CardUI>
+  );
+}

src/hooks/common.ts

@@ -50,6 +50,21 @@ export function buildMultisigWallet(
       }
     });
   }
+  if (wallet.signersDRepKeys && wallet.signersDRepKeys.length > 0) {
+    wallet.signersDRepKeys.forEach((dRepKey, i) => {
+      if (dRepKey) {
+        try {
+          keys.push({
+            keyHash: dRepKey,
+            role: 3,
+            name: wallet.signersDescriptions[i] || "",
+          });
+        } catch (e) {
+          console.warn(`Invalid dRep address at index ${i}:`, dRepKey);
+        }
+      }
+    });
+  }
   if (keys.length === 0) return;
   const multisigWallet = new MultisigWallet(
     wallet.name,

src/utils/multisigSDK.ts

@@ -102,7 +102,7 @@ export class MultisigWallet {
       );
       return undefined;
     }
-    return getScript(paymentScript, this.network, ).scriptCbor;
+    return getScript(paymentScript, this.network).scriptCbor;
   }
 
   getStakingScript(): string | undefined {
@@ -183,6 +183,16 @@ export class MultisigWallet {
     );
   }
 
+  isGovernanceEnabled(): boolean {
+    const paymentKeys = this.getKeysByRole(0);
+    const dRepKeys = this.getKeysByRole(3);
+    if (!paymentKeys || !dRepKeys || dRepKeys.length === 0) return false;
+    console.log(
+      `Governance enabled: ${dRepKeys.length} dRep keys and ${paymentKeys.length} payment keys`
+    );
+    return dRepKeys.length === paymentKeys.length;
+  }
+
   getDRepId(): string | undefined {
     return getDRepIds(this.getDRepId105()!).cip129;
   }
@@ -192,6 +202,13 @@ export class MultisigWallet {
    * built from role-3 keys (DRep keys). Throws if no script is built.
    */
   getDRepId105(): string | undefined {
+    // Check if all signers have supplied DRep keys
+    if (this.isGovernanceEnabled()) {
+      const dRepScript = this.buildScript(3);
+      if (dRepScript) {
+        return resolveScriptHashDRepId(resolveNativeScriptHash(dRepScript));
+      }
+    }
     return resolveScriptHashDRepId(
       resolveNativeScriptHash(this.buildScript(0)!), // Still Wrong should be 3 -> for drep keys.
     );
@@ -288,7 +305,7 @@ function getScript(
     script,
     stakeCredentialHash,
     network,
-    enabled
+    enabled,
   );
   if (!scriptCbor) {
     throw new Error("Failed to serialize multisig script");