;;; lang/go/config.el -*- lexical-binding: t; -*-

;;
;;; Packages

(defun +go-common-config (mode)
  (set-docsets! mode "Go")
  (set-repl-handler! mode #'gorepl-run)
  (set-lookup-handlers! mode
    :documentation #'godoc-at-point)

  (when (modulep! +lsp)
    (add-hook (intern (format "%s-local-vars-hook" mode)) #'lsp! 'append))

  (map! :map ,(symbol-value (intern (format "%s-map" mode)))
        :localleader
        "a" #'go-tag-add
        "d" #'go-tag-remove
        "e" #'+go/play-buffer-or-region
        "i" #'go-goto-imports      ; Go to imports
        (:prefix ("h" . "help")
          "." #'godoc-at-point)    ; Lookup in godoc
        (:prefix ("ri" . "imports")
          "a" #'go-import-add)
        (:prefix ("b" . "build")
          :desc "go run ." "r" (cmd! (compile "go run ."))
          :desc "go build" "b" (cmd! (compile "go build"))
          :desc "go clean" "c" (cmd! (compile "go clean")))
        (:prefix ("g" . "generate")
          "f" #'+go/generate-file
          "d" #'+go/generate-dir
          "a" #'+go/generate-all)
        (:prefix ("t" . "test")
          "t" #'+go/test-rerun
          "a" #'+go/test-all
          "s" #'+go/test-single
          "n" #'+go/test-nested
          "f" #'+go/test-file
          "g" #'go-gen-test-dwim
          "G" #'go-gen-test-all
          "e" #'go-gen-test-exported
          (:prefix ("b" . "bench")
            "s" #'+go/bench-single
            "a" #'+go/bench-all))))


(after! go-mode
  (+go-common-config 'go-mode))


(use-package! go-ts-mode
  :when (modulep! +tree-sitter)
  :when (fboundp 'go-ts-mode) ; 31.1+ only
  :defer t
  :init
  (set-tree-sitter! 'go-mode 'go-ts-mode
    '((go :url "https://github.com/tree-sitter/tree-sitter-go" :ref "v0.23.4")
      (gomod :url "https://github.com/camdencheek/tree-sitter-go-mod" :ref "v1.1.0")
      (gowork :url "https://github.com/omertuc/tree-sitter-go-work")))
  :config
  (+go-common-config 'go-ts-mode))


(use-package! gorepl-mode
  :commands gorepl-run-load-current-file)


(use-package! flycheck-golangci-lint
  :when (modulep! :checkers syntax -flymake)
  :hook (go-mode . flycheck-golangci-lint-setup)
  :hook (go-ts-mode . flycheck-golangci-lint-setup))