test_parser.lua

local ltreesitter = require("ltreesitter")

-- Load TypeScript parser
local parser = ltreesitter.load("tree-sitter/typescript.so", "typescript")
print("Parser loaded:", parser)

local source = [[
function login(username, password) {
    const user = fetchUser(username);
    if (user.isValid(password)) {
        return createSession(user);
    }
    throw new Error("Invalid credentials");
}

const logout = () => {
    clearSession();
    redirect("/home");
};
]]

local tree = parser:parse_string(source)
print("Tree root:", tree:root():type())

-- Test query: find function declarations
local q = parser:query("(function_declaration name: (identifier) @name) @def")
print("\nFunction declarations:")
for match in q:match(tree:root()) do
    print("  Match id:", match.id)
    for name, node in pairs(match.captures) do
        print("    capture:", name, "type:", node:type(), "source:", node:source():sub(1, 50))
    end
end

-- Test query: find call expressions
local q2 = parser:query("(call_expression function: (identifier) @call_name) @call_expr")
print("\nCall expressions:")
for match in q2:match(tree:root()) do
    for name, node in pairs(match.captures) do
        print("  capture:", name, "type:", node:type(), "text:", node:source():sub(1, 50))
    end
end

-- Test: find arrow functions
local q3 = parser:query([[
(variable_declarator
    name: (identifier) @name
    value: (arrow_function) @def)
]])
print("\nArrow functions:")
for match in q3:match(tree:root()) do
    for name, node in pairs(match.captures) do
        print("  capture:", name, "type:", node:type(), "text:", node:source():sub(1, 50))
    end
end

-- Test node position API
print("\nNode position API:")
local root = tree:root()
local first_child = root:child(0)
print("  start_point:", first_child:start_point())
print("  end_point:", first_child:end_point())
print("  start_byte:", first_child:start_byte())
print("  end_byte:", first_child:end_byte())
print("  named_child_count:", first_child:named_child_count())

print("\nSUCCESS: All API tests passed!")