src/storage/image.zig


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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);
}