import { describe, it, expect } from 'vitest';
import {
  parseTokensFromScss,
  parseScssVariables,
  resolveTokenValues,
} from '../generate-tokens';

describe('parseScssVariables', () => {
  it('extracts SCSS variable declarations', () => {
    const scss = '$pri: #C4623A;\n$bg: #F5F0EB;';
    const vars = parseScssVariables(scss);
    expect(vars).toEqual({ pri: '#C4623A', bg: '#F5F0EB' });
  });

  it('ignores comments and non-variable lines', () => {
    const scss = '// comment\n$pri: #C4623A;';
    const vars = parseScssVariables(scss);
    expect(vars).toEqual({ pri: '#C4623A' });
  });

  it('handles values with spaces and units', () => {
    const scss = '$shadow-sm: 0 1px 4px rgba(45, 42, 38, 0.06);\n$r: 16px;';
    const vars = parseScssVariables(scss);
    expect(vars['shadow-sm']).toBe('0 1px 4px rgba(45, 42, 38, 0.06)');
    expect(vars['r']).toBe('16px');
  });
});

describe('parseTokensFromScss', () => {
  it('extracts --tk-* CSS variables from page selector', () => {
    const scss = `page {
  --tk-font-h1: 28px;
  --tk-font-body: 16px;
}`;
    const tokens = parseTokensFromScss(scss, 'page');
    expect(tokens).toEqual({ 'font-h1': '28px', 'font-body': '16px' });
  });

  it('extracts from class selector', () => {
    const scss = `.elder-mode {
  --tk-font-h1: 32px;
}`;
    const tokens = parseTokensFromScss(scss, '.elder-mode');
    expect(tokens).toEqual({ 'font-h1': '32px' });
  });

  it('captures SCSS variable references as-is for later resolution', () => {
    const scss = `page {
  --tk-pri: #{$pri};
  --tk-font-body: 16px;
}`;
    const tokens = parseTokensFromScss(scss, 'page');
    expect(tokens['pri']).toBe('#{$pri}');
    expect(tokens['font-body']).toBe('16px');
  });

  it('returns empty object when selector not found', () => {
    const scss = `page { --tk-font-h1: 28px; }`;
    const tokens = parseTokensFromScss(scss, '.nonexistent');
    expect(tokens).toEqual({});
  });
});

describe('resolveTokenValues', () => {
  it('replaces SCSS variable references with actual values', () => {
    const tokensScss = `page {
  --tk-pri: #{$pri};
}`;
    const varsScss = '$pri: #C4623A;';
    const result = resolveTokenValues(tokensScss, varsScss, 'page');
    expect(result).toEqual({ pri: '#C4623A' });
  });

  it('handles plain values without SCSS references', () => {
    const tokensScss = `page {
  --tk-card-radius: 16px;
  --tk-font-body: 16px;
}`;
    const varsScss = '$r: 16px;';
    const result = resolveTokenValues(tokensScss, varsScss, 'page');
    expect(result['card-radius']).toBe('16px');
    expect(result['font-body']).toBe('16px');
  });

  it('handles multiple variable references', () => {
    const tokensScss = `page {
  --tk-pri: #{$pri};
  --tk-pri-l: #{$pri-l};
  --tk-font-body: 16px;
}`;
    const varsScss = '$pri: #C4623A;\n$pri-l: #F0DDD4;';
    const result = resolveTokenValues(tokensScss, varsScss, 'page');
    expect(result['pri']).toBe('#C4623A');
    expect(result['pri-l']).toBe('#F0DDD4');
    expect(result['font-body']).toBe('16px');
  });
});