From 4683835fbd4f495f223c5c4b88d354e074b398c4 Mon Sep 17 00:00:00 2001
From: ray-x <rayx.cn@gmail.com>
Date: Thu, 29 Feb 2024 09:55:04 +1100
Subject: [PATCH] mockgen updates

---
 lua/go/mockgen.lua            | 72 +++++++++++++++++------------------
 lua/tests/go_mockgen_spec.lua |  4 +-
 2 files changed, 38 insertions(+), 38 deletions(-)

diff --git a/lua/go/mockgen.lua b/lua/go/mockgen.lua
index 6bc5794..fed065e 100644
--- a/lua/go/mockgen.lua
+++ b/lua/go/mockgen.lua
@@ -1,41 +1,41 @@
 -- local ts_utils = require 'nvim-treesitter.ts_utils'
-local utils = require("go.utils")
+local utils = require('go.utils')
 local log = utils.log
 local vfn = vim.fn
-local mockgen = "mockgen" -- GoMock f *Foo io.Writer
+local mockgen = 'mockgen' -- GoMock f *Foo io.Writer
 
 -- use ts to get name
 local function get_interface_name()
-  local name = require("go.ts.go").get_interface_node_at_pos()
+  local name = require('go.ts.go').get_interface_node_at_pos()
   if name == nil then
     return nil
   end
   utils.log(name)
   if name == nil then
-    return ""
+    return ''
   end
   local node_name = name.name
   -- let move the cursor to end of line of struct name
   local dim = name.dim.e
   -- let move cursor
   local r, c = dim.r, dim.c
-  utils.log("move cusror to ", r, c)
+  utils.log('move cusror to ', r, c)
   vim.api.nvim_win_set_cursor(0, { r, c })
   return node_name
 end
 
 local run = function(opts)
-  require("go.install").install(mockgen)
+  require('go.install').install(mockgen)
 
   local long_opts = {
-    package = "p",
-    source = "s",
-    destination = "d",
-    interface = "i",
+    package = 'p',
+    source = 's',
+    destination = 'd',
+    interface = 'i',
   }
 
-  local getopt = require("go.alt_getopt")
-  local short_opts = "p:d:i:s"
+  local getopt = require('go.alt_getopt')
+  local short_opts = 'p:d:i:s'
   local args = opts.fargs or {}
   log(args)
 
@@ -43,52 +43,52 @@ local run = function(opts)
   local mockgen_cmd = { mockgen }
   utils.log(arg, reminder)
 
-  local sep = require("go.utils").sep()
+  local sep = require('go.utils').sep()
 
   local ifname = get_interface_name()
 
-  if optarg["i"] ~= nil and #optarg["i"] > 0 then
-    ifname = optarg["i"]
+  if optarg['i'] ~= nil and #optarg['i'] > 0 then
+    ifname = optarg['i']
   end
 
-  if optarg["s"] ~= nil then
-    ifname = ""
+  if optarg['s'] ~= nil then
+    ifname = ''
   end
   local fpath = utils.rel_path(true) -- rel/path/only
   log(fpath, mockgen_cmd)
-  local sname = vfn.expand("%:t") -- name.go only
+  local sname = vfn.expand('%:t') -- name.go only
 
-  if fpath ~= "" then
+  if fpath ~= '' then
     fpath = fpath .. sep
   end
 
-  if ifname == "" or ifname == nil then
+  local pkgname = optarg['p'] or 'mocks'
+  table.insert(mockgen_cmd, '-package')
+  table.insert(mockgen_cmd, pkgname)
+
+  local dname = fpath .. pkgname .. sep .. 'mock_' .. sname
+  table.insert(mockgen_cmd, '-destination')
+  table.insert(mockgen_cmd, dname)
+
+  if ifname == '' or ifname == nil then
     -- source mode default
-    table.insert(mockgen_cmd, "-source")
+    table.insert(mockgen_cmd, '-source')
     table.insert(mockgen_cmd, fpath .. sname)
   else
-    log("interface ", ifname)
+    log('interface ', ifname)
     -- need to get the import path
     local bufnr = vim.api.nvim_get_current_buf()
 
-    local pkg = require("go.package").pkg_from_path(nil, bufnr)
-    if pkg ~= nil and type(pkg) == "table" and pkg[1] then
+    local pkg = require('go.package').pkg_from_path(nil, bufnr)
+    if pkg ~= nil and type(pkg) == 'table' and pkg[1] then
       table.insert(mockgen_cmd, pkg[1])
     else
-      utils.notify("no package found, using .")
+      utils.notify('no package found, using .')
       table.insert(mockgen_cmd, '.')
     end
     table.insert(mockgen_cmd, ifname)
   end
 
-  local pkgname = optarg["p"] or "mocks"
-  table.insert(mockgen_cmd, "-package")
-  table.insert(mockgen_cmd, pkgname)
-
-  local dname = fpath .. pkgname .. sep .. "mock_" .. sname
-  table.insert(mockgen_cmd, "-destination")
-  table.insert(mockgen_cmd, dname)
-
   log(mockgen_cmd)
 
   utils.log(mockgen_cmd)
@@ -100,18 +100,18 @@ local run = function(opts)
         -- utils.warn("mockgen failed" .. vim.inspect(data))
         return
       end
-      data = vim.split(data, "\n")
+      data = vim.split(data, '\n')
       data = utils.handle_job_data(data)
       if not data then
         return
       end
       --
       vim.schedule(function()
-        utils.info(vfn.join(mockgen_cmd, " ") .. " finished " .. vfn.join(data, " "))
+        utils.info(vfn.join(mockgen_cmd, ' ') .. ' finished ' .. vfn.join(data, ' '))
       end)
     end,
   }
-  local runner = require("go.runner")
+  local runner = require('go.runner')
   runner.run(mockgen_cmd, mock_opts)
   return mockgen_cmd
 end
diff --git a/lua/tests/go_mockgen_spec.lua b/lua/tests/go_mockgen_spec.lua
index 3062b60..a2eeb1f 100644
--- a/lua/tests/go_mockgen_spec.lua
+++ b/lua/tests/go_mockgen_spec.lua
@@ -31,12 +31,12 @@ describe("should run mockgen", function()
 
     local expected_cmd = {
       "mockgen",
-      "-source",
-      "interfaces.go",
       "-package",
       "mocks",
       "-destination",
       "mocks/mock_interfaces.go",
+      "-source",
+      "interfaces.go",
     }
     eq(cmd, expected_cmd)
   end)