Files
lua-config-vim/current_config/lua/alpha/php/gateway.lua
2023-12-13 00:27:19 +03:00

138 lines
4.2 KiB
Lua

local pickers = require "telescope.pickers"
local finders = require "telescope.finders"
local conf = require("telescope.config").values
local action_state = require "telescope.actions.state"
local action_set = require "telescope.actions.set"
local actions = require "telescope.actions"
local get_node_text = vim.treesitter.get_node_text
local gateway = {}
local get_definition_buffer = function(type)
local uri = vim.uri_from_fname(vim.fn.getcwd() .. "/config/graphql/_" .. type .. ".yaml")
local buff = vim.uri_to_bufnr(uri)
vim.fn.bufload(buff)
return buff
end
local get_all_definitions = function()
local mutations_definitions = gateway.get_graphql_definitions(get_definition_buffer "mutation")
local queries_definitions = gateway.get_graphql_definitions(get_definition_buffer "query")
for _, definition in ipairs(queries_definitions) do
table.insert(mutations_definitions, definition)
end
return mutations_definitions
end
-- "'@=mutation(\"App\\\\GraphQL\\\\Mutations\\\\Mutation::action\", ...)'",
local open_resolver = function(resolver)
-- need to parse to get this values
local args = string.match(resolver, "'@=mutation%((.+)%)'")
if args == nil then
args = string.match(resolver, "'@=query%((.+)%)'")
end
if args == nil then
return
end
local args_tbl = vim.split(args, ", ")
local fqn = string.match(args_tbl[1], '"(.+)"')
fqn = string.gsub(fqn, "\\\\", "\\")
-- P(fqn)
require("alpha.php.lsp").find(fqn, "phpactor")
end
--- @param bufnr integer Buff id
gateway.get_graphql_definitions = function(bufnr)
vim.treesitter.query.set(
"yaml",
"GraphQL_endpoints",
[[
(document (block_node (block_mapping (block_mapping_pair
value: (block_node (block_mapping (block_mapping_pair
value: (block_node (block_mapping (block_mapping_pair
value: (block_node (block_mapping (block_mapping_pair
key: (flow_node) @mutation
value: (block_node (block_mapping (block_mapping_pair
key: (flow_node) @resolve (#eq? @resolve "resolve")
value: (flow_node) @resolver
)))
)))
)))
)))
))))
]]
)
local query = vim.treesitter.query.get("yaml", "GraphQL_endpoints")
local elements = {}
local parsers = require "nvim-treesitter.parsers"
local parser = parsers.get_parser(bufnr)
local tree = parser:parse()[1]
for id, node in query:iter_captures(tree:root(), bufnr) do
if query.captures[id] == "mutation" then
table.insert(elements, {
value = get_node_text(node, bufnr),
lnum = node:start() + 1,
buffer = bufnr,
})
elseif query.captures[id] == "resolver" then
elements[#elements].resolver = get_node_text(node, bufnr)
end
end
return elements
end
--- gets the graphql function defined on a given buffer
--- @param opts table: options to pass to the picker
gateway.graphql_definitions = function(opts)
opts = opts or {}
pickers
.new(opts, {
promt_title = "Queries",
finder = finders.new_table {
results = get_all_definitions(),
entry_maker = function(entry)
-- buffer = 43,
-- lnum = 82,
-- resolver = "'@=mutation(\"App\\\\GraphQL\\\\Mutation\\\\Mutation::action\", ...)'",
-- value = "deleteVehicleImage"
return {
value = entry,
display = entry.value,
ordinal = entry.value,
path = vim.api.nvim_buf_get_name(entry.buffer),
lnum = entry.lnum,
resolver = entry.resolver,
}
end,
},
sorter = conf.generic_sorter(opts),
previewer = conf.grep_previewer(opts),
attach_mappings = function(prompt_bufnr, map)
action_set.select:enhance {
post = function()
local selection = action_state.get_selected_entry()
vim.api.nvim_win_set_cursor(0, { selection.lnum, 0 })
end,
}
map("i", "<c-o>", function()
actions.close(prompt_bufnr)
local selection = action_state.get_selected_entry()
open_resolver(selection.resolver)
end)
return true
end,
})
:find()
end
return gateway