zig/std/os/path.zig

1027 lines
36 KiB
Zig
Raw Normal View History

const builtin = @import("builtin");
const Os = builtin.Os;
const debug = @import("../debug.zig");
const assert = debug.assert;
const mem = @import("../mem.zig");
const fmt = @import("../fmt/index.zig");
const Allocator = mem.Allocator;
const os = @import("index.zig");
const math = @import("../math/index.zig");
const posix = os.posix;
const windows = os.windows;
const c = @import("../c/index.zig");
const cstr = @import("../cstr.zig");
2017-10-05 21:27:15 -07:00
pub const sep_windows = '\\';
pub const sep_posix = '/';
pub const sep = if (is_windows) sep_windows else sep_posix;
pub const delimiter_windows = ';';
pub const delimiter_posix = ':';
pub const delimiter = if (is_windows) delimiter_windows else delimiter_posix;
const is_windows = builtin.os == builtin.Os.windows;
/// Naively combines a series of paths with the native path seperator.
/// Allocates memory for the result, which must be freed by the caller.
pub fn join(allocator: &Allocator, paths: ...) -> %[]u8 {
2017-10-05 21:27:15 -07:00
if (is_windows) {
return joinWindows(allocator, paths);
} else {
return joinPosix(allocator, paths);
}
}
pub fn joinWindows(allocator: &Allocator, paths: ...) -> %[]u8 {
return mem.join(allocator, sep_windows, paths);
}
pub fn joinPosix(allocator: &Allocator, paths: ...) -> %[]u8 {
return mem.join(allocator, sep_posix, paths);
}
test "os.path.join" {
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, %%joinWindows(&debug.global_allocator, "c:\\a\\b", "c"), "c:\\a\\b\\c"));
assert(mem.eql(u8, %%joinWindows(&debug.global_allocator, "c:\\a\\b\\", "c"), "c:\\a\\b\\c"));
assert(mem.eql(u8, %%joinWindows(&debug.global_allocator, "c:\\", "a", "b\\", "c"), "c:\\a\\b\\c"));
assert(mem.eql(u8, %%joinWindows(&debug.global_allocator, "c:\\a\\", "b\\", "c"), "c:\\a\\b\\c"));
assert(mem.eql(u8, %%joinWindows(&debug.global_allocator,
"c:\\home\\andy\\dev\\zig\\build\\lib\\zig\\std", "io.zig"),
"c:\\home\\andy\\dev\\zig\\build\\lib\\zig\\std\\io.zig"));
assert(mem.eql(u8, %%joinPosix(&debug.global_allocator, "/a/b", "c"), "/a/b/c"));
assert(mem.eql(u8, %%joinPosix(&debug.global_allocator, "/a/b/", "c"), "/a/b/c"));
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, %%joinPosix(&debug.global_allocator, "/", "a", "b/", "c"), "/a/b/c"));
assert(mem.eql(u8, %%joinPosix(&debug.global_allocator, "/a/", "b/", "c"), "/a/b/c"));
2017-04-24 09:01:19 -07:00
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, %%joinPosix(&debug.global_allocator, "/home/andy/dev/zig/build/lib/zig/std", "io.zig"),
2017-04-24 09:01:19 -07:00
"/home/andy/dev/zig/build/lib/zig/std/io.zig"));
}
pub fn isAbsolute(path: []const u8) -> bool {
2017-10-05 21:27:15 -07:00
if (is_windows) {
return isAbsoluteWindows(path);
} else {
return isAbsolutePosix(path);
}
}
pub fn isAbsoluteWindows(path: []const u8) -> bool {
if (path[0] == '/')
return true;
if (path[0] == '\\') {
return true;
}
if (path.len < 3) {
return false;
}
if (path[1] == ':') {
if (path[2] == '/')
return true;
if (path[2] == '\\')
return true;
}
return false;
}
pub fn isAbsolutePosix(path: []const u8) -> bool {
return path[0] == sep_posix;
}
test "os.path.isAbsoluteWindows" {
testIsAbsoluteWindows("/", true);
testIsAbsoluteWindows("//", true);
testIsAbsoluteWindows("//server", true);
testIsAbsoluteWindows("//server/file", true);
testIsAbsoluteWindows("\\\\server\\file", true);
testIsAbsoluteWindows("\\\\server", true);
testIsAbsoluteWindows("\\\\", true);
testIsAbsoluteWindows("c", false);
testIsAbsoluteWindows("c:", false);
testIsAbsoluteWindows("c:\\", true);
testIsAbsoluteWindows("c:/", true);
testIsAbsoluteWindows("c://", true);
testIsAbsoluteWindows("C:/Users/", true);
testIsAbsoluteWindows("C:\\Users\\", true);
testIsAbsoluteWindows("C:cwd/another", false);
testIsAbsoluteWindows("C:cwd\\another", false);
testIsAbsoluteWindows("directory/directory", false);
testIsAbsoluteWindows("directory\\directory", false);
}
test "os.path.isAbsolutePosix" {
testIsAbsolutePosix("/home/foo", true);
testIsAbsolutePosix("/home/foo/..", true);
testIsAbsolutePosix("bar/", false);
testIsAbsolutePosix("./baz", false);
}
fn testIsAbsoluteWindows(path: []const u8, expected_result: bool) {
assert(isAbsoluteWindows(path) == expected_result);
}
fn testIsAbsolutePosix(path: []const u8, expected_result: bool) {
assert(isAbsolutePosix(path) == expected_result);
}
pub fn drive(path: []const u8) -> ?[]const u8 {
if (path.len < 2)
return null;
if (path[1] != ':')
return null;
return path[0..2];
}
pub fn networkShare(path: []const u8) -> ?[]const u8 {
if (path.len < "//a/b".len)
return null;
2017-10-06 14:21:21 -07:00
// TODO when I combined these together with `inline for` the compiler crashed
2017-10-05 21:27:15 -07:00
{
const this_sep = '/';
const two_sep = []u8{this_sep, this_sep};
if (mem.startsWith(u8, path, two_sep)) {
if (path[2] == this_sep)
return null;
2017-10-06 14:21:21 -07:00
var it = mem.split(path, []u8{this_sep});
_ = (it.next() ?? return null);
_ = (it.next() ?? return null);
return path[0..it.index];
2017-10-05 21:27:15 -07:00
}
}
{
const this_sep = '\\';
const two_sep = []u8{this_sep, this_sep};
if (mem.startsWith(u8, path, two_sep)) {
if (path[2] == this_sep)
return null;
2017-10-06 14:21:21 -07:00
var it = mem.split(path, []u8{this_sep});
_ = (it.next() ?? return null);
_ = (it.next() ?? return null);
return path[0..it.index];
2017-10-05 21:27:15 -07:00
}
}
return null;
}
test "os.path.networkShare" {
assert(mem.eql(u8, ??networkShare("//a/b"), "//a/b"));
assert(mem.eql(u8, ??networkShare("\\\\a\\b"), "\\\\a\\b"));
assert(networkShare("\\\\a\\") == null);
}
2017-10-08 18:40:33 -07:00
pub fn diskDesignator(path: []const u8) -> []const u8 {
if (is_windows) {
return diskDesignatorWindows(path);
} else {
2017-10-08 18:40:33 -07:00
return "";
}
}
2017-10-05 21:27:15 -07:00
pub fn diskDesignatorWindows(path: []const u8) -> []const u8 {
2017-10-05 21:27:15 -07:00
return drive(path) ?? (networkShare(path) ?? []u8{});
}
2017-10-08 18:16:10 -07:00
// TODO ASCII is wrong, we actually need full unicode support to compare paths.
fn networkShareServersEql(ns1: []const u8, ns2: []const u8) -> bool {
const sep1 = ns1[0];
const sep2 = ns2[0];
var it1 = mem.split(ns1, []u8{sep1});
var it2 = mem.split(ns2, []u8{sep2});
return asciiEqlIgnoreCase(??it1.next(), ??it2.next());
2017-10-05 21:27:15 -07:00
}
2017-10-06 14:21:21 -07:00
fn asciiUpper(byte: u8) -> u8 {
2017-10-05 21:27:15 -07:00
return switch (byte) {
2017-10-06 14:21:21 -07:00
'a' ... 'z' => 'A' + (byte - 'a'),
2017-10-05 21:27:15 -07:00
else => byte,
};
}
2017-10-08 18:16:10 -07:00
fn asciiEqlIgnoreCase(s1: []const u8, s2: []const u8) -> bool {
if (s1.len != s2.len)
return false;
var i: usize = 0;
while (i < s1.len) : (i += 1) {
if (asciiUpper(s1[i]) != asciiUpper(s2[i]))
return false;
}
return true;
}
2017-10-06 14:21:21 -07:00
/// Converts the command line arguments into a slice and calls `resolveSlice`.
pub fn resolve(allocator: &Allocator, args: ...) -> %[]u8 {
var paths: [args.len][]const u8 = undefined;
comptime var arg_i = 0;
inline while (arg_i < args.len) : (arg_i += 1) {
paths[arg_i] = args[arg_i];
}
return resolveSlice(allocator, paths);
}
2017-10-06 14:21:21 -07:00
/// On Windows, this calls `resolveWindows` and on POSIX it calls `resolvePosix`.
pub fn resolveSlice(allocator: &Allocator, paths: []const []const u8) -> %[]u8 {
2017-10-05 21:27:15 -07:00
if (is_windows) {
return resolveWindows(allocator, paths);
} else {
return resolvePosix(allocator, paths);
}
}
2017-10-06 14:21:21 -07:00
/// This function is like a series of `cd` statements executed one after another.
/// It resolves "." and "..".
/// The result does not have a trailing path separator.
/// If all paths are relative it uses the current working directory as a starting point.
/// Each drive has its own current working directory.
/// Path separators are canonicalized to '\\' and drives are canonicalized to capital letters.
2017-10-05 21:27:15 -07:00
pub fn resolveWindows(allocator: &Allocator, paths: []const []const u8) -> %[]u8 {
if (paths.len == 0) {
assert(is_windows); // resolveWindows called on non windows can't use getCwd
return os.getCwd(allocator);
}
// determine which drive we want to result with
2017-10-06 14:21:21 -07:00
var result_drive_upcase: ?u8 = null;
2017-10-05 21:27:15 -07:00
var have_abs = false;
var first_index: usize = 0;
var max_size: usize = 0;
for (paths) |p, i| {
const is_abs = isAbsoluteWindows(p);
if (is_abs) {
have_abs = true;
first_index = i;
max_size = 0;
}
if (drive(p)) |d| {
2017-10-06 14:21:21 -07:00
result_drive_upcase = asciiUpper(d[0]);
} else if (networkShare(p)) |_| {
result_drive_upcase = null;
2017-10-05 21:27:15 -07:00
}
max_size += p.len + 1;
}
2017-10-06 14:21:21 -07:00
2017-10-05 21:27:15 -07:00
// if we will result with a drive, loop again to determine
// which is the first time the drive is absolutely specified, if any
// and count up the max bytes for paths related to this drive
2017-10-06 14:21:21 -07:00
if (result_drive_upcase) |res_dr| {
2017-10-05 21:27:15 -07:00
have_abs = false;
first_index = 0;
2017-10-06 14:21:21 -07:00
max_size = "_:".len;
2017-10-05 21:27:15 -07:00
var correct_drive = false;
for (paths) |p, i| {
if (drive(p)) |dr| {
2017-10-06 14:21:21 -07:00
correct_drive = asciiUpper(dr[0]) == res_dr;
} else if (networkShare(p)) |_| {
continue;
2017-10-05 21:27:15 -07:00
}
if (!correct_drive) {
continue;
}
const is_abs = isAbsoluteWindows(p);
if (is_abs) {
first_index = i;
2017-10-06 14:21:21 -07:00
max_size = "_:".len;
have_abs = true;
2017-10-05 21:27:15 -07:00
}
max_size += p.len + 1;
}
}
2017-10-06 14:21:21 -07:00
var drive_buf = "_:";
2017-10-05 21:27:15 -07:00
var result: []u8 = undefined;
var result_index: usize = 0;
2017-10-06 14:21:21 -07:00
var root_slice: []const u8 = undefined;
2017-10-05 21:27:15 -07:00
if (have_abs) {
result = %return allocator.alloc(u8, max_size);
2017-10-06 14:21:21 -07:00
if (result_drive_upcase) |res_dr| {
drive_buf[0] = res_dr;
root_slice = drive_buf[0..];
mem.copy(u8, result, root_slice);
result_index += root_slice.len;
} else {
// We know it looks like //a/b or \\a\b because of earlier code
var it = mem.split(paths[first_index], "/\\");
const server_name = ??it.next();
const other_name = ??it.next();
result[result_index] = '\\';
result_index += 1;
result[result_index] = '\\';
result_index += 1;
mem.copy(u8, result[result_index..], server_name);
result_index += server_name.len;
result[result_index] = '\\';
result_index += 1;
mem.copy(u8, result[result_index..], other_name);
result_index += other_name.len;
root_slice = result[0..result_index];
}
2017-10-05 21:27:15 -07:00
} else {
assert(is_windows); // resolveWindows called on non windows can't use getCwd
// TODO get cwd for result_drive if applicable
const cwd = %return os.getCwd(allocator);
defer allocator.free(cwd);
result = %return allocator.alloc(u8, max_size + cwd.len + 1);
mem.copy(u8, result, cwd);
result_index += cwd.len;
2017-10-06 14:21:21 -07:00
root_slice = diskDesignatorWindows(result[0..result_index]);
2017-10-05 21:27:15 -07:00
}
%defer allocator.free(result);
2017-10-06 14:21:21 -07:00
var correct_drive = true;
2017-10-05 21:27:15 -07:00
for (paths[first_index..]) |p, i| {
2017-10-06 14:21:21 -07:00
if (result_drive_upcase) |res_dr| {
2017-10-05 21:27:15 -07:00
if (drive(p)) |dr| {
2017-10-06 14:21:21 -07:00
correct_drive = asciiUpper(dr[0]) == res_dr;
} else if (networkShare(p)) |_| {
continue;
2017-10-05 21:27:15 -07:00
}
if (!correct_drive) {
continue;
}
}
var it = mem.split(p[diskDesignatorWindows(p).len..], "/\\");
2017-10-05 21:27:15 -07:00
while (it.next()) |component| {
if (mem.eql(u8, component, ".")) {
continue;
} else if (mem.eql(u8, component, "..")) {
while (true) {
2017-10-06 14:21:21 -07:00
if (result_index == 0 or result_index == root_slice.len)
2017-10-05 21:27:15 -07:00
break;
result_index -= 1;
if (result[result_index] == '\\' or result[result_index] == '/')
break;
}
} else {
2017-10-06 14:21:21 -07:00
result[result_index] = sep_windows;
2017-10-05 21:27:15 -07:00
result_index += 1;
mem.copy(u8, result[result_index..], component);
result_index += component.len;
}
}
2017-05-22 21:26:12 -07:00
}
2017-10-05 21:27:15 -07:00
2017-10-06 14:21:21 -07:00
if (result_index == root_slice.len) {
result[result_index] = '\\';
result_index += 1;
}
2017-10-05 21:27:15 -07:00
return result[0..result_index];
}
2017-10-06 14:21:21 -07:00
/// This function is like a series of `cd` statements executed one after another.
/// It resolves "." and "..".
/// The result does not have a trailing path separator.
/// If all paths are relative it uses the current working directory as a starting point.
2017-10-05 21:27:15 -07:00
pub fn resolvePosix(allocator: &Allocator, paths: []const []const u8) -> %[]u8 {
if (paths.len == 0) {
assert(!is_windows); // resolvePosix called on windows can't use getCwd
return os.getCwd(allocator);
2017-10-05 21:27:15 -07:00
}
var first_index: usize = 0;
var have_abs = false;
var max_size: usize = 0;
for (paths) |p, i| {
2017-10-05 21:27:15 -07:00
if (isAbsolutePosix(p)) {
first_index = i;
have_abs = true;
max_size = 0;
}
max_size += p.len + 1;
}
var result: []u8 = undefined;
var result_index: usize = 0;
if (have_abs) {
result = %return allocator.alloc(u8, max_size);
} else {
2017-10-05 21:27:15 -07:00
assert(!is_windows); // resolvePosix called on windows can't use getCwd
const cwd = %return os.getCwd(allocator);
defer allocator.free(cwd);
result = %return allocator.alloc(u8, max_size + cwd.len + 1);
mem.copy(u8, result, cwd);
result_index += cwd.len;
}
%defer allocator.free(result);
for (paths[first_index..]) |p, i| {
2017-10-05 21:27:15 -07:00
var it = mem.split(p, "/");
while (it.next()) |component| {
if (mem.eql(u8, component, ".")) {
continue;
} else if (mem.eql(u8, component, "..")) {
while (true) {
if (result_index == 0)
break;
result_index -= 1;
if (result[result_index] == '/')
break;
}
} else {
result[result_index] = '/';
result_index += 1;
mem.copy(u8, result[result_index..], component);
result_index += component.len;
}
}
}
if (result_index == 0) {
result[0] = '/';
result_index += 1;
}
return result[0..result_index];
}
test "os.path.resolve" {
2017-10-05 21:27:15 -07:00
const cwd = %%os.getCwd(&debug.global_allocator);
if (is_windows) {
assert(mem.eql(u8, testResolveWindows([][]const u8{"."}), cwd));
} else {
assert(mem.eql(u8, testResolvePosix([][]const u8{"a/b/c/", "../../.."}), cwd));
assert(mem.eql(u8, testResolvePosix([][]const u8{"."}), cwd));
}
}
test "os.path.resolveWindows" {
2017-10-06 14:21:21 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/blah\\blah", "d:/games", "c:../a"}), "C:\\blah\\a"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/blah\\blah", "d:/games", "C:../a"}), "C:\\blah\\a"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/ignore", "d:\\a/b\\c/d", "\\e.exe"}), "D:\\e.exe"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/ignore", "c:/some/file"}), "C:\\some\\file"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"d:/ignore", "d:some/dir//"}), "D:\\ignore\\some\\dir"));
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"//server/share", "..", "relative\\"}), "\\\\server\\share\\relative"));
2017-10-06 14:21:21 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/", "//"}), "C:\\"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/", "//dir"}), "C:\\dir"));
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/", "//server/share"}), "\\\\server\\share\\"));
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/", "//server//share"}), "\\\\server\\share\\"));
2017-10-06 14:21:21 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"c:/", "///some//dir"}), "C:\\some\\dir"));
2017-10-05 21:27:15 -07:00
assert(mem.eql(u8, testResolveWindows([][]const u8{"C:\\foo\\tmp.3\\", "..\\tmp.3\\cycles\\root.js"}),
"C:\\foo\\tmp.3\\cycles\\root.js"));
}
test "os.path.resolvePosix" {
assert(mem.eql(u8, testResolvePosix([][]const u8{"/a/b", "c"}), "/a/b/c"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/a/b", "c", "//d", "e///"}), "/d/e"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/a/b/c", "..", "../"}), "/a"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/", "..", ".."}), "/"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/a/b/c/"}), "/a/b/c"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/var/lib", "../", "file/"}), "/var/file"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/var/lib", "/../", "file/"}), "/file"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/some/dir", ".", "/absolute/"}), "/absolute"));
assert(mem.eql(u8, testResolvePosix([][]const u8{"/foo/tmp.3/", "../tmp.3/cycles/root.js"}), "/foo/tmp.3/cycles/root.js"));
}
fn testResolveWindows(paths: []const []const u8) -> []u8 {
return %%resolveWindows(&debug.global_allocator, paths);
}
2017-10-05 21:27:15 -07:00
fn testResolvePosix(paths: []const []const u8) -> []u8 {
return %%resolvePosix(&debug.global_allocator, paths);
}
pub fn dirname(path: []const u8) -> []const u8 {
2017-10-05 21:27:15 -07:00
if (is_windows) {
return dirnameWindows(path);
} else {
return dirnamePosix(path);
}
}
pub fn dirnameWindows(path: []const u8) -> []const u8 {
if (path.len == 0)
return path[0..0];
const root_slice = diskDesignatorWindows(path);
2017-10-06 14:21:21 -07:00
if (path.len == root_slice.len)
2017-10-05 21:27:15 -07:00
return path;
2017-10-06 14:21:21 -07:00
const have_root_slash = path.len > root_slice.len and (path[root_slice.len] == '/' or path[root_slice.len] == '\\');
2017-10-05 21:27:15 -07:00
var end_index: usize = path.len - 1;
2017-10-06 14:21:21 -07:00
while ((path[end_index] == '/' or path[end_index] == '\\') and end_index > root_slice.len) {
2017-10-05 21:27:15 -07:00
if (end_index == 0)
return path[0..0];
end_index -= 1;
}
2017-10-06 14:21:21 -07:00
while (path[end_index] != '/' and path[end_index] != '\\' and end_index > root_slice.len) {
2017-10-05 21:27:15 -07:00
if (end_index == 0)
return path[0..0];
end_index -= 1;
}
2017-10-06 14:21:21 -07:00
if (have_root_slash and end_index == root_slice.len) {
2017-10-05 21:27:15 -07:00
end_index += 1;
2017-05-22 21:26:12 -07:00
}
2017-10-05 21:27:15 -07:00
return path[0..end_index];
}
pub fn dirnamePosix(path: []const u8) -> []const u8 {
if (path.len == 0)
return path[0..0];
2017-10-05 21:27:15 -07:00
var end_index: usize = path.len - 1;
while (path[end_index] == '/') {
if (end_index == 0)
return path[0..1];
end_index -= 1;
}
while (path[end_index] != '/') {
if (end_index == 0)
return path[0..0];
end_index -= 1;
}
if (end_index == 0 and path[end_index] == '/')
return path[0..1];
return path[0..end_index];
}
2017-10-05 21:27:15 -07:00
test "os.path.dirnamePosix" {
testDirnamePosix("/a/b/c", "/a/b");
testDirnamePosix("/a/b/c///", "/a/b");
testDirnamePosix("/a", "/");
testDirnamePosix("/", "/");
testDirnamePosix("////", "/");
testDirnamePosix("", "");
testDirnamePosix("a", "");
testDirnamePosix("a/", "");
testDirnamePosix("a//", "");
}
test "os.path.dirnameWindows" {
testDirnameWindows("c:\\", "c:\\");
testDirnameWindows("c:\\foo", "c:\\");
testDirnameWindows("c:\\foo\\", "c:\\");
testDirnameWindows("c:\\foo\\bar", "c:\\foo");
testDirnameWindows("c:\\foo\\bar\\", "c:\\foo");
testDirnameWindows("c:\\foo\\bar\\baz", "c:\\foo\\bar");
testDirnameWindows("\\", "\\");
testDirnameWindows("\\foo", "\\");
testDirnameWindows("\\foo\\", "\\");
testDirnameWindows("\\foo\\bar", "\\foo");
testDirnameWindows("\\foo\\bar\\", "\\foo");
testDirnameWindows("\\foo\\bar\\baz", "\\foo\\bar");
testDirnameWindows("c:", "c:");
testDirnameWindows("c:foo", "c:");
testDirnameWindows("c:foo\\", "c:");
testDirnameWindows("c:foo\\bar", "c:foo");
testDirnameWindows("c:foo\\bar\\", "c:foo");
testDirnameWindows("c:foo\\bar\\baz", "c:foo\\bar");
testDirnameWindows("file:stream", "");
testDirnameWindows("dir\\file:stream", "dir");
testDirnameWindows("\\\\unc\\share", "\\\\unc\\share");
testDirnameWindows("\\\\unc\\share\\foo", "\\\\unc\\share\\");
testDirnameWindows("\\\\unc\\share\\foo\\", "\\\\unc\\share\\");
testDirnameWindows("\\\\unc\\share\\foo\\bar", "\\\\unc\\share\\foo");
testDirnameWindows("\\\\unc\\share\\foo\\bar\\", "\\\\unc\\share\\foo");
testDirnameWindows("\\\\unc\\share\\foo\\bar\\baz", "\\\\unc\\share\\foo\\bar");
testDirnameWindows("/a/b/", "/a");
testDirnameWindows("/a/b", "/a");
testDirnameWindows("/a", "/");
testDirnameWindows("", "");
testDirnameWindows("/", "/");
testDirnameWindows("////", "/");
testDirnameWindows("foo", "");
}
2017-10-05 21:27:15 -07:00
fn testDirnamePosix(input: []const u8, expected_output: []const u8) {
assert(mem.eql(u8, dirnamePosix(input), expected_output));
}
fn testDirnameWindows(input: []const u8, expected_output: []const u8) {
assert(mem.eql(u8, dirnameWindows(input), expected_output));
}
pub fn basename(path: []const u8) -> []const u8 {
if (is_windows) {
return basenameWindows(path);
} else {
return basenamePosix(path);
2017-05-22 21:26:12 -07:00
}
}
pub fn basenamePosix(path: []const u8) -> []const u8 {
if (path.len == 0)
return []u8{};
var end_index: usize = path.len - 1;
while (path[end_index] == '/') {
if (end_index == 0)
return []u8{};
end_index -= 1;
}
var start_index: usize = end_index;
end_index += 1;
while (path[start_index] != '/') {
if (start_index == 0)
return path[0..end_index];
start_index -= 1;
}
return path[start_index + 1..end_index];
}
pub fn basenameWindows(path: []const u8) -> []const u8 {
if (path.len == 0)
return []u8{};
var end_index: usize = path.len - 1;
while (true) {
const byte = path[end_index];
if (byte == '/' or byte == '\\') {
if (end_index == 0)
return []u8{};
end_index -= 1;
continue;
}
if (byte == ':' and end_index == 1) {
return []u8{};
}
break;
}
var start_index: usize = end_index;
end_index += 1;
while (path[start_index] != '/' and path[start_index] != '\\' and
!(path[start_index] == ':' and start_index == 1))
{
if (start_index == 0)
return path[0..end_index];
start_index -= 1;
}
return path[start_index + 1..end_index];
}
test "os.path.basename" {
testBasename("", "");
testBasename("/", "");
testBasename("/dir/basename.ext", "basename.ext");
testBasename("/basename.ext", "basename.ext");
testBasename("basename.ext", "basename.ext");
testBasename("basename.ext/", "basename.ext");
testBasename("basename.ext//", "basename.ext");
testBasename("/aaa/bbb", "bbb");
testBasename("/aaa/", "aaa");
testBasename("/aaa/b", "b");
testBasename("/a/b", "b");
testBasename("//a", "a");
testBasenamePosix("\\dir\\basename.ext", "\\dir\\basename.ext");
testBasenamePosix("\\basename.ext", "\\basename.ext");
testBasenamePosix("basename.ext", "basename.ext");
testBasenamePosix("basename.ext\\", "basename.ext\\");
testBasenamePosix("basename.ext\\\\", "basename.ext\\\\");
testBasenamePosix("foo", "foo");
testBasenameWindows("\\dir\\basename.ext", "basename.ext");
testBasenameWindows("\\basename.ext", "basename.ext");
testBasenameWindows("basename.ext", "basename.ext");
testBasenameWindows("basename.ext\\", "basename.ext");
testBasenameWindows("basename.ext\\\\", "basename.ext");
testBasenameWindows("foo", "foo");
testBasenameWindows("C:", "");
testBasenameWindows("C:.", ".");
testBasenameWindows("C:\\", "");
testBasenameWindows("C:\\dir\\base.ext", "base.ext");
testBasenameWindows("C:\\basename.ext", "basename.ext");
testBasenameWindows("C:basename.ext", "basename.ext");
testBasenameWindows("C:basename.ext\\", "basename.ext");
testBasenameWindows("C:basename.ext\\\\", "basename.ext");
testBasenameWindows("C:foo", "foo");
testBasenameWindows("file:stream", "file:stream");
}
fn testBasename(input: []const u8, expected_output: []const u8) {
assert(mem.eql(u8, basename(input), expected_output));
}
fn testBasenamePosix(input: []const u8, expected_output: []const u8) {
assert(mem.eql(u8, basenamePosix(input), expected_output));
}
fn testBasenameWindows(input: []const u8, expected_output: []const u8) {
assert(mem.eql(u8, basenameWindows(input), expected_output));
}
2017-10-08 18:16:10 -07:00
/// Returns the relative path from `from` to `to`. If `from` and `to` each
/// resolve to the same path (after calling `resolve` on each), a zero-length
/// string is returned.
2017-10-08 18:16:10 -07:00
/// On Windows this canonicalizes the drive to a capital letter and paths to `\\`.
pub fn relative(allocator: &Allocator, from: []const u8, to: []const u8) -> %[]u8 {
2017-10-05 21:27:15 -07:00
if (is_windows) {
2017-10-08 18:16:10 -07:00
return relativeWindows(allocator, from, to);
2017-10-05 21:27:15 -07:00
} else {
2017-10-08 18:16:10 -07:00
return relativePosix(allocator, from, to);
2017-05-22 21:26:12 -07:00
}
2017-10-05 21:27:15 -07:00
}
2017-10-08 18:16:10 -07:00
pub fn relativeWindows(allocator: &Allocator, from: []const u8, to: []const u8) -> %[]u8 {
const resolved_from = %return resolveWindows(allocator, [][]const u8{from});
defer allocator.free(resolved_from);
var clean_up_resolved_to = true;
const resolved_to = %return resolveWindows(allocator, [][]const u8{to});
defer if (clean_up_resolved_to) allocator.free(resolved_to);
const result_is_to = if (drive(resolved_to)) |to_drive| {
if (drive(resolved_from)) |from_drive| {
asciiUpper(from_drive[0]) != asciiUpper(to_drive[0])
} else {
true
}
} else if (networkShare(resolved_to)) |to_ns| {
if (networkShare(resolved_from)) |from_ns| {
!networkShareServersEql(to_ns, from_ns)
} else {
true
}
} else {
unreachable
};
if (result_is_to) {
clean_up_resolved_to = false;
return resolved_to;
}
var from_it = mem.split(resolved_from, "/\\");
var to_it = mem.split(resolved_to, "/\\");
while (true) {
const from_component = from_it.next() ?? return mem.dupe(allocator, u8, to_it.rest());
const to_rest = to_it.rest();
if (to_it.next()) |to_component| {
// TODO ASCII is wrong, we actually need full unicode support to compare paths.
if (asciiEqlIgnoreCase(from_component, to_component))
continue;
}
var up_count: usize = 1;
while (from_it.next()) |_| {
up_count += 1;
}
const up_index_end = up_count * "..\\".len;
const result = %return allocator.alloc(u8, up_index_end + to_rest.len);
%defer allocator.free(result);
var result_index: usize = 0;
while (result_index < up_index_end) {
result[result_index] = '.';
result_index += 1;
result[result_index] = '.';
result_index += 1;
result[result_index] = '\\';
result_index += 1;
}
// shave off the trailing slash
result_index -= 1;
var rest_it = mem.split(to_rest, "/\\");
while (rest_it.next()) |to_component| {
result[result_index] = '\\';
result_index += 1;
mem.copy(u8, result[result_index..], to_component);
result_index += to_component.len;
}
return result[0..result_index];
}
return []u8{};
2017-10-05 21:27:15 -07:00
}
2017-10-08 18:16:10 -07:00
pub fn relativePosix(allocator: &Allocator, from: []const u8, to: []const u8) -> %[]u8 {
const resolved_from = %return resolvePosix(allocator, [][]const u8{from});
defer allocator.free(resolved_from);
2017-10-08 18:16:10 -07:00
const resolved_to = %return resolvePosix(allocator, [][]const u8{to});
defer allocator.free(resolved_to);
2017-10-06 14:21:21 -07:00
var from_it = mem.split(resolved_from, "/");
var to_it = mem.split(resolved_to, "/");
while (true) {
const from_component = from_it.next() ?? return mem.dupe(allocator, u8, to_it.rest());
const to_rest = to_it.rest();
if (to_it.next()) |to_component| {
if (mem.eql(u8, from_component, to_component))
continue;
}
var up_count: usize = 1;
while (from_it.next()) |_| {
up_count += 1;
}
const up_index_end = up_count * "../".len;
const result = %return allocator.alloc(u8, up_index_end + to_rest.len);
%defer allocator.free(result);
var result_index: usize = 0;
while (result_index < up_index_end) {
result[result_index] = '.';
result_index += 1;
result[result_index] = '.';
result_index += 1;
result[result_index] = '/';
result_index += 1;
}
if (to_rest.len == 0) {
// shave off the trailing slash
return result[0..result_index - 1];
}
mem.copy(u8, result[result_index..], to_rest);
return result;
}
return []u8{};
}
test "os.path.relative" {
2017-10-08 18:16:10 -07:00
testRelativeWindows("c:/blah\\blah", "d:/games", "D:\\games");
testRelativeWindows("c:/aaaa/bbbb", "c:/aaaa", "..");
testRelativeWindows("c:/aaaa/bbbb", "c:/cccc", "..\\..\\cccc");
testRelativeWindows("c:/aaaa/bbbb", "c:/aaaa/bbbb", "");
testRelativeWindows("c:/aaaa/bbbb", "c:/aaaa/cccc", "..\\cccc");
testRelativeWindows("c:/aaaa/", "c:/aaaa/cccc", "cccc");
testRelativeWindows("c:/", "c:\\aaaa\\bbbb", "aaaa\\bbbb");
testRelativeWindows("c:/aaaa/bbbb", "d:\\", "D:\\");
testRelativeWindows("c:/AaAa/bbbb", "c:/aaaa/bbbb", "");
testRelativeWindows("c:/aaaaa/", "c:/aaaa/cccc", "..\\aaaa\\cccc");
testRelativeWindows("C:\\foo\\bar\\baz\\quux", "C:\\", "..\\..\\..\\..");
testRelativeWindows("C:\\foo\\test", "C:\\foo\\test\\bar\\package.json", "bar\\package.json");
testRelativeWindows("C:\\foo\\bar\\baz-quux", "C:\\foo\\bar\\baz", "..\\baz");
testRelativeWindows("C:\\foo\\bar\\baz", "C:\\foo\\bar\\baz-quux", "..\\baz-quux");
testRelativeWindows("\\\\foo\\bar", "\\\\foo\\bar\\baz", "baz");
testRelativeWindows("\\\\foo\\bar\\baz", "\\\\foo\\bar", "..");
testRelativeWindows("\\\\foo\\bar\\baz-quux", "\\\\foo\\bar\\baz", "..\\baz");
testRelativeWindows("\\\\foo\\bar\\baz", "\\\\foo\\bar\\baz-quux", "..\\baz-quux");
testRelativeWindows("C:\\baz-quux", "C:\\baz", "..\\baz");
testRelativeWindows("C:\\baz", "C:\\baz-quux", "..\\baz-quux");
testRelativeWindows("\\\\foo\\baz-quux", "\\\\foo\\baz", "..\\baz");
testRelativeWindows("\\\\foo\\baz", "\\\\foo\\baz-quux", "..\\baz-quux");
testRelativeWindows("C:\\baz", "\\\\foo\\bar\\baz", "\\\\foo\\bar\\baz");
testRelativeWindows("\\\\foo\\bar\\baz", "C:\\baz", "C:\\baz");
testRelativePosix("/var/lib", "/var", "..");
testRelativePosix("/var/lib", "/bin", "../../bin");
testRelativePosix("/var/lib", "/var/lib", "");
testRelativePosix("/var/lib", "/var/apache", "../apache");
testRelativePosix("/var/", "/var/lib", "lib");
testRelativePosix("/", "/var/lib", "var/lib");
testRelativePosix("/foo/test", "/foo/test/bar/package.json", "bar/package.json");
testRelativePosix("/Users/a/web/b/test/mails", "/Users/a/web/b", "../..");
testRelativePosix("/foo/bar/baz-quux", "/foo/bar/baz", "../baz");
testRelativePosix("/foo/bar/baz", "/foo/bar/baz-quux", "../baz-quux");
testRelativePosix("/baz-quux", "/baz", "../baz");
testRelativePosix("/baz", "/baz-quux", "../baz-quux");
}
fn testRelativePosix(from: []const u8, to: []const u8, expected_output: []const u8) {
const result = %%relativePosix(&debug.global_allocator, from, to);
assert(mem.eql(u8, result, expected_output));
}
fn testRelativeWindows(from: []const u8, to: []const u8, expected_output: []const u8) {
const result = %%relativeWindows(&debug.global_allocator, from, to);
assert(mem.eql(u8, result, expected_output));
}
error AccessDenied;
error FileNotFound;
error NotSupported;
error NotDir;
error NameTooLong;
error SymLinkLoop;
error InputOutput;
/// Return the canonicalized absolute pathname.
/// Expands all symbolic links and resolves references to `.`, `..`, and
/// extra `/` characters in ::pathname.
/// Caller must deallocate result.
pub fn real(allocator: &Allocator, pathname: []const u8) -> %[]u8 {
switch (builtin.os) {
Os.windows => {
const pathname_buf = %return allocator.alloc(u8, pathname.len + 1);
defer allocator.free(pathname_buf);
mem.copy(u8, pathname_buf, pathname);
pathname_buf[pathname.len] = 0;
const h_file = windows.CreateFileA(pathname_buf.ptr,
windows.GENERIC_READ, windows.FILE_SHARE_READ, null, windows.OPEN_EXISTING,
windows.FILE_ATTRIBUTE_NORMAL, null);
if (h_file == windows.INVALID_HANDLE_VALUE) {
const err = windows.GetLastError();
return switch (err) {
windows.ERROR.FILE_NOT_FOUND => error.FileNotFound,
windows.ERROR.ACCESS_DENIED => error.AccessDenied,
windows.ERROR.FILENAME_EXCED_RANGE => error.NameTooLong,
else => os.unexpectedErrorWindows(err),
};
}
defer os.close(h_file);
var buf = %return allocator.alloc(u8, 256);
%defer allocator.free(buf);
while (true) {
const buf_len = math.cast(windows.DWORD, buf.len) %% return error.NameTooLong;
const result = windows.GetFinalPathNameByHandleA(h_file, buf.ptr, buf_len, windows.VOLUME_NAME_DOS);
if (result == 0) {
const err = windows.GetLastError();
return switch (err) {
windows.ERROR.PATH_NOT_FOUND => error.FileNotFound,
windows.ERROR.NOT_ENOUGH_MEMORY => error.OutOfMemory,
windows.ERROR.INVALID_PARAMETER => unreachable,
else => os.unexpectedErrorWindows(err),
};
}
if (result > buf.len) {
buf = %return allocator.realloc(u8, buf, result);
continue;
}
// windows returns \\?\ prepended to the path
// we strip it because nobody wants \\?\ prepended to their path
const final_len = if (result > 4 and mem.startsWith(u8, buf, "\\\\?\\")) {
var i: usize = 4;
while (i < result) : (i += 1) {
buf[i - 4] = buf[i];
}
result - 4
} else {
result
};
return allocator.shrink(u8, buf, final_len);
}
},
Os.darwin, Os.macosx, Os.ios => {
// TODO instead of calling the libc function here, port the implementation
// to Zig, and then remove the NameTooLong error possibility.
const pathname_buf = %return allocator.alloc(u8, pathname.len + 1);
defer allocator.free(pathname_buf);
const result_buf = %return allocator.alloc(u8, posix.PATH_MAX);
%defer allocator.free(result_buf);
mem.copy(u8, pathname_buf, pathname);
pathname_buf[pathname.len] = 0;
const err = posix.getErrno(posix.realpath(pathname_buf.ptr, result_buf.ptr));
if (err > 0) {
return switch (err) {
posix.EINVAL => unreachable,
posix.EBADF => unreachable,
posix.EFAULT => unreachable,
posix.EACCES => error.AccessDenied,
posix.ENOENT => error.FileNotFound,
posix.ENOTSUP => error.NotSupported,
posix.ENOTDIR => error.NotDir,
posix.ENAMETOOLONG => error.NameTooLong,
posix.ELOOP => error.SymLinkLoop,
posix.EIO => error.InputOutput,
else => os.unexpectedErrorPosix(err),
};
}
return allocator.shrink(u8, result_buf, cstr.len(result_buf.ptr));
},
Os.linux => {
const fd = %return os.posixOpen(pathname, posix.O_PATH|posix.O_NONBLOCK|posix.O_CLOEXEC, 0, allocator);
defer os.close(fd);
var buf: ["/proc/self/fd/-2147483648".len]u8 = undefined;
const proc_path = fmt.bufPrint(buf[0..], "/proc/self/fd/{}", fd);
return os.readLink(allocator, proc_path);
},
else => @compileError("TODO implement os.path.real for " ++ @enumTagName(builtin.os)),
2017-05-22 21:26:12 -07:00
}
}
test "os.path.real" {
// at least call it so it gets compiled
_ = real(&debug.global_allocator, "some_path");
}