import * as grok from 'datagrok-api/grok';
import * as ui from 'datagrok-api/ui';
import * as DG from 'datagrok-api/dg';

import {before, category, test} from '@datagrok-libraries/test/src/test';
import {ALIGNMENT, ALPHABET, NOTATION} from '@datagrok-libraries/bio/src/utils/macromolecule';
import {ISeqHelper, getSeqHelper} from '@datagrok-libraries/bio/src/utils/seq-helper';

import {_testNeg, _testPos, DfReaderFunc} from './utils/detectors-utils';


category('detectors.weak-and-likely', () => {
  let seqHelper: ISeqHelper;

  before(async () => {
    seqHelper = await getSeqHelper();
  });

  const enum csvTests {
    fastaDnaWeak1 = 'fastaDnaWeak1',
    fastaDnaWeak1LikelyName = 'fastaDnaWeak1LikelyName',

    fastaRnaWeak1 = 'fastaRnaWeak1',
    fastaRnaWeak1LikelyName = 'fastaRnaWeak1LikelyName',

    fastaPtWeak1 = 'fastaPtWeak1',
    fastaPtWeak1LikelyName = 'fastaPtWeak1LikelyName',

    /* Notation 'fasta' alphabet 'UN' is forbidden for likely columns too. */
    fastaUn1 = 'fastaUn1',
    fastaUn1LikelyName = 'fastaUn1LikelyName',
    fastaUn2LikelyName = 'fastaUn2LikelyName',
    fastaUnMsa1LikelyName = 'fastaUnMsa1LikelyName',
  }

  const csvData: { [name: string]: string } = {
    [csvTests.fastaDnaWeak1]: `id,colName
1,TTTTTTTTTT
2,TTTTTTTTTT
3,TTTTTTTTTT
4,TTTTTTTTTT`,
    [csvTests.fastaDnaWeak1LikelyName]: `id,seq
1,TTTTTTT
2,TTTTTTT
3,TTTTTTT
4,TTTTTTT`,
    [csvTests.fastaRnaWeak1]: `id,colName
1,UUUUUUUUUU
2,UUUUUUUUUU
3,UUUUUUUUUU
4,UUUUUUUUUU`,
    [csvTests.fastaRnaWeak1LikelyName]: `id,seq
1,UUUUUUU
2,UUUUUUU
3,UUUUUUU
4,UUUUUUU`,
    [csvTests.fastaPtWeak1]: `id,colName
1,SLSLSPGKSLSLSPGK
2,SLSLSPGKSLSLSPGK
3,SLSLSPGKSLSLSPGK
4,SLSLSPGKSLSLSPGK`,
    [csvTests.fastaPtWeak1LikelyName]: `id,seq
1,SLSLSPGKSLSLSPGK
2,SLSLSPGKSLSLSPGK
3,SLSLSPGKSLSLSPGK
4,SLSLSPGKSLSLSPGK`,
    [csvTests.fastaUn1]: `id,colName
1,word
2,other
3,some
4,another`,
    [csvTests.fastaUn1LikelyName]: `id,seq
1,word
2,other
3,some
4,another`,
    [csvTests.fastaUn2LikelyName]: `protein
Boombastic
Megafantastic
"just-a-random-thought,oy!"`,
    [csvTests.fastaUnMsa1LikelyName]: `id,seq
1,word
2,male
3,bare
4,core`,
  };

  const readCsv: (key: csvTests) => DfReaderFunc = (key: keyof typeof csvData) => {
    return async () => {
      // Always recreate test data frame from CSV for reproducible detector behavior in tests.
      const csv: string = csvData[key];
      const df: DG.DataFrame = DG.DataFrame.fromCsv(csv);
      await grok.data.detectSemanticTypes(df);
      return df;
    };
  };

  test(csvTests.fastaDnaWeak1, async () => {
    await _testNeg(readCsv(csvTests.fastaDnaWeak1), 'colName');
  });
  test(csvTests.fastaDnaWeak1LikelyName, async () => {
    await _testPos(readCsv(csvTests.fastaDnaWeak1LikelyName), 'seq', seqHelper,
      NOTATION.FASTA, ALIGNMENT.SEQ_MSA, ALPHABET.DNA, 4, false);
  });

  test(csvTests.fastaRnaWeak1, async () => {
    await _testNeg(readCsv(csvTests.fastaRnaWeak1), 'colName');
  });
  test(csvTests.fastaRnaWeak1LikelyName, async () => {
    await _testPos(readCsv(csvTests.fastaRnaWeak1LikelyName), 'seq', seqHelper,
      NOTATION.FASTA, ALIGNMENT.SEQ_MSA, ALPHABET.RNA, 4, false);
  });

  test(csvTests.fastaPtWeak1, async () => {
    await _testNeg(readCsv(csvTests.fastaPtWeak1), 'colName');
  });
  test(csvTests.fastaPtWeak1LikelyName, async () => {
    await _testPos(readCsv(csvTests.fastaPtWeak1LikelyName), 'seq', seqHelper,
      NOTATION.FASTA, ALIGNMENT.SEQ_MSA, ALPHABET.PT, 20, false);
  });

  test(csvTests.fastaUn1, async () => {
    await _testNeg(readCsv(csvTests.fastaUn1), 'colName');
  });
  test(csvTests.fastaUn1LikelyName, async () => {
    await _testNeg(readCsv(csvTests.fastaUn1LikelyName), 'seq');
  });
  test(csvTests.fastaUn2LikelyName, async () => {
    await _testNeg(readCsv(csvTests.fastaUn2LikelyName), 'protein');
  });
  test(csvTests.fastaUnMsa1LikelyName, async () => {
    await _testNeg(readCsv(csvTests.fastaUnMsa1LikelyName), 'seq');
  });
});