Merge pull request #33 from dev-five-git/optimize-export

Optimize export

Author: owjs3901

bun.lock

@@ -16,8 +16,8 @@
   "license": "",
   "devDependencies": {
     "@figma/plugin-typings": "^1.123",
-    "@rspack/cli": "^1.7.7",
-    "@rspack/core": "^1.7.7",
+    "@rspack/cli": "^1.7.8",
+    "@rspack/core": "^1.7.8",
 
     "husky": "^9.1",
     "typescript": "^5.9",

package.json

@@ -23,6 +23,7 @@ import {
   extractCustomComponentImports,
   extractImports,
 } from './commands/exportPagesAndComponents'
+
 export { extractCustomComponentImports, extractImports }
 
 import { getComponentName, resetTextStyleCache } from './utils'

src/code-impl.ts

@@ -55,6 +55,25 @@ export function resetGlobalBuildTreeCache(): void {
   globalBuildTreeCache.clear()
 }
 
+// Global asset node registry populated during buildTree().
+// Tracks nodes classified as SVG/PNG assets so callers (e.g. export commands)
+// can collect them without re-walking the Figma node tree via IPC.
+const globalAssetNodes = new Map<
+  string,
+  { node: SceneNode; type: 'svg' | 'png' }
+>()
+
+export function resetGlobalAssetNodes(): void {
+  globalAssetNodes.clear()
+}
+
+export function getGlobalAssetNodes(): ReadonlyMap<
+  string,
+  { node: SceneNode; type: 'svg' | 'png' }
+> {
+  return globalAssetNodes
+}
+
 /**
  * Get componentPropertyReferences from a node (if available).
  */
@@ -373,6 +392,11 @@ export class Codegen {
     // Handle asset nodes (images/SVGs)
     const assetNode = checkAssetNode(node)
     if (assetNode) {
+      // Register in global asset registry for export commands
+      const assetKey = `${assetNode}/${node.name}`
+      if (!globalAssetNodes.has(assetKey)) {
+        globalAssetNodes.set(assetKey, { node, type: assetNode })
+      }
       const props = await getProps(node)
       props.src = `/${assetNode === 'svg' ? 'icons' : 'images'}/${node.name}.${assetNode}`
       if (assetNode === 'svg') {

src/codegen/Codegen.ts

@@ -1,6 +1,11 @@
 import { afterAll, beforeEach, describe, expect, test } from 'bun:test'
 import { resetTextStyleCache } from '../../utils'
-import { Codegen, resetGlobalBuildTreeCache } from '../Codegen'
+import {
+  Codegen,
+  getGlobalAssetNodes,
+  resetGlobalAssetNodes,
+  resetGlobalBuildTreeCache,
+} from '../Codegen'
 import { resetGetPropsCache } from '../props'
 import { resetSelectorPropsCache } from '../props/selector'
 import { ResponsiveCodegen } from '../responsive/ResponsiveCodegen'
@@ -47,19 +52,36 @@ import { ResponsiveCodegen } from '../responsive/ResponsiveCodegen'
 
 beforeEach(() => {
   resetGlobalBuildTreeCache()
+  resetGlobalAssetNodes()
   resetGetPropsCache()
   resetSelectorPropsCache()
   resetTextStyleCache()
 })
 
 afterAll(() => {
   resetGlobalBuildTreeCache()
+  resetGlobalAssetNodes()
   resetGetPropsCache()
   resetSelectorPropsCache()
   resetTextStyleCache()
   ;(globalThis as { figma?: unknown }).figma = undefined
 })
 
+describe('globalAssetNodes', () => {
+  test('resetGlobalAssetNodes clears and getGlobalAssetNodes returns empty map', () => {
+    resetGlobalAssetNodes()
+    const assets = getGlobalAssetNodes()
+    expect(assets.size).toBe(0)
+  })
+
+  test('getGlobalAssetNodes returns readonly map', () => {
+    const assets = getGlobalAssetNodes()
+    expect(typeof assets.get).toBe('function')
+    expect(typeof assets.has).toBe('function')
+    expect(typeof assets.size).toBe('number')
+  })
+})
+
 function createComponentNode(
   name: string,
   variantProperties: Record<string, string>,