From 2505fa3a5499c943944153284e7ae1b5c3a59a02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20Wo=C5=BAniak?= <y0rune@aol.com>
Date: Wed, 21 Sep 2022 20:30:34 +0200
Subject: [PATCH] Added the custom function for gofile
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Marcin Woźniak <y0rune@aol.com>
---
 .config/nvim/init.vim | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/.config/nvim/init.vim b/.config/nvim/init.vim
index 4f58bf9..b3cb014 100644
--- a/.config/nvim/init.vim
+++ b/.config/nvim/init.vim
@@ -533,6 +533,26 @@ map <F4> :setlocal spell! spelllang=pl<CR>
 :command! Q q
 :command! Wq wq
 
+""""""""""""""""""""""""""""""""
+" Custom functions
+""""""""""""""""""""""""""""""""
+lua <<EOF
+  function go_org_imports(wait_ms)
+    local params = vim.lsp.util.make_range_params()
+    params.context = {only = {"source.organizeImports"}}
+    local result = vim.lsp.buf_request_sync(0, "textDocument/codeAction", params, wait_ms)
+    for cid, res in pairs(result or {}) do
+      for _, r in pairs(res.result or {}) do
+        if r.edit then
+          local enc = (vim.lsp.get_client_by_id(cid) or {}).offset_encoding or "utf-16"
+          vim.lsp.util.apply_workspace_edit(r.edit, enc)
+        end
+      end
+    end
+  end
+EOF
+
+
 """"""""""""""""""""""""""""""""
 " Files
 """"""""""""""""""""""""""""""""
@@ -575,6 +595,7 @@ autocmd BufRead,BufNewFile *.yaml let g:indentLine_char = '⦙'
 
 " Go
 autocmd BufRead *.go set noexpandtab
+autocmd BufWritePre *.go lua go_org_imports()
 
 " Conf
 au BufNewFile,BufRead *.conf setfiletype conf