2022-02-21 20:51:41 +01:00
|
|
|
module server
|
2022-01-11 14:24:29 +01:00
|
|
|
|
|
|
|
import web
|
|
|
|
import os
|
|
|
|
import repo
|
2022-01-11 18:19:41 +01:00
|
|
|
import time
|
2022-01-19 21:24:46 +01:00
|
|
|
import rand
|
2022-02-21 20:51:41 +01:00
|
|
|
import util
|
2022-02-25 21:47:28 +01:00
|
|
|
import net.http
|
2022-01-11 18:19:41 +01:00
|
|
|
|
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.
|
2022-01-31 11:11:45 +01:00
|
|
|
['/health'; get]
|
|
|
|
pub fn (mut app App) healthcheck() web.Result {
|
2022-01-31 21:14:43 +01:00
|
|
|
return app.text('Healthy')
|
2022-01-31 11:11:45 +01:00
|
|
|
}
|
|
|
|
|
2022-03-27 17:00:11 +02:00
|
|
|
['/:repo/:arch/:filename'; get; head]
|
|
|
|
fn (mut app App) get_repo_file(repo string, arch string, filename string) web.Result {
|
2022-01-11 21:01:09 +01:00
|
|
|
mut full_path := ''
|
|
|
|
|
2022-03-27 17:00:11 +02:00
|
|
|
db_exts := ['.db', '.files', '.db.tar.gz', '.files.tar.gz']
|
|
|
|
|
|
|
|
if db_exts.any(filename.ends_with(it)) {
|
|
|
|
full_path = os.join_path(app.repo.data_dir, repo, arch, filename)
|
|
|
|
|
|
|
|
// repo-add does this using symlinks, but we just change the requested
|
|
|
|
// path
|
|
|
|
if !full_path.ends_with('.tar.gz') {
|
|
|
|
full_path += '.tar.gz'
|
|
|
|
}
|
2022-01-11 21:01:09 +01:00
|
|
|
} else {
|
2022-01-31 21:14:43 +01:00
|
|
|
full_path = os.join_path_single(app.repo.pkg_dir, filename)
|
2022-01-11 21:01:09 +01:00
|
|
|
}
|
|
|
|
|
2022-02-25 21:47:28 +01:00
|
|
|
// Scuffed way to respond to HEAD requests
|
|
|
|
if app.req.method == http.Method.head {
|
|
|
|
if os.exists(full_path) {
|
|
|
|
return app.ok('')
|
|
|
|
}
|
|
|
|
|
|
|
|
return app.not_found()
|
|
|
|
}
|
|
|
|
|
2022-01-11 21:01:09 +01:00
|
|
|
return app.file(full_path)
|
|
|
|
}
|
|
|
|
|
2022-03-27 17:00:11 +02:00
|
|
|
['/:repo/publish'; post]
|
|
|
|
fn (mut app App) put_package(repo string) web.Result {
|
2022-01-19 21:24:46 +01:00
|
|
|
if !app.is_authorized() {
|
|
|
|
return app.text('Unauthorized.')
|
|
|
|
}
|
2022-01-11 21:01:09 +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
|
|
|
}
|
2022-01-11 16:08:11 +01:00
|
|
|
|
2022-02-21 22:22:36 +01:00
|
|
|
app.ldebug("Uploading $length bytes (${util.pretty_bytes(length.int())}) to '$pkg_path'.")
|
2022-01-11 18:19:41 +01:00
|
|
|
|
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-01-11 18:19:41 +01:00
|
|
|
|
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-01-19 21:24:46 +01:00
|
|
|
return app.text('Failed to upload file.')
|
|
|
|
}
|
2022-01-11 18:19:41 +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.')
|
|
|
|
return app.text("Content-Type header isn't set.")
|
|
|
|
}
|
2022-01-11 14:24:29 +01:00
|
|
|
|
2022-03-27 17:00:11 +02:00
|
|
|
res := app.repo.add_pkg_from_path(repo, 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-01-30 14:13:37 +01:00
|
|
|
|
2022-01-19 21:24:46 +01:00
|
|
|
return app.text('Failed to add package.')
|
|
|
|
}
|
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-01-30 14:13:37 +01:00
|
|
|
|
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-01-19 21:24:46 +01:00
|
|
|
return app.text('File already exists.')
|
|
|
|
}
|
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-01-19 21:24:46 +01:00
|
|
|
return app.text('Package added successfully.')
|
|
|
|
}
|