import { Arr, Fun, Optional, Optionals } from '@ephox/katamari';
import { Attribute, Css, type SugarElement, SugarNode } from '@ephox/sugar';

import { getAttrValue } from '../util/CellUtils';
import type { CellElement, CompElm, RowElement } from '../util/TableTypes';

export interface RowData {
  readonly element: SugarElement<RowElement>;
}

export interface CellData {
  readonly element: SugarElement<CellElement>;
  readonly colspan: number;
  readonly rowspan: number;
}

export interface Generators {
  readonly cell: (cellData: CellData) => SugarElement<HTMLTableCellElement>;
  readonly row: (rowData: RowData) => SugarElement<HTMLTableRowElement>;
  readonly replace: (cell: SugarElement<HTMLTableCellElement>, tag: 'td' | 'th', attrs: Record<string, string | number | boolean | null>) => SugarElement<HTMLTableCellElement>;
  readonly gap: () => SugarElement<HTMLTableCellElement>;
  readonly colGap: () => SugarElement<HTMLTableColElement>;
  readonly col: (prev: CellData) => SugarElement<HTMLTableColElement>;
  readonly colgroup: (prev: RowData) => SugarElement<HTMLTableColElement>;
}

export interface SimpleGenerators extends Generators {
  readonly cell: () => SugarElement<HTMLTableCellElement>;
  readonly row: () => SugarElement<HTMLTableRowElement>;
  readonly replace: (cell: SugarElement<HTMLTableCellElement>) => SugarElement<HTMLTableCellElement>;
  readonly gap: () => SugarElement<HTMLTableCellElement>;
  readonly col: () => SugarElement<HTMLTableColElement>;
  readonly colgroup: () => SugarElement<HTMLTableColElement>;
}

export interface GeneratorsWrapper {}

export interface GeneratorsModification extends GeneratorsWrapper {
  readonly getOrInit: <T extends RowElement | CellElement>(element: SugarElement<T>, comparator: CompElm) => SugarElement<T>;
}

export interface GeneratorsTransform extends GeneratorsWrapper {
  readonly replaceOrInit: <T extends RowElement | CellElement>(element: SugarElement<T>, comparator: CompElm) => SugarElement<T>;
}

export interface GeneratorsMerging extends GeneratorsWrapper {
  readonly unmerge: (cell: SugarElement<HTMLTableCellElement>) => () => SugarElement<HTMLTableCellElement>;
  readonly merge: (cells: SugarElement<HTMLTableCellElement>[]) => () => SugarElement<HTMLTableCellElement>;
}

interface Recent {
  readonly item: SugarElement<CellElement>;
  readonly replacement: SugarElement<CellElement>;
}

interface Item {
  readonly item: SugarElement<HTMLTableCellElement>;
  readonly sub: SugarElement<HTMLTableCellElement>;
}

const isCol = SugarNode.isTag('col');
const isColgroup = SugarNode.isTag('colgroup');

const isRow = (element: SugarElement<RowElement | CellElement>): element is SugarElement<RowElement> =>
  SugarNode.name(element) === 'tr' || isColgroup(element);

const elementToData = (element: SugarElement<CellElement>): CellData => {
  const colspan = getAttrValue(element, 'colspan', 1);
  const rowspan = getAttrValue(element, 'rowspan', 1);
  return {
    element,
    colspan,
    rowspan
  };
};

// note that `toData` seems to be only for testing
const modification = (generators: Generators, toData = elementToData): GeneratorsModification => {

  const nuCell = (data: CellData) =>
    isCol(data.element) ? generators.col(data) : generators.cell(data);

  const nuRow = (data: RowData) =>
    isColgroup(data.element) ? generators.colgroup(data) : generators.row(data);

  const add = (element: SugarElement<RowElement | CellElement>) => {
    if (isRow(element)) {
      return nuRow({ element });
    } else {
      const cell = element as SugarElement<CellElement>;
      const replacement = nuCell(toData(cell));
      recent = Optional.some({ item: cell, replacement });
      return replacement;
    }
  };

  let recent = Optional.none<Recent>();
  const getOrInit = <T extends RowElement | CellElement>(element: SugarElement<T>, comparator: CompElm): SugarElement<T> => {
    return recent.fold(() => {
      return add(element);
    }, (p) => {
      return comparator(element, p.item) ? p.replacement : add(element);
    }) as SugarElement<T>;
  };

  return {
    getOrInit
  };
};

const transform = (tag: 'td' | 'th') => {
  return (generators: Generators): GeneratorsTransform => {
    const list: Item[] = [];

    const find = (element: SugarElement<RowElement | CellElement>, comparator: CompElm) => {
      return Arr.find(list, (x) => {
        return comparator(x.item, element);
      });
    };

    const makeNew = (element: SugarElement<HTMLTableCellElement>) => {
      // Ensure scope is never set on a td element as it's a deprecated attribute
      const attrs: Record<string, string | number | null> = tag === 'td' ? { scope: null } : {};
      const cell = generators.replace(element, tag, attrs);
      list.push({
        item: element,
        sub: cell
      });
      return cell;
    };

    const replaceOrInit = <T extends RowElement | CellElement>(element: SugarElement<T>, comparator: CompElm): SugarElement<T> => {
      if (isRow(element) || isCol(element)) {
        return element;
      } else {
        const cell = element as SugarElement<HTMLTableCellElement>;
        return find(cell, comparator).fold(() => {
          return makeNew(cell);
        }, (p) => {
          return comparator(element, p.item) ? p.sub : makeNew(cell);
        }) as SugarElement<T>;
      }
    };

    return {
      replaceOrInit
    };
  };
};

const getScopeAttribute = (cell: SugarElement<HTMLElement>) =>
  Attribute.getOpt(cell, 'scope').map(
    // Attribute can be col, colgroup, row, and rowgroup.
    // As col and colgroup are to be treated as if they are the same, lob off everything after the first three characters and there is no difference.
    (attribute) => attribute.substr(0, 3)
  );

const merging = (generators: Generators): GeneratorsMerging => {
  const unmerge = (cell: SugarElement<HTMLTableCellElement>) => {
    const scope = getScopeAttribute(cell);

    scope.each((attribute) => Attribute.set(cell, 'scope', attribute));

    return () => {
      const raw = generators.cell({
        element: cell,
        colspan: 1,
        rowspan: 1
      });
      // Remove any width calculations because they are no longer relevant.
      Css.remove(raw, 'width');
      Css.remove(cell, 'width');

      scope.each((attribute) => Attribute.set(raw, 'scope', attribute));

      return raw;
    };
  };

  const merge = (cells: SugarElement<HTMLTableCellElement>[]) => {
    const getScopeProperty = () => {

      const stringAttributes = Optionals.cat(
        Arr.map(cells, getScopeAttribute)
      );

      if (stringAttributes.length === 0) {
        return Optional.none<string>();
      } else {
        const baseScope = stringAttributes[0];
        const scopes = [ 'row', 'col' ];

        const isMixed = Arr.exists(stringAttributes, (attribute) => {
          return attribute !== baseScope && Arr.contains(scopes, attribute);
        });

        return isMixed ? Optional.none<string>() : Optional.from(baseScope);
      }
    };

    Css.remove(cells[0], 'width');

    getScopeProperty().fold(
      () => Attribute.remove(cells[0], 'scope'),
      (attribute) => Attribute.set(cells[0], 'scope', attribute + 'group')
    );

    return Fun.constant(cells[0]);
  };

  return {
    unmerge,
    merge
  };
};

export const Generators = {
  modification,
  transform,
  merging
};
