Add missing code

Author: owjs3901

src/__tests__/code.test.ts

@@ -1498,6 +1498,18 @@ describe('registerCodegen with usage output', () => {
     expect(usageCode).toContain('<PrimaryButton')
     expect(usageCode).toContain('variant="primary"')
     expect(usageCode).toContain('size="lg"')
+
+    // Should also include the component definition
+    const definitionResult = result.find(
+      (r: unknown) =>
+        typeof r === 'object' &&
+        r !== null &&
+        'title' in r &&
+        (r as { title: string }).title === 'PrimaryButton',
+    )
+    expect(definitionResult).toBeDefined()
+    const definitionCode = (definitionResult as { code: string }).code
+    expect(definitionCode).toContain('export function PrimaryButton')
   })
 
   it('should generate usage for positioned INSTANCE node (absolute)', async () => {

src/code-impl.ts

@@ -422,13 +422,49 @@ export function registerCodegen(ctx: typeof figma) {
             ...responsiveComponentsCodes,
           ]
 
+          // For INSTANCE nodes, include the referenced component definition(s)
+          // alongside Usage so developers see both how to use AND the implementation.
+          const componentDefinitionResults: {
+            title: string
+            language: 'TYPESCRIPT' | 'BASH'
+            code: string
+          }[] = []
+          if (node.type === 'INSTANCE' && componentsCodes.length > 0) {
+            const importStatement = generateImportStatements(componentsCodes)
+            const combinedCode = componentsCodes
+              .map(([, code]) => code)
+              .join('\n\n')
+            const label =
+              componentsCodes.length === 1
+                ? componentsCodes[0][0]
+                : `${node.name} - Components`
+            componentDefinitionResults.push(
+              {
+                title: label,
+                language: 'TYPESCRIPT',
+                code: importStatement + combinedCode,
+              },
+              {
+                title: `${label} - CLI (Bash)`,
+                language: 'BASH',
+                code: generateBashCLI(componentsCodes),
+              },
+              {
+                title: `${label} - CLI (PowerShell)`,
+                language: 'BASH',
+                code: generatePowerShellCLI(componentsCodes),
+              },
+            )
+          }
+
           // For COMPONENT nodes, show both the single-variant code AND Usage.
           // For COMPONENT_SET and INSTANCE, show only Usage.
           // For all other types, show the main code.
           const showMainCode = !isComponentType || node.type === 'COMPONENT'
 
           return [
             ...usageResults,
+            ...componentDefinitionResults,
             ...(showMainCode
               ? [
                   {