From 17038dc33c536688adfcd02e2a22d4da51f59d61 Mon Sep 17 00:00:00 2001
From: Jef Roosens <roosensjef@gmail.com>
Date: Sun, 9 Jan 2022 22:18:04 +0100
Subject: [PATCH 1/2] First working upload streaming!

---
 vieter/main.v    | 32 ++++++++++++++------------------
 vieter/web/web.v | 14 +++++++-------
 2 files changed, 21 insertions(+), 25 deletions(-)

diff --git a/vieter/main.v b/vieter/main.v
index 2978298..7d9a1e2 100644
--- a/vieter/main.v
+++ b/vieter/main.v
@@ -6,6 +6,7 @@ import log
 import io
 
 const port = 8000
+
 const buf_size = 1_000_000
 
 struct App {
@@ -21,7 +22,7 @@ fn exit_with_message(code int, msg string) {
 	exit(code)
 }
 
-fn reader_to_file(mut reader io.BufferedReader, path string) ? {
+fn reader_to_file(mut reader io.BufferedReader, length int, path string) ? {
 	// Open up a file for writing to
 	mut file := os.create(path) ?
 	defer {
@@ -29,36 +30,27 @@ fn reader_to_file(mut reader io.BufferedReader, path string) ? {
 	}
 
 	mut buf := []byte{len: buf_size}
+	mut bytes_left := length
 
 	// Repeat as long as the stream still has data
-	for {
-		// TODO don't just endlessly loop if reading keeps failing
-		println('heey')
+	for bytes_left > 0 {
 		// TODO check if just breaking here is safe
-		bytes_read := reader.read(mut &buf) or {
-			println('youre here')
-			break
-		}
-		println(bytes_read)
+		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 {
-				println("$err.msg")
-				continue
-			}
+			bytes_written := file.write(buf[bytes_read - to_write..bytes_read]) or { continue }
 			println(bytes_written)
 
 			to_write = to_write - bytes_written
 		}
 	}
-
-	println('File complete!')
 }
 
-[put; '/pkgs/:pkg']
+['/pkgs/:pkg'; put]
 fn (mut app App) put_package(pkg string) web.Result {
 	full_path := os.join_path_single(app.repo_dir, pkg)
 
@@ -66,8 +58,12 @@ fn (mut app App) put_package(pkg string) web.Result {
 		return app.text('File already exists.')
 	}
 
-	reader_to_file(mut app.reader, full_path) or {
-		return app.text('Failed to upload file.')
+	if length := app.req.header.get(.content_length) {
+		reader_to_file(mut app.reader, length.int(), full_path) or {
+			return app.text('Failed to upload file.')
+		}
+	} else {
+		return app.text("Content-Type header isn't set.")
 	}
 
 	return app.text('just stop')
diff --git a/vieter/web/web.v b/vieter/web/web.v
index 4e13f3f..e98ef9b 100644
--- a/vieter/web/web.v
+++ b/vieter/web/web.v
@@ -447,13 +447,13 @@ fn handle_conn<T>(mut conn net.TcpConn, mut app T, routes map[string]Route) {
 		return
 	}
 
-// 	req := http.parse_request(mut reader) or {
-// 		// Prevents errors from being thrown when BufferedReader is empty
-// 		if '$err' != 'none' {
-// 			eprintln('error parsing request: $err')
-// 		}
-// 		return
-// 	}
+	// 	req := http.parse_request(mut reader) or {
+	// 		// Prevents errors from being thrown when BufferedReader is empty
+	// 		if '$err' != 'none' {
+	// 			eprintln('error parsing request: $err')
+	// 		}
+	// 		return
+	// 	}
 
 	// URL Parse
 	url := urllib.parse(head.url) or {

From 93c0dc9c4f69cde6e91d270c43c3f0e8b92234d3 Mon Sep 17 00:00:00 2001
From: Jef Roosens <roosensjef@gmail.com>
Date: Sun, 9 Jan 2022 22:30:07 +0100
Subject: [PATCH 2/2] Added start of repo-add functionality

---
 vieter/main.v | 12 ++++++++++--
 vieter/repo.v | 12 ++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)
 create mode 100644 vieter/repo.v

diff --git a/vieter/main.v b/vieter/main.v
index 7d9a1e2..d7a1b7d 100644
--- a/vieter/main.v
+++ b/vieter/main.v
@@ -4,11 +4,14 @@ import web
 import os
 import log
 import io
+import repo
 
 const port = 8000
 
 const buf_size = 1_000_000
 
+const db_name = 'pieter.db.tar.gz'
+
 struct App {
 	web.Context
 	api_key  string  [required; web_global]
@@ -43,7 +46,6 @@ fn reader_to_file(mut reader io.BufferedReader, length int, path string) ? {
 		for to_write > 0 {
 			// TODO don't just loop infinitely here
 			bytes_written := file.write(buf[bytes_read - to_write..bytes_read]) or { continue }
-			println(bytes_written)
 
 			to_write = to_write - bytes_written
 		}
@@ -66,7 +68,13 @@ fn (mut app App) put_package(pkg string) web.Result {
 		return app.text("Content-Type header isn't set.")
 	}
 
-	return app.text('just stop')
+	repo.add_package(os.join_path_single(app.repo_dir, db_name), full_path) or {
+		os.rm(full_path) or { println('Failed to remove $full_path') }
+
+		return app.text('Failed to add package to repo.')
+	}
+
+	return app.text('Package added successfully.')
 }
 
 // ['/publish'; post]
diff --git a/vieter/repo.v b/vieter/repo.v
new file mode 100644
index 0000000..54dfc38
--- /dev/null
+++ b/vieter/repo.v
@@ -0,0 +1,12 @@
+module repo
+
+import os
+
+pub fn add_package(db_path string, pkg_path string) ? {
+	res := os.execute("repo-add '$db_path' '$pkg_path'")
+
+	if res.exit_code != 0 {
+		println(res.output)
+		return error('repo-add failed.')
+	}
+}