backend: add image upload

1 files changed, 39 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);
+}