import Decimal from 'decimal.js';

const zhFractions = ['角', '分'];

const zhDigits = ['零', '壹', '贰', '叁', '肆', '伍', '陆', '柒', '捌', '玖'];

// 大单位
const zhUnits = [
  '', // 10^0
  '万', // 10^4
  '亿', // 10^8
  '兆', // 10^12
  '京', // 10^16
  '垓', // 10^20
  '秭', // 10^24
  '穰', // 10^28
  '沟', // 10^32
  '涧', // 10^36
  '正', // 10^40
  '载', // 10^44
  '极', // 10^48
  '恒河沙', // 10^52
  '阿僧祗', // 10^56
  '那由他', // 10^60
  '不可思议', // 10^64
  '无量', // 10^68
  '大数', // 巨大数目
];

// 最大到那由他
const limitNum = new Decimal(10).toPower(64);

// 局部单位
const unitl = ['', '拾', '佰', '仟'];

interface LocalResult {
  num: number;
  numHan: string;
  unit: string;
}

interface Thunk {
  localResult: LocalResult[];
  unit: string;
  unitId: number;
  localStack: LocalResult[];
  value: number;
  unSpell?: boolean;
}

interface DecimalThunk {
  num: number;
  numHan: string;
  unit: string;
  unitId: number;
}

function extractDigits(int: Decimal) {
  const digits: number[] = [];
  const base = 10;

  let number = int;
  while (true) {
    // 余数
    const remainder = number.mod(base);
    digits.push(remainder.toNumber());

    // 商
    const quotient = number.divToInt(base);
    if (quotient.isZero()) break;
    number = quotient;
  }
  return digits;
}

// 数组分块
function splitArray<T>(array: T[], chunkSize: number) {
  let result: T[][] = [];
  for (let i = 0; i < array.length; i += chunkSize) {
    result.push(array.slice(i, i + chunkSize));
  }
  return result;
}

function first<T>(arr: T[]) {
  return arr[0];
}

function last<T>(arr: T[]) {
  return arr[arr.length - 1];
}

function trimZeroLocal(locals: LocalResult[]) {
  const localStack: LocalResult[] = [];
  for (let i = 0; i < locals.length; i++) {
    const local = locals[i];

    const top = last(localStack);
    if (top?.num === 0 && local.num === 0) continue;

    localStack.push(local);
  }

  while (localStack.length > 0) {
    if (first(localStack).num !== 0) break;
    // if (localStack.length <= 1) break;
    localStack.shift();
  }

  while (localStack.length > 0) {
    if (last(localStack).num !== 0) break;
    // if (localStack.length <= 1) break;
    localStack.pop();
  }

  // console.log(thunk);
  // console.log("localstack:", localStack);
  return localStack;
}

function spell(localStack: LocalResult[]) {
  let words;
  if (localStack.length === 0) {
    words = zhDigits[0];
  } else {
    words = localStack
      .map(e => {
        return e.num !== 0 ? `${e.numHan}${e.unit}` : `${e.numHan}`;
      })
      .reverse()
      .join('');
  }
  return words;
}

function parseIntegerPart(integerPart: Decimal) {
  // 整数部分各个单位的值
  // 1008610086
  // [ 1,0,0,8,6,0,0,8,6 ]
  const digits = extractDigits(integerPart);

  // 将整个数字拆分到对应的大单位上
  // [ [1], [0,0,8,6], [0,0,8,6] ]
  //    亿   千百十万   千百十个
  const digitsThunks = splitArray(digits, 4);

  const thunkStack: Thunk[] = [];

  // 针对每个trunk进行处理
  for (let thunkIdx = 0; thunkIdx < digitsThunks.length; thunkIdx++) {
    const digits = digitsThunks[thunkIdx];
    // 单位
    const thunkUnit = zhUnits[thunkIdx];
    const unitId = thunkIdx;

    // 局部处理
    let localResult: LocalResult[] = [];
    for (let digitIdx = 0; digitIdx < digits.length; digitIdx++) {
      const num = digits[digitIdx];
      const numHan = zhDigits[num];
      const localUnit = unitl[digitIdx];

      localResult.push({ num, numHan, unit: localUnit });
    }

    // 局部移除不必要0
    const localStack = trimZeroLocal(localResult);

    // 局部的值
    let thunkValue = 0;
    for (let q = 0; q < localResult.length; q++) {
      thunkValue += localResult[q].num * Math.pow(10, q);
    }

    thunkStack.push({
      localResult,
      unit: thunkUnit,
      unitId,
      localStack,
      value: thunkValue,
    });
  }

  // 读数
  const spelledThunk: Thunk[] = [];
  const allWords: string[] = [];

  for (let i = thunkStack.length - 1; i >= 0; i--) {
    const aThunk = thunkStack[i];
    if (aThunk.value === 0) continue;

    const words = spell(aThunk.localStack);

    const lastThunk = last(spelledThunk);
    if (lastThunk) {
      if (lastThunk.unitId - aThunk.unitId > 1) {
        // 要加0，因为上一个空的被跳过了
        allWords.push('零');
      } else {
        // 如果是相邻的，则要判断当前的最高位是否为空
        // 能到达这里说明当前的最高位应该是千位，
        // 因为能到这里说明肯定在更高的位置上有更大的单位阿斯顿发山东发生的
        if (last(aThunk.localResult).num === 0) {
          allWords.push('零');
        }
      }
    }

    allWords.push(`${words}${aThunk.unit}`);
    spelledThunk.push(aThunk);
  }

  return {
    allWords,
    thunkStack,
    isZero: integerPart.isZero(),
  };
}

function spellFraction(result: any[]) {
  const words = result.map(e => {
    if (e.num === 0) return '';
    return `${e.numHan}${e.unit}`;
  });
  return words;
}

function parseDecimalPart(decimalPart: Decimal) {
  const thunks: DecimalThunk[] = [];

  let isZero = true;

  let number = decimalPart;
  for (let i = 0; i < zhFractions.length; i++) {
    number = number.mul(10);

    const remainder = number.mod(10).floor();
    const num = remainder.toNumber();

    thunks.push({
      num,
      numHan: zhDigits[num],
      unit: zhFractions[i],
      unitId: i,
    });

    if (num != 0) {
      isZero = false;
    }
  }

  const allWords = spellFraction(thunks);
  return { allWords, thunks, isZero };
}

function dxrmb(value: string | number) {
  let sign = Decimal.sign(value);

  // 负数后面在考虑
  let number = Decimal.abs(value);

  if (number.cmp(limitNum) >= 0) {
    throw new Error('超过最大支持10的64次方');
  }

  // 获取整数部分
  const integerPart = Decimal.floor(number);
  const decimalPart = Decimal.sub(number, integerPart);

  // 解析
  const integer = parseIntegerPart(integerPart);
  const decimal = parseDecimalPart(decimalPart);

  const words: string[] = [];
  // 读元
  if (!integerPart.isZero()) {
    words.push(integer.allWords.join(''), '元');
  }

  if (decimal.thunks[0].num !== 0) {
    words.push(decimal.thunks[0].numHan, '角');
  }

  if (decimal.thunks[1].num !== 0) {
    // if (integerPart !== 0 && decimal.thunks[0].num === 0) {
    //   words.push("零");
    // }
    words.push(decimal.thunks[1].numHan, '分');
  }

  if (words.length === 0) {
    return '零元整';
  }

  // 如果有元而且没分就以整结尾
  if (!integerPart.isZero() && decimal.thunks[1].num === 0) {
    words.push('整');
  }

  if (sign < 0) {
    words.unshift('负');
  }

  return words.join('');
}

export default dxrmb;
