diff --git a/src/QueryCache.zig b/src/QueryCache.zig index a9dad05..816d8bc 100644 --- a/src/QueryCache.zig +++ b/src/QueryCache.zig @@ -24,7 +24,8 @@ const CacheEntry = struct { query: ?*Query, query_arena: ?*std.heap.ArenaAllocator, query_type: QueryType, - file_type: *const FileType, + file_type_name: []const u8, + lang_fn: FileType.LangFn, fn destroy(self: *@This(), allocator: std.mem.Allocator) void { if (self.query_arena) |a| { @@ -101,7 +102,7 @@ fn release_cache_entry_hash_map(allocator: std.mem.Allocator, hash_map: *std.Str hash_map.deinit(allocator); } -fn get_cache_entry(self: *Self, file_type: *const FileType, comptime query_type: QueryType) CacheError!*CacheEntry { +fn get_cache_entry(self: *Self, file_type: FileType, comptime query_type: QueryType) CacheError!*CacheEntry { if (self.mutex) |*mtx| mtx.lock(); defer if (self.mutex) |*mtx| mtx.unlock(); @@ -119,7 +120,8 @@ fn get_cache_entry(self: *Self, file_type: *const FileType, comptime query_type: .query = null, .query_arena = null, .mutex = if (self.mutex) |_| .{} else null, - .file_type = file_type, + .lang_fn = file_type.lang_fn, + .file_type_name = file_type.name, .query_type = query_type, }; entry_.value_ptr.* = q; @@ -133,8 +135,8 @@ fn get_cached_query(self: *Self, entry: *CacheEntry) Error!?*Query { defer if (entry.mutex) |*mtx| mtx.unlock(); return if (entry.query) |query| query else blk: { - const lang = entry.file_type.lang_fn() orelse std.debug.panic("tree-sitter parser function failed for language: {s}", .{entry.file_type.name}); - const queries = FileType.queries.get(entry.file_type.name) orelse return null; + const lang = entry.lang_fn() orelse std.debug.panic("tree-sitter parser function failed for language: {s}", .{entry.file_type_name}); + const queries = FileType.queries.get(entry.file_type_name) orelse return null; const query_bin = switch (entry.query_type) { .highlights => queries.highlights_bin, .errors => queries.errors_bin, @@ -166,7 +168,7 @@ fn ReturnType(comptime query_type: QueryType) type { }; } -pub fn get(self: *Self, file_type: *const FileType, comptime query_type: QueryType) Error!ReturnType(query_type) { +pub fn get(self: *Self, file_type: FileType, comptime query_type: QueryType) Error!ReturnType(query_type) { const query = try self.get_cached_query(try self.get_cache_entry(file_type, query_type)); self.add_ref_locked(); return switch (@typeInfo(ReturnType(query_type))) { diff --git a/src/file_type.zig b/src/file_type.zig index 4c45c64..c59a922 100644 --- a/src/file_type.zig +++ b/src/file_type.zig @@ -20,43 +20,45 @@ comment: []const u8, formatter: ?[]const []const u8, language_server: ?[]const []const u8, -pub fn get_by_name(name: []const u8) ?*const FileType { - for (file_types) |*file_type| - if (std.mem.eql(u8, file_type.name, name)) +pub fn get_by_name_static(name: []const u8) ?FileType { + return FileType.static_file_types.get(name); +} + +pub fn get_all() []const FileType { + return FileType.static_file_types.values(); +} + +pub fn guess_static(file_path: ?[]const u8, content: []const u8) ?FileType { + if (guess_first_line_static(content)) |ft| return ft; + for (static_file_types.values()) |*file_type| + if (file_path) |fp| if (match_file_type(file_type.extensions, fp)) return file_type; return null; } -pub fn guess(file_path: ?[]const u8, content: []const u8) ?*const FileType { - if (guess_first_line(content)) |ft| return ft; - for (file_types) |*file_type| - if (file_path) |fp| if (match_file_type(file_type, fp)) - return file_type; - return null; -} - -fn guess_first_line(content: []const u8) ?*const FileType { +fn guess_first_line_static(content: []const u8) ?FileType { const first_line = if (std.mem.indexOf(u8, content, "\n")) |pos| content[0..pos] else content; - for (file_types) |*file_type| + for (static_file_types) |*file_type| if (file_type.first_line_matches) |match| - if (match_first_line(match, first_line)) + if (match_first_line(match.prefix, match.content, first_line)) return file_type; return null; } -fn match_first_line(match: FirstLineMatch, first_line: []const u8) bool { - if (match.prefix) |prefix| +pub fn match_first_line(match_prefix: ?[]const u8, match_content: ?[]const u8, first_line: []const u8) bool { + if (match_prefix == null and match_content == null) return false; + if (match_prefix) |prefix| if (prefix.len > first_line.len or !std.mem.eql(u8, first_line[0..prefix.len], prefix)) return false; - if (match.content) |content| + if (match_content) |content| if (std.mem.indexOf(u8, first_line, content)) |_| {} else return false; return true; } -fn match_file_type(file_type: *const FileType, file_path: []const u8) bool { +pub fn match_file_type(extensions: []const []const u8, file_path: []const u8) bool { const basename = std.fs.path.basename(file_path); const extension = std.fs.path.extension(file_path); - return for (file_type.extensions) |ext| { + return for (extensions) |ext| { if (ext.len == basename.len and std.mem.eql(u8, ext, basename)) return true; if (extension.len > 0 and ext.len == extension.len - 1 and std.mem.eql(u8, ext, extension[1..])) @@ -85,14 +87,15 @@ fn ft_func_name(comptime lang: []const u8) []const u8 { return &func_name; } -const LangFn = *const fn () callconv(.C) ?*const treez.Language; +pub const LangFn = *const fn () callconv(.C) ?*const treez.Language; pub const FirstLineMatch = struct { prefix: ?[]const u8 = null, content: ?[]const u8 = null, }; -pub const file_types = load_file_types(@import("file_types.zig")); +const static_file_type_list = load_file_types(@import("file_types.zig")); +const static_file_types = std.StaticStringMap(FileType).initComptime(static_file_type_list); fn vec(comptime args: anytype) []const []const u8 { var cmd: []const []const u8 = &[_][]const u8{}; @@ -102,7 +105,9 @@ fn vec(comptime args: anytype) []const []const u8 { return cmd; } -fn load_file_types(comptime Namespace: type) []const FileType { +const ListEntry = struct { []const u8, FileType }; + +fn load_file_types(comptime Namespace: type) []const ListEntry { comptime switch (@typeInfo(Namespace)) { .@"struct" => |info| { var count = 0; @@ -110,12 +115,12 @@ fn load_file_types(comptime Namespace: type) []const FileType { // @compileLog(decl.name, @TypeOf(@field(Namespace, decl.name))); count += 1; } - var construct_types: [count]FileType = undefined; + var construct_types: [count]ListEntry = undefined; var i = 0; for (info.decls) |decl| { const lang = decl.name; const args = @field(Namespace, lang); - construct_types[i] = .{ + construct_types[i] = .{ lang, .{ .color = if (@hasField(@TypeOf(args), "color")) args.color else 0xffffff, .icon = if (@hasField(@TypeOf(args), "icon")) args.icon else "󱀫", .name = lang, @@ -126,7 +131,7 @@ fn load_file_types(comptime Namespace: type) []const FileType { .first_line_matches = if (@hasField(@TypeOf(args), "first_line_matches")) args.first_line_matches else null, .formatter = if (@hasField(@TypeOf(args), "formatter")) vec(args.formatter) else null, .language_server = if (@hasField(@TypeOf(args), "language_server")) vec(args.language_server) else null, - }; + } }; i += 1; } const types = construct_types; @@ -142,7 +147,7 @@ pub const FileTypeQueries = struct { injections_bin: ?[]const u8, }; -pub const queries = std.static_string_map.StaticStringMap(FileTypeQueries).initComptime(load_queries()); +pub const queries = std.StaticStringMap(FileTypeQueries).initComptime(load_queries()); fn load_queries() []const struct { []const u8, FileTypeQueries } { if (!build_options.use_tree_sitter) return &.{}; diff --git a/src/file_types.zig b/src/file_types.zig index e19b5e5..efb8ca2 100644 --- a/src/file_types.zig +++ b/src/file_types.zig @@ -88,6 +88,7 @@ pub const css = .{ .icon = "󰌜", .extensions = .{"css"}, .comment = "//", + .language_server = .{ "vscode-css-language-server", "--stdio" }, }; pub const diff = .{ @@ -230,15 +231,16 @@ pub const javascript = .{ .extensions = .{"js"}, .comment = "//", .injections = "tree-sitter-javascript/queries/injections.scm", - .language_server = .{ "deno", "lsp" }, + .language_server = .{ "typescript-language-server", "--stdio" }, + .formatter = .{ "prettier", "--parser", "typescript" }, }; pub const json = .{ .description = "JSON", .extensions = .{"json"}, .comment = "//", - .language_server = .{ "deno", "lsp" }, - .formatter = .{ "hjson", "-j" }, + .language_server = .{ "vscode-json-language-server", "--stdio" }, + .formatter = .{ "prettier", "--parser", "json" }, }; pub const julia = .{ @@ -293,7 +295,8 @@ pub const markdown = .{ .comment = "