;; Forked from https://github.com/nvim-treesitter/nvim-treesitter/blob/master/queries/go/highlights.scm
;; Licensed under the Apache License 2.0
; Identifiers
(type_identifier) @type

(type_spec
  name: (type_identifier) @type.definition
)

(field_identifier) @property

(identifier) @variable

(package_identifier) @namespace

(parameter_declaration
  (identifier) @parameter
)

(variadic_parameter_declaration
  (identifier) @parameter
)

(label_name) @label

(
  (identifier) @constant
  (#eq? @constant "_")
)

(const_spec
  name: (identifier) @constant
)

; Function calls
(call_expression
  function: (identifier) @function.call
)

(call_expression
  function: (selector_expression
    field: (field_identifier) @method.call
  )
)

; Function definitions
(function_declaration
  name: (identifier) @function
)

(method_declaration
  name: (field_identifier) @method
)

(method_spec
  name: (field_identifier) @method
)

; Constructors
(
  (call_expression
    (identifier) @constructor
  )
  (#lua-match? @constructor "^[nN]ew.+$")
)

(
  (call_expression
    (identifier) @constructor
  )
  (#lua-match? @constructor "^[mM]ake.+$")
)

; Operators
[
  "--"
  "-"
  "-="
  ":="
  "!"
  "!="
  "..."
  "*"
  "*"
  "*="
  "/"
  "/="
  "&"
  "&&"
  "&="
  "&^"
  "&^="
  "%"
  "%="
  "^"
  "^="
  "+"
  "++"
  "+="
  "<-"
  "<"
  "<<"
  "<<="
  "<="
  "="
  "=="
  ">"
  ">="
  ">>"
  ">>="
  "|"
  "|="
  "||"
  ; crates.io skip
  "~"
] @operator

; Keywords
[
  "break"
  "const"
  "continue"
  "default"
  "defer"
  "goto"
  "interface"
  "range"
  "select"
  "struct"
  "type"
  "var"
  "fallthrough"
] @keyword

"func" @keyword.function

"return" @keyword.return

"go" @keyword.coroutine

"for" @repeat

[
  "import"
  "package"
] @include

[
  "else"
  "case"
  "switch"
  "if"
] @conditional

;; Builtin types
[
  "chan"
  "map"
] @type.builtin

(
  (type_identifier) @type.builtin
  (#any-of?
    @type.builtin
    "any"
    "bool"
    "byte"
    "comparable"
    "complex128"
    "complex64"
    "error"
    "float32"
    "float64"
    "int"
    "int16"
    "int32"
    "int64"
    "int8"
    "rune"
    "string"
    "uint"
    "uint16"
    "uint32"
    "uint64"
    "uint8"
    "uintptr"
  )
)

;; Builtin functions
(
  (identifier) @function.builtin
  (#any-of?
    @function.builtin
    "append"
    "cap"
    "clear"
    "close"
    "complex"
    "copy"
    "delete"
    "imag"
    "len"
    "make"
    "new"
    "panic"
    "print"
    "println"
    "real"
    "recover"
  )
)

; Delimiters
"." @punctuation.delimiter

"," @punctuation.delimiter

":" @punctuation.delimiter

";" @punctuation.delimiter

"(" @punctuation.bracket

")" @punctuation.bracket

"{" @punctuation.bracket

"}" @punctuation.bracket

"[" @punctuation.bracket

"]" @punctuation.bracket

; Literals
(interpreted_string_literal) @string

(raw_string_literal) @string @spell

(rune_literal) @string

(escape_sequence) @string.escape

(int_literal) @number

(float_literal) @float

(imaginary_literal) @number

[
  (true)
  (false)
] @boolean

[
  (nil)
  (iota)
] @constant.builtin

(keyed_element
  .
  (literal_element
    (identifier) @field
  )
)

(field_declaration
  name: (field_identifier) @field
)

; Comments
(comment) @comment @spell

;; Doc Comments
(source_file
  .
  (comment)+ @comment.documentation
)

(source_file
  (comment)+ @comment.documentation
  .
  (const_declaration)
)

(source_file
  (comment)+ @comment.documentation
  .
  (function_declaration)
)

(source_file
  (comment)+ @comment.documentation
  .
  (type_declaration)
)

(source_file
  (comment)+ @comment.documentation
  .
  (var_declaration)
)

; Errors
(ERROR) @error

; Spell
(
  (interpreted_string_literal) @spell
  (#not-has-parent? @spell import_spec)
)