mirror of
https://github.com/oven-sh/bun
synced 2026-02-09 18:38:55 +00:00
61 lines
2.5 KiB
Zig
61 lines
2.5 KiB
Zig
const std = @import("std");
|
|
const bun = @import("bun");
|
|
fn isValid(buf: *[bun.MAX_PATH_BYTES]u8, segment: []const u8, bin: []const u8) ?u16 {
|
|
std.mem.copy(u8, buf, segment);
|
|
buf[segment.len] = std.fs.path.sep;
|
|
std.mem.copy(u8, buf[segment.len + 1 ..], bin);
|
|
buf[segment.len + 1 + bin.len ..][0] = 0;
|
|
const filepath = buf[0 .. segment.len + 1 + bin.len :0];
|
|
if (!checkPath(filepath)) return null;
|
|
return @intCast(u16, filepath.len);
|
|
}
|
|
|
|
fn checkPath(filepath: [:0]const u8) bool {
|
|
// TODO: is there a single system call for executable AND file?
|
|
std.os.accessZ(filepath, std.os.X_OK) catch return false;
|
|
std.os.accessZ(filepath, std.os.F_OK) catch return false;
|
|
return true;
|
|
}
|
|
|
|
// Like /usr/bin/which but without needing to exec a child process
|
|
// Remember to resolve the symlink if necessary
|
|
pub fn which(buf: *[bun.MAX_PATH_BYTES]u8, path: []const u8, cwd: []const u8, bin: []const u8) ?[:0]const u8 {
|
|
if (bin.len == 0) return null;
|
|
|
|
// handle absolute paths
|
|
if (std.fs.path.isAbsolute(bin)) {
|
|
std.mem.copy(u8, buf, bin);
|
|
buf[bin.len] = 0;
|
|
var binZ: [:0]u8 = buf[0..bin.len :0];
|
|
if (checkPath(binZ)) return binZ;
|
|
|
|
// note that directories are often executable
|
|
// TODO: should we return null here? What about the case where ytou have
|
|
// /foo/bar/baz as a path and you're in /home/jarred?
|
|
}
|
|
|
|
if (isValid(buf, std.mem.trimRight(u8, cwd, std.fs.path.sep_str), bin)) |len| {
|
|
return buf[0..len :0];
|
|
}
|
|
|
|
var path_iter = std.mem.tokenize(u8, path, ":");
|
|
while (path_iter.next()) |segment| {
|
|
if (isValid(buf, segment, bin)) |len| {
|
|
return buf[0..len :0];
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
test "which" {
|
|
var buf: [bun.MAX_PATH_BYTES]u8 = undefined;
|
|
var realpath = bun.getenvZ("PATH") orelse unreachable;
|
|
var whichbin = which(&buf, realpath, try std.process.getCwdAlloc(std.heap.c_allocator), "which");
|
|
try std.testing.expectEqualStrings(whichbin orelse return std.debug.assert(false), "/usr/bin/which");
|
|
try std.testing.expect(null == which(&buf, realpath, try std.process.getCwdAlloc(std.heap.c_allocator), "baconnnnnn"));
|
|
try std.testing.expect(null != which(&buf, realpath, try std.process.getCwdAlloc(std.heap.c_allocator), "zig"));
|
|
try std.testing.expect(null == which(&buf, realpath, try std.process.getCwdAlloc(std.heap.c_allocator), "bin"));
|
|
try std.testing.expect(null == which(&buf, realpath, try std.process.getCwdAlloc(std.heap.c_allocator), "usr"));
|
|
}
|