Skip to content

Commit d1fb0c1

Browse files
committed
feat: Add support for Powershell/ps1
1 parent e1de22f commit d1fb0c1

File tree

2 files changed

+28
-14
lines changed

2 files changed

+28
-14
lines changed

lua/debugprint/filetypes.lua

+6
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,12 @@ return {
8686
mid_var = "$",
8787
right_var = '\\n");',
8888
},
89+
["ps1"] = {
90+
left = 'Write-Error "',
91+
right = '"',
92+
mid_var = "$",
93+
right_var = '"',
94+
},
8995
["python"] = {
9096
left = 'print(f"',
9197
right = '", file=sys.stderr)',

lua/debugprint/utils.lua

+22-14
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,30 @@
11
local M = {}
22

33
local get_node_at_cursor = function()
4-
if vim.treesitter.get_node then
5-
-- Supported as of NeoVim 0.9?
6-
return vim.treesitter.get_node()
7-
else
8-
local function requiref(module)
9-
require(module)
10-
end
11-
12-
local ts_utils_test = pcall(requiref, "nvim-treesitter.ts_utils")
13-
14-
if not ts_utils_test then
15-
return nil
4+
local success, is_node = pcall(vim.treesitter.get_node)
5+
6+
-- This will fail if this language is not supported by Treesitter, e.g.
7+
-- Powershell/ps1
8+
if success then
9+
if is_node then
10+
-- Supported as of NeoVim 0.9?
11+
return vim.treesitter.get_node()
1612
else
17-
local ts_utils = require("nvim-treesitter.ts_utils")
18-
return ts_utils.get_node_at_cursor()
13+
local function requiref(module)
14+
require(module)
15+
end
16+
17+
local ts_utils_test = pcall(requiref, "nvim-treesitter.ts_utils")
18+
19+
if not ts_utils_test then
20+
return nil
21+
else
22+
local ts_utils = require("nvim-treesitter.ts_utils")
23+
return ts_utils.get_node_at_cursor()
24+
end
1925
end
26+
else
27+
return nil
2028
end
2129
end
2230

0 commit comments

Comments
 (0)