Update programLinkNode name when updating programNode name

metaplex-foundation · Jan 1, 2024 · 4a1aaad · 4a1aaad
1 parent e7c378f
commit 4a1aaad
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 15 deletions.
diff --git a/src/renderers/js-experimental/fragments/accountPdaHelpers.ts b/src/renderers/js-experimental/fragments/accountPdaHelpers.ts
@@ -1,5 +1,5 @@
-import { LinkableDictionary } from 'src/shared';
 import { AccountNode, ProgramNode, isNodeFilter } from '../../../nodes';
+import { LinkableDictionary } from '../../../shared';
 import { NameApi } from '../nameTransformers';
 import { Fragment, fragment, fragmentFromTemplate } from './common';
 

diff --git a/src/visitors/getByteSizeVisitor.ts b/src/visitors/getByteSizeVisitor.ts
@@ -1,10 +1,10 @@
-import { LinkableDictionary } from 'src/shared';
 import {
   REGISTERED_TYPE_NODE_KEYS,
   RegisteredTypeNodes,
   isNode,
   isScalarEnum,
 } from '../nodes';
+import { LinkableDictionary } from '../shared';
 import { mergeVisitor } from './mergeVisitor';
 import { Visitor, visit } from './visitor';
 

diff --git a/src/visitors/updateDefinedTypesVisitor.ts b/src/visitors/updateDefinedTypesVisitor.ts
@@ -30,7 +30,7 @@ export function updateDefinedTypesVisitor(
             ? mainCase(updates.name)
             : undefined;
 
-        const transforms: BottomUpNodeTransformerWithSelector[] = [
+        const transformers: BottomUpNodeTransformerWithSelector[] = [
           {
             select: `${selectorStack.join('.')}.[definedTypeNode]${name}`,
             transform: (node) => {
@@ -56,7 +56,7 @@ export function updateDefinedTypesVisitor(
         ];
 
         if (newName) {
-          transforms.push({
+          transformers.push({
             select: `${selectorStack.join('.')}.[definedTypeLinkNode]${name}`,
             transform: (node) => {
               assertIsNode(node, 'definedTypeLinkNode');
@@ -66,7 +66,7 @@ export function updateDefinedTypesVisitor(
           });
         }
 
-        return transforms;
+        return transformers;
       }
     )
   );

diff --git a/src/visitors/updateProgramsVisitor.ts b/src/visitors/updateProgramsVisitor.ts
@@ -1,4 +1,10 @@
-import { ProgramNodeInput, assertIsNode, programNode } from '../nodes';
+import {
+  ProgramNodeInput,
+  assertIsNode,
+  programLinkNode,
+  programNode,
+} from '../nodes';
+import { mainCase } from '../shared';
 import {
   BottomUpNodeTransformerWithSelector,
   bottomUpTransformerVisitor,
@@ -15,15 +21,37 @@ export type ProgramUpdates =
 
 export function updateProgramsVisitor(map: Record<string, ProgramUpdates>) {
   return bottomUpTransformerVisitor(
-    Object.entries(map).map(
-      ([name, updates]): BottomUpNodeTransformerWithSelector => ({
-        select: `[programNode]${name}`,
-        transform: (node) => {
-          assertIsNode(node, 'programNode');
-          if ('delete' in updates) return null;
-          return programNode({ ...node, ...updates });
-        },
-      })
+    Object.entries(map).flatMap(
+      ([name, updates]): BottomUpNodeTransformerWithSelector[] => {
+        const newName =
+          typeof updates === 'object' && 'name' in updates && updates.name
+            ? mainCase(updates.name)
+            : undefined;
+
+        const transformers: BottomUpNodeTransformerWithSelector[] = [
+          {
+            select: `[programNode]${name}`,
+            transform: (node) => {
+              assertIsNode(node, 'programNode');
+              if ('delete' in updates) return null;
+              return programNode({ ...node, ...updates });
+            },
+          },
+        ];
+
+        if (newName) {
+          transformers.push({
+            select: `[programLinkNode]${name}`,
+            transform: (node) => {
+              assertIsNode(node, 'programLinkNode');
+              if (node.importFrom) return node;
+              return programLinkNode(newName);
+            },
+          });
+        }
+
+        return transformers;
+      }
     )
   );
 }