import { createHighlightSpan, isLeafNode } from '../utils';

import { BasePlugin } from './base';
import { Options } from '../type';

export class ContentNodePlugin extends BasePlugin {
  process(currNode: HTMLElement, opt: Options) {
    if (isLeafNode(currNode) && opt.getContentNodeInfo && opt.getContentNodeInfo(currNode)) {
      const contentInfo = opt.getContentNodeInfo(currNode);
      currNode._contentInfo = contentInfo
      const { text, len } = contentInfo;
      currNode._text = text;
      if (this.state.lastContentNode) {
        this.state.lastContentNode._next = currNode;
        currNode._prev = this.state.lastContentNode;
      }
      const newOffset = currNode._wordoffset + len;
      this.state.textNodeArr.length = Math.max(newOffset, this.state.textNodeArr.length);
      this.state.textNodeArr.fill(currNode, currNode._wordoffset, newOffset);
      this.state.lastContentNode = currNode;
    }

    return true;
  }

  resolveNode(currNode: HTMLElement, startOffset: number, endOffset: number, props: any, opt: Options) {
    if (currNode._contentInfo) {
      const highlightSpan = createHighlightSpan(opt.tag, props);
      currNode.parentElement!.insertBefore(highlightSpan, currNode);
      highlightSpan.appendChild(currNode);
      return highlightSpan;
    }
  }
}