92 lines
2.7 KiB
Zig
92 lines
2.7 KiB
Zig
const std = @import("std");
|
|
|
|
const reserved_chars = &[_]u8{
|
|
'!', '#', '$', '%', '&', '\'',
|
|
'(', ')', '*', '+', ',', ':',
|
|
';', '=', '?', '@', '[', ']',
|
|
};
|
|
|
|
/// Returns a URI from a path, caller owns the memory allocated with `allocator`
|
|
pub fn fromPath(allocator: *std.mem.Allocator, path: []const u8) ![]const u8 {
|
|
if (path.len == 0) return "";
|
|
const prefix = if (std.builtin.os.tag == .windows) "file:///" else "file://";
|
|
|
|
var buf = std.ArrayList(u8).init(allocator);
|
|
try buf.appendSlice(prefix);
|
|
|
|
const out_stream = buf.writer();
|
|
|
|
for (path) |char| {
|
|
if (char == std.fs.path.sep) {
|
|
try buf.append('/');
|
|
} else if (std.mem.indexOfScalar(u8, reserved_chars, char) != null) {
|
|
// Write '%' + hex with uppercase
|
|
try buf.append('%');
|
|
try std.fmt.format(out_stream, "{X}", .{char});
|
|
} else {
|
|
try buf.append(char);
|
|
}
|
|
}
|
|
|
|
// On windows, we need to lowercase the drive name.
|
|
if (std.builtin.os.tag == .windows) {
|
|
if (buf.items.len > prefix.len + 1 and
|
|
std.ascii.isAlpha(buf.items[prefix.len]) and
|
|
std.mem.startsWith(u8, buf.items[prefix.len + 1 ..], "%3A"))
|
|
{
|
|
buf.items[prefix.len] = std.ascii.toLower(buf.items[prefix.len]);
|
|
}
|
|
}
|
|
|
|
return buf.toOwnedSlice();
|
|
}
|
|
|
|
pub const UriParseError = error{
|
|
UriBadScheme,
|
|
UriBadHexChar,
|
|
UriBadEscape,
|
|
OutOfMemory,
|
|
};
|
|
|
|
// Original code: https://github.com/andersfr/zig-lsp/blob/master/uri.zig
|
|
fn parseHex(c: u8) !u8 {
|
|
return switch (c) {
|
|
'0'...'9' => c - '0',
|
|
'a'...'f' => c - 'a' + 10,
|
|
'A'...'F' => c - 'A' + 10,
|
|
else => return error.UriBadHexChar,
|
|
};
|
|
}
|
|
|
|
/// Caller should free memory
|
|
pub fn parse(allocator: *std.mem.Allocator, str: []const u8) ![]u8 {
|
|
if (str.len < 7 or !std.mem.eql(u8, "file://", str[0..7])) return error.UriBadScheme;
|
|
|
|
const uri = try allocator.alloc(u8, str.len - (if (std.fs.path.sep == '\\') 8 else 7));
|
|
errdefer allocator.free(uri);
|
|
|
|
const path = if (std.fs.path.sep == '\\') str[8..] else str[7..];
|
|
|
|
var i: usize = 0;
|
|
var j: usize = 0;
|
|
while (j < path.len) : (i += 1) {
|
|
if (path[j] == '%') {
|
|
if (j + 2 >= path.len) return error.UriBadEscape;
|
|
const upper = try parseHex(path[j + 1]);
|
|
const lower = try parseHex(path[j + 2]);
|
|
uri[i] = (upper << 4) + lower;
|
|
j += 3;
|
|
} else {
|
|
uri[i] = if (path[j] == '/') std.fs.path.sep else path[j];
|
|
j += 1;
|
|
}
|
|
}
|
|
|
|
// Remove trailing separator
|
|
if (i > 0 and uri[i - 1] == std.fs.path.sep) {
|
|
i -= 1;
|
|
}
|
|
|
|
return allocator.shrink(uri, i);
|
|
}
|