first

.gitignore

@@ -0,0 +1 @@
+/target

Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "fserve"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+rocket = "0.4.10"
+rocket_upload = "0.1.0"
+rocket_contrib = { version = "0.4.10", default-features = false, features = [ "handlebars_templates" ]}
+serde = { version = "1.0", features = ["derive"] }
+rand = "0.8.5"
+lazy_static = "1.4.0"
+

Dockerfile

@@ -0,0 +1,23 @@
+FROM rust:1.59 as builder
+
+WORKDIR /build
+
+COPY ./src/ /build/src
+COPY ./Cargo.toml /build/Cargo.toml
+
+RUN rustup update nightly && rustup default nightly && \
+    cargo clean && cargo build --release
+
+FROM debian:buster-slim
+
+WORKDIR /app
+
+COPY --from=builder /build/target/release/fastcloud /app/fastcloud
+
+RUN mkdir /app/service
+
+COPY ./templates/ /app/templates
+
+ENV FSERVE_DIR /app/service
+
+CMD [ "/app/fastcloud" ]

src/main.rs

@@ -0,0 +1,114 @@
+#![feature(proc_macro_hygiene, decl_macro)]
+
+#[macro_use]
+extern crate rocket;
+use rand::Rng;
+use rocket::http::{ContentType, Cookie, Cookies, Status};
+use rocket::request;
+use rocket::request::{Form, FromRequest, Request};
+use rocket::response::{NamedFile, Redirect};
+use rocket_contrib::templates::Template;
+use rocket_upload::MultipartDatas;
+use serde::Deserialize;
+use std::collections::HashMap;
+use std::env;
+use std::fs;
+use std::path::Path;
+use std::path::PathBuf;
+
+#[macro_use]
+extern crate lazy_static;
+
+lazy_static! {
+    static ref FSERVE_PASSWD: String =
+        env::var("FSERVE_PASSWD").unwrap_or_else(|_| String::from(""));
+    static ref DIR: PathBuf = Path::new(
+        env::var("FSERVE_DIR")
+            .unwrap_or_else(|_| String::from(""))
+            .as_str()
+    )
+    .to_path_buf();
+}
+
+struct AdminUser;
+
+impl<'a, 'r> FromRequest<'a, 'r> for AdminUser {
+    type Error = ();
+
+    fn from_request(request: &'a Request<'r>) -> request::Outcome<Self, Self::Error> {
+        match request.cookies().get("key") {
+            Some(key) => {
+                if key.value() == FSERVE_PASSWD.as_str() {
+                    request::Outcome::Success(AdminUser)
+                } else {
+                    request::Outcome::Failure((Status::BadRequest, ()))
+                }
+            }
+
+            None => request::Outcome::Failure((Status::BadRequest, ())),
+        }
+    }
+}
+
+#[get("/f/<path..>")]
+fn get_file(path: PathBuf) -> Option<NamedFile> {
+    let dir = DIR.as_path().join(path);
+    rocket::response::NamedFile::open(dir.as_path()).ok()
+}
+
+fn upload(data: MultipartDatas) -> Vec<String> {
+    let mut rng = rand::thread_rng();
+    let mut v = Vec::new();
+    for f in data.files {
+        let mut hex_str = format!("{:016x}", rng.gen::<u64>());
+        while Path::new(&format!("{}/{}", DIR.to_str().unwrap(), hex_str)).exists() {
+            hex_str = format!("{:016x}", rng.gen::<u64>());
+        }
+        let dirloc = &format!("{}/{}", DIR.to_str().unwrap(), hex_str);
+        fs::create_dir_all(Path::new(dirloc)).unwrap();
+        let dir = Path::new(dirloc);
+        v.push(format!("/{}/{}", hex_str, f.filename));
+        f.persist(dir);
+    }
+    v
+}
+
+#[post("/upload", data = "<data>")]
+fn upload_file(_content_type: &ContentType, data: MultipartDatas, _guard: AdminUser) -> Redirect {
+    let resp = upload(data);
+    Redirect::to(uri!(get_file: Path::new(&resp[0])))
+}
+
+#[get("/up")]
+fn up(_guard: AdminUser) -> Template {
+    let map: HashMap<String, String> = HashMap::new();
+    Template::render("up", map)
+}
+
+#[derive(Debug, FromForm, Deserialize)]
+struct GenericUser {
+    key: String,
+}
+
+#[post("/login", data = "<form>")]
+fn login(form: Form<GenericUser>, mut cookies: Cookies) -> Redirect {
+    let key = form.into_inner().key;
+    let mut cook = Cookie::new("key", key);
+    cook.set_http_only(true);
+    cook.set_secure(true);
+    cookies.add(cook);
+    Redirect::to("/up")
+}
+
+#[get("/")]
+fn root() -> Template {
+    let map: HashMap<String, String> = HashMap::new();
+    Template::render("home", map)
+}
+
+fn main() {
+    rocket::ignite()
+        .attach(Template::fairing())
+        .mount("/", routes![get_file, upload_file, root, login, up])
+        .launch();
+}

templates/home.html.hbs

@@ -0,0 +1,23 @@
+<html>
+    <head>
+        <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-1BmE4kWBq78iYhFldvKuhfTAU6auU8tT94WrHftjDbrCEXSU1oBoqyl2QvZ6jIW3" crossorigin="anonymous">
+        <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.8.1/font/bootstrap-icons.css">
+    </head>
+    <body>
+        <div class="vstack gap-3">
+                <div class="bg-light border rounded-3 p-2">
+                        <div class="d-grid gap-2">
+                            <form class='form-inline' action="/login" method="post" id = "login-form" enctype="application/x-ww-form-urlencoded">
+                                <div class="input-group mb-3 flex-nowrap col-sm-6">
+                                    <input type="password" class="form-control" id="key" name="key" placeholder = "Enter Key"/>
+                                </div>
+                            </form>
+
+                            <script>
+                                document.getElementById("login").addEventListener("keyup", function (event) { if (event.keyCode == 13) { document.getElementById("login-form").submit(); }});
+                            </script>
+                        </div>
+                </div>
+        </div>
+    </body>
+</html>

templates/up.html.hbs

@@ -0,0 +1,21 @@
+
+<html>
+    <head>
+        <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-1BmE4kWBq78iYhFldvKuhfTAU6auU8tT94WrHftjDbrCEXSU1oBoqyl2QvZ6jIW3" crossorigin="anonymous">
+        <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.8.1/font/bootstrap-icons.css">
+    </head>
+    <body>
+        <div class="vstack gap-3">
+                <div class="bg-light border rounded-3 p-2">
+                        <div class="d-grid gap-2">
+                            <form action="/upload" method="post" enctype="multipart/form-data">
+                                <input type="file" id="file" name="upload" onchange="this.form.submit()" hidden/>
+                            </form>
+                            <button class="btn btn-outline-success text-start col-sm-1 ms-1 me-1" onclick="document.getElementById('file').click()">
+                                <i class="bi bi-file-earmark-plus"></i>
+                            </button>
+                        </div>
+                </div>
+        </div>
+    </body>
+</html>