import { describe, it, expect } from 'vitest';
import {
  exportToJson,
  exportToCsv,
  importFromJson,
  importFromCsv,
  importFromTxt,
  importFromContent,
  detectExportFormat,
  flattenImportResult,
} from '../../src/core/import-export';
import type { PathEntry } from '../../src/core/path-entry';

function pe(s: string, enabled: boolean = true): PathEntry {
  return { path: s, enabled };
}

const sampleData = {
  system: [pe('C:\\Windows'), pe('C:\\Program Files')],
  user: [pe('C:\\Users\\me\\AppData')],
};

describe('exportToJson', () => {
  it('导出结构化 JSON', () => {
    const json = exportToJson(sampleData);
    const parsed = JSON.parse(json);
    expect(parsed.version).toBe('5.1.0');
    expect(parsed.timestamp).toBeDefined();
    expect(parsed.system.map((e: { path: string }) => e.path)).toEqual(sampleData.system.map(e => e.path));
    expect(parsed.user.map((e: { path: string }) => e.path)).toEqual(sampleData.user.map(e => e.path));
    expect(parsed.system[0].enabled).toBe(true);
    expect(parsed.user[0].enabled).toBe(true);
  });
});

describe('importFromJson', () => {
  it('正确导入 JSON', () => {
    const json = JSON.stringify({
      system: sampleData.system.map(e => e.path),
      user: sampleData.user.map(e => e.path),
    });
    const result = importFromJson(json);
    expect(result.system).toEqual(sampleData.system);
    expect(result.user).toEqual(sampleData.user);
  });

  it('过滤空字符串', () => {
    const json = JSON.stringify({ system: ['C:\\', '', '  '], user: [] });
    const result = importFromJson(json);
    expect(result.system).toEqual([pe('C:\\')]);
  });
});

describe('exportToCsv', () => {
  it('导出 CSV 含 BOM', () => {
    const csv = exportToCsv(sampleData);
    expect(csv.startsWith('')).toBe(true);
    expect(csv).toContain('type,path,enabled');
    expect(csv).toContain('system,C:\\Windows,true');
    expect(csv).toContain('user,C:\\Users\\me\\AppData,true');
  });

  it('CSV 字段转义', () => {
    const data = { system: [pe('C:\\Path,with,commas')], user: [] };
    const csv = exportToCsv(data);
    expect(csv).toContain('"C:\\Path,with,commas",true');
  });

  it('CSV 双引号转义', () => {
    const data = { system: [pe('Path with "quotes"')], user: [] };
    const csv = exportToCsv(data);
    expect(csv).toContain('"Path with ""quotes""",true');
  });
});

describe('importFromCsv', () => {
  it('正确导入 CSV', () => {
    const csv = 'type,path\nsystem,C:\\Windows\nuser,C:\\AppData\n';
    const result = importFromCsv(csv);
    expect(result.system).toEqual([pe('C:\\Windows')]);
    expect(result.user).toEqual([pe('C:\\AppData')]);
  });

  it('跳过未知类型', () => {
    const csv = 'type,path\nother,C:\\Unknown\nsystem,C:\\Valid';
    const result = importFromCsv(csv);
    expect(result.system.length).toBe(1);
    expect(result.user.length).toBe(0);
  });

  it('处理带引号的 CSV 字段', () => {
    const csv = 'type,path\nsystem,"C:\\Path,With,Commas"';
    const result = importFromCsv(csv);
    expect(result.system).toEqual([pe('C:\\Path,With,Commas')]);
  });
});

describe('importFromTxt', () => {
  it('逐行导入，跳过注释和空行', () => {
    const txt = '# 这是注释\nC:\\Windows\n\nD:\\Projects\n# 另一个注释';
    const paths = importFromTxt(txt);
    expect(paths).toEqual([pe('C:\\Windows'), pe('D:\\Projects')]);
  });

  it('跳过 BOM', () => {
    const txt = 'C:\\Windows';
    const paths = importFromTxt(txt);
    expect(paths).toEqual([pe('C:\\Windows')]);
  });
});

describe('importFromContent', () => {
  it('根据扩展名选择格式', () => {
    const csvContent = 'type,path\nsystem,C:\\Test';
    const jsonContent = JSON.stringify({ system: ['C:\\Test'], user: [] });
    const txtContent = 'C:\\Test';

    expect(importFromContent(csvContent, 'test.csv').system).toEqual([pe('C:\\Test')]);
    expect(importFromContent(jsonContent, 'test.json').system).toEqual([pe('C:\\Test')]);
    expect(importFromContent(txtContent, 'test.txt').system).toEqual([pe('C:\\Test')]);
  });
});

describe('detectExportFormat', () => {
  it('.csv 检测为 CSV', () => {
    expect(detectExportFormat('data.CSV')).toBe('csv');
  });
  it('.txt 检测为 TXT', () => {
    expect(detectExportFormat('data.txt')).toBe('txt');
  });
  it('其他扩展名检测为 JSON', () => {
    expect(detectExportFormat('data.json')).toBe('json');
  });
});

describe('flattenImportResult', () => {
  const data = { system: [pe('S1')], user: [pe('U1')] };

  it('仅系统', () => {
    const r = flattenImportResult(data, 'system');
    expect(r.system).toEqual([pe('S1')]);
    expect(r.user).toEqual([]);
  });

  it('仅用户', () => {
    const r = flattenImportResult(data, 'user');
    expect(r.system).toEqual([]);
    expect(r.user).toEqual([pe('U1')]);
  });

  it('两者都导入', () => {
    const r = flattenImportResult(data, 'both');
    expect(r.system).toEqual([pe('S1')]);
    expect(r.user).toEqual([pe('U1')]);
  });
});