vieter/src/server/routes.v

98 lines
2.7 KiB
Coq
Raw Normal View History

2022-02-21 20:51:41 +01:00
module server
2022-01-11 14:24:29 +01:00
import web
import os
import repo
import time
2022-01-19 21:24:46 +01:00
import rand
2022-02-21 20:51:41 +01:00
import util
import net.http
2022-02-02 13:34:28 +01:00
// healthcheck just returns a string, but can be used to quickly check if the
// server is still responsive.
['/health'; get]
pub fn (mut app App) healthcheck() web.Result {
2022-03-28 14:24:26 +02:00
return app.json(200, new_response('Healthy.'))
}
2022-01-14 22:46:04 +01:00
// get_root handles a GET request for a file on the root
['/:filename'; get; head]
fn (mut app App) get_root(filename string) web.Result {
mut full_path := ''
if filename.ends_with('.db') || filename.ends_with('.files') {
full_path = os.join_path_single(app.repo.repo_dir, '${filename}.tar.gz')
} else if filename.ends_with('.db.tar.gz') || filename.ends_with('.files.tar.gz') {
full_path = os.join_path_single(app.repo.repo_dir, '$filename')
} else {
full_path = os.join_path_single(app.repo.pkg_dir, filename)
}
// Scuffed way to respond to HEAD requests
if app.req.method == http.Method.head {
if os.exists(full_path) {
2022-03-28 14:24:26 +02:00
return app.status(200)
}
return app.not_found()
}
return app.file(full_path)
}
2022-01-19 21:24:46 +01:00
['/publish'; post]
fn (mut app App) put_package() web.Result {
if !app.is_authorized() {
2022-03-28 14:24:26 +02:00
return app.json(401, new_response('Unauthorized.'))
2022-01-19 21:24:46 +01:00
}
2022-01-19 21:24:46 +01:00
mut pkg_path := ''
2022-01-11 14:24:29 +01:00
2022-01-19 21:24:46 +01:00
if length := app.req.header.get(.content_length) {
// Generate a random filename for the temp file
2022-02-21 20:40:13 +01:00
pkg_path = os.join_path_single(app.conf.download_dir, rand.uuid_v4())
2022-01-11 14:24:29 +01:00
2022-01-19 21:24:46 +01:00
for os.exists(pkg_path) {
2022-02-21 20:40:13 +01:00
pkg_path = os.join_path_single(app.conf.download_dir, rand.uuid_v4())
2022-01-19 21:24:46 +01:00
}
app.ldebug("Uploading $length bytes (${util.pretty_bytes(length.int())}) to '$pkg_path'.")
2022-01-19 21:24:46 +01:00
// This is used to time how long it takes to upload a file
mut sw := time.new_stopwatch(time.StopWatchOptions{ auto_start: true })
2022-02-21 20:51:41 +01:00
util.reader_to_file(mut app.reader, length.int(), pkg_path) or {
2022-01-19 21:24:46 +01:00
app.lwarn("Failed to upload '$pkg_path'")
2022-01-11 14:24:29 +01:00
2022-03-28 14:24:26 +02:00
return app.json(500, new_response('Failed to upload file.'))
2022-01-19 21:24:46 +01:00
}
2022-01-19 21:24:46 +01:00
sw.stop()
app.ldebug("Upload of '$pkg_path' completed in ${sw.elapsed().seconds():.3}s.")
} else {
app.lwarn('Tried to upload package without specifying a Content-Length.')
2022-03-28 14:24:26 +02:00
// length required
return app.status(411)
2022-01-19 21:24:46 +01:00
}
2022-01-11 14:24:29 +01:00
2022-02-02 13:24:31 +01:00
res := app.repo.add_from_path(pkg_path) or {
2022-01-20 19:39:38 +01:00
app.lerror('Error while adding package: $err.msg')
2022-01-11 14:24:29 +01:00
2022-02-02 13:24:31 +01:00
os.rm(pkg_path) or { app.lerror("Failed to remove download '$pkg_path': $err.msg") }
2022-03-28 14:24:26 +02:00
return app.json(500, new_response('Failed to add package.'))
2022-01-19 21:24:46 +01:00
}
2022-02-02 13:24:31 +01:00
if !res.added {
os.rm(pkg_path) or { app.lerror("Failed to remove download '$pkg_path': $err.msg") }
2022-02-02 13:24:31 +01:00
app.lwarn("Duplicate package '$res.pkg.full_name()'.")
2022-01-11 14:40:25 +01:00
2022-03-28 14:24:26 +02:00
return app.json(400, new_response('File already exists.'))
2022-01-19 21:24:46 +01:00
}
2022-01-13 21:47:14 +01:00
2022-02-02 13:24:31 +01:00
app.linfo("Added '$res.pkg.full_name()' to repository.")
2022-01-11 14:24:29 +01:00
2022-03-28 14:24:26 +02:00
return app.json(200, new_response('Package added successfully.'))
2022-01-19 21:24:46 +01:00
}