2427-renaming-tokens-in-the-other-type-breaks-references (#2440)

https://github.com/tokens-studio/figma-plugin/assets/103296157/f897fb11-9e83-4bab-a376-fd1e56f75bcf
tokens-studio · Jan 8, 2024 · 1424c9f · 1424c9f
1 parent c306ddf
commit 1424c9f
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/.changeset/moody-carrots-agree.md b/.changeset/moody-carrots-agree.md
@@ -0,0 +1,5 @@
+---
+'@tokens-studio/figma-plugin': patch
+---
+
+Fix the renaming tokens in the other type breaks references
diff --git a/src/app/store/models/tokenState.tsx b/src/app/store/models/tokenState.tsx
@@ -42,6 +42,7 @@ import { CompareStateType, findDifferentState } from '@/utils/findDifferentState
 import { RenameTokensAcrossSetsPayload } from '@/types/payloads/RenameTokensAcrossSets';
 import { wrapTransaction } from '@/profiling/transaction';
 import addIdPropertyToTokens from '@/utils/addIdPropertyToTokens';
+import { ColorModifier } from '@/types/Modifier';
 
 export interface TokenState {
   tokens: Record<string, AnyTokenList>;
@@ -502,6 +503,29 @@ export const tokenState = createModel<RootModel>()({
               } as SingleToken;
             }
 
+            if (token.$extensions?.['studio.tokens'] && token.$extensions?.['studio.tokens']?.modify && token.$extensions?.['studio.tokens'].modify.value) {
+              const updatedModify = Object.entries(token.$extensions?.['studio.tokens'].modify).reduce<ColorModifier>((modify, [key, value]: string[]) => {
+                modify = {
+                  ...modify,
+                  [key]: value.replace(data.oldName, data.newName),
+                };
+                return modify;
+              }, {} as ColorModifier);
+
+              return {
+                ...token,
+                $extensions: {
+                  ...token.$extensions,
+                  'studio.tokens': {
+                    ...token.$extensions['studio.tokens'],
+                    modify: {
+                      ...updatedModify,
+                    },
+                  },
+                },
+              } as SingleToken;
+            }
+
             return {
               ...token,
               value: replaceReferences(token.value.toString(), data.oldName, data.newName),