feat: CMC post-install to authorize to create canisters in subnet (#43)

* feat: CMC post-install to authorize to create canisters in subnet

* feat: neuronId constant was moved

* fix: nullish

* fix: principal self authenticating

Author: peterpeterparker

cli/src/modules/cmc/cmc.post-install.ts

@@ -0,0 +1,55 @@
+import {IDL} from '@dfinity/candid';
+import {GovernanceCanister, NnsFunction, type MakeProposalRequest} from '@dfinity/nns';
+import {Principal} from '@dfinity/principal';
+import {arrayBufferToUint8Array, createAgent} from '@dfinity/utils';
+import {MAIN_IDENTITY_KEY} from '../../constants/constants';
+import {NEURON_ID} from '../../constants/modules.constants';
+import type {ModuleInstallParams} from '../../types/module';
+
+export const makeAuthorizedSubnetworksProposal = async ({
+  identities
+}: Pick<ModuleInstallParams, 'identities'>) => {
+  const {[MAIN_IDENTITY_KEY]: identity} = identities;
+
+  const agent = await createAgent({
+    identity,
+    host: 'http://127.0.0.1:5987',
+    fetchRootKey: true
+  });
+
+  const {makeProposal} = GovernanceCanister.create({
+    agent
+  });
+
+  const subnetId = Principal.selfAuthenticating(arrayBufferToUint8Array(agent.rootKey));
+
+  const arg = IDL.encode(
+    [
+      IDL.Record({
+        who: IDL.Opt(IDL.Principal),
+        subnets: IDL.Vec(IDL.Principal)
+      })
+    ],
+    [
+      {
+        who: [],
+        subnets: [subnetId]
+      }
+    ]
+  );
+
+  const request: MakeProposalRequest = {
+    neuronId: BigInt(NEURON_ID),
+    url: 'https://forum.dfinity.org',
+    title: 'Authorize CMC to create canisters in subnets',
+    summary: 'The lack of documentation makes developing anything with the CMC canister a pain.',
+    action: {
+      ExecuteNnsFunction: {
+        nnsFunctionId: NnsFunction.SetAuthorizedSubnetworks,
+        payloadBytes: arg
+      }
+    }
+  };
+
+  await makeProposal(request);
+};

cli/src/modules/cmc/index.ts

@@ -1,6 +1,11 @@
+import {assertNonNullish} from '@dfinity/utils';
+import kleur from 'kleur';
 import {Module} from '../../services/modules.services';
 import type {ModuleDescription, ModuleInstallParams} from '../../types/module';
 import {prepareCmcArgs} from './cmc.install';
+import {makeAuthorizedSubnetworksProposal} from './cmc.post-install';
+
+const {green, cyan} = kleur;
 
 const CMC: ModuleDescription = {
   key: 'cmc',
@@ -19,6 +24,23 @@ class CmcModule extends Module {
       ...rest
     });
   }
+
+  override async postInstall(context: ModuleInstallParams): Promise<void> {
+    await makeAuthorizedSubnetworksProposal(context);
+
+    const {state} = context;
+
+    const metadata = state.getModule(this.key);
+
+    assertNonNullish(
+      metadata,
+      'Module has not been installed and therefore cannot be post-installed!'
+    );
+
+    const {name, canisterId} = metadata;
+
+    console.log(`🫠  ${green(name)} post-install. ID: ${cyan(canisterId.toString())}`);
+  }
 }
 
 export const cmc = new CmcModule(CMC);