╭ progress.zig ─────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
🬐 🬠 🬞🬂🬏 🬟 🬑 🬟 🬑 ╭cc3ada4e..╮ cc3ada4e.. see-em: 🬋🬋🬋 🬋🬋🬋 🬋 🬋 │files │ commits refs discuss issues new login 🬐 🬠 🬐 🬠 🬑 🬟 🬰 ╰──────────╯ (558) user
╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╮ │http│ ssh │ │ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╯
.. embed/ event/ test/ ui/ command.zig diff3.zig event.zig find.zig fork.zig lib.zig main.zig main_wasm.zig patch.zig ╭─────────────────────────────────╮ │progress.zig │ ╰─────────────────────────────────╯ push.zig search.zig search_commit.zig search_thread.zig serve.zig serve_common.zig serve_http.zig serve_ssh.zig serve_ssh_protocol.zig test.zig testnet.zig try.zig ui.zig web.zig
1 const std = @import("std"); 2 const xit = @import("xit"); 3 const rp = xit.repo; 4 const ssh = @import("./serve_ssh_protocol.zig"); 5 6 // reporting is optional and best effort: a failed report must not fail the work 7 pub fn report(comptime repo_opts: rp.RepoOpts(.xit), io: std.Io, progress_ctx_maybe: ?repo_opts.ProgressCtx, event: rp.ProgressEvent) void { 8 if (repo_opts.ProgressCtx != void) { 9 if (progress_ctx_maybe) |progress_ctx| progress_ctx.run(io, event) catch {}; 10 } 11 } 12 13 // a push's or a clone's progress, sent on the pack stream's sideband 14 pub const Sideband = struct { 15 // a push reports through its deferred response, which knows whether the 16 // client wanted sideband at all. a clone has none and writes band 2 itself. 17 response: ?*xit.net_server_receive_pack.Response = null, 18 writer: *std.Io.Writer, 19 // the client to probe, absent when nothing is listening 20 sess: ?*ssh.SessionCtx = null, 21 // set once it has hung up, so the phase in flight can stop too 22 gone: bool = false, 23 label: []const u8 = "Processing commits", 24 count: usize = 0, 25 total: usize = 0, 26 // the last reported step, so a line is sent only when it changes 27 step: usize = 0, 28 29 // how many items a phase with no total advances between reports 30 const report_every = 1000; 31 32 // the phases a client is told about, listed so a new kind has to be 33 // decided on rather than inherit whatever label was last set 34 fn reported(kind: rp.ProgressKind) bool { 35 return switch (kind) { 36 .writing_object_from_pack, .checking_object, .walking_commit, .enumerating_object, .compressing_object, .writing_patch => true, 37 .writing_object, .sending_bytes, .receiving_bytes => false, 38 }; 39 } 40 41 pub fn run(self: *Sideband, _: std.Io, event: rp.ProgressEvent) !void { 42 switch (event) { 43 .start => |start| { 44 if (!reported(start.kind)) return; 45 // a pack phase carries no label of its own; the rest send a 46 // .text first 47 switch (start.kind) { 48 .writing_object_from_pack => self.label = "Unpacking objects", 49 .checking_object => self.label = "Checking connectivity", 50 .walking_commit => self.label = "Walking commits", 51 .enumerating_object => self.label = "Enumerating objects", 52 .compressing_object => self.label = "Compressing objects", 53 else => {}, 54 } 55 self.total = start.estimated_total_items; 56 self.count = 0; 57 self.step = 0; 58 }, 59 .complete_one => |kind| { 60 if (!reported(kind)) return; 61 self.count += 1; 62 }, 63 .end => |kind| { 64 if (!reported(kind)) return; 65 }, 66 // each phase names itself before reporting its count 67 .text => |text| { 68 self.label = text; 69 return; 70 }, 71 else => return, 72 } 73 // a phase that knows no total counts instead, and stays quiet until it 74 // has something to count 75 if (self.total == 0 and self.count == 0) return; 76 77 // every report flushes, so one is sent per percent, or per block of 78 // items when there is no total to turn into one 79 const step: usize = if (self.total > 0) @intCast(@as(u128, self.count) * 100 / self.total) else self.count / report_every; 80 if (event == .complete_one and step == self.step) return; 81 self.step = step; 82 83 // reporting is the only moment a transaction looks up from its work 84 if (self.sess) |sess| if (sess.peerGone()) { 85 self.gone = true; 86 return error.ClientGone; 87 }; 88 89 var buffer: [128]u8 = undefined; 90 const suffix = if (event == .end) "\n" else "\r"; 91 const text = if (self.total > 0) 92 try std.fmt.bufPrint(&buffer, "{s}: {d}% ({d}/{d}){s}", .{ self.label, step, self.count, self.total, suffix }) 93 else 94 try std.fmt.bufPrint(&buffer, "{s}: {d}{s}", .{ self.label, self.count, suffix }); 95 if (self.response) |response| { 96 try response.progress(self.writer, text); 97 } else { 98 try xit.net_server_pkt.sendSideband(self.writer, 2, text); 99 try self.writer.flush(); 100 } 101 } 102 }; 103 104 // whether the client the work is reporting to has hung up. only a sideband 105 // reporter probes, so any other context never cancels. 106 pub fn cancelled(comptime repo_opts: rp.RepoOpts(.xit), progress_ctx_maybe: ?repo_opts.ProgressCtx) bool { 107 if (repo_opts.ProgressCtx != *Sideband) return false; 108 const progress_ctx = progress_ctx_maybe orelse return false; 109 return progress_ctx.gone; 110 } 111