fixup! feat: added regex support to bulk remap

tokens-studio · Feb 23, 2024 · 8467f02 · 8467f02
1 parent 53eaa5f
commit 8467f02
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/plugin/asyncMessageHandlers/__tests__/bulkRemapTokens.test.ts b/src/plugin/asyncMessageHandlers/__tests__/bulkRemapTokens.test.ts
@@ -95,7 +95,7 @@ describe('bulkRemapTokens', () => {
     ]));
     await bulkRemapTokens({
       type: AsyncMessageTypes.BULK_REMAP_TOKENS,
-      oldName: 'old-size.*',
+      oldName: '/old-size.*/g',
       newName: 'new-size.1000',
       updateMode: UpdateMode.SELECTION,
     });

diff --git a/src/plugin/asyncMessageHandlers/bulkRemapTokens.ts b/src/plugin/asyncMessageHandlers/bulkRemapTokens.ts
@@ -27,11 +27,14 @@ export const bulkRemapTokens: AsyncMessageChannelHandlers[AsyncMessageTypes.BULK
 
     const tracker = new ProgressTracker(BackgroundJobs.PLUGIN_UPDATEPLUGINDATA);
     const promises: Set<Promise<void>> = new Set();
+    const regexPattern = /^\/(.*)\/([gimsuy]*)$/;
 
     allWithData.forEach(({ node, tokens }) => {
       promises.add(defaultWorker.schedule(async () => {
         Object.entries(tokens).forEach(([key, value]) => {
-          const pattern = (/^\/.*\/$/.test(oldName)) ? new RegExp(oldName.slice(1, -1)) : new RegExp(oldName);
+          const regexTest = oldName.match(regexPattern);
+          // If the pattern passed is a regex, use it, otherwise use the old name with a global flag
+          const pattern = regexTest ? new RegExp(regexTest[1], regexTest[2]) : new RegExp(oldName, 'g');
           if (pattern.test(value)) {
             const newValue = value.replace(pattern, newName);
             const jsonValue = JSON.stringify(newValue);