RGBCube/ZTerm
1
Fork 0
mirror of https://github.com/RGBCube/ZTerm synced 2025-07-29 08:57:46 +00:00
Code Issues Activity Actions

Refactor Spinner

Browse source
This commit is contained in:
RGBCube 2023-05-17 23:23:56 +03:00
parent cc538aa18a
commit e9e4ba5529
2 changed files with 33 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/Spinner.zig
View file

@ -1,60 +1,55 @@
const std = @import("std"); const std = @import("std");
const AtomicBool = std.atomic.Atomic(bool); const Atomic = std.atomic.Atomic;
const Mutex = Thread.Mutex; const Mutex = Thread.Mutex;
const Thread = std.Thread; const Thread = std.Thread;
const time = std.time; const time = std.time;
const Spinner = @This(); const Spinner = @This();
const default_framerate = 100 * time.ns_per_ms; const default_loading_charset = [_][]const u8{ "|", "/", "-", "\\" };
const default_charset = [_][]const u8{ "|", "/", "-", "\\" }; const default_finished_charset = "✓";
keep_going: AtomicBool, keep_going: Atomic(bool) = Atomic(bool).init(false),
spinner_thread: ?Thread, spinner_thread: ?Thread = null,
// The number of nanoseconds to wait between frames. framerate_ns: u64 = 100 * time.ns_per_ms,
framerate_ns: u64,
charset: []const []const u8, loading_charset: []const []const u8 = &default_loading_charset,
message: []const u8, finished_charset: []const u8 = default_finished_charset,
pub fn new(framerate_ns: ?u64, charset: ?[]const []const u8, message: ?[]const u8) Spinner { loading_message: []const u8 = "",
return Spinner{ finished_message: []const u8 = "",
.keep_going = AtomicBool.init(false),
.spinner_thread = null,
.framerate_ns = framerate_ns orelse default_framerate,
.charset = charset orelse &default_charset,
.message = message orelse "",
};
}
pub fn start(sp: *Spinner) !void { pub fn start(sp: *Spinner) !void {
sp.keep_going.store(true, .SeqCst); sp.keep_going.store(true, .SeqCst);
sp.spinner_thread = try Thread.spawn(.{}, writer, .{sp}); sp.spinner_thread = try Thread.spawn(.{}, writer, .{sp});
} }
pub fn stop(sp: *Spinner) void { pub fn stop(sp: *Spinner) !void {
sp.keep_going.store(false, .SeqCst); sp.keep_going.store(false, .SeqCst);
if (sp.spinner_thread) |*thread| thread.join(); if (sp.spinner_thread) |*thread| thread.join();
}
fn animateSpinnerOnce(sp: *Spinner) !void {
var stdOut = std.io.getStdOut(); var stdOut = std.io.getStdOut();
for (sp.charset) |frame| { _ = try stdOut.write("\r");
_ = try stdOut.write(sp.finished_charset);
_ = try stdOut.write(" ");
_ = try stdOut.write(sp.finished_message);
}
fn writer(sp: *Spinner) !void {
var stdOut = std.io.getStdOut();
var current_char_idx: usize = 0;
while (true) : (current_char_idx += 1) {
if (!sp.keep_going.load(.SeqCst)) break;
if (current_char_idx >= sp.loading_charset.len - 1) current_char_idx = 0;
_ = try stdOut.write("\r"); _ = try stdOut.write("\r");
_ = try stdOut.write(frame); _ = try stdOut.write(sp.loading_charset[current_char_idx]);
_ = try stdOut.write(" "); _ = try stdOut.write(" ");
_ = try stdOut.write(sp.message); _ = try stdOut.write(sp.loading_message);
time.sleep(sp.framerate_ns); time.sleep(sp.framerate_ns);
} }
} }
fn writer(sp: *Spinner) !void {
while (true) {
if (!sp.keep_going.load(.SeqCst)) break;
try sp.animateSpinnerOnce();
}
}

9
zterm.zig
View file

@ -5,10 +5,13 @@ const std = @import("std");
const time = std.time; const time = std.time;
pub fn main() !void { pub fn main() !void {
var sp = Spinner.new(100 * time.ns_per_ms, null, "Loading..."); var sp = Spinner{
.loading_message = "Loading",
.finished_message = "Done",
};
try sp.start(); try sp.start();
time.sleep(5 * time.ns_per_s); time.sleep(3 * time.ns_per_s);
sp.stop(); try sp.stop();
} }