1 files changed, 49 insertions, 0 deletions
diff --git a/plugins/gofmt.lua b/plugins/gofmt.lua
new file mode 100644
index 0000000..11c90b9
--- /dev/null
+++ b/plugins/gofmt.lua
@@ -0,0 +1,49 @@
+local core = require "core"
+local command = require "core.command"
+local keymap = require "core.keymap"
+
+local function exec(cmd)
+  local fp = io.popen(cmd, "r")
+  local res = fp:read("*a")
+  local success = fp:close()
+  return res:gsub("%\n$", ""), success
+end
+
+local function get_cmd_text(cmd, doc)
+  local active_filename = doc and system.absolute_path(doc.filename or "")
+  return exec(string.format("%s %s", cmd, active_filename))
+end
+
+local function update_doc(cmd, doc)
+  local text, success = get_cmd_text(cmd, doc)
+  if success == nil then
+    local err_text = "Command '%s' not found in the system"
+    core.error(string.format(err_text, cmd))
+    return
+  end
+
+  local sel = { doc:get_selection() }
+  doc:remove(1, 1, math.huge, math.huge)
+  doc:insert(1, 1, text)
+  doc:set_selection(table.unpack(sel))
+end
+
+command.add("core.docview", {
+  ["gofmt:gofmt"] = function()
+    update_doc("gofmt", core.active_view.doc)
+  end,
+
+  ["gofmt:goimports"] = function()
+    update_doc("goimports", core.active_view.doc)
+  end,
+
+  ["gofmt:goreturns"] = function()
+    update_doc("goreturns", core.active_view.doc)
+  end,
+})
+
+keymap.add {
+  ["ctrl+i"] = "gofmt:gofmt",
+  ["ctrl+h"] = "gofmt:goimports",
+  ["ctrl+u"] = "gofmt:goreturns",
+}