import { Transaction } from 'prosemirror-state';
import { Node as ProsemirrorNode, Mark as ProsemirrorMark, Schema } from 'prosemirror-model';
import { setTextAlign } from './text_align';
import { setTextLineHeight } from './line_height';

const enum FormatMarks {
  bold = 'bold',
  italic = 'italic',
  underline = 'underline',
  strike = 'strike',
  link = 'link',
  textColor = 'text_color',
  textHighlight = 'text_highlight',
}

const FORMAT_MARK_NAMES: FormatMarks[] = [
  FormatMarks.bold,
  FormatMarks.italic,
  FormatMarks.underline,
  FormatMarks.strike,
  FormatMarks.link,
  FormatMarks.textColor,
  FormatMarks.textHighlight,
];

interface ClearMarksTask {
  node: ProsemirrorNode,
  mark: ProsemirrorMark,
  pos: number,
}

export function clearMarks(tr: Transaction, schema: Schema): Transaction {
  const { doc, selection } = tr;
  if (!selection || !doc) return tr;

  const { from, to, empty } = selection;
  if (empty) return tr;

  const markTypesToRemove = new Set(
    FORMAT_MARK_NAMES.map(n => schema.marks[n]).filter(Boolean)
  );

  if (!markTypesToRemove.size) return tr;

  const tasks: ClearMarksTask[] = [];
  doc.nodesBetween(from, to, (node, pos) => {
    if (node.marks && node.marks.length) {
      node.marks.some(mark => {
        if (markTypesToRemove.has(mark.type)) {
          tasks.push({ node, pos, mark });
        }
      });
      return true;
    }
    return true;
  });

  tasks.forEach(job => {
    const { node, mark, pos } = job;
    tr = tr.removeMark(pos, pos + node.nodeSize, mark.type);
  });

  tr = setTextAlign(tr, null);
  tr = setTextLineHeight(tr, null);

  return tr;
}