/**
 * @license
 * Copyright Google LLC All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.io/license
 */

import {digest, serializeNodes} from '@angular/compiler/src/i18n/digest';
import {extractMessages} from '@angular/compiler/src/i18n/extractor_merger';
import {Message} from '@angular/compiler/src/i18n/i18n_ast';
import {HtmlParser} from '@angular/compiler/src/ml_parser/html_parser';
import {DEFAULT_INTERPOLATION_CONFIG} from '@angular/compiler/src/ml_parser/interpolation_config';

{
  describe('I18nParser', () => {
    describe('elements', () => {
      it('should extract from elements', () => {
        expect(_humanizeMessages('<div i18n="m|d">text</div>')).toEqual([
          [['text'], 'm', 'd', ''],
        ]);
      });

      it('should extract from nested elements', () => {
        expect(_humanizeMessages('<div i18n="m|d">text<span><b>nested</b></span></div>')).toEqual([
          [
            [
              'text',
              '<ph tag name="START_TAG_SPAN"><ph tag name="START_BOLD_TEXT">nested</ph name="CLOSE_BOLD_TEXT"></ph name="CLOSE_TAG_SPAN">'
            ],
            'm', 'd', ''
          ],
        ]);
      });

      it('should not create a message for empty elements', () => {
        expect(_humanizeMessages('<div i18n="m|d"></div>')).toEqual([]);
      });

      it('should not create a message for plain elements', () => {
        expect(_humanizeMessages('<div></div>')).toEqual([]);
      });

      it('should support void elements', () => {
        expect(_humanizeMessages('<div i18n="m|d"><p><br></p></div>')).toEqual([
          [
            [
              '<ph tag name="START_PARAGRAPH"><ph tag name="LINE_BREAK"/></ph name="CLOSE_PARAGRAPH">'
            ],
            'm', 'd', ''
          ],
        ]);
      });

      it('should trim whitespace from custom ids (but not meanings)', () => {
        expect(_humanizeMessages('<div i18n="\n   m|d@@id\n   ">text</div>')).toEqual([
          [['text'], '\n   m', 'd', 'id'],
        ]);
      });
    });

    describe('attributes', () => {
      it('should extract from attributes outside of translatable section', () => {
        expect(_humanizeMessages('<div i18n-title="m|d" title="msg"></div>')).toEqual([
          [['msg'], 'm', 'd', ''],
        ]);
      });

      it('should extract from attributes in translatable element', () => {
        expect(_humanizeMessages('<div i18n><p><b i18n-title="m|d" title="msg"></b></p></div>'))
            .toEqual([
              [
                [
                  '<ph tag name="START_PARAGRAPH"><ph tag name="START_BOLD_TEXT"></ph name="CLOSE_BOLD_TEXT"></ph name="CLOSE_PARAGRAPH">'
                ],
                '', '', ''
              ],
              [['msg'], 'm', 'd', ''],
            ]);
      });

      it('should extract from attributes in translatable block', () => {
        expect(_humanizeMessages(
                   '<!-- i18n --><p><b i18n-title="m|d" title="msg"></b></p><!-- /i18n -->'))
            .toEqual([
              [['msg'], 'm', 'd', ''],
              [
                [
                  '<ph tag name="START_PARAGRAPH"><ph tag name="START_BOLD_TEXT"></ph name="CLOSE_BOLD_TEXT"></ph name="CLOSE_PARAGRAPH">'
                ],
                '', '', ''
              ],
            ]);
      });

      it('should extract from attributes in translatable ICU', () => {
        expect(
            _humanizeMessages(
                '<!-- i18n -->{count, plural, =0 {<p><b i18n-title="m|d" title="msg"></b></p>}}<!-- /i18n -->'))
            .toEqual([
              [['msg'], 'm', 'd', ''],
              [
                [
                  '{count, plural, =0 {[<ph tag name="START_PARAGRAPH"><ph tag name="START_BOLD_TEXT"></ph name="CLOSE_BOLD_TEXT"></ph name="CLOSE_PARAGRAPH">]}}'
                ],
                '', '', ''
              ],
            ]);
      });

      it('should extract from attributes in non translatable ICU', () => {
        expect(
            _humanizeMessages('{count, plural, =0 {<p><b i18n-title="m|d" title="msg"></b></p>}}'))
            .toEqual([
              [['msg'], 'm', 'd', ''],
            ]);
      });

      it('should not create a message for empty attributes', () => {
        expect(_humanizeMessages('<div i18n-title="m|d" title></div>')).toEqual([]);
      });
    });

    describe('interpolation', () => {
      it('should replace interpolation with placeholder', () => {
        expect(_humanizeMessages('<div i18n="m|d">before{{ exp }}after</div>')).toEqual([
          [['[before, <ph name="INTERPOLATION"> exp </ph>, after]'], 'm', 'd', ''],
        ]);
      });

      it('should support named interpolation', () => {
        expect(_humanizeMessages('<div i18n="m|d">before{{ exp //i18n(ph="teSt") }}after</div>'))
            .toEqual([
              [['[before, <ph name="TEST"> exp //i18n(ph="teSt") </ph>, after]'], 'm', 'd', ''],
            ]);

        expect(
            _humanizeMessages('<div i18n=\'m|d\'>before{{ exp //i18n(ph=\'teSt\') }}after</div>'))
            .toEqual([
              [[`[before, <ph name="TEST"> exp //i18n(ph='teSt') </ph>, after]`], 'm', 'd', ''],
            ]);
      });
    });

    describe('blocks', () => {
      it('should extract from blocks', () => {
        expect(_humanizeMessages(`<!-- i18n: meaning1|desc1 -->message1<!-- /i18n -->
         <!-- i18n: desc2 -->message2<!-- /i18n -->
         <!-- i18n -->message3<!-- /i18n -->`))
            .toEqual([
              [['message1'], 'meaning1', 'desc1', ''],
              [['message2'], '', 'desc2', ''],
              [['message3'], '', '', ''],
            ]);
      });

      it('should extract all siblings', () => {
        expect(_humanizeMessages(`<!-- i18n -->text<p>html<b>nested</b></p><!-- /i18n -->`)).toEqual([
          [
            [
              'text',
              '<ph tag name="START_PARAGRAPH">html, <ph tag name="START_BOLD_TEXT">nested</ph name="CLOSE_BOLD_TEXT"></ph name="CLOSE_PARAGRAPH">'
            ],
            '', '', ''
          ],
        ]);
      });
    });

    describe('ICU messages', () => {
      it('should extract as ICU when single child of an element', () => {
        expect(_humanizeMessages('<div i18n="m|d">{count, plural, =0 {zero}}</div>')).toEqual([
          [['{count, plural, =0 {[zero]}}'], 'm', 'd', ''],
        ]);
      });

      it('should extract as ICU + ph when not single child of an element', () => {
        expect(_humanizeMessages('<div i18n="m|d">b{count, plural, =0 {zero}}a</div>')).toEqual([
          [['b', '<ph icu name="ICU">{count, plural, =0 {[zero]}}</ph>', 'a'], 'm', 'd', ''],
          [['{count, plural, =0 {[zero]}}'], '', '', ''],
        ]);
      });

      it('should extract as ICU + ph when wrapped in whitespace in an element', () => {
        expect(_humanizeMessages('<div i18n="m|d"> {count, plural, =0 {zero}} </div>')).toEqual([
          [[' ', '<ph icu name="ICU">{count, plural, =0 {[zero]}}</ph>', ' '], 'm', 'd', ''],
          [['{count, plural, =0 {[zero]}}'], '', '', ''],
        ]);
      });

      it('should extract as ICU when single child of a block', () => {
        expect(_humanizeMessages('<!-- i18n:m|d -->{count, plural, =0 {zero}}<!-- /i18n -->'))
            .toEqual([
              [['{count, plural, =0 {[zero]}}'], 'm', 'd', ''],
            ]);
      });

      it('should extract as ICU + ph when not single child of a block', () => {
        expect(_humanizeMessages('<!-- i18n:m|d -->b{count, plural, =0 {zero}}a<!-- /i18n -->'))
            .toEqual([
              [['{count, plural, =0 {[zero]}}'], '', '', ''],
              [['b', '<ph icu name="ICU">{count, plural, =0 {[zero]}}</ph>', 'a'], 'm', 'd', ''],
            ]);
      });

      it('should not extract nested ICU messages', () => {
        expect(_humanizeMessages(
                   '<div i18n="m|d">b{count, plural, =0 {{sex, select, male {m}}}}a</div>'))
            .toEqual([
              [
                [
                  'b', '<ph icu name="ICU">{count, plural, =0 {[{sex, select, male {[m]}}]}}</ph>',
                  'a'
                ],
                'm', 'd', ''
              ],
              [['{count, plural, =0 {[{sex, select, male {[m]}}]}}'], '', '', ''],
            ]);
      });
    });

    describe('implicit elements', () => {
      it('should extract from implicit elements', () => {
        expect(_humanizeMessages('<b>bold</b><i>italic</i>', ['b'])).toEqual([
          [['bold'], '', '', ''],
        ]);
      });
    });

    describe('implicit attributes', () => {
      it('should extract implicit attributes', () => {
        expect(_humanizeMessages(
                   '<b title="bb">bold</b><i title="ii">italic</i>', [], {'b': ['title']}))
            .toEqual([
              [['bb'], '', '', ''],
            ]);
      });
    });

    describe('placeholders', () => {
      it('should reuse the same placeholder name for tags', () => {
        const html = '<div i18n="m|d"><p>one</p><p>two</p><p other>three</p></div>';
        expect(_humanizeMessages(html)).toEqual([
          [
            [
              '<ph tag name="START_PARAGRAPH">one</ph name="CLOSE_PARAGRAPH">',
              '<ph tag name="START_PARAGRAPH">two</ph name="CLOSE_PARAGRAPH">',
              '<ph tag name="START_PARAGRAPH_1">three</ph name="CLOSE_PARAGRAPH">',
            ],
            'm', 'd', ''
          ],
        ]);

        expect(_humanizePlaceholders(html)).toEqual([
          'START_PARAGRAPH=<p>, CLOSE_PARAGRAPH=</p>, START_PARAGRAPH_1=<p other>',
        ]);
      });

      it('should reuse the same placeholder name for interpolations', () => {
        const html = '<div i18n="m|d">{{ a }}{{ a }}{{ b }}</div>';
        expect(_humanizeMessages(html)).toEqual([
          [
            [
              '[<ph name="INTERPOLATION"> a </ph>, <ph name="INTERPOLATION"> a </ph>, <ph name="INTERPOLATION_1"> b </ph>]'
            ],
            'm', 'd', ''
          ],
        ]);

        expect(_humanizePlaceholders(html)).toEqual([
          'INTERPOLATION={{ a }}, INTERPOLATION_1={{ b }}',
        ]);
      });

      it('should reuse the same placeholder name for icu messages', () => {
        const html =
            '<div i18n="m|d">{count, plural, =0 {0}}{count, plural, =0 {0}}{count, plural, =1 {1}}</div>';

        expect(_humanizeMessages(html)).toEqual([
          [
            [
              '<ph icu name="ICU">{count, plural, =0 {[0]}}</ph>',
              '<ph icu name="ICU">{count, plural, =0 {[0]}}</ph>',
              '<ph icu name="ICU_1">{count, plural, =1 {[1]}}</ph>',
            ],
            'm', 'd', ''
          ],
          [['{count, plural, =0 {[0]}}'], '', '', ''],
          [['{count, plural, =0 {[0]}}'], '', '', ''],
          [['{count, plural, =1 {[1]}}'], '', '', ''],
        ]);

        expect(_humanizePlaceholders(html)).toEqual([
          '',
          'VAR_PLURAL=count',
          'VAR_PLURAL=count',
          'VAR_PLURAL=count',
        ]);

        expect(_humanizePlaceholdersToMessage(html)).toEqual([
          'ICU=f0f76923009914f1b05f41042a5c7231b9496504, ICU_1=73693d1f78d0fc882f0bcbce4cb31a0aa1995cfe',
          '',
          '',
          '',
        ]);
      });
    });
  });
}

export function _humanizeMessages(
    html: string, implicitTags: string[] = [],
    implicitAttrs: {[k: string]: string[]} = {}): [string[], string, string, string][] {
  // clang-format off
  // https://github.com/angular/clang-format/issues/35
  return _extractMessages(html, implicitTags, implicitAttrs).map(
    message => [serializeNodes(message.nodes), message.meaning, message.description, message.id]) as [string[], string, string, string][];
  // clang-format on
}

function _humanizePlaceholders(
    html: string, implicitTags: string[] = [],
    implicitAttrs: {[k: string]: string[]} = {}): string[] {
  // clang-format off
  // https://github.com/angular/clang-format/issues/35
  return _extractMessages(html, implicitTags, implicitAttrs).map(
    msg => Object.keys(msg.placeholders).map((name) => `${name}=${msg.placeholders[name].text}`).join(', '));
  // clang-format on
}

function _humanizePlaceholdersToMessage(
    html: string, implicitTags: string[] = [],
    implicitAttrs: {[k: string]: string[]} = {}): string[] {
  // clang-format off
  // https://github.com/angular/clang-format/issues/35
  return _extractMessages(html, implicitTags, implicitAttrs).map(
    msg => Object.keys(msg.placeholderToMessage).map(k => `${k}=${digest(msg.placeholderToMessage[k])}`).join(', '));
  // clang-format on
}


export function _extractMessages(
    html: string, implicitTags: string[] = [],
    implicitAttrs: {[k: string]: string[]} = {}): Message[] {
  const htmlParser = new HtmlParser();
  const parseResult = htmlParser.parse(html, 'extractor spec', {tokenizeExpansionForms: true});
  if (parseResult.errors.length > 1) {
    throw Error(`unexpected parse errors: ${parseResult.errors.join('\n')}`);
  }

  return extractMessages(
             parseResult.rootNodes, DEFAULT_INTERPOLATION_CONFIG, implicitTags, implicitAttrs)
      .messages;
}