import { Gene, TestUniverse, TextGene } from '@ephox/boss';
import { Arr } from '@ephox/katamari';
import * as Extract from 'ephox/phoenix/api/general/Extract';
import * as Finder from 'ephox/phoenix/test/Finder';
import * as TestRenders from 'ephox/phoenix/test/TestRenders';
import { UnitTest, assert } from '@ephox/bedrock';

UnitTest.test('api.Extract.(from,all,extract,extractTo)', function () {
  const doc = TestUniverse(
    Gene('root', 'root', [
      Gene('1', 'div', [
        Gene('1.1', 'p', [
          Gene('1.1.1', 'img', []),
          TextGene('1.1.2', 'post-image text')
        ]),
        Gene('1.2', 'p', [
          TextGene('1.2.1', 'This is text'),
          Gene('1.2.2', 'span', [
            TextGene('1.2.2.1', 'inside a span')
          ]),
          TextGene('1.2.3', 'More text'),
          Gene('1.2.4', 'em', [
            TextGene('1.2.4.1', 'Inside em')
          ]),
          TextGene('1.2.5', 'Last piece of text')
        ])
      ])
    ])
  );

  const check = function (expected: string[], extract: typeof Extract.from, initial: string) {
    const start = Finder.get(doc, initial);
    const actual = extract(doc, start);
    assert.eq(expected, Arr.map(actual, TestRenders.typeditem));
  };

  const checkFrom = function (expected: string[], initial: string) {
    check(expected, Extract.from, initial);
  };

  const checkAll = function (expected: string[], initial: string) {
    const start = Finder.get(doc, initial);
    const actual = Extract.all(doc, start);
    assert.eq(expected, Arr.map(actual, function (a) {
      return a.id;
    }));
  };

  //
  // const extract = function (universe, child, offset) {
  const checkExtract = function (expected: {id: string; offset: number}, childId: string, offset: number) {
    const child = Finder.get(doc, childId);
    const actual = Extract.extract(doc, child, offset);
    assert.eq(expected.id, actual.element().id);
    assert.eq(expected.offset, actual.offset());
  };

  const checkExtractTo = function (expected: {id: string; offset: number}, childId: string, offset: number, pred: (e: any) => boolean) {
    const child = Finder.get(doc, childId);
    const actual = Extract.extractTo(doc, child, offset, pred);
    assert.eq(expected.id, actual.element().id);
    assert.eq(expected.offset, actual.offset());
  };

  checkFrom([
    'boundary(1)',
    'boundary(1.1)',
    'empty(1.1.1)',
    'text("post-image text")',
    'boundary(1.1)',
    'boundary(1.2)',
    'text("This is text")',
    'text("inside a span")',
    'text("More text")',
    'text("Inside em")',
    'text("Last piece of text")',
    'boundary(1.2)',
    'boundary(1)'
  ], 'root');

  checkAll([
    '1', '1.1', '1.1.1', '1.1.2', '1.1', '1.2', '1.2.1', '1.2.2.1',
    '1.2.3', '1.2.4.1', '1.2.5', '1.2', '1'
  ], 'root');

  checkExtract({ id: '1.2', offset: 3 }, '1.2.1', 3);
  checkExtract({ id: '1.2', offset: 'This is textinside a span'.length }, '1.2.3', 0);
  checkExtract({
    id: '1.2',
    offset: 'This is textinside a spanMore textInside em'.length
  }, '1.2.5', 0);
  checkExtract({ id: '1.1', offset: 1 }, '1.1.2', 0);

  checkExtractTo({ id: '1.2', offset: 'This is textinside a spanMore text'.length + 2 }, '1.2.4.1', 2, function (item) {
    return item.name === 'p';
  });
});