tree-sitter-d2/grammar.js

const spaces = repeat(" ");

module.exports = grammar({
  name: "d2",

  extras: ($) => [],

  word: ($) => $._word,

  conflicts: ($) => [[$.identifier], [$.arrow]],

  rules: {
    source_file: ($) => repeat($._definition),

    _definition: ($) =>
      choice($._end, $._root_attribute, $.connection, $.shape),

    connection: ($) =>
      seq(
        $._identifier,
        repeat1(seq($._arrow, $._identifier)),
        optional(
          choice(seq($.dot, $._connection_attribute), seq($._colon, $.label))
        ),
        $._end
      ),

    shape: ($) =>
      seq(
        $._identifier,
        repeat(seq($.dot, $._identifier)),
        optional(
          choice(seq($.dot, $._shape_attribute), seq($._colon, $.label))
        ),
        $._end
      ),

    label: ($) => choice($.string, $._unquoted_string),

    attr_value: ($) => choice($.string, $._unquoted_string),

    _root_attribute: ($) =>
      seq(alias($._root_attr_key, $.attr_key), $._colon, $.attr_value, $._end),

    _root_attr_key: ($) => "direction",

    _style_attribute: ($) =>
      seq(
        alias("style", $.attr_key),
        $.dot,
        alias($._style_attr_key, $.attr_key),
        $._colon,
        $.attr_value
      ),

    _shape_attribute: ($) =>
      choice(
        seq(alias($._shape_attr_key, $.attr_key), $._colon, $.attr_value),
        $._style_attribute
      ),

    _connection_attribute: ($) =>
      seq(alias($._connection_attr_key, $.attr_key), $._colon, $.attr_value),

    _shape_attr_key: ($) =>
      choice(
        "shape",
        "label",
        // sql
        "constraint",
        // image
        "icon",
        "width",
        "height"
      ),

    _style_attr_key: ($) =>
      choice(
        "opacity",
        "fill",
        "stroke",
        "stroke-width",
        "stroke-dash",
        "border-radius",
        "font-color",
        "shadow",
        "multiple",
        "animated",
        "3d",
        "link"
      ),

    _text_attr_key: ($) => "near",

    _connection_attr_key: ($) => choice("source-arrowhead", "target-arrowhead"),

    _identifier: ($) => seq(spaces, $.identifier),

    identifier: ($) =>
      seq(
        optional($._dash),
        choice(
          $._word,
          repeat1(seq($._word, choice(spaces, $._dash), $._word))
        ),
        optional($._dash)
      ),

    _colon: ($) => seq(spaces, ":", spaces),

    _arrow: ($) => seq(spaces, $.arrow),

    arrow: ($) =>
      choice(
        seq("--", repeat($._dash)),
        seq("<-", repeat($._dash)),
        seq("<-", repeat($._dash), ">"),
        seq(repeat($._dash), "->")
      ),

    _dash: ($) => token.immediate("-"),

    dot: ($) => token.immediate("."),

    _unquoted_string: ($) => token.immediate(/[^'"`\n;{]+/),

    string: ($) =>
      choice(
        seq("'", repeat(token.immediate(/[^'\n]+/)), "'"),
        seq('"', repeat(token.immediate(/[^"\n]+/)), '"'),
        seq("`", repeat(token.immediate(/[^`\n]+/)), "`")
      ),

    _word: ($) => /[\w\d]+/,

    _eof: ($) => choice("\n", "\0"),
    _end: ($) => seq(spaces, choice(";", $._eof)),
  },
});