| ... | ... | @@ -54,3 +54,43 @@ pub fn fromStd(writer_ptr: *std.Io.Writer) AnyWritable { |
| 54 | 54 | .state = @ptrCast(@constCast(writer_ptr)), |
| 55 | 55 | }; |
| 56 | 56 | } |
| 57 | |
| 58 | pub fn toStd(r: AnyWritable, buf: []u8) StdWriter { |
| 59 | const S = struct { |
| 60 | fn drain(sw: *std.Io.Writer, data: []const []const u8, splat: usize) error{WriteFailed}!usize { |
| 61 | const w: *StdWriter = @alignCast(@fieldParentPtr("sw", sw)); |
| 62 | while (true) { |
| 63 | const rem = w.sw.buffered(); |
| 64 | const n = w.aw.write(rem) catch return error.WriteFailed; |
| 65 | const l = w.sw.consume(n); |
| 66 | if (l == 0) break; |
| 67 | } |
| 68 | var n: usize = 0; |
| 69 | const slice = data[0 .. data.len - 1]; |
| 70 | w.aw.writevAll(slice) catch return error.WriteFailed; |
| 71 | for (slice) |x| n += x.len; |
| 72 | const pattern = data[slice.len]; |
| 73 | w.aw.writeNTimes(pattern, splat) catch return error.WriteFailed; |
| 74 | n += pattern.len * splat; |
| 75 | return n; |
| 76 | } |
| 77 | fn flush(sw: *std.Io.Writer) error{WriteFailed}!void { |
| 78 | const w: *StdWriter = @alignCast(@fieldParentPtr("sw", sw)); |
| 79 | return w.aw.writeAll(w.sw.buffered()) catch return error.WriteFailed; |
| 80 | } |
| 81 | }; |
| 82 | return .{ |
| 83 | .aw = r, |
| 84 | .sw = .{ |
| 85 | .vtable = &.{ |
| 86 | .drain = S.drain, |
| 87 | .flush = S.flush, |
| 88 | }, |
| 89 | .buffer = buf, |
| 90 | }, |
| 91 | }; |
| 92 | } |
| 93 | const StdWriter = struct { |
| 94 | aw: AnyWritable, |
| 95 | sw: std.Io.Writer, |
| 96 | }; |