diff options
Diffstat (limited to 'src/storage')
| -rw-r--r-- | src/storage/image.zig | 39 | ||||
| -rw-r--r-- | src/storage/root.zig | 1 |
2 files changed, 40 insertions, 0 deletions
diff --git a/src/storage/image.zig b/src/storage/image.zig new file mode 100644 index 0000000..73bca84 --- /dev/null +++ b/src/storage/image.zig @@ -0,0 +1,39 @@ +const std = @import("std"); +const Storage = @import("root.zig"); + +const id_size = 32; + +fn new_id() [id_size]u8 { + var buffer: [id_size]u8 = undefined; + var raw_buffer: [id_size / 2]u8 = undefined; + std.crypto.random.bytes(&raw_buffer); + + var writer = std.Io.Writer.fixed(buffer[0..]); + writer.print("{x}", .{raw_buffer}) catch unreachable; + + return buffer[0..].*; +} + +pub fn new( + storage: *Storage, + reader: *std.Io.Reader, + size: usize, +) !void { + var dir = storage.dir.openDir("image", .{}) catch blk: { + try storage.dir.makeDir("image"); + break :blk try storage.dir.openDir("image", .{}); + }; + defer dir.close(); + + var file_name: [id_size+4]u8 = undefined; + @memcpy(file_name[0..id_size], &new_id()); + @memcpy(file_name[id_size..], ".jpg"); + + var file = try dir.createFile(&file_name, .{}); + defer file.close(); + + var buffer: [1024]u8 = undefined; + var file_writer = file.writer(&buffer); + + try reader.streamExact(&file_writer.interface, size); +} diff --git a/src/storage/root.zig b/src/storage/root.zig index f1a753c..c121ca9 100644 --- a/src/storage/root.zig +++ b/src/storage/root.zig @@ -3,6 +3,7 @@ const config = @import("../config.zig"); const prompt = @import("../prompt.zig"); pub const User = @import("user.zig"); +pub const Image = @import("image.zig"); pub const SessionManager = @import("session-manager/root.zig"); pub const Session = SessionManager.Session; |