Compare commits
4 commits
c00cde50ed
...
43abed7e80
Author | SHA1 | Date | |
---|---|---|---|
43abed7e80 | |||
48da2945d8 | |||
866fee1502 | |||
368675acd7 |
4 changed files with 91 additions and 0 deletions
3
.gitignore
vendored
3
.gitignore
vendored
|
@ -1 +1,4 @@
|
||||||
zig-cache/
|
zig-cache/
|
||||||
|
deps.zig
|
||||||
|
images/
|
||||||
|
.zigmod/
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
const std = @import("std");
|
const std = @import("std");
|
||||||
|
const deps = @import("./deps.zig");
|
||||||
|
|
||||||
pub fn build(b: *std.build.Builder) void {
|
pub fn build(b: *std.build.Builder) void {
|
||||||
// Standard target options allows the person running `zig build` to choose
|
// Standard target options allows the person running `zig build` to choose
|
||||||
|
@ -14,6 +15,7 @@ pub fn build(b: *std.build.Builder) void {
|
||||||
const exe = b.addExecutable("yet-another-pomf-clone", "src/main.zig");
|
const exe = b.addExecutable("yet-another-pomf-clone", "src/main.zig");
|
||||||
exe.setTarget(target);
|
exe.setTarget(target);
|
||||||
exe.setBuildMode(mode);
|
exe.setBuildMode(mode);
|
||||||
|
deps.addAllTo(exe);
|
||||||
exe.install();
|
exe.install();
|
||||||
|
|
||||||
const run_cmd = exe.run();
|
const run_cmd = exe.run();
|
||||||
|
|
82
src/main.zig
82
src/main.zig
|
@ -1,5 +1,87 @@
|
||||||
const std = @import("std");
|
const std = @import("std");
|
||||||
|
const http = @import("apple_pie");
|
||||||
|
|
||||||
|
const images_dir_path = "./images";
|
||||||
|
|
||||||
pub fn main() anyerror!void {
|
pub fn main() anyerror!void {
|
||||||
std.log.info("All your codebase are belong to us.", .{});
|
std.log.info("All your codebase are belong to us.", .{});
|
||||||
|
|
||||||
|
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
||||||
|
defer _ = gpa.deinit();
|
||||||
|
|
||||||
|
const bind_addr = try std.net.Address.parseIp("0.0.0.0", 8080);
|
||||||
|
std.log.info("serving on {}", .{bind_addr});
|
||||||
|
|
||||||
|
// TODO: configurable path via env var
|
||||||
|
try std.fs.cwd().makePath(images_dir_path);
|
||||||
|
|
||||||
|
try http.listenAndServe(
|
||||||
|
&gpa.allocator,
|
||||||
|
bind_addr,
|
||||||
|
comptime http.router.router(&[_]http.router.Route{
|
||||||
|
http.router.get("/", index),
|
||||||
|
http.router.post("/api/upload", uploadFile),
|
||||||
|
http.router.get("/i/:filename", fetchFile),
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn index(response: *http.Response, request: http.Request) !void {
|
||||||
|
try response.writer().writeAll("Hello Zig!");
|
||||||
|
}
|
||||||
|
|
||||||
|
fn generateImageId(buffer: []u8) []const u8 {
|
||||||
|
var i: usize = 0;
|
||||||
|
|
||||||
|
const seed = @truncate(u64, @bitCast(u128, std.time.nanoTimestamp()));
|
||||||
|
var r = std.rand.DefaultPrng.init(seed);
|
||||||
|
|
||||||
|
while (i < 16) : (i += 1) {
|
||||||
|
// random ascii lowercase char
|
||||||
|
var idx = @intCast(u8, r.random.uintLessThan(u5, 24));
|
||||||
|
var letter = @as(u8, 97) + idx;
|
||||||
|
buffer[i] = letter;
|
||||||
|
}
|
||||||
|
|
||||||
|
return buffer[0..i];
|
||||||
|
}
|
||||||
|
|
||||||
|
fn uploadFile(response: *http.Response, request: http.Request) !void {
|
||||||
|
std.log.info("upload! got {d} bytes", .{request.body.len});
|
||||||
|
|
||||||
|
var image_id_buffer: [256]u8 = undefined;
|
||||||
|
const image_id = generateImageId(&image_id_buffer);
|
||||||
|
|
||||||
|
var image_path_buffer: [512]u8 = undefined;
|
||||||
|
const image_path = try std.fmt.bufPrint(
|
||||||
|
&image_path_buffer,
|
||||||
|
"{s}/{s}.jpg",
|
||||||
|
.{ images_dir_path, image_id },
|
||||||
|
);
|
||||||
|
|
||||||
|
const image_file = try std.fs.cwd().createFile(image_path, .{});
|
||||||
|
try image_file.writer().writeAll(request.body);
|
||||||
|
|
||||||
|
try response.writer().writeAll(image_path);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn fetchFile(response: *http.Response, request: http.Request, filename: []const u8) !void {
|
||||||
|
std.log.info("got name: {s}", .{filename});
|
||||||
|
var image_path_buffer: [512]u8 = undefined;
|
||||||
|
|
||||||
|
const images_dir = try std.fs.cwd().openDir(images_dir_path, .{});
|
||||||
|
const image_path = try std.fmt.bufPrint(
|
||||||
|
&image_path_buffer,
|
||||||
|
"{s}/{s}",
|
||||||
|
.{ images_dir_path, filename },
|
||||||
|
);
|
||||||
|
|
||||||
|
// TODO return 404 on error
|
||||||
|
const image_file = try std.fs.cwd().openFile(image_path, .{ .read = false });
|
||||||
|
while (true) {
|
||||||
|
var file_write_buffer: [1024]u8 = undefined;
|
||||||
|
const bytes_read = try image_file.read(&file_write_buffer);
|
||||||
|
if (bytes_read == 0) return;
|
||||||
|
try response.writer().writeAll(&file_write_buffer);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
4
zig.mod
4
zig.mod
|
@ -2,3 +2,7 @@ id: n5imv64i14btgrqm6qbzyhqs2xc2e7w3hxk1l9b9s5qmtfz0
|
||||||
name: yet-another-pomf-clone
|
name: yet-another-pomf-clone
|
||||||
main: src/main.zig
|
main: src/main.zig
|
||||||
dependencies:
|
dependencies:
|
||||||
|
|
||||||
|
- src: git https://github.com/Luukdegram/apple_pie
|
||||||
|
name: apple_pie
|
||||||
|
main: src/apple_pie.zig
|
||||||
|
|
Loading…
Reference in a new issue