feat(io): File.isTTY

buzz-language · Sep 11, 2023 · 8bfaa6d · 8bfaa6d
1 parent a667a81
commit 8bfaa6d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/lib/buzz_io.zig b/src/lib/buzz_io.zig
@@ -19,6 +19,16 @@ export fn getStdErr(ctx: *api.NativeCtx) c_int {
     return 1;
 }
 
+export fn FileIsTTY(ctx: api.NativeCtx) c_int {
+    const handle: std.fs.File.Handle = @intCast(
+        ctx.vm.bz_peek(0).integer(),
+    );
+
+    ctx.vm.bz_pushBool(std.os.isatty(handle));
+
+    return 1;
+}
+
 fn handleFileOpenError(ctx: *api.NativeCtx, err: anytype) void {
     switch (err) {
         error.AccessDenied,

diff --git a/src/lib/io.buzz b/src/lib/io.buzz
@@ -18,6 +18,8 @@ extern fun getStdIn() > int;
 extern fun getStdOut() > int;
 || @private
 extern fun getStdErr() > int;
+|| @private
+extern fun FileIsTTY(int fd) > bool;
 
 || File mode with which you can open a file
 export enum FileMode {
@@ -70,6 +72,11 @@ export object File {
     fun write(str bytes) > void !> FileSystemError, ReadWriteError, UnexpectedError {
         FileWrite(this.fd, bytes: bytes);
     }
+
+    || @return true if file is TTY
+    fun isTTY() > bool {
+        return FileIsTTY(this.fd);
+    }
 }
 
 || stdin