forked from vieter-v/vieter
Finally enabled web server again
parent
57fe767a70
commit
d04ba06ad3
2
Makefile
2
Makefile
|
@ -38,7 +38,7 @@ c:
|
||||||
# Run the server in the default 'data' directory
|
# Run the server in the default 'data' directory
|
||||||
.PHONY: run
|
.PHONY: run
|
||||||
run: vieter
|
run: vieter
|
||||||
API_KEY=test REPO_DIR=data LOG_LEVEL=DEBUG ./vieter
|
API_KEY=test DOWNLOAD_DIR=data/downloads REPO_DIR=data/repo PKG_DIR=data/pkgs LOG_LEVEL=DEBUG ./vieter
|
||||||
|
|
||||||
.PHONY: run-prod
|
.PHONY: run-prod
|
||||||
run-prod: prod
|
run-prod: prod
|
||||||
|
|
121
src/main.v
121
src/main.v
|
@ -16,6 +16,7 @@ struct App {
|
||||||
web.Context
|
web.Context
|
||||||
pub:
|
pub:
|
||||||
api_key string [required; web_global]
|
api_key string [required; web_global]
|
||||||
|
dl_dir string [required; web_global]
|
||||||
pub mut:
|
pub mut:
|
||||||
repo repo.Repo [required; web_global]
|
repo repo.Repo [required; web_global]
|
||||||
}
|
}
|
||||||
|
@ -53,65 +54,67 @@ fn reader_to_file(mut reader io.BufferedReader, length int, path string) ? {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// fn main2() {
|
|
||||||
// // Configure logger
|
|
||||||
// log_level_str := os.getenv_opt('LOG_LEVEL') or { 'WARN' }
|
|
||||||
// log_level := log.level_from_tag(log_level_str) or {
|
|
||||||
// exit_with_message(1, 'Invalid log level. The allowed values are FATAL, ERROR, WARN, INFO & DEBUG.')
|
|
||||||
// }
|
|
||||||
// log_file := os.getenv_opt('LOG_FILE') or { 'vieter.log' }
|
|
||||||
|
|
||||||
// mut logger := log.Log{
|
|
||||||
// level: log_level
|
|
||||||
// }
|
|
||||||
|
|
||||||
// logger.set_full_logpath(log_file)
|
|
||||||
// logger.log_to_console_too()
|
|
||||||
|
|
||||||
// defer {
|
|
||||||
// logger.info('Flushing log file')
|
|
||||||
// logger.flush()
|
|
||||||
// logger.close()
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // Configure web server
|
|
||||||
// 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.')
|
|
||||||
// }
|
|
||||||
|
|
||||||
// repo := repo.Repo{
|
|
||||||
// dir: repo_dir
|
|
||||||
// name: db_name
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // We create the upload directory during startup
|
|
||||||
// if !os.is_dir(repo.pkg_dir()) {
|
|
||||||
// os.mkdir_all(repo.pkg_dir()) or {
|
|
||||||
// exit_with_message(2, "Failed to create repo directory '$repo.pkg_dir()'.")
|
|
||||||
// }
|
|
||||||
|
|
||||||
// logger.info("Created package directory '$repo.pkg_dir()'.")
|
|
||||||
// }
|
|
||||||
|
|
||||||
// web.run(&App{
|
|
||||||
// logger: logger
|
|
||||||
// api_key: key
|
|
||||||
// repo: repo
|
|
||||||
// }, port)
|
|
||||||
// }
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
r := repo.new('data/repo', 'data/pkgs') or { return }
|
// Configure logger
|
||||||
print(r.add_from_path('test/homebank-5.5.1-1-x86_64.pkg.tar.zst') or { panic('you fialed') })
|
log_level_str := os.getenv_opt('LOG_LEVEL') or { 'WARN' }
|
||||||
|
log_level := log.level_from_tag(log_level_str) or {
|
||||||
|
exit_with_message(1, 'Invalid log level. The allowed values are FATAL, ERROR, WARN, INFO & DEBUG.')
|
||||||
|
}
|
||||||
|
log_file := os.getenv_opt('LOG_FILE') or { 'vieter.log' }
|
||||||
|
|
||||||
// archive.list_filenames()
|
mut logger := log.Log{
|
||||||
// res := pkg.read_pkg('test/jjr-joplin-desktop-2.6.10-4-x86_64.pkg.tar.zst') or {
|
level: log_level
|
||||||
// eprintln(err.msg)
|
}
|
||||||
// return
|
|
||||||
// }
|
logger.set_full_logpath(log_file)
|
||||||
// println(info)
|
logger.log_to_console_too()
|
||||||
// println('hey')
|
|
||||||
// print(res.to_desc())
|
defer {
|
||||||
// print(res.to_files())
|
logger.info('Flushing log file')
|
||||||
|
logger.flush()
|
||||||
|
logger.close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Configure web server
|
||||||
|
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.')
|
||||||
|
}
|
||||||
|
pkg_dir := os.getenv_opt('PKG_DIR') or {
|
||||||
|
exit_with_message(1, 'No package directory was configured.')
|
||||||
|
}
|
||||||
|
dl_dir := os.getenv_opt('DOWNLOAD_DIR') or {
|
||||||
|
exit_with_message(1, 'No download directory was configured.')
|
||||||
|
}
|
||||||
|
|
||||||
|
// This also creates the directories if needed
|
||||||
|
repo := repo.new(repo_dir, pkg_dir) or {
|
||||||
|
exit_with_message(1, 'Failed to create required directories.')
|
||||||
|
}
|
||||||
|
|
||||||
|
os.mkdir_all(dl_dir) or { exit_with_message(1, 'Failed to create download directory.') }
|
||||||
|
|
||||||
|
web.run(&App{
|
||||||
|
logger: logger
|
||||||
|
api_key: key
|
||||||
|
dl_dir: dl_dir
|
||||||
|
repo: repo
|
||||||
|
}, port)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// fn main() {
|
||||||
|
// r := repo.new('data/repo', 'data/pkgs') or { return }
|
||||||
|
// print(r.add_from_path('test/jjr-joplin-desktop-2.6.10-4-x86_64.pkg.tar.zst') or {
|
||||||
|
// panic('you fialed')
|
||||||
|
// })
|
||||||
|
|
||||||
|
// // archive.list_filenames()
|
||||||
|
// // res := pkg.read_pkg('test/jjr-joplin-desktop-2.6.10-4-x86_64.pkg.tar.zst') or {
|
||||||
|
// // eprintln(err.msg)
|
||||||
|
// // return
|
||||||
|
// // }
|
||||||
|
// // println(info)
|
||||||
|
// // println('hey')
|
||||||
|
// // print(res.to_desc())
|
||||||
|
// // print(res.to_files())
|
||||||
|
// }
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
module package
|
module package
|
||||||
|
|
||||||
import time
|
|
||||||
import os
|
import os
|
||||||
import util
|
import util
|
||||||
|
|
||||||
|
@ -93,6 +92,7 @@ pub mut:
|
||||||
checkdepends []string
|
checkdepends []string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// checksum calculates the md5 & sha256 hash of the package
|
||||||
pub fn (p &Pkg) checksum() ?(string, string) {
|
pub fn (p &Pkg) checksum() ?(string, string) {
|
||||||
return util.hash_file(p.path)
|
return util.hash_file(p.path)
|
||||||
}
|
}
|
||||||
|
@ -214,6 +214,7 @@ fn format_entry(key string, value string) string {
|
||||||
return '\n%$key%\n$value\n'
|
return '\n%$key%\n$value\n'
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// filename returns the correct filename of the package file
|
||||||
pub fn (pkg &Pkg) filename() string {
|
pub fn (pkg &Pkg) filename() string {
|
||||||
p := pkg.info
|
p := pkg.info
|
||||||
|
|
||||||
|
|
|
@ -26,6 +26,7 @@ pub:
|
||||||
pkg_dir string [required]
|
pkg_dir string [required]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// new creates a new Repo & creates the directories as needed
|
||||||
pub fn new(repo_dir string, pkg_dir string) ?Repo {
|
pub fn new(repo_dir string, pkg_dir string) ?Repo {
|
||||||
if !os.is_dir(repo_dir) {
|
if !os.is_dir(repo_dir) {
|
||||||
os.mkdir_all(repo_dir) or { return error('Failed to create repo directory.') }
|
os.mkdir_all(repo_dir) or { return error('Failed to create repo directory.') }
|
||||||
|
|
75
src/routes.v
75
src/routes.v
|
@ -4,6 +4,7 @@ import web
|
||||||
import os
|
import os
|
||||||
import repo
|
import repo
|
||||||
import time
|
import time
|
||||||
|
import rand
|
||||||
|
|
||||||
const prefixes = ['B', 'KB', 'MB', 'GB']
|
const prefixes = ['B', 'KB', 'MB', 'GB']
|
||||||
|
|
||||||
|
@ -38,57 +39,55 @@ fn (mut app App) get_root(filename string) web.Result {
|
||||||
return app.file(full_path)
|
return app.file(full_path)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ['/pkgs/:pkg'; put]
|
['/publish'; post]
|
||||||
// fn (mut app App) put_package(pkg string) web.Result {
|
fn (mut app App) put_package() web.Result {
|
||||||
// if !app.is_authorized() {
|
if !app.is_authorized() {
|
||||||
// return app.text('Unauthorized.')
|
return app.text('Unauthorized.')
|
||||||
// }
|
}
|
||||||
|
|
||||||
// if !is_pkg_name(pkg) {
|
mut pkg_path := ''
|
||||||
// app.lwarn("Invalid package name '$pkg'.")
|
|
||||||
|
|
||||||
// return app.text('Invalid filename.')
|
if length := app.req.header.get(.content_length) {
|
||||||
// }
|
// Generate a random filename for the temp file
|
||||||
|
pkg_path = os.join_path_single(app.dl_dir, rand.uuid_v4())
|
||||||
|
|
||||||
// if app.repo.exists(pkg) {
|
for os.exists(pkg_path) {
|
||||||
// app.lwarn("Duplicate package '$pkg'")
|
pkg_path = os.join_path_single(app.dl_dir, rand.uuid_v4())
|
||||||
|
}
|
||||||
|
|
||||||
// return app.text('File already exists.')
|
app.ldebug("Uploading $length (${pretty_bytes(length.int())}) bytes to '$pkg_path'.")
|
||||||
// }
|
|
||||||
|
|
||||||
// pkg_path := app.repo.pkg_path(pkg)
|
// This is used to time how long it takes to upload a file
|
||||||
|
mut sw := time.new_stopwatch(time.StopWatchOptions{ auto_start: true })
|
||||||
|
|
||||||
// if length := app.req.header.get(.content_length) {
|
reader_to_file(mut app.reader, length.int(), pkg_path) or {
|
||||||
// app.ldebug("Uploading $length (${pretty_bytes(length.int())}) bytes to package '$pkg'.")
|
app.lwarn("Failed to upload '$pkg_path'")
|
||||||
|
|
||||||
// // This is used to time how long it takes to upload a file
|
return app.text('Failed to upload file.')
|
||||||
// mut sw := time.new_stopwatch(time.StopWatchOptions{ auto_start: true })
|
}
|
||||||
|
|
||||||
// reader_to_file(mut app.reader, length.int(), pkg_path) or {
|
sw.stop()
|
||||||
// app.lwarn("Failed to upload package '$pkg'")
|
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.")
|
||||||
|
}
|
||||||
|
|
||||||
// return app.text('Failed to upload file.')
|
added := app.repo.add_from_path(pkg_path) or {
|
||||||
// }
|
app.lerror('Error while adding package.')
|
||||||
|
|
||||||
// sw.stop()
|
return app.text('Failed to add package.')
|
||||||
// app.ldebug("Upload of package '$pkg' completed in ${sw.elapsed().seconds():.3}s.")
|
}
|
||||||
// } else {
|
if !added {
|
||||||
// app.lwarn("Tried to upload package '$pkg' without specifying a Content-Length.")
|
app.lwarn('Duplicate package.')
|
||||||
// return app.text("Content-Type header isn't set.")
|
|
||||||
// }
|
|
||||||
|
|
||||||
// app.repo.add_package(pkg_path) or {
|
return app.text('File already exists.')
|
||||||
// app.lwarn("Failed to add package '$pkg' to database.")
|
}
|
||||||
|
|
||||||
// os.rm(pkg_path) or { println('Failed to remove $pkg_path') }
|
app.linfo("Added '$pkg_path' to repository.")
|
||||||
|
|
||||||
// return app.text('Failed to add package to repo.')
|
return app.text('Package added successfully.')
|
||||||
// }
|
}
|
||||||
|
|
||||||
// app.linfo("Added '$pkg' to repository.")
|
|
||||||
|
|
||||||
// return app.text('Package added successfully.')
|
|
||||||
// }
|
|
||||||
|
|
||||||
// add_package PUT a new package to the server
|
// add_package PUT a new package to the server
|
||||||
['/add'; put]
|
['/add'; put]
|
||||||
|
|
Loading…
Reference in New Issue