Initial commit

src/main.rs

@@ -0,0 +1,67 @@
+use actix_web::{HttpServer, App, get, web, Responder, HttpResponse};
+use serde::Deserialize;
+
+pub static HTML: &'static str = include_str!("./index.html");
+pub static JSON: &'static str = include_str!("./prov.json");
+
+#[derive(Deserialize)]
+struct Params {
+    redirect: Option<String>,
+    title: Option<String>,
+    description: Option<String>,
+    sitename: Option<String>,
+    image: Option<String>,
+    color: Option<String>,
+    title_url: Option<String>,
+    sitename_url: Option<String>,
+}
+
+#[get("/")]
+async fn index(info: web::Query<Params>) -> impl Responder {
+    let info = info.into_inner();
+
+    HttpResponse::Ok().content_type("text/html").body(HTML
+        .replace("{{REDIRECT}}", &info.redirect.unwrap_or("https://davidon.top".to_string()))
+        .replace("{{TITLE}}", &info.title.unwrap_or_default())
+        .replace("{{DESCRIPTION}}", &info.description.unwrap_or_default())
+        .replace("{{PROV}}", &info.sitename.unwrap_or_default())
+        .replace("{{IMAGE}}", &info.image.unwrap_or_default())
+        .replace("{{COLOR}}", &info.color.unwrap_or_default())
+        .replace("{{TITURL}}", &info.title_url.unwrap_or_default())
+        .replace("{{PROVURL}}", &info.sitename_url.unwrap_or_default())
+    )
+}
+
+#[derive(Deserialize)]
+struct ParamsJson {
+    title: Option<String>,
+    title_url: Option<String>,
+    provider: Option<String>,
+    provider_url: Option<String>,
+}
+
+#[get("/json")]
+async fn json(info: web::Query<ParamsJson>) -> impl Responder {
+    let info = info.into_inner();
+
+    HttpResponse::Ok().content_type("application/json").body(JSON
+        .replace("{{TITLE}}", &info.title.unwrap_or_default())
+        .replace("{{TITURL}}", &info.title_url.unwrap_or_default())
+        .replace("{{PROV}}", &info.provider.unwrap_or_default())
+        .replace("{{PROVURL}}", &info.provider_url.unwrap_or_default())
+    )
+}
+
+#[actix_web::main]
+async fn main() -> std::io::Result<()> {
+    tracing_subscriber::fmt::init();
+
+    HttpServer::new(|| {
+        App::new()
+            .service(index)
+            .service(json)
+    })
+    .bind(("127.0.0.1", 3596))?
+    .run()
+    .await
+}