Added --config-path option

This commit is contained in:
Alexandros Naskos 2021-05-03 16:07:53 +03:00
parent f8b9d3bfcb
commit 937471d27b
No known key found for this signature in database
GPG Key ID: 02BF2E72B0EA32D2

View File

@ -1195,17 +1195,17 @@ fn documentSymbol(arena: *std.heap.ArenaAllocator, id: types.RequestId, handle:
}); });
} }
fn loadConfig(folder_path: []const u8) ?Config { fn loadConfigFile(file_path: []const u8) ?Config {
var folder = std.fs.cwd().openDir(folder_path, .{}) catch return null; var file = std.fs.cwd().openFile(file_path, .{}) catch |err| {
defer folder.close();
const file_buf = folder.readFileAlloc(allocator, "zls.json", 0x1000000) catch |err| {
if (err != error.FileNotFound) if (err != error.FileNotFound)
logger.warn("Error while reading configuration file: {}", .{err}); logger.warn("Error while reading configuration file: {}", .{err});
return null; return null;
}; };
defer allocator.free(file_buf);
defer file.close();
const file_buf = file.readToEndAlloc(allocator, 0x1000000) catch return null;
defer allocator.free(file_buf);
@setEvalBranchQuota(2000); @setEvalBranchQuota(2000);
// TODO: Better errors? Doesn't seem like std.json can provide us positions or context. // TODO: Better errors? Doesn't seem like std.json can provide us positions or context.
var config = std.json.parse(Config, &std.json.TokenStream.init(file_buf), std.json.ParseOptions{ .allocator = allocator }) catch |err| { var config = std.json.parse(Config, &std.json.TokenStream.init(file_buf), std.json.ParseOptions{ .allocator = allocator }) catch |err| {
@ -1224,6 +1224,15 @@ fn loadConfig(folder_path: []const u8) ?Config {
return config; return config;
} }
fn loadConfigInFolder(folder_path: []const u8) ?Config {
const full_path = std.fs.path.resolve(allocator, &.{
folder_path,
"zls.json",
}) catch return null;
defer allocator.free(full_path);
return loadConfigFile(full_path);
}
fn initializeHandler(arena: *std.heap.ArenaAllocator, id: types.RequestId, req: requests.Initialize, config: Config) !void { fn initializeHandler(arena: *std.heap.ArenaAllocator, id: types.RequestId, req: requests.Initialize, config: Config) !void {
for (req.params.capabilities.offsetEncoding.value) |encoding| { for (req.params.capabilities.offsetEncoding.value) |encoding| {
if (std.mem.eql(u8, encoding, "utf-8")) { if (std.mem.eql(u8, encoding, "utf-8")) {
@ -1728,13 +1737,25 @@ pub fn main() anyerror!void {
const prog_name = try args_it.next(allocator) orelse @panic("Could not find self argument"); const prog_name = try args_it.next(allocator) orelse @panic("Could not find self argument");
allocator.free(prog_name); allocator.free(prog_name);
var config_path: ?[]const u8 = null;
var next_arg_config_path = false;
while (args_it.next(allocator)) |maybe_arg| { while (args_it.next(allocator)) |maybe_arg| {
const arg = try maybe_arg; const arg = try maybe_arg;
defer allocator.free(arg); defer allocator.free(arg);
if (next_arg_config_path) {
config_path = try allocator.dupe(u8, arg);
next_arg_config_path = false;
continue;
}
if (std.mem.eql(u8, arg, "--debug-log")) { if (std.mem.eql(u8, arg, "--debug-log")) {
actual_log_level = .debug; actual_log_level = .debug;
std.debug.print("Enabled debug logging\n", .{}); std.debug.print("Enabled debug logging\n", .{});
} else if (std.mem.eql(u8, arg, "config")) { } else if (std.mem.eql(u8, arg, "--config-path")) {
next_arg_config_path = true;
continue;
} else if (std.mem.eql(u8, arg, "config") or std.mem.eql(u8, arg, "configure")) {
try setup.wizard(allocator); try setup.wizard(allocator);
return; return;
} else { } else {
@ -1742,6 +1763,10 @@ pub fn main() anyerror!void {
std.os.exit(1); std.os.exit(1);
} }
} }
if (next_arg_config_path) {
std.debug.print("Expected configuration file path after --config-path argument\n", .{});
return;
}
// Init global vars // Init global vars
const reader = std.io.getStdIn().reader(); const reader = std.io.getStdIn().reader();
@ -1753,16 +1778,25 @@ pub fn main() anyerror!void {
defer std.json.parseFree(Config, config, config_parse_options); defer std.json.parseFree(Config, config, config_parse_options);
config_read: { config_read: {
if (config_path) |path| {
defer allocator.free(path);
if (loadConfigFile(path)) |conf| {
config = conf;
break :config_read;
}
std.debug.print("Could not open configuration file '{s}'\n", .{path});
std.debug.print("Falling back to a lookup in the local and global configuration folders\n", .{});
}
if (try known_folders.getPath(allocator, .local_configuration)) |path| { if (try known_folders.getPath(allocator, .local_configuration)) |path| {
defer allocator.free(path); defer allocator.free(path);
if (loadConfig(path)) |conf| { if (loadConfigInFolder(path)) |conf| {
config = conf; config = conf;
break :config_read; break :config_read;
} }
} }
if (try known_folders.getPath(allocator, .global_configuration)) |path| { if (try known_folders.getPath(allocator, .global_configuration)) |path| {
defer allocator.free(path); defer allocator.free(path);
if (loadConfig(path)) |conf| { if (loadConfigInFolder(path)) |conf| {
config = conf; config = conf;
break :config_read; break :config_read;
} }