forked from vieter-v/vieter
69 lines
1.4 KiB
V
69 lines
1.4 KiB
V
module main
|
|
|
|
import web
|
|
import os
|
|
import io
|
|
import repo
|
|
|
|
const port = 8000
|
|
|
|
const buf_size = 1_000_000
|
|
|
|
struct App {
|
|
web.Context
|
|
pub:
|
|
api_key string [required; web_global]
|
|
dl_dir string [required; web_global]
|
|
pub mut:
|
|
repo repo.Repo [required; web_global]
|
|
}
|
|
|
|
[noreturn]
|
|
fn exit_with_message(code int, msg string) {
|
|
eprintln(msg)
|
|
exit(code)
|
|
}
|
|
|
|
fn reader_to_file(mut reader io.BufferedReader, length int, path string) ? {
|
|
mut file := os.create(path) ?
|
|
defer {
|
|
file.close()
|
|
}
|
|
|
|
mut buf := []byte{len: buf_size}
|
|
mut bytes_left := length
|
|
|
|
// Repeat as long as the stream still has data
|
|
for bytes_left > 0 {
|
|
// TODO check if just breaking here is safe
|
|
bytes_read := reader.read(mut buf) or { break }
|
|
bytes_left -= bytes_read
|
|
|
|
mut to_write := bytes_read
|
|
|
|
for to_write > 0 {
|
|
// TODO don't just loop infinitely here
|
|
bytes_written := file.write(buf[bytes_read - to_write..bytes_read]) or { continue }
|
|
|
|
to_write = to_write - bytes_written
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
key := os.getenv_opt('API_KEY') or { exit_with_message(1, 'No API key was provided.') }
|
|
repo_dir := os.getenv_opt('REPO_DIR') or {
|
|
exit_with_message(1, 'No repo directory was configured.')
|
|
}
|
|
|
|
if os.args.len == 1 {
|
|
exit_with_message(1, 'No action provided.')
|
|
}
|
|
|
|
match os.args[1] {
|
|
'server' { server(key, repo_dir) }
|
|
'build' { build(key, repo_dir) ? }
|
|
else { exit_with_message(1, 'Unknown action: ${os.args[1]}') }
|
|
}
|
|
}
|