extract-instance-variant-props.test.ts

import { describe, expect, test } from 'bun:test'
import { extractInstanceVariantProps } from '../extract-instance-variant-props'

describe('extractInstanceVariantProps', () => {
  test('extracts VARIANT type props with sanitized keys', () => {
    // Figma componentProperties keys include node IDs like "property#nodeId:uniqueId"
    const node = {
      componentProperties: {
        'status#123:456': { type: 'VARIANT', value: 'scroll' },
        'size#789:012': { type: 'VARIANT', value: 'Md' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    // sanitizePropertyName strips "#nodeId:uniqueId" suffix
    expect(result).toEqual({
      status: 'scroll',
      size: 'Md',
    })
  })

  test('extracts BOOLEAN type props when value is true', () => {
    const node = {
      componentProperties: {
        'status#123:456': { type: 'VARIANT', value: 'active' },
        'label#789:012': { type: 'TEXT', value: 'Click me' },
        'visible#345:678': { type: 'BOOLEAN', value: true },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'active',
      visible: true,
    })
  })

  test('ignores BOOLEAN type props when value is false', () => {
    const node = {
      componentProperties: {
        'status#123:456': { type: 'VARIANT', value: 'active' },
        'hidden#345:678': { type: 'BOOLEAN', value: false },
        'label#789:012': { type: 'TEXT', value: 'Click me' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'active',
    })
  })

  test('returns empty object when componentProperties is undefined', () => {
    const node = {
      componentProperties: undefined,
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({})
  })

  test('returns empty object when componentProperties is null', () => {
    const node = {
      componentProperties: null,
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({})
  })

  test('handles Korean property names (속성 -> property)', () => {
    const node = {
      componentProperties: {
        '속성 1#789:012': { type: 'VARIANT', value: '값1' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    // sanitizePropertyName strips "#789:012" suffix, then converts "속성 1" to "property1"
    expect(result.property1).toBe('값1')
  })

  test('converts values to string', () => {
    const node = {
      componentProperties: {
        count: { type: 'VARIANT', value: 5 },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result.count).toBe('5')
    expect(typeof result.count).toBe('string')
  })

  test('handles simple property names without node IDs', () => {
    const node = {
      componentProperties: {
        status: { type: 'VARIANT', value: 'active' },
        variant: { type: 'VARIANT', value: 'primary' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'active',
      variant: 'primary',
    })
  })

  test('filters out reserved "effect" variant key', () => {
    const node = {
      componentProperties: {
        status: { type: 'VARIANT', value: 'active' },
        effect: { type: 'VARIANT', value: 'hover' },
        'Effect#123:456': { type: 'VARIANT', value: 'pressed' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'active',
    })
    expect(result.effect).toBeUndefined()
    expect(result.Effect).toBeUndefined()
  })

  test('filters out reserved "viewport" variant key', () => {
    const node = {
      componentProperties: {
        status: { type: 'VARIANT', value: 'active' },
        viewport: { type: 'VARIANT', value: 'desktop' },
        'Viewport#123:456': { type: 'VARIANT', value: 'mobile' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'active',
    })
    expect(result.viewport).toBeUndefined()
    expect(result.Viewport).toBeUndefined()
  })

  test('returns empty object when componentProperties getter throws', () => {
    const node = {
      get componentProperties(): never {
        throw new Error(
          'in get_componentProperties: Component set for node has existing errors',
        )
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({})
  })

  test('filters out both effect and viewport but keeps other variants', () => {
    const node = {
      componentProperties: {
        status: { type: 'VARIANT', value: 'scroll' },
        size: { type: 'VARIANT', value: 'Md' },
        effect: { type: 'VARIANT', value: 'default' },
        viewport: { type: 'VARIANT', value: 'desktop' },
      },
    } as unknown as InstanceNode

    const result = extractInstanceVariantProps(node)

    expect(result).toEqual({
      status: 'scroll',
      size: 'Md',
    })
  })
})