aboutsummaryrefslogtreecommitdiffstats
path: root/lua/alpha/snips/ft/go.lua
diff options
context:
space:
mode:
authorAriel D'Alessandro <dalessandro.ariel@gmail.com>2023-01-31 09:44:13 +0100
committerAriel D'Alessandro <dalessandro.ariel@gmail.com>2023-01-31 09:44:13 +0100
commit669561e21073e6efafddda25ba3ace69fb43819e (patch)
treef94ebd60ad4cab480b5cbcb7e65dd2060696c339 /lua/alpha/snips/ft/go.lua
parent5044c70e9f62cdd06ee91e754423b07c1c18c12d (diff)
downloadnvim-conf-669561e21073e6efafddda25ba3ace69fb43819e.tar.gz
nvim-conf-669561e21073e6efafddda25ba3ace69fb43819e.zip
feat: apply stylua
Diffstat (limited to 'lua/alpha/snips/ft/go.lua')
-rw-r--r--lua/alpha/snips/ft/go.lua190
1 files changed, 95 insertions, 95 deletions
diff --git a/lua/alpha/snips/ft/go.lua b/lua/alpha/snips/ft/go.lua
index 6038d52..58c6bfc 100644
--- a/lua/alpha/snips/ft/go.lua
+++ b/lua/alpha/snips/ft/go.lua
@@ -1,4 +1,4 @@
-local ls = require("luasnip")
+local ls = require "luasnip"
local snippet_from_nodes = ls.sn
local c = ls.choice_node
@@ -8,15 +8,15 @@ local i = ls.insert_node
local rep = require("luasnip.extras").rep
local fmt = require("luasnip.extras.fmt").fmt
-local ts_locals = require("nvim-treesitter.locals")
-local ts_utils = require("nvim-treesitter.ts_utils")
+local ts_locals = require "nvim-treesitter.locals"
+local ts_utils = require "nvim-treesitter.ts_utils"
local get_node_text = vim.treesitter.get_node_text
vim.treesitter.set_query(
- "go",
- "LuaSnip_Result",
- [[
+ "go",
+ "LuaSnip_Result",
+ [[
[
(method_declaration result: (_) @id)
(function_declaration result: (_) @id)
@@ -26,107 +26,107 @@ vim.treesitter.set_query(
)
local transform = function(text, info)
- if text == "int" then
- return t("0")
- elseif text == "error" then
- if info then
- info.index = info.index + 1
-
- return c(info.index, {
- t(string.format('fmt.Errorf("%s %%w", %s)', info.func_name, info.err_name)),
- t(info.err_name),
- })
- else
- return t("err")
- end
- elseif text == "bool" then
- return t("false")
- elseif text == "string" then
- return t('""')
- elseif string.find(text, "*", 1, true) then
- return t("nil")
- end
-
- return t(text)
+ if text == "int" then
+ return t "0"
+ elseif text == "error" then
+ if info then
+ info.index = info.index + 1
+
+ return c(info.index, {
+ t(string.format('fmt.Errorf("%s %%w", %s)', info.func_name, info.err_name)),
+ t(info.err_name),
+ })
+ else
+ return t "err"
+ end
+ elseif text == "bool" then
+ return t "false"
+ elseif text == "string" then
+ return t '""'
+ elseif string.find(text, "*", 1, true) then
+ return t "nil"
+ end
+
+ return t(text)
end
local handlers = {
- ["parameter_list"] = function(node, info)
- local result = {}
-
- local count = node:named_child_count()
- for idx = 0, count - 1 do
- table.insert(result, transform(get_node_text(node:named_child(idx), 0), info))
- if idx ~= count - 1 then
- table.insert(result, t({ ", " }))
- end
- end
-
- return result
- end,
-
- ["type_identifier"] = function(node, info)
- local text = get_node_text(node, 0)
- return { transform(text, info) }
- end,
+ ["parameter_list"] = function(node, info)
+ local result = {}
+
+ local count = node:named_child_count()
+ for idx = 0, count - 1 do
+ table.insert(result, transform(get_node_text(node:named_child(idx), 0), info))
+ if idx ~= count - 1 then
+ table.insert(result, t { ", " })
+ end
+ end
+
+ return result
+ end,
+
+ ["type_identifier"] = function(node, info)
+ local text = get_node_text(node, 0)
+ return { transform(text, info) }
+ end,
}
local function go_result_type(info)
- local cursor_node = ts_utils.get_node_at_cursor()
- local scope = ts_locals.get_scope_tree(cursor_node, 0)
-
- local function_node
- for _, v in ipairs(scope) do
- if v:type() == "function_declaration" or v:type() == "method_declaration" or v:type() == "func_literal" then
- function_node = v
- break
- end
- end
-
- local query = vim.treesitter.get_query("go", "LuaSnip_Result")
- for _, node in query:iter_captures(function_node, 0) do
- if handlers[node:type()] then
- return handlers[node:type()](node, info)
- end
- end
+ local cursor_node = ts_utils.get_node_at_cursor()
+ local scope = ts_locals.get_scope_tree(cursor_node, 0)
+
+ local function_node
+ for _, v in ipairs(scope) do
+ if v:type() == "function_declaration" or v:type() == "method_declaration" or v:type() == "func_literal" then
+ function_node = v
+ break
+ end
+ end
+
+ local query = vim.treesitter.get_query("go", "LuaSnip_Result")
+ for _, node in query:iter_captures(function_node, 0) do
+ if handlers[node:type()] then
+ return handlers[node:type()](node, info)
+ end
+ end
end
local go_ret_vals = function(args)
- local info = { index = 0, err_name = args[1][1], func_name = args[2][1] }
- return snippet_from_nodes(nil, go_result_type(info))
+ local info = { index = 0, err_name = args[1][1], func_name = args[2][1] }
+ return snippet_from_nodes(nil, go_result_type(info))
end
local M = {
- main = {
- t({ "func main() {", "\t" }),
- i(0),
- t({ "", "}" }),
- },
- ef = {
- i(1, { "val" }),
- t(", err := "),
- i(2, { "f" }),
- t("("),
- i(3),
- t(")"),
- i(0),
- },
- efi = {
- i(1, { "val" }),
- ", ",
- i(2, { "err" }),
- " := ",
- i(3, { "f" }),
- "(",
- i(4),
- ")",
- t({ "", "if " }),
- rep(2),
- t({ " != nil {", "\treturn " }),
- d(5, go_ret_vals, { 2, 3 }),
- t({ "", "}" }),
- i(0),
- },
+ main = {
+ t { "func main() {", "\t" },
+ i(0),
+ t { "", "}" },
+ },
+ ef = {
+ i(1, { "val" }),
+ t ", err := ",
+ i(2, { "f" }),
+ t "(",
+ i(3),
+ t ")",
+ i(0),
+ },
+ efi = {
+ i(1, { "val" }),
+ ", ",
+ i(2, { "err" }),
+ " := ",
+ i(3, { "f" }),
+ "(",
+ i(4),
+ ")",
+ t { "", "if " },
+ rep(2),
+ t { " != nil {", "\treturn " },
+ d(5, go_ret_vals, { 2, 3 }),
+ t { "", "}" },
+ i(0),
+ },
}
M.f = fmt("func {}({}) {} {{\n\t{}\n}}", { i(1, "name"), i(2), i(3), i(0) })