added url shrtner && moved the whole project to rocket.rs

Author: plexx-dev

.gitignore

@@ -0,0 +1,17 @@
+# Generated by Cargo
+# will have compiled files and executables
+/target/
+
+# Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
+# More information here https://doc.rust-lang.org/cargo/guide/cargo-toml-vs-cargo-lock.html
+Cargo.lock
+
+# These are backup files generated by rustfmt
+**/*.rs.bk
+
+
+# Added by cargo
+
+/target
+
+.env

404.html

@@ -0,0 +1,17 @@
+[package]
+name = "plexx-dev"
+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.5.0-rc.2"
+rocket_dyn_templates = { version = "0.1.0-rc.2", features = ["tera"] }
+sha2 = "*"
+hex = "*"
+validator = { version = "0.15", features = ["derive"] }
+
+[dependencies.rocket_db_pools]
+version = "0.1.0-rc.2"
+features = ["sqlx_mysql"]

Cargo.toml

@@ -1,3 +1,12 @@
-FROM nginx:alpine
+FROM rust:latest as builder
+WORKDIR /app
+COPY . .
+RUN cargo install --path .
 
-COPY . /usr/share/nginx/html/
+
+FROM debian:buster-slim as runner
+WORKDIR /app
+COPY --from=builder /usr/local/cargo/bin/plexx-dev /usr/local/bin/plexx-dev
+COPY . .
+RUN ls
+CMD ["plexx-dev"]

Dockerfile

@@ -0,0 +1,14 @@
+[debug]
+port = 5000
+
+[release]
+cli_colors = false
+port = 80
+address = "0.0.0.0"
+
+[global]
+min_connections = 64
+max_connections = 1024
+connect_timeout = 5
+idle_timeout = 120
+limits = { form = "64 kB", json = "1 MiB" }

JetBrainsMono_var_weight.ttf

@@ -0,0 +1,23 @@
+pub fn get_error_message(code: u16) -> String {
+    match code {
+        300 => "Multiple Choices".to_string(),
+        301 => "Moved Permanently".to_string(),
+        304 => "Not Modified".to_string(),
+        307 => "Temporary Redirect (since HTTP/1.1)".to_string(),
+        308 => "Permanent Redirect".to_string(),
+
+        400 => "Bad Request".to_string(),
+        401 => "Unauthorized".to_string(),
+        403 => "Forbidden".to_string(),
+        404 => "Not found".to_string(),
+        405 => "Method not allowed".to_string(),
+        408 => "Request Timeout".to_string(),
+        418 => "I'm a Teapot".to_string(),
+
+        500 => "Internal Server Error".to_string(),
+        502 => "Bad Gateway".to_string(),
+        503 => "Service Unavailable".to_string(),
+
+        _ => "WIP ERROR CODE".to_string(),
+    }
+}

LICENSE

@@ -0,0 +1,45 @@
+use rocket::*;
+use rocket::fs::FileServer;
+use rocket::http::Status;
+
+use rocket_dyn_templates::{Template, context};
+
+use rocket_db_pools::Database;
+use rocket_db_pools::sqlx;
+
+mod urlshrtner;
+mod error;
+
+#[derive(Database)]
+#[database("urls")]
+pub struct Urls(sqlx::MySqlPool);
+
+#[get("/")]
+async fn index() -> Template {
+    Template::render("index", context! {})
+}
+
+#[catch(default)]
+async fn default_catcher(status: Status, _: &Request<'_>) -> Template {
+    Template::render("error", context! {
+        code: status.code,
+        message: error::get_error_message(status.code)
+    })
+}
+
+#[launch]
+async fn rocket() -> _ {
+    rocket::build()
+        .mount("/", routes![
+            index
+        ])
+        .mount("/", routes![
+            urlshrtner::routes::get_url,
+            urlshrtner::routes::index,
+            urlshrtner::routes::submit,           
+        ])
+        .mount("/static", FileServer::from("static"))
+        .register("/", catchers![default_catcher])
+        .attach(Template::fairing())
+        .attach(Urls::init())
+}

Rocket.toml

@@ -0,0 +1,2 @@
+pub mod routes;
+pub mod short;

index.html

@@ -0,0 +1,85 @@
+use rocket::*;
+
+use rocket::form::Form;
+use rocket::response::Redirect;
+use rocket_dyn_templates::{Template, context};
+
+#[allow(unused_imports)]
+use rocket_db_pools::{Connection, Database};
+use rocket_db_pools::sqlx::{self, Row};
+
+use validator::{Validate, validate_url};
+
+use crate::Urls;
+
+use super::short;
+
+#[get("/<url_hash>")]
+pub async fn get_url(mut db: Connection<Urls>, url_hash: String) -> Redirect {
+    let fallback_url = String::from("https://plexx.dev");
+    
+    let url: Option<String> = match sqlx::query("SELECT url FROM urls WHERE url_hash = ?")
+        .bind(&url_hash)
+        .fetch_one(&mut*db)
+        .await {
+            Ok(row) => row,
+            Err(error) => {
+                println!("{}", error);
+                return Redirect::to(fallback_url)
+            },
+        }
+        .try_get(0)
+       .ok();
+
+    let url_str = match url {
+        Some(url_str) => url_str,
+        None => {
+            println!("failed");
+            fallback_url
+        },
+    };
+
+    Redirect::to(url_str)
+}
+
+#[get("/short")]
+pub async fn index() -> Template {
+    Template::render("urlshrtner/index", context! {})
+}
+
+#[derive(FromForm, Debug, Validate)]
+pub struct Url {
+    #[validate(url)]
+    pub url: String
+}
+
+#[allow(unused_assignments)]
+#[post("/short", data = "<form>")]
+pub async fn submit(mut db: Connection<Urls>, form: Form<Url>) -> Template {
+    let mut url = form.into_inner().url;
+
+    let mut mod_url = String::new();
+    let is_valid = validate_url(&url);
+    if !is_valid {
+        mod_url = "https://".to_string();
+        mod_url.push_str(url.as_str());
+        if validate_url(&mod_url) {
+            url = mod_url;
+        } else {
+            return Template::render("urlshrtner/index", context! {
+                url: "URL is not valid"
+            });
+        }
+    }
+
+    let url_hash = short::hash(&url);
+
+    sqlx::query("INSERT INTO urls (url, url_hash) VALUES (?, ?)").bind(&url).bind(&url_hash)
+        .execute(&mut *db)
+        .await
+        .ok();
+
+    Template::render("urlshrtner/result", context! {
+        url: format!("plexx.dev/{}", url_hash)
+    })
+}

src/error.rs

@@ -0,0 +1,9 @@
+use sha2::{Sha256, Digest};
+
+pub fn hash(input: &String) -> String {
+    let mut h = Sha256::new();
+    h.update(&input);
+    let hash_u32 = &h.finalize()[..];
+
+    format!("{:.6}", hex::encode(hash_u32))
+}

src/main.rs

@@ -1,19 +1,15 @@
 @font-face{
-    font-family: "JetBrains Mono";
-    src: url("JetBrainsMono_var_weight.ttf");
-    src: url("JetBrainsMono_var_weight.ttf") format("ttf"),
-    url("JetBrainsMono_var_weight.ttf") format("opentype"),
-    url("JetBrainsMono_var_weight.ttf") format("svg");
+    font-family: "Source Code Pro";
+    src: url("SourceCodePro-VariableFont_wght.ttf");
+    src: url("SourceCodePro-VariableFont_wght.ttf") format("ttf"),
+    url("SourceCodePro-VariableFont_wght.ttf") format("opentype"),
+    url("SourceCodePro-VariableFont_wght.ttf") format("svg");
   }
-  
+
   * {
-    font-family: "JetBrains Mono";
+    font-family: "Source Code Pro";
   }
 
-html {
-    height: 100%;
-}
-
 body {
     background-color: rgb(0,0,0);
     color: #fff;
@@ -49,7 +45,7 @@ nav {
 
 a {
     padding: .75rem;
-    color: inherit;
+    color: #fff;
     font-weight: 515;
     font-size: 1.25rem;
     text-decoration: none;
@@ -58,3 +54,11 @@ a {
 .link-text {
     text-decoration: underline;
 }
+
+.form {
+    vertical-align: baseline;
+    padding-top: 24vh;
+    margin: 0 auto 24vh;
+    max-width: fit-content;
+    max-height: fit-content;
+}

src/urlshrtner/mod.rs

@@ -0,0 +1,13 @@
+{% extends 'layout' %}
+{% block content %}
+
+<p class="title">{{ code }}</p>
+<p class="text">{{ message }}</p>
+<div class="text">
+    <br>
+    <a href="/" class="text">
+    <i class="fas fa-home"></i>
+    <text class="link-text">Back</text></a>
+</div>
+
+{% endblock content %}

src/urlshrtner/routes.rs

@@ -0,0 +1,16 @@
+{% extends 'layout' %}
+{% block content %}
+
+<p class="title">plexx</p>
+<p class="text">Professional™ Developer</p>
+<nav>
+    <a href="https://github.com/plexx-dev"><i class="fab fa-github"></i> <text class="link-text">Github</text></a>
+    <a href="mailto:[email protected]"><i class="fas fa-envelope"></i> <text class="link-text">Email</text></a>
+    <a href="https://steamcommunity.com/id/plexx_dev/"><i class="fab fa-steam"></i> <text class="link-text">Steam</text></a>
+    <a href="https://status.plexx.dev"><i class="fas fa-server"></i> <text class="link-text">Server-Status</text></a>
+    <a href="/short"><i class="fas fa-link"></i> <text class="link-text">UrlShrtner</text></a>
+    <a href="/static/pgp.txt"><i class="fas fa-key"></i> <text class="link-text">PGP-Key</text></a>
+    <a><i class="fab fa-discord"></i> <text class="link-text">plexx#0928</text></a>
+</nav>
+
+{% endblock content %}

src/urlshrtner/short.rs

@@ -0,0 +1,16 @@
+<!DOCTYPE html>
+<html>
+    <head>
+        <meta charset="utf-8">
+        <link rel="stylesheet" href="/static/main.css">
+        <link rel="stylesheet" href="/static/all.min.css">
+        <title>plexx.dev</title>
+        <link rel="shortcut icon" href="/static/favicon.ico">
+    </head>
+    <body>
+        <main>
+            {% block content %}
+            {% endblock %}
+        </main>
+    </body>
+</html>