/**
 * 个人信息生成器
 * 包含：中文姓名、邮箱、手机号、身份证号、中文地址、年龄
 */

import type { GeneratorDefinition } from '@/types/testDataGenerator';
import { randomInt, randomPick } from './random';

// 中文姓氏
const SURNAMES = [
  '王',
  '李',
  '张',
  '刘',
  '陈',
  '杨',
  '赵',
  '黄',
  '周',
  '吴',
  '徐',
  '孙',
  '胡',
  '朱',
  '高',
  '林',
  '何',
  '郭',
  '马',
  '罗',
  '梁',
  '宋',
  '郑',
  '谢',
  '韩',
  '唐',
  '冯',
  '于',
  '董',
  '萧',
  '程',
  '曹',
  '袁',
  '邓',
  '许',
  '傅',
  '沈',
  '曾',
  '彭',
  '吕',
  '苏',
  '卢',
  '蒋',
  '蔡',
  '贾',
  '丁',
  '魏',
  '薛',
  '叶',
  '阎',
];

// 中文名字常用字
const NAME_CHARS = [
  '伟',
  '芳',
  '娜',
  '秀英',
  '敏',
  '静',
  '丽',
  '强',
  '磊',
  '洋',
  '艳',
  '勇',
  '军',
  '杰',
  '娟',
  '涛',
  '明',
  '超',
  '秀兰',
  '霞',
  '平',
  '刚',
  '桂英',
  '文',
  '华',
  '飞',
  '鑫',
  '浩',
  '凯',
  '宁',
  '建',
  '峰',
  '辉',
  '成',
  '宇',
  '博',
  '泽',
  '思',
  '睿',
  '晨',
  '阳',
  '雪',
  '冰',
  '琳',
  '瑶',
  '婷',
  '欣',
  '悦',
  '佳',
  '慧',
];

// 邮箱域名
const EMAIL_DOMAINS = ['qq.com', '163.com', '126.com', 'gmail.com', 'outlook.com', 'hotmail.com'];

// 省份
const PROVINCES = [
  '北京市',
  '天津市',
  '上海市',
  '重庆市',
  '河北省',
  '山西省',
  '辽宁省',
  '吉林省',
  '黑龙江省',
  '江苏省',
  '浙江省',
  '安徽省',
  '福建省',
  '江西省',
  '山东省',
  '河南省',
  '湖北省',
  '湖南省',
  '广东省',
  '海南省',
  '四川省',
  '贵州省',
  '云南省',
  '陕西省',
  '甘肃省',
  '青海省',
  '台湾省',
  '内蒙古自治区',
  '广西壮族自治区',
  '西藏自治区',
  '宁夏回族自治区',
  '新疆维吾尔自治区',
];

// 城市
const CITIES: Record<string, string[]> = {
  北京市: ['东城区', '西城区', '朝阳区', '海淀区'],
  上海市: ['黄浦区', '徐汇区', '长宁区', '静安区'],
  广东省: ['广州市', '深圳市', '东莞市', '佛山市'],
  浙江省: ['杭州市', '宁波市', '温州市', '嘉兴市'],
  江苏省: ['南京市', '苏州市', '无锡市', '常州市'],
  四川省: ['成都市', '绵阳市', '德阳市', '宜宾市'],
};

/**
 * 中文姓名生成器
 */
export const chineseName: GeneratorDefinition = {
  id: 'chineseName',
  name: '中文姓名',
  description: '生成随机中文姓名',
  categoryId: 'personal',
  params: [],
  generate: () => {
    const surname = randomPick(SURNAMES);
    const name = randomPick(NAME_CHARS);
    return surname + name;
  },
};

/**
 * 邮箱生成器
 */
export const email: GeneratorDefinition = {
  id: 'email',
  name: '邮箱',
  description: '生成随机邮箱地址',
  categoryId: 'personal',
  params: [
    {
      key: 'domain',
      label: '邮箱域名',
      type: 'select',
      defaultValue: 'random',
      description: '选择邮箱域名',
      options: [
        { label: '随机', value: 'random' },
        ...EMAIL_DOMAINS.map((d) => ({ label: d, value: d })),
      ],
    },
  ],
  generate: (params) => {
    const domain =
      params.domain === 'random' ? randomPick(EMAIL_DOMAINS) : (params.domain as string);
    const username = Math.random().toString(36).substring(2, 10);
    return `${username}@${domain}`;
  },
};

/**
 * 手机号生成器
 */
export const chinesePhone: GeneratorDefinition = {
  id: 'chinesePhone',
  name: '手机号',
  description: '生成随机中国手机号码',
  categoryId: 'personal',
  params: [
    {
      key: 'prefix',
      label: '号段前缀',
      type: 'select',
      defaultValue: 'random',
      description: '选择手机号前缀',
      options: [
        { label: '随机', value: 'random' },
        { label: '138', value: '138' },
        { label: '139', value: '139' },
        { label: '158', value: '158' },
        { label: '188', value: '188' },
        { label: '177', value: '177' },
      ],
    },
  ],
  generate: (params) => {
    const prefixes = ['138', '139', '158', '188', '177', '136', '159', '186', '135', '150'];
    const prefix = params.prefix === 'random' ? randomPick(prefixes) : (params.prefix as string);
    const suffix = String(randomInt(10000000, 99999999));
    return prefix + suffix;
  },
};

/**
 * 身份证号生成器
 */
export const idCard: GeneratorDefinition = {
  id: 'idCard',
  name: '身份证号',
  description: '生成随机身份证号码',
  categoryId: 'personal',
  params: [
    {
      key: 'region',
      label: '地区',
      type: 'select',
      defaultValue: 'random',
      description: '选择身份证前6位地区码',
      options: [
        { label: '随机', value: 'random' },
        { label: '北京', value: '110101' },
        { label: '上海', value: '310101' },
        { label: '广州', value: '440103' },
        { label: '深圳', value: '440305' },
      ],
    },
  ],
  generate: (params) => {
    const regions = ['110101', '310101', '440103', '440305', '510104', '330102'];
    const region = params.region === 'random' ? randomPick(regions) : (params.region as string);

    // 出生日期 1960-2005年
    const year = randomInt(1960, 2005);
    const month = String(randomInt(1, 12)).padStart(2, '0');
    const day = String(randomInt(1, 28)).padStart(2, '0');
    const birthday = `${year}${month}${day}`;

    // 顺序码
    const sequence = String(randomInt(1, 999)).padStart(3, '0');

    // 前17位
    const prefix17 = region + birthday + sequence;

    // 校验码
    const weights = [7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2];
    const checkCodes = ['1', '0', 'X', '9', '8', '7', '6', '5', '4', '3', '2'];
    let sum = 0;
    for (let i = 0; i < 17; i++) {
      sum += parseInt(prefix17[i]) * weights[i];
    }
    const checkCode = checkCodes[sum % 11];

    return prefix17 + checkCode;
  },
  generateAtIndex: (params, index) => {
    const regions = ['110101', '310101', '440103', '440305', '510104', '330102'];
    const region =
      params.region === 'random' ? regions[index % regions.length] : (params.region as string);

    const year = 1990;
    const month = String(randomInt(1, 12)).padStart(2, '0');
    const day = String(randomInt(1, 28)).padStart(2, '0');
    const birthday = `${year}${month}${day}`;

    const sequence = String(index + 1).padStart(3, '0');
    const prefix17 = region + birthday + sequence;

    const weights = [7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2];
    const checkCodes = ['1', '0', 'X', '9', '8', '7', '6', '5', '4', '3', '2'];
    let sum = 0;
    for (let i = 0; i < 17; i++) {
      sum += parseInt(prefix17[i]) * weights[i];
    }
    const checkCode = checkCodes[sum % 11];

    return prefix17 + checkCode;
  },
};

/**
 * 中文地址生成器
 */
export const chineseAddress: GeneratorDefinition = {
  id: 'chineseAddress',
  name: '中文地址',
  description: '生成随机中文地址',
  categoryId: 'personal',
  params: [
    {
      key: 'includeDetail',
      label: '包含详细地址',
      type: 'boolean',
      defaultValue: true,
      description: '是否包含街道门牌号',
    },
  ],
  generate: (params) => {
    const province = randomPick(PROVINCES);
    const cities = CITIES[province] || ['市区'];
    const city = randomPick(cities);
    const district = `区${randomInt(1, 20)}号`;

    if (params.includeDetail) {
      const street = `路${randomInt(1, 200)}号`;
      return `${province}${city}${district}${street}`;
    }
    return `${province}${city}${district}`;
  },
};

/**
 * 年龄生成器
 */
export const age: GeneratorDefinition = {
  id: 'age',
  name: '年龄',
  description: '生成随机年龄',
  categoryId: 'personal',
  params: [
    {
      key: 'min',
      label: '最小年龄',
      type: 'number',
      defaultValue: 18,
      min: 0,
      max: 150,
      description: '年龄最小值',
    },
    {
      key: 'max',
      label: '最大年龄',
      type: 'number',
      defaultValue: 65,
      min: 0,
      max: 150,
      description: '年龄最大值',
    },
    {
      key: 'distribution',
      label: '分布方式',
      type: 'select',
      defaultValue: 'uniform',
      description: '年龄分布方式',
      options: [
        { label: '均匀分布', value: 'uniform' },
        { label: '正态分布', value: 'normal' },
        { label: '人口统计分布', value: 'demographic' },
      ],
    },
  ],
  generate: (params) => {
    const min = (params.min as number) || 18;
    const max = (params.max as number) || 65;
    const distribution = (params.distribution as string) || 'uniform';

    if (distribution === 'normal') {
      return String(normalRandom(min, max));
    } else if (distribution === 'demographic') {
      return String(demographicRandom());
    }
    return String(randomInt(min, max));
  },
};

/**
 * 正态分布随机数
 */
function normalRandom(min: number, max: number): number {
  const mean = (min + max) / 2;
  const stdDev = (max - min) / 6;
  let u = 0,
    v = 0;
  while (u === 0) u = Math.random();
  while (v === 0) v = Math.random();
  const num = Math.sqrt(-2.0 * Math.log(u)) * Math.cos(2.0 * Math.PI * v);
  const value = Math.round(mean + num * stdDev);
  return Math.max(min, Math.min(max, value));
}

/**
 * 人口统计分布随机数（模拟真实年龄分布）
 */
function demographicRandom(): number {
  const ageGroups = [
    { min: 0, max: 14, weight: 0.18 },
    { min: 15, max: 24, weight: 0.12 },
    { min: 25, max: 34, weight: 0.18 },
    { min: 35, max: 44, weight: 0.17 },
    { min: 45, max: 54, weight: 0.16 },
    { min: 55, max: 64, weight: 0.12 },
    { min: 65, max: 100, weight: 0.07 },
  ];

  const random = Math.random();
  let cumulative = 0;
  for (const group of ageGroups) {
    cumulative += group.weight;
    if (random <= cumulative) {
      return randomInt(group.min, group.max);
    }
  }
  return randomInt(25, 34);
}

export const personalGenerators: GeneratorDefinition[] = [
  chineseName,
  email,
  chinesePhone,
  idCard,
  chineseAddress,
  age,
];