Support custom dfx canister principals (#284)

* Support 'type: custom' dfx canisters * Add virtual Candid files in LSP directory * 0.16.4
dfinity · Jun 11, 2024 · 2192518 · 2192518
1 parent d03e364
commit 2192518
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -2,7 +2,7 @@
     "name": "vscode-motoko",
     "displayName": "Motoko",
     "description": "Motoko language support",
-    "version": "0.16.3",
+    "version": "0.16.4",
     "publisher": "dfinity-foundation",
     "repository": "https://github.com/dfinity/vscode-motoko",
     "engines": {

diff --git a/src/server/dfx.ts b/src/server/dfx.ts
@@ -4,6 +4,12 @@ import { dirname } from 'path';
 interface DfxCanister {
     type?: string;
     main?: string;
+    remote?: {
+        candid?: string;
+        id?: {
+            local?: string;
+        };
+    };
 }
 
 interface DfxConfig {

diff --git a/src/server/server.ts b/src/server/server.ts
@@ -438,7 +438,34 @@ function notifyDfxChange() {
                                 },
                             );
                         }
+                        Object.entries(dfxConfig.canisters).forEach(
+                            ([name, canister]) => {
+                                if (!aliases.hasOwnProperty(name)) {
+                                    const id = canister.remote?.id?.local;
+                                    if (id) {
+                                        aliases[name] = id;
+                                        const candidPath =
+                                            canister.remote?.candid;
+                                        if (candidPath) {
+                                            // Add Candid as virtual file in LSP directory
+                                            const candid = readFileSync(
+                                                resolve(projectDir, candidPath),
+                                                'utf8',
+                                            );
+                                            writeVirtual(
+                                                resolveVirtualPath(
+                                                    candidUri,
+                                                    `${id}.did`,
+                                                ),
+                                                candid,
+                                            );
+                                        }
+                                    }
+                                }
+                            },
+                        );
                         allContexts().forEach(({ motoko }) => {
+                            console.log('Actor aliases:', aliases);
                             motoko.setAliases(
                                 resolveVirtualPath(candidUri),
                                 aliases,