From c11b12a03f4fe578791f0684add9a6a897eadc02 Mon Sep 17 00:00:00 2001
From: alemi <me@alemi.dev>
Date: Sun, 15 Oct 2023 04:27:34 +0200
Subject: [PATCH] fix: initial impl

---
 Cargo.toml  | 19 ++++++++++++
 src/main.rs | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)
 create mode 100644 Cargo.toml
 create mode 100644 src/main.rs

diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..848ff53
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,19 @@
+[package]
+name = "suggestions-bot"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+async-trait = "0.1.73"
+axum = "0.6.20"
+clap = { version = "4.4.6", features = ["derive"] }
+html-escape = "0.2.13"
+lazy_static = "1.4.0"
+serde = { version = "1.0.189", features = ["derive"] }
+serde_json = "1.0.107"
+teloxide = { version = "0.12.2", features = ["macros"] }
+tokio = { version = "1.33.0", features = ["macros", "rt-multi-thread"] }
+tracing = "0.1.39"
+tracing-subscriber = "0.3.17"
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..89a34fb
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,83 @@
+use std::net::SocketAddr;
+use axum::{routing::{put, post}, http::StatusCode, Json, Router, Form, response::Response, extract::FromRequest};
+use serde::{Deserialize, Serialize};
+use clap::Parser;
+use teloxide::prelude::*;
+use tokio::sync::RwLock;
+
+lazy_static::lazy_static! {
+	static ref BOT : Bot = Bot::from_env();
+	static ref CHAT_ID : RwLock<ChatId> = RwLock::new(ChatId(0)); // TODO ewwwwwww
+}
+
+#[derive(Debug, Clone, Parser)]
+/// api for sending anonymous telegram messages to a specific user
+struct CliArgs {
+	/// chat id of target user
+	target: i64,
+
+	#[arg(long, short, default_value = "127.0.0.1:37812")]
+	/// host to bind onto
+	addr: String,
+}
+
+#[tokio::main]
+async fn main() {
+	tracing_subscriber::fmt::init();
+
+	let args = CliArgs::parse();
+
+	*CHAT_ID.write().await = ChatId(args.target);
+
+	let app = Router::new()
+		.route("/suggestion", post(suggestion_form))
+		.route("/suggestion", put(suggestion_json));
+
+	let addr : SocketAddr = args.addr.parse().expect("invalid host provided");
+
+	tracing::info!("listening on {}", addr);
+
+	axum::Server::bind(&addr)
+		.serve(app.into_make_service())
+		.await
+		.unwrap();
+}
+
+async fn suggestion_json(Json(payload): Json<Suggestion>) -> (StatusCode, Json<Acknowledgement>) {
+	suggestion_inner(payload).await
+}
+
+async fn suggestion_form(Form(payload): Form<Suggestion>) -> (StatusCode, Json<Acknowledgement>) {
+	suggestion_inner(payload).await
+}
+
+async fn suggestion_inner(payload: Suggestion) -> (StatusCode, Json<Acknowledgement>) {
+	let message = format!(
+		"[<code>{}</code>] <i>{}</i> | {}",
+		html_escape::encode_text(payload.author.as_deref().unwrap_or("anon")),
+		html_escape::encode_text(payload.contact.as_deref().unwrap_or("N/A")),
+		html_escape::encode_text(&payload.body)
+	);
+
+	match BOT
+		.send_message(*CHAT_ID.read().await, message)
+		.parse_mode(teloxide::types::ParseMode::Html)
+		.await
+	{
+		Ok(x) => (StatusCode::OK, Json(Acknowledgement::Sent(x.text().unwrap_or("").into()))),
+		Err(e) => (StatusCode::INTERNAL_SERVER_ERROR, Json(Acknowledgement::Refused(e.to_string()))),
+	}
+}
+
+#[derive(Deserialize)]
+struct Suggestion {
+	author: Option<String>,
+	contact: Option<String>,
+	body: String,
+}
+
+#[derive(Serialize)]
+enum Acknowledgement {
+	Sent(String),
+	Refused(String),
+}