figma-console-mcp 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (118) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +328 -0
  3. package/dist/browser/base.d.ts +50 -0
  4. package/dist/browser/base.d.ts.map +1 -0
  5. package/dist/browser/base.js +6 -0
  6. package/dist/browser/base.js.map +1 -0
  7. package/dist/browser/local.d.ts +66 -0
  8. package/dist/browser/local.d.ts.map +1 -0
  9. package/dist/browser/local.js +223 -0
  10. package/dist/browser/local.js.map +1 -0
  11. package/dist/cloudflare/browser/base.js +5 -0
  12. package/dist/cloudflare/browser/cloudflare.js +156 -0
  13. package/dist/cloudflare/browser-manager.js +157 -0
  14. package/dist/cloudflare/core/config.js +161 -0
  15. package/dist/cloudflare/core/console-monitor.js +382 -0
  16. package/dist/cloudflare/core/enrichment/enrichment-service.js +272 -0
  17. package/dist/cloudflare/core/enrichment/index.js +7 -0
  18. package/dist/cloudflare/core/enrichment/relationship-mapper.js +351 -0
  19. package/dist/cloudflare/core/enrichment/style-resolver.js +326 -0
  20. package/dist/cloudflare/core/figma-api.js +273 -0
  21. package/dist/cloudflare/core/figma-desktop-connector.js +383 -0
  22. package/dist/cloudflare/core/figma-style-extractor.js +311 -0
  23. package/dist/cloudflare/core/figma-tools.js +2299 -0
  24. package/dist/cloudflare/core/logger.js +53 -0
  25. package/dist/cloudflare/core/snippet-injector.js +96 -0
  26. package/dist/cloudflare/core/types/enriched.js +5 -0
  27. package/dist/cloudflare/core/types/index.js +4 -0
  28. package/dist/cloudflare/index.js +1059 -0
  29. package/dist/cloudflare/test-browser.js +88 -0
  30. package/dist/config.d.ts +17 -0
  31. package/dist/config.d.ts.map +1 -0
  32. package/dist/config.js +141 -0
  33. package/dist/config.js.map +1 -0
  34. package/dist/core/config.d.ts +17 -0
  35. package/dist/core/config.d.ts.map +1 -0
  36. package/dist/core/config.js +162 -0
  37. package/dist/core/config.js.map +1 -0
  38. package/dist/core/console-monitor.d.ts +81 -0
  39. package/dist/core/console-monitor.d.ts.map +1 -0
  40. package/dist/core/console-monitor.js +383 -0
  41. package/dist/core/console-monitor.js.map +1 -0
  42. package/dist/core/enrichment/enrichment-service.d.ts +52 -0
  43. package/dist/core/enrichment/enrichment-service.d.ts.map +1 -0
  44. package/dist/core/enrichment/enrichment-service.js +273 -0
  45. package/dist/core/enrichment/enrichment-service.js.map +1 -0
  46. package/dist/core/enrichment/index.d.ts +8 -0
  47. package/dist/core/enrichment/index.d.ts.map +1 -0
  48. package/dist/core/enrichment/index.js +8 -0
  49. package/dist/core/enrichment/index.js.map +1 -0
  50. package/dist/core/enrichment/relationship-mapper.d.ts +106 -0
  51. package/dist/core/enrichment/relationship-mapper.d.ts.map +1 -0
  52. package/dist/core/enrichment/relationship-mapper.js +352 -0
  53. package/dist/core/enrichment/relationship-mapper.js.map +1 -0
  54. package/dist/core/enrichment/style-resolver.d.ts +80 -0
  55. package/dist/core/enrichment/style-resolver.d.ts.map +1 -0
  56. package/dist/core/enrichment/style-resolver.js +327 -0
  57. package/dist/core/enrichment/style-resolver.js.map +1 -0
  58. package/dist/core/figma-api.d.ts +137 -0
  59. package/dist/core/figma-api.d.ts.map +1 -0
  60. package/dist/core/figma-api.js +274 -0
  61. package/dist/core/figma-api.js.map +1 -0
  62. package/dist/core/figma-desktop-connector.d.ts +52 -0
  63. package/dist/core/figma-desktop-connector.d.ts.map +1 -0
  64. package/dist/core/figma-desktop-connector.js +384 -0
  65. package/dist/core/figma-desktop-connector.js.map +1 -0
  66. package/dist/core/figma-style-extractor.d.ts +76 -0
  67. package/dist/core/figma-style-extractor.d.ts.map +1 -0
  68. package/dist/core/figma-style-extractor.js +312 -0
  69. package/dist/core/figma-style-extractor.js.map +1 -0
  70. package/dist/core/figma-tools.d.ts +15 -0
  71. package/dist/core/figma-tools.d.ts.map +1 -0
  72. package/dist/core/figma-tools.js +2300 -0
  73. package/dist/core/figma-tools.js.map +1 -0
  74. package/dist/core/logger.d.ts +22 -0
  75. package/dist/core/logger.d.ts.map +1 -0
  76. package/dist/core/logger.js +54 -0
  77. package/dist/core/logger.js.map +1 -0
  78. package/dist/core/snippet-injector.d.ts +24 -0
  79. package/dist/core/snippet-injector.d.ts.map +1 -0
  80. package/dist/core/snippet-injector.js +97 -0
  81. package/dist/core/snippet-injector.js.map +1 -0
  82. package/dist/core/types/enriched.d.ts +213 -0
  83. package/dist/core/types/enriched.d.ts.map +1 -0
  84. package/dist/core/types/enriched.js +6 -0
  85. package/dist/core/types/enriched.js.map +1 -0
  86. package/dist/core/types/index.d.ts +112 -0
  87. package/dist/core/types/index.d.ts.map +1 -0
  88. package/dist/core/types/index.js +5 -0
  89. package/dist/core/types/index.js.map +1 -0
  90. package/dist/index.d.ts +8 -0
  91. package/dist/index.d.ts.map +1 -0
  92. package/dist/index.js +72 -0
  93. package/dist/index.js.map +1 -0
  94. package/dist/local.d.ts +57 -0
  95. package/dist/local.d.ts.map +1 -0
  96. package/dist/local.js +668 -0
  97. package/dist/local.js.map +1 -0
  98. package/dist/logger.d.ts +22 -0
  99. package/dist/logger.d.ts.map +1 -0
  100. package/dist/logger.js +45 -0
  101. package/dist/logger.js.map +1 -0
  102. package/dist/server.d.ts +40 -0
  103. package/dist/server.d.ts.map +1 -0
  104. package/dist/server.js +99 -0
  105. package/dist/server.js.map +1 -0
  106. package/dist/tools/index.d.ts +15 -0
  107. package/dist/tools/index.d.ts.map +1 -0
  108. package/dist/tools/index.js +184 -0
  109. package/dist/tools/index.js.map +1 -0
  110. package/dist/types/index.d.ts +102 -0
  111. package/dist/types/index.d.ts.map +1 -0
  112. package/dist/types/index.js +6 -0
  113. package/dist/types/index.js.map +1 -0
  114. package/figma-desktop-bridge/README.md +232 -0
  115. package/figma-desktop-bridge/code.js +133 -0
  116. package/figma-desktop-bridge/manifest.json +13 -0
  117. package/figma-desktop-bridge/ui.html +200 -0
  118. package/package.json +77 -0
@@ -0,0 +1,383 @@
1
+ /**
2
+ * Figma Desktop Connector
3
+ *
4
+ * This service connects directly to Figma Desktop's plugin context
5
+ * to execute code with access to the full Figma Plugin API,
6
+ * including variables without Enterprise access.
7
+ *
8
+ * Uses Puppeteer's Worker API to directly access plugin workers,
9
+ * bypassing CDP context enumeration limitations.
10
+ */
11
+ import { logger } from './logger.js';
12
+ export class FigmaDesktopConnector {
13
+ constructor(page) {
14
+ this.page = page;
15
+ }
16
+ /**
17
+ * Initialize connection to Figma Desktop's plugin context
18
+ * No setup needed - Puppeteer handles worker access automatically
19
+ */
20
+ async initialize() {
21
+ logger.info('Figma Desktop connector initialized (using Puppeteer Worker API)');
22
+ }
23
+ /**
24
+ * Execute code in Figma's plugin context where the figma API is available
25
+ * Uses Puppeteer's direct worker access instead of CDP context enumeration
26
+ */
27
+ async executeInPluginContext(code) {
28
+ try {
29
+ // Use Puppeteer's worker API directly - this can access plugin workers
30
+ // that CDP's Runtime.getExecutionContexts cannot enumerate
31
+ const workers = this.page.workers();
32
+ // Log to browser console so MCP can capture it
33
+ await this.page.evaluate((count, urls) => {
34
+ console.log(`[DESKTOP_CONNECTOR] Found ${count} workers via Puppeteer API:`, urls);
35
+ }, workers.length, workers.map(w => w.url()));
36
+ logger.info({
37
+ workerCount: workers.length,
38
+ workerUrls: workers.map(w => w.url())
39
+ }, 'Found workers via Puppeteer API');
40
+ // Try each worker to find one with figma API
41
+ for (const worker of workers) {
42
+ try {
43
+ // Log to browser console
44
+ await this.page.evaluate((url) => {
45
+ console.log(`[DESKTOP_CONNECTOR] Checking worker: ${url}`);
46
+ }, worker.url());
47
+ // Check if this worker has the figma API
48
+ // Use string evaluation to avoid TypeScript errors about figma global
49
+ const hasFigmaApi = await worker.evaluate('typeof figma !== "undefined"');
50
+ // Log result to browser console
51
+ await this.page.evaluate((url, hasApi) => {
52
+ console.log(`[DESKTOP_CONNECTOR] Worker ${url} has figma API: ${hasApi}`);
53
+ }, worker.url(), hasFigmaApi);
54
+ if (hasFigmaApi) {
55
+ logger.info({ workerUrl: worker.url() }, 'Found worker with Figma API');
56
+ await this.page.evaluate((url) => {
57
+ console.log(`[DESKTOP_CONNECTOR] ✅ SUCCESS! Found worker with Figma API: ${url}`);
58
+ }, worker.url());
59
+ // Execute the code in this worker context
60
+ // Wrap the code in a function to ensure proper evaluation
61
+ const wrappedCode = `(${code})`;
62
+ const result = await worker.evaluate(wrappedCode);
63
+ return result;
64
+ }
65
+ }
66
+ catch (workerError) {
67
+ // This worker doesn't have figma API or evaluation failed, try next
68
+ await this.page.evaluate((url, err) => {
69
+ console.error(`[DESKTOP_CONNECTOR] ❌ Worker ${url} check failed:`, err);
70
+ }, worker.url(), workerError instanceof Error ? workerError.message : String(workerError));
71
+ logger.error({ error: workerError, workerUrl: worker.url() }, 'Worker check failed, trying next');
72
+ continue;
73
+ }
74
+ }
75
+ // If no worker found with figma API, throw error
76
+ throw new Error('No plugin worker found with Figma API. Make sure a plugin is running in Figma Desktop.');
77
+ }
78
+ catch (error) {
79
+ logger.error({ error, code: code.substring(0, 200) }, 'Failed to execute in plugin context');
80
+ throw error;
81
+ }
82
+ }
83
+ /**
84
+ * Get Figma variables from plugin UI window object
85
+ * This bypasses Figma's plugin sandbox security restrictions
86
+ * by accessing data that the plugin posted to its UI iframe
87
+ */
88
+ async getVariablesFromPluginUI(fileKey) {
89
+ try {
90
+ // Log to browser console
91
+ await this.page.evaluate((key) => {
92
+ console.log(`[DESKTOP_CONNECTOR] 🚀 getVariablesFromPluginUI() called, fileKey: ${key}`);
93
+ }, fileKey);
94
+ logger.info({ fileKey }, 'Getting variables from plugin UI iframe');
95
+ // Get all frames (iframes) in the page
96
+ const frames = this.page.frames();
97
+ await this.page.evaluate((count) => {
98
+ console.log(`[DESKTOP_CONNECTOR] Found ${count} frames (iframes)`);
99
+ }, frames.length);
100
+ logger.info({ frameCount: frames.length }, 'Found frames in page');
101
+ // Try to find plugin UI iframe with variables data
102
+ for (const frame of frames) {
103
+ try {
104
+ const frameUrl = frame.url();
105
+ await this.page.evaluate((url) => {
106
+ console.log(`[DESKTOP_CONNECTOR] Checking frame: ${url}`);
107
+ }, frameUrl);
108
+ // Check if this frame has our variables data
109
+ const hasData = await frame.evaluate('typeof window.__figmaVariablesData !== "undefined" && window.__figmaVariablesReady === true');
110
+ await this.page.evaluate((url, has) => {
111
+ console.log(`[DESKTOP_CONNECTOR] Frame ${url} has variables data: ${has}`);
112
+ }, frameUrl, hasData);
113
+ if (hasData) {
114
+ logger.info({ frameUrl }, 'Found frame with variables data');
115
+ await this.page.evaluate((url) => {
116
+ console.log(`[DESKTOP_CONNECTOR] ✅ SUCCESS! Found plugin UI with variables data: ${url}`);
117
+ }, frameUrl);
118
+ // Get the data from window object
119
+ const result = await frame.evaluate('window.__figmaVariablesData');
120
+ logger.info({
121
+ variableCount: result.variables?.length,
122
+ collectionCount: result.variableCollections?.length
123
+ }, 'Successfully retrieved variables from plugin UI');
124
+ await this.page.evaluate((varCount, collCount) => {
125
+ console.log(`[DESKTOP_CONNECTOR] ✅ Retrieved ${varCount} variables in ${collCount} collections`);
126
+ }, result.variables?.length || 0, result.variableCollections?.length || 0);
127
+ return result;
128
+ }
129
+ }
130
+ catch (frameError) {
131
+ await this.page.evaluate((url, err) => {
132
+ console.log(`[DESKTOP_CONNECTOR] Frame ${url} check failed: ${err}`);
133
+ }, frame.url(), frameError instanceof Error ? frameError.message : String(frameError));
134
+ logger.debug({ error: frameError, frameUrl: frame.url() }, 'Frame check failed, trying next');
135
+ continue;
136
+ }
137
+ }
138
+ // If no frame found with data, throw error
139
+ throw new Error('No plugin UI found with variables data. Make sure the Variables Exporter (Persistent) plugin is running.');
140
+ }
141
+ catch (error) {
142
+ logger.error({ error }, 'Failed to get variables from plugin UI');
143
+ await this.page.evaluate((msg) => {
144
+ console.error('[DESKTOP_CONNECTOR] ❌ getVariablesFromPluginUI failed:', msg);
145
+ }, error instanceof Error ? error.message : String(error));
146
+ throw error;
147
+ }
148
+ }
149
+ /**
150
+ * Get component data by node ID from plugin UI window object
151
+ * This bypasses the REST API bug where descriptions are missing
152
+ * by accessing data from the Desktop Bridge plugin via its UI iframe
153
+ */
154
+ async getComponentFromPluginUI(nodeId) {
155
+ try {
156
+ // Log to browser console
157
+ await this.page.evaluate((id) => {
158
+ console.log(`[DESKTOP_CONNECTOR] 🎯 getComponentFromPluginUI() called, nodeId: ${id}`);
159
+ }, nodeId);
160
+ logger.info({ nodeId }, 'Getting component from plugin UI iframe');
161
+ // Get all frames (iframes) in the page
162
+ const frames = this.page.frames();
163
+ await this.page.evaluate((count) => {
164
+ console.log(`[DESKTOP_CONNECTOR] Found ${count} frames (iframes)`);
165
+ }, frames.length);
166
+ logger.info({ frameCount: frames.length }, 'Found frames in page');
167
+ // Try to find plugin UI iframe with requestComponentData function
168
+ for (const frame of frames) {
169
+ try {
170
+ const frameUrl = frame.url();
171
+ await this.page.evaluate((url) => {
172
+ console.log(`[DESKTOP_CONNECTOR] Checking frame: ${url}`);
173
+ }, frameUrl);
174
+ // Check if this frame has our requestComponentData function
175
+ const hasFunction = await frame.evaluate('typeof window.requestComponentData === "function"');
176
+ await this.page.evaluate((url, has) => {
177
+ console.log(`[DESKTOP_CONNECTOR] Frame ${url} has requestComponentData: ${has}`);
178
+ }, frameUrl, hasFunction);
179
+ if (hasFunction) {
180
+ logger.info({ frameUrl }, 'Found frame with requestComponentData function');
181
+ await this.page.evaluate((url) => {
182
+ console.log(`[DESKTOP_CONNECTOR] ✅ SUCCESS! Found plugin UI with requestComponentData: ${url}`);
183
+ }, frameUrl);
184
+ // Call the function with the nodeId - it returns a Promise
185
+ // Use JSON.stringify to safely pass the nodeId as a string literal
186
+ const result = await frame.evaluate(`window.requestComponentData(${JSON.stringify(nodeId)})`);
187
+ logger.info({
188
+ nodeId,
189
+ componentName: result.component?.name,
190
+ hasDescription: !!result.component?.description
191
+ }, 'Successfully retrieved component from plugin UI');
192
+ await this.page.evaluate((name, hasDesc) => {
193
+ console.log(`[DESKTOP_CONNECTOR] ✅ Retrieved component "${name}", has description: ${hasDesc}`);
194
+ }, result.component?.name, !!result.component?.description);
195
+ return result;
196
+ }
197
+ }
198
+ catch (frameError) {
199
+ await this.page.evaluate((url, err) => {
200
+ console.log(`[DESKTOP_CONNECTOR] Frame ${url} check failed: ${err}`);
201
+ }, frame.url(), frameError instanceof Error ? frameError.message : String(frameError));
202
+ logger.debug({ error: frameError, frameUrl: frame.url() }, 'Frame check failed, trying next');
203
+ continue;
204
+ }
205
+ }
206
+ // If no frame found with function, throw error
207
+ throw new Error('No plugin UI found with requestComponentData function. Make sure the Desktop Bridge plugin is running.');
208
+ }
209
+ catch (error) {
210
+ logger.error({ error, nodeId }, 'Failed to get component from plugin UI');
211
+ await this.page.evaluate((msg) => {
212
+ console.error('[DESKTOP_CONNECTOR] ❌ getComponentFromPluginUI failed:', msg);
213
+ }, error instanceof Error ? error.message : String(error));
214
+ throw error;
215
+ }
216
+ }
217
+ /**
218
+ * Get Figma variables using the desktop connection
219
+ * This bypasses the Enterprise requirement!
220
+ */
221
+ async getVariables(fileKey) {
222
+ // Log to browser console
223
+ await this.page.evaluate((key) => {
224
+ console.log(`[DESKTOP_CONNECTOR] 🚀 getVariables() called, fileKey: ${key}`);
225
+ }, fileKey);
226
+ logger.info({ fileKey }, 'Getting variables via Desktop connection');
227
+ const code = `
228
+ (async () => {
229
+ try {
230
+ // Check if we're in the right context
231
+ if (typeof figma === 'undefined') {
232
+ throw new Error('Figma API not available in this context');
233
+ }
234
+
235
+ // Get variables just like the official MCP does
236
+ const variables = await figma.variables.getLocalVariablesAsync();
237
+ const collections = await figma.variables.getLocalVariableCollectionsAsync();
238
+
239
+ // Format the response
240
+ const result = {
241
+ success: true,
242
+ timestamp: Date.now(),
243
+ variables: variables.map(v => ({
244
+ id: v.id,
245
+ name: v.name,
246
+ key: v.key,
247
+ resolvedType: v.resolvedType,
248
+ valuesByMode: v.valuesByMode,
249
+ variableCollectionId: v.variableCollectionId,
250
+ scopes: v.scopes,
251
+ description: v.description,
252
+ hiddenFromPublishing: v.hiddenFromPublishing
253
+ })),
254
+ variableCollections: collections.map(c => ({
255
+ id: c.id,
256
+ name: c.name,
257
+ key: c.key,
258
+ modes: c.modes,
259
+ defaultModeId: c.defaultModeId,
260
+ variableIds: c.variableIds
261
+ }))
262
+ };
263
+
264
+ return result;
265
+ } catch (error) {
266
+ return {
267
+ success: false,
268
+ error: error.message
269
+ };
270
+ }
271
+ })()
272
+ `;
273
+ try {
274
+ const result = await this.executeInPluginContext(code);
275
+ if (!result.success) {
276
+ throw new Error(result.error || 'Failed to get variables');
277
+ }
278
+ logger.info({
279
+ variableCount: result.variables?.length,
280
+ collectionCount: result.variableCollections?.length
281
+ }, 'Successfully retrieved variables via Desktop');
282
+ return result;
283
+ }
284
+ catch (error) {
285
+ logger.error({ error }, 'Failed to get variables via Desktop');
286
+ throw error;
287
+ }
288
+ }
289
+ /**
290
+ * Clean up resources (no-op since we use Puppeteer's built-in worker management)
291
+ */
292
+ /**
293
+ * Get component data by node ID using Plugin API
294
+ * This bypasses the REST API bug where descriptions are missing
295
+ */
296
+ async getComponentByNodeId(nodeId) {
297
+ await this.page.evaluate((id) => {
298
+ console.log(`[DESKTOP_CONNECTOR] 🎯 getComponentByNodeId() called, nodeId: ${id}`);
299
+ }, nodeId);
300
+ logger.info({ nodeId }, 'Getting component via Desktop Plugin API');
301
+ const code = `
302
+ (async () => {
303
+ try {
304
+ // Check if we're in the right context
305
+ if (typeof figma === 'undefined') {
306
+ throw new Error('Figma API not available in this context');
307
+ }
308
+
309
+ // Get the node by ID
310
+ const node = figma.getNodeById('${nodeId}');
311
+
312
+ if (!node) {
313
+ throw new Error('Node not found with ID: ${nodeId}');
314
+ }
315
+
316
+ // Check if it's a component-like node
317
+ if (node.type !== 'COMPONENT' && node.type !== 'COMPONENT_SET' && node.type !== 'INSTANCE') {
318
+ throw new Error('Node is not a component, component set, or instance. Type: ' + node.type);
319
+ }
320
+
321
+ // Extract component data including description fields
322
+ const result = {
323
+ success: true,
324
+ timestamp: Date.now(),
325
+ component: {
326
+ id: node.id,
327
+ name: node.name,
328
+ type: node.type,
329
+ description: node.description || null,
330
+ descriptionMarkdown: node.descriptionMarkdown || null,
331
+ // Include other useful properties
332
+ visible: node.visible,
333
+ locked: node.locked,
334
+ // For components with properties
335
+ componentPropertyDefinitions: node.type === 'COMPONENT' || node.type === 'COMPONENT_SET'
336
+ ? node.componentPropertyDefinitions
337
+ : undefined,
338
+ // Get children info (lightweight)
339
+ children: node.children ? node.children.map(child => ({
340
+ id: child.id,
341
+ name: child.name,
342
+ type: child.type
343
+ })) : undefined
344
+ }
345
+ };
346
+
347
+ return result;
348
+ } catch (error) {
349
+ return {
350
+ success: false,
351
+ error: error.message,
352
+ stack: error.stack
353
+ };
354
+ }
355
+ })()
356
+ `;
357
+ try {
358
+ const result = await this.executeInPluginContext(code);
359
+ if (!result.success) {
360
+ throw new Error(result.error || 'Failed to get component data');
361
+ }
362
+ logger.info({
363
+ nodeId,
364
+ componentName: result.component?.name,
365
+ hasDescription: !!result.component?.description
366
+ }, 'Successfully retrieved component via Desktop Plugin API');
367
+ await this.page.evaluate((name, hasDesc) => {
368
+ console.log(`[DESKTOP_CONNECTOR] ✅ Retrieved component "${name}", has description: ${hasDesc}`);
369
+ }, result.component?.name, !!result.component?.description);
370
+ return result;
371
+ }
372
+ catch (error) {
373
+ logger.error({ error, nodeId }, 'Failed to get component via Desktop Plugin API');
374
+ await this.page.evaluate((id, err) => {
375
+ console.error(`[DESKTOP_CONNECTOR] ❌ getComponentByNodeId failed for ${id}:`, err);
376
+ }, nodeId, error instanceof Error ? error.message : String(error));
377
+ throw error;
378
+ }
379
+ }
380
+ async dispose() {
381
+ logger.info('Figma Desktop connector disposed');
382
+ }
383
+ }
@@ -0,0 +1,311 @@
1
+ /**
2
+ * Figma Style Extractor
3
+ *
4
+ * Extracts style information (colors, typography, spacing) from Figma files
5
+ * using the REST API /files endpoint. This provides an alternative to the
6
+ * Enterprise-only Variables API by parsing style data directly from nodes.
7
+ *
8
+ * Based on the approach used by Figma-Context-MCP
9
+ */
10
+ import { logger } from './logger';
11
+ export class FigmaStyleExtractor {
12
+ constructor() {
13
+ this.extractedVariables = new Map();
14
+ this.colorIndex = 0;
15
+ this.typographyIndex = 0;
16
+ this.spacingIndex = 0;
17
+ this.radiusIndex = 0;
18
+ }
19
+ /**
20
+ * Extract style "variables" from Figma file data
21
+ * This mimics what users would see as variables in Figma
22
+ */
23
+ async extractStylesFromFile(fileData) {
24
+ try {
25
+ logger.info('Extracting styles from Figma file data');
26
+ this.extractedVariables.clear();
27
+ this.colorIndex = 0;
28
+ this.typographyIndex = 0;
29
+ this.spacingIndex = 0;
30
+ this.radiusIndex = 0;
31
+ // Process the document tree
32
+ if (fileData.document) {
33
+ this.processNode(fileData.document);
34
+ }
35
+ // Also process components for more style data
36
+ if (fileData.components) {
37
+ Object.values(fileData.components).forEach((component) => {
38
+ if (component.node) {
39
+ this.processNode(component.node);
40
+ }
41
+ });
42
+ }
43
+ // Process styles if available
44
+ if (fileData.styles) {
45
+ this.processStyles(fileData.styles);
46
+ }
47
+ const variables = Array.from(this.extractedVariables.values());
48
+ logger.info({
49
+ colorCount: variables.filter(v => v.type === 'COLOR').length,
50
+ typographyCount: variables.filter(v => v.type === 'TYPOGRAPHY').length,
51
+ spacingCount: variables.filter(v => v.type === 'SPACING').length,
52
+ radiusCount: variables.filter(v => v.type === 'RADIUS').length,
53
+ totalCount: variables.length
54
+ }, 'Extracted style variables from file');
55
+ return variables;
56
+ }
57
+ catch (error) {
58
+ logger.error({ error }, 'Failed to extract styles from file');
59
+ throw error;
60
+ }
61
+ }
62
+ /**
63
+ * Process a single node and extract style information
64
+ */
65
+ processNode(node, depth = 0) {
66
+ if (!node || depth > 10)
67
+ return; // Limit depth to prevent infinite recursion
68
+ // Extract colors from fills
69
+ if (node.fills && Array.isArray(node.fills)) {
70
+ node.fills.forEach(fill => {
71
+ if (fill.type === 'SOLID' && fill.color && fill.visible !== false) {
72
+ this.extractColor(fill.color, fill.opacity, node);
73
+ }
74
+ });
75
+ }
76
+ // Extract colors from strokes
77
+ if (node.strokes && Array.isArray(node.strokes)) {
78
+ node.strokes.forEach(stroke => {
79
+ if (stroke.type === 'SOLID' && stroke.color && stroke.visible !== false) {
80
+ this.extractColor(stroke.color, stroke.opacity, node, 'stroke');
81
+ }
82
+ });
83
+ }
84
+ // Extract typography styles
85
+ if (node.type === 'TEXT' && node.style) {
86
+ this.extractTypography(node.style, node);
87
+ }
88
+ // Extract spacing from auto-layout
89
+ if (node.layoutMode) {
90
+ if (node.itemSpacing !== undefined && node.itemSpacing > 0) {
91
+ this.extractSpacing('spacing', node.itemSpacing, node);
92
+ }
93
+ if (node.paddingLeft !== undefined && node.paddingLeft > 0) {
94
+ this.extractSpacing('padding', node.paddingLeft, node);
95
+ }
96
+ }
97
+ // Extract corner radius
98
+ if (node.cornerRadius !== undefined && node.cornerRadius > 0) {
99
+ this.extractRadius(node.cornerRadius, node);
100
+ }
101
+ else if (node.rectangleCornerRadii && node.rectangleCornerRadii.length > 0) {
102
+ const uniqueRadii = [...new Set(node.rectangleCornerRadii)];
103
+ uniqueRadii.forEach(radius => {
104
+ if (radius > 0) {
105
+ this.extractRadius(radius, node);
106
+ }
107
+ });
108
+ }
109
+ // Process children recursively
110
+ if (node.children && Array.isArray(node.children)) {
111
+ node.children.forEach(child => {
112
+ this.processNode(child, depth + 1);
113
+ });
114
+ }
115
+ }
116
+ /**
117
+ * Extract color variable
118
+ */
119
+ extractColor(color, opacity = 1, node, type = 'fill') {
120
+ const r = Math.round(color.r * 255);
121
+ const g = Math.round(color.g * 255);
122
+ const b = Math.round(color.b * 255);
123
+ const a = opacity * (color.a || 1);
124
+ const hex = '#' + ((1 << 24) + (r << 16) + (g << 8) + b).toString(16).slice(1).toUpperCase();
125
+ const rgba = a < 1 ? `rgba(${r}, ${g}, ${b}, ${a})` : hex;
126
+ // Create a unique key based on the color value
127
+ const key = `color_${hex}_${a}`;
128
+ if (!this.extractedVariables.has(key)) {
129
+ // Generate a meaningful name based on the node
130
+ const category = this.inferColorCategory(node.name);
131
+ const name = this.generateColorName(category, type, this.colorIndex++);
132
+ this.extractedVariables.set(key, {
133
+ id: key,
134
+ name,
135
+ value: rgba,
136
+ type: 'COLOR',
137
+ category,
138
+ description: `Extracted from ${node.name || 'unnamed node'}`,
139
+ nodeId: node.id
140
+ });
141
+ }
142
+ }
143
+ /**
144
+ * Extract typography variable
145
+ */
146
+ extractTypography(style, node) {
147
+ const key = `typography_${style.fontFamily}_${style.fontSize}_${style.fontWeight}`;
148
+ if (!this.extractedVariables.has(key)) {
149
+ const name = this.generateTypographyName(node.name, this.typographyIndex++);
150
+ const value = [
151
+ `font-family: "${style.fontFamily || 'Inter'}"`,
152
+ style.fontSize ? `font-size: ${style.fontSize}px` : '',
153
+ style.fontWeight ? `font-weight: ${style.fontWeight}` : '',
154
+ style.lineHeightPx ? `line-height: ${style.lineHeightPx}px` : '',
155
+ style.letterSpacing ? `letter-spacing: ${style.letterSpacing}px` : ''
156
+ ].filter(Boolean).join(', ');
157
+ this.extractedVariables.set(key, {
158
+ id: key,
159
+ name,
160
+ value,
161
+ type: 'TYPOGRAPHY',
162
+ category: 'text',
163
+ description: `Extracted from ${node.name || 'unnamed text'}`,
164
+ nodeId: node.id
165
+ });
166
+ }
167
+ }
168
+ /**
169
+ * Extract spacing variable
170
+ */
171
+ extractSpacing(type, value, node) {
172
+ const key = `spacing_${type}_${value}`;
173
+ if (!this.extractedVariables.has(key)) {
174
+ const name = `${type}/${Math.round(value / 4) * 4 || value}`; // Round to nearest 4px
175
+ this.extractedVariables.set(key, {
176
+ id: key,
177
+ name,
178
+ value: `${value}px`,
179
+ type: 'SPACING',
180
+ category: type,
181
+ description: `Extracted from ${node.name || 'unnamed node'}`,
182
+ nodeId: node.id
183
+ });
184
+ }
185
+ }
186
+ /**
187
+ * Extract radius variable
188
+ */
189
+ extractRadius(value, node) {
190
+ const key = `radius_${value}`;
191
+ if (!this.extractedVariables.has(key)) {
192
+ const name = `radius/${this.categorizeRadius(value)}`;
193
+ this.extractedVariables.set(key, {
194
+ id: key,
195
+ name,
196
+ value: `${value}px`,
197
+ type: 'RADIUS',
198
+ category: 'border',
199
+ description: `Extracted from ${node.name || 'unnamed node'}`,
200
+ nodeId: node.id
201
+ });
202
+ }
203
+ }
204
+ /**
205
+ * Process Figma styles object
206
+ */
207
+ processStyles(styles) {
208
+ Object.entries(styles).forEach(([styleId, styleData]) => {
209
+ const { name, description, styleType } = styleData;
210
+ if (styleType === 'FILL' || styleType === 'TEXT') {
211
+ // These are named styles that could be considered variables
212
+ const variable = {
213
+ id: styleId,
214
+ name: name || styleId,
215
+ value: styleId, // We don't have the actual value here
216
+ type: styleType === 'FILL' ? 'COLOR' : 'TYPOGRAPHY',
217
+ description: description || `Style: ${name}`,
218
+ category: 'style'
219
+ };
220
+ this.extractedVariables.set(`style_${styleId}`, variable);
221
+ }
222
+ });
223
+ }
224
+ /**
225
+ * Helper to infer color category from node name
226
+ */
227
+ inferColorCategory(nodeName) {
228
+ if (!nodeName)
229
+ return 'color';
230
+ const name = nodeName.toLowerCase();
231
+ if (name.includes('background') || name.includes('bg'))
232
+ return 'background';
233
+ if (name.includes('text') || name.includes('label') || name.includes('title'))
234
+ return 'text';
235
+ if (name.includes('border') || name.includes('stroke'))
236
+ return 'border';
237
+ if (name.includes('primary') || name.includes('secondary') || name.includes('accent'))
238
+ return 'theme';
239
+ if (name.includes('success') || name.includes('error') || name.includes('warning'))
240
+ return 'semantic';
241
+ return 'color';
242
+ }
243
+ /**
244
+ * Generate a meaningful color name
245
+ */
246
+ generateColorName(category, type, index) {
247
+ const tier = index < 5 ? 'primary' : index < 10 ? 'secondary' : 'tertiary';
248
+ return `${category}/${tier}-${type}`;
249
+ }
250
+ /**
251
+ * Generate a meaningful typography name
252
+ */
253
+ generateTypographyName(nodeName, index) {
254
+ if (nodeName) {
255
+ const name = nodeName.toLowerCase();
256
+ if (name.includes('heading') || name.includes('h1') || name.includes('h2')) {
257
+ return `heading/${name.replace(/[^a-z0-9]/g, '-')}`;
258
+ }
259
+ if (name.includes('body') || name.includes('paragraph')) {
260
+ return `body/${name.replace(/[^a-z0-9]/g, '-')}`;
261
+ }
262
+ if (name.includes('caption') || name.includes('label')) {
263
+ return `caption/${name.replace(/[^a-z0-9]/g, '-')}`;
264
+ }
265
+ }
266
+ return `text/style-${index}`;
267
+ }
268
+ /**
269
+ * Categorize radius values
270
+ */
271
+ categorizeRadius(value) {
272
+ if (value === 0)
273
+ return 'none';
274
+ if (value <= 2)
275
+ return 'xs';
276
+ if (value <= 4)
277
+ return 'sm';
278
+ if (value <= 8)
279
+ return 'md';
280
+ if (value <= 16)
281
+ return 'lg';
282
+ if (value <= 24)
283
+ return 'xl';
284
+ return 'xxl';
285
+ }
286
+ /**
287
+ * Format the extracted variables for output
288
+ */
289
+ formatVariablesAsOutput(variables) {
290
+ // Group variables by type and category
291
+ const grouped = {};
292
+ variables.forEach(variable => {
293
+ const key = variable.name;
294
+ grouped[key] = variable.value;
295
+ });
296
+ // Add metadata about extraction method
297
+ grouped['_metadata'] = {
298
+ extractionMethod: 'REST_API_STYLES',
299
+ note: 'These are extracted style properties, not true Figma Variables (which require Enterprise)',
300
+ timestamp: new Date().toISOString(),
301
+ counts: {
302
+ colors: variables.filter(v => v.type === 'COLOR').length,
303
+ typography: variables.filter(v => v.type === 'TYPOGRAPHY').length,
304
+ spacing: variables.filter(v => v.type === 'SPACING').length,
305
+ radius: variables.filter(v => v.type === 'RADIUS').length,
306
+ total: variables.length
307
+ }
308
+ };
309
+ return grouped;
310
+ }
311
+ }