tree-sitter-fastbuild 0.0.1

let precedence = {
  or: 2,
  and: 3,
  compare: 5,
  lookup: 6,
  not: 7,
  function: 8,
};

function preprocessor($, name, ...args) {
  return seq(alias(seq("#", name), "#" + name), ...args, $._preprocessor_end);
}

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

  extras: ($) => [
    $.comment,
    $._whitespace,
    $._separator,
    $._newline,
    $.define,
    $.undef,
    $.if,
    $.import,
    $.include,
    $.once,
    $.else,
    $.endif,
    $.unknown,
  ],

  word: ($) => $.identifier,

  supertypes: ($) => [
    $.literal,
    $.expression,
    $.statement,
    $.number,
    $.preprocessor_expression,
  ],

  inline: ($) => [],

  externals: ($) => [$._preprocessor_end],

  rules: {
    source_file: ($) => repeat($.statement),
    _separator: (_) => ",",
    _whitespace: (_) => token(/[\s\u00A0\uFEFF\u3000]+/),
    // http://stackoverflow.com/questions/13014947/regex-to-match-a-c-style-multiline-comment/36328890#36328890
    comment: (_) => token(choice(/;.*/, /\/\/.*/)),
    _newline: (_) => token(/\r?\n/),
    decimal: (_) => token(/[\d_]+/),
    filename: (_) => token(/[\/\.\w\- ]+/),
    identifier: (_) => token(/[a-zA-Z_][\w_]*/),
    single_quoted: (_) => token.immediate(prec(1, /[^^$%']+?/)),
    double_quoted: (_) => token.immediate(prec(1, /[^^$%"]+?/)),
    escape_sequence: (_) => token.immediate(/\^./),
    string: ($) =>
      choice(
        seq("'", repeat($._single_inner), "'"),
        seq('"', repeat($._double_inner), '"'),
      ),
    _single_inner: ($) =>
      choice(
        field("single_quoted", $.single_quoted),
        field("interpolation", $.interpolation),
        field("placeholder", $.placeholder),
        field("escape_sequence", $.escape_sequence),
      ),
    _double_inner: ($) =>
      choice(
        field("double_quoted", $.double_quoted),
        field("interpolation", $.interpolation),
        field("placeholder", $.placeholder),
        field("escape_sequence", $.escape_sequence),
      ),
    interpolation: ($) =>
      seq(token.immediate("$"), field("variable", $.identifier), "$"),
    placeholder: ($) =>
      seq(token.immediate("%"), field("argument", choice($.decimal, "@", "*"))),

    unknown: ($) => seq("#", $.identifier, $._preprocessor_end),
    define: ($) => preprocessor($, "define", field("variable", $.identifier)),
    undef: ($) => preprocessor($, "undef", field("variable", $.identifier)),
    import: ($) => preprocessor($, "import", field("variable", $.identifier)),
    include: ($) => preprocessor($, "include", field("filename", $.string)),
    if: ($) =>
      preprocessor($, "if", field("condition", $.preprocessor_expression)),
    else: ($) => preprocessor($, "else"),
    endif: ($) => preprocessor($, "endif"),
    once: ($) => preprocessor($, "once"),
    preprocessor_expression: ($) =>
      choice(
        $.literal,
        $.identifier,
        alias($.preprocessor_not, $.not),
        alias($.preprocessor_and, $.and),
        alias($.preprocessor_or, $.or),
        alias($.preprocessor_call, $.call),
      ),
    preprocessor_call: ($) =>
      seq(
        field("name", $.identifier),
        "(",
        field("arguments", repeat($.preprocessor_expression)),
        ")",
      ),
    preprocessor_or: ($) =>
      prec.left(
        precedence.or,
        seq(
          field("left", $.preprocessor_expression),
          "||",
          field("right", $.preprocessor_expression),
        ),
      ),
    preprocessor_and: ($) =>
      prec.left(
        precedence.and,
        seq(
          field("left", $.preprocessor_expression),
          "&&",
          field("right", $.preprocessor_expression),
        ),
      ),
    preprocessor_not: ($) =>
      prec.right(
        precedence.not,
        seq("!", field("right", $.preprocessor_expression)),
      ),

    usage: ($) =>
      seq(
        field("scope", choice(".", "^")),
        field("variable", choice($.identifier, $.string)),
      ),

    array: ($) => seq("{", repeat($.statement), "}"),
    struct: ($) => seq("[", repeat($.statement), "]"),
    arguments: ($) => seq("(", repeat($.statement), ")"),
    parenthesis: ($) => seq("(", $.statement, ")"),

    call: ($) =>
      prec.right(
        precedence.function,
        seq(
          field("name", $.identifier),
          field("arguments", optional($.arguments)),
          field("body", optional($.array)),
        ),
      ),

    function_definition: ($) =>
      prec.right(
        precedence.function,
        seq(
          "function",
          field("name", $.identifier),
          field("arguments", optional($.arguments)),
          field("body", optional($.array)),
        ),
      ),

    statement: ($) => choice($.compound, $.logic, $.expression),

    expression: ($) =>
      choice(
        $.literal,
        $.usage,
        $.array,
        $.struct,
        $.parenthesis,
        $.not,
        $.in,
        $.not_in,
        $.compare,
        $.call,
        $.function_definition,
      ),

    compound: ($) =>
      seq(
        field("left", $.usage),
        repeat1(choice($.assign, $.concatenate, $.subtract)),
      ),
    assign: ($) => seq("=", field("right", choice($.expression, $.logic))),
    concatenate: ($) => seq("+", field("right", choice($.expression, $.logic))),
    subtract: ($) => seq("-", field("right", choice($.expression, $.logic))),

    logic: ($) =>
      seq(field("left", $.expression), repeat1(choice($.and, $.or))),
    and: ($) => seq("&&", field("right", $.expression)),
    or: ($) => seq("||", field("right", $.expression)),

    not: ($) =>
      prec.right(precedence.not, seq("!", field("right", $.expression))),

    compare: ($) =>
      prec.left(
        precedence.compare,
        seq(
          field("left", $.expression),
          field("operator", choice("==", "!=", ">", "<", ">=", "<=")),
          field("right", $.expression),
        ),
      ),

    in: ($) =>
      prec.left(
        precedence.lookup,
        seq(
          field("left", $.expression),
          field("operator", "in"),
          field("right", $.expression),
        ),
      ),

    not_in: ($) =>
      prec.left(
        precedence.lookup,
        seq(
          field("left", $.expression),
          field("operator", alias(seq("not", "in"), "not in")),
          field("right", $.expression),
        ),
      ),

    literal: ($) => choice($.string, $.number, $.boolean),

    number: ($) => $.decimal,
    boolean: (_) => choice("true", "false"),
  },
});