module.exports = grammar({
  name: 'html',

  extras: $ => [
    $.comment,
    /\s+/,
  ],

  externals: $ => [
    $._open_start_tag,
    $._open_raw_start_tag,
    $._close_start_tag,
    $._self_close_start_tag,
    $.end_tag,
    $._implicit_end_tag,
    $._erroneous_end_tag,
    $._raw_text,
    $.comment,
  ],

  rules: {
    fragment: $ => repeat($._node),

    doctype: $ => seq(
      '<!',
      /[Dd][Oo][Cc][Tt][Yy][Pp][Ee]/,
      /[^>]+/,
      '>'
    ),

    _node: $ => choice(
      $.doctype,
      $.text,
      $._erroneous_end_tag,
      $.element,
      $.raw_element
    ),

    element: $ => choice(
      seq(
        $.start_tag,
        repeat($._node),
        choice($.end_tag, $._implicit_end_tag)
      ),
      $.self_closing_tag
    ),

    raw_element: $ => seq(
      alias($._raw_start_tag, $.start_tag),
      optional($._raw_text),
      $.end_tag
    ),

    start_tag: $ => seq(
      $._open_start_tag,
      repeat($.attribute),
      $._close_start_tag
    ),

    _raw_start_tag: $ => seq(
      $._open_raw_start_tag,
      repeat($.attribute),
      $._close_start_tag
    ),

    self_closing_tag: $ => seq(
      $._open_start_tag,
      repeat($.attribute),
      $._self_close_start_tag
    ),

    attribute: $ => seq(
      $.attribute_name,
      optional(seq(
        '=',
        choice(
          $.attribute_value,
          $.quoted_attribute_value
        )
      ))
    ),

    attribute_name: $ => /[^<>"'/=\s]+/,

    attribute_value: $ => /[^<>"'=\s]+/,

    quoted_attribute_value: $ => choice(
      seq("'", optional(alias(/[^']+/, $.attribute_value)), "'"),
      seq('"', optional(alias(/[^"]+/, $.attribute_value)), '"')
    ),

    text: $ => /[^<>]+/
  }
});