Initial (full)
This commit is contained in:
parent
94cec33b46
commit
9dd708d276
8 changed files with 4810 additions and 0 deletions
123
src/main.rs
Normal file
123
src/main.rs
Normal file
|
|
@ -0,0 +1,123 @@
|
|||
#![doc = include_str!("../README.md")]
|
||||
mod model;
|
||||
mod routes;
|
||||
|
||||
use axum::{Extension, Router};
|
||||
use nanoneo::core::element::Render;
|
||||
use std::{collections::HashMap, env::var, net::SocketAddr, process::exit, sync::Arc};
|
||||
use tera::{Tera, Value};
|
||||
use tetratto_core::sdk::DataClient;
|
||||
use tokio::sync::RwLock;
|
||||
use tower_http::{
|
||||
catch_panic::CatchPanicLayer,
|
||||
trace::{self, TraceLayer},
|
||||
};
|
||||
use tracing::{Level, info};
|
||||
|
||||
pub(crate) type InnerState = (DataClient, Tera);
|
||||
pub(crate) type State = Arc<RwLock<InnerState>>;
|
||||
|
||||
fn render_markdown(value: &Value, _: &HashMap<String, Value>) -> tera::Result<Value> {
|
||||
Ok(
|
||||
tetratto_shared::markdown::render_markdown(value.as_str().unwrap())
|
||||
.replace("\\@", "@")
|
||||
.replace("%5C@", "@")
|
||||
.into(),
|
||||
)
|
||||
}
|
||||
|
||||
fn remove_script_tags(value: &Value, _: &HashMap<String, Value>) -> tera::Result<Value> {
|
||||
Ok(value
|
||||
.as_str()
|
||||
.unwrap()
|
||||
.replace("</script>", "</script>")
|
||||
.into())
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! create_dir_if_not_exists {
|
||||
($dir_path:expr) => {
|
||||
if !std::fs::exists(&$dir_path).unwrap() {
|
||||
std::fs::create_dir($dir_path).unwrap();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
#[tokio::main(flavor = "multi_thread")]
|
||||
async fn main() {
|
||||
dotenv::dotenv().ok();
|
||||
tracing_subscriber::fmt()
|
||||
.with_target(false)
|
||||
.compact()
|
||||
.init();
|
||||
|
||||
let port = match var("PORT") {
|
||||
Ok(port) => port.parse::<u16>().expect("port should be a u16"),
|
||||
Err(_) => 9119,
|
||||
};
|
||||
|
||||
// ...
|
||||
let database = DataClient::new(
|
||||
Some(var("TETRATTO").unwrap_or("https://tetratto.com".to_string())),
|
||||
var("API_KEY").expect("API_KEY environment variable required"),
|
||||
);
|
||||
|
||||
// build lisp
|
||||
create_dir_if_not_exists!("./templates_build");
|
||||
for x in glob::glob("./templates_src/**/*").expect("failed to read pattern") {
|
||||
match x {
|
||||
Ok(x) => std::fs::write(
|
||||
x.to_str()
|
||||
.unwrap()
|
||||
.replace("templates_src/", "templates_build/"),
|
||||
nanoneo::parse(&std::fs::read_to_string(x).expect("failed to read template"))
|
||||
.render(&mut HashMap::new()),
|
||||
)
|
||||
.expect("failed to write template"),
|
||||
Err(e) => panic!("{e}"),
|
||||
}
|
||||
}
|
||||
|
||||
// ...
|
||||
let mut tera = match Tera::new(&format!("./templates_build/**/*")) {
|
||||
Ok(t) => t,
|
||||
Err(e) => {
|
||||
println!("{e}");
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
tera.register_filter("markdown", render_markdown);
|
||||
tera.register_filter("remove_script_tags", remove_script_tags);
|
||||
|
||||
// create app
|
||||
let app = Router::new()
|
||||
.merge(routes::routes())
|
||||
.layer(Extension(Arc::new(RwLock::new((database, tera)))))
|
||||
.layer(axum::extract::DefaultBodyLimit::max(
|
||||
var("BODY_LIMIT")
|
||||
.unwrap_or("8388608".to_string())
|
||||
.parse::<usize>()
|
||||
.unwrap(),
|
||||
))
|
||||
.layer(
|
||||
TraceLayer::new_for_http()
|
||||
.make_span_with(trace::DefaultMakeSpan::new().level(Level::INFO))
|
||||
.on_response(trace::DefaultOnResponse::new().level(Level::INFO)),
|
||||
)
|
||||
.layer(CatchPanicLayer::new());
|
||||
|
||||
// ...
|
||||
let listener = tokio::net::TcpListener::bind(format!("0.0.0.0:{}", port))
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
info!("🦌 attobin.");
|
||||
info!("listening on http://0.0.0.0:{}", port);
|
||||
axum::serve(
|
||||
listener,
|
||||
app.into_make_service_with_connect_info::<SocketAddr>(),
|
||||
)
|
||||
.await
|
||||
.unwrap();
|
||||
}
|
||||
11
src/model.rs
Normal file
11
src/model.rs
Normal file
|
|
@ -0,0 +1,11 @@
|
|||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
pub struct Entry {
|
||||
pub slug: String,
|
||||
pub edit_code: String,
|
||||
pub salt: String,
|
||||
pub created: usize,
|
||||
pub edited: usize,
|
||||
pub content: String,
|
||||
}
|
||||
390
src/routes.rs
Normal file
390
src/routes.rs
Normal file
|
|
@ -0,0 +1,390 @@
|
|||
use crate::{State, model::Entry};
|
||||
use axum::{
|
||||
Extension, Json, Router,
|
||||
extract::Path,
|
||||
response::{Html, IntoResponse},
|
||||
routing::{get, get_service, post},
|
||||
};
|
||||
use serde::Deserialize;
|
||||
use tera::Context;
|
||||
use tetratto_core::{
|
||||
model::{
|
||||
ApiReturn, Error,
|
||||
apps::{AppDataQueryResult, AppDataSelectMode, AppDataSelectQuery},
|
||||
},
|
||||
sdk::{DataClient, SimplifiedQuery},
|
||||
};
|
||||
use tetratto_shared::{
|
||||
hash::{hash, salt},
|
||||
unix_epoch_timestamp,
|
||||
};
|
||||
|
||||
pub fn routes() -> Router {
|
||||
Router::new()
|
||||
.nest_service(
|
||||
"/public",
|
||||
get_service(tower_http::services::ServeDir::new("./public")),
|
||||
)
|
||||
.fallback(not_found_request)
|
||||
// pages
|
||||
.route("/", get(index_request))
|
||||
.route("/{slug}", get(view_request))
|
||||
.route("/{slug}/edit", get(editor_request))
|
||||
// api
|
||||
.route("/api/v1/render", post(render_request))
|
||||
.route("/api/v1/entries", post(create_request))
|
||||
.route("/api/v1/entries/{slug}", post(edit_request))
|
||||
}
|
||||
|
||||
fn default_context(data: &DataClient) -> Context {
|
||||
let mut ctx = Context::new();
|
||||
ctx.insert(
|
||||
"name",
|
||||
&std::env::var("NAME").unwrap_or("Attobin".to_string()),
|
||||
);
|
||||
ctx.insert("tetratto", &data.host);
|
||||
ctx
|
||||
}
|
||||
|
||||
// pages
|
||||
async fn not_found_request(Extension(data): Extension<State>) -> impl IntoResponse {
|
||||
let (ref data, ref tera) = *data.read().await;
|
||||
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert(
|
||||
"error",
|
||||
&Error::GeneralNotFound("page".to_string()).to_string(),
|
||||
);
|
||||
return Html(tera.render("error.lisp", &ctx).unwrap());
|
||||
}
|
||||
|
||||
async fn index_request(Extension(data): Extension<State>) -> impl IntoResponse {
|
||||
let (ref data, ref tera) = *data.read().await;
|
||||
Html(tera.render("index.lisp", &default_context(&data)).unwrap())
|
||||
}
|
||||
|
||||
async fn view_request(
|
||||
Extension(data): Extension<State>,
|
||||
Path(slug): Path<String>,
|
||||
) -> impl IntoResponse {
|
||||
let (ref data, ref tera) = *data.read().await;
|
||||
let entry = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => serde_json::from_str::<Entry>(&r.value).unwrap(),
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(_) => {
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert(
|
||||
"error",
|
||||
&Error::GeneralNotFound("entry".to_string()).to_string(),
|
||||
);
|
||||
|
||||
return Html(tera.render("error.lisp", &ctx).unwrap());
|
||||
}
|
||||
};
|
||||
|
||||
let (views_id, views) = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries.views('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => (r.id, r.value.parse::<usize>().unwrap()),
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(e) => {
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert("error", &e.to_string());
|
||||
|
||||
return Html(tera.render("error.lisp", &ctx).unwrap());
|
||||
}
|
||||
};
|
||||
|
||||
// count view
|
||||
if let Err(e) = data.update(views_id, (views + 1).to_string()).await {
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert("error", &e.to_string());
|
||||
|
||||
return Html(tera.render("error.lisp", &ctx).unwrap());
|
||||
}
|
||||
|
||||
// ...
|
||||
let mut ctx = default_context(&data);
|
||||
|
||||
ctx.insert("entry", &entry);
|
||||
ctx.insert("views", &views);
|
||||
|
||||
Html(tera.render("view.lisp", &ctx).unwrap())
|
||||
}
|
||||
|
||||
async fn editor_request(
|
||||
Extension(data): Extension<State>,
|
||||
Path(slug): Path<String>,
|
||||
) -> impl IntoResponse {
|
||||
let (ref data, ref tera) = *data.read().await;
|
||||
let entry = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => serde_json::from_str::<Entry>(&r.value).unwrap(),
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(_) => {
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert(
|
||||
"error",
|
||||
&Error::GeneralNotFound("entry".to_string()).to_string(),
|
||||
);
|
||||
|
||||
return Html(tera.render("error.lisp", &ctx).unwrap());
|
||||
}
|
||||
};
|
||||
|
||||
// ...
|
||||
let mut ctx = default_context(&data);
|
||||
ctx.insert("entry", &entry);
|
||||
|
||||
Html(tera.render("edit.lisp", &ctx).unwrap())
|
||||
}
|
||||
|
||||
// api
|
||||
#[derive(Deserialize)]
|
||||
struct RenderMarkdown {
|
||||
content: String,
|
||||
}
|
||||
|
||||
async fn render_request(Json(req): Json<RenderMarkdown>) -> impl IntoResponse {
|
||||
tetratto_shared::markdown::render_markdown(&req.content)
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct CreateEntry {
|
||||
content: String,
|
||||
#[serde(default = "default_random")]
|
||||
slug: String,
|
||||
#[serde(default = "default_random")]
|
||||
edit_code: String,
|
||||
}
|
||||
|
||||
fn default_random() -> String {
|
||||
salt()
|
||||
}
|
||||
|
||||
async fn create_request(
|
||||
Extension(data): Extension<State>,
|
||||
Json(req): Json<CreateEntry>,
|
||||
) -> impl IntoResponse {
|
||||
let (ref data, _) = *data.read().await;
|
||||
|
||||
// check lengths
|
||||
if req.slug.len() < 2 {
|
||||
return Json(Error::DataTooShort("slug".to_string()).into());
|
||||
}
|
||||
|
||||
if req.slug.len() > 32 {
|
||||
return Json(Error::DataTooLong("slug".to_string()).into());
|
||||
}
|
||||
|
||||
// check for existing
|
||||
if data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries('{}')", req.slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
.is_ok()
|
||||
{
|
||||
return Json(Error::MiscError("Slug already in use".to_string()).into());
|
||||
}
|
||||
|
||||
// create
|
||||
let created = unix_epoch_timestamp();
|
||||
let salt = salt();
|
||||
|
||||
if let Err(e) = data
|
||||
.insert(
|
||||
format!("entries('{}')", req.slug),
|
||||
serde_json::to_string(&Entry {
|
||||
slug: req.slug.clone(),
|
||||
edit_code: hash(req.edit_code.clone() + &salt),
|
||||
salt,
|
||||
created,
|
||||
edited: created,
|
||||
content: req.content,
|
||||
})
|
||||
.unwrap(),
|
||||
)
|
||||
.await
|
||||
{
|
||||
return Json(e.into());
|
||||
}
|
||||
|
||||
if let Err(e) = data
|
||||
.insert(format!("entries.views('{}')", req.slug), 0.to_string())
|
||||
.await
|
||||
{
|
||||
return Json(e.into());
|
||||
}
|
||||
|
||||
// return
|
||||
Json(ApiReturn {
|
||||
ok: true,
|
||||
message: "Success".to_string(),
|
||||
payload: Some((req.slug, req.edit_code)),
|
||||
})
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct EditEntry {
|
||||
content: String,
|
||||
edit_code: String,
|
||||
#[serde(default)]
|
||||
new_slug: Option<String>,
|
||||
#[serde(default)]
|
||||
new_edit_code: Option<String>,
|
||||
#[serde(default)]
|
||||
delete: bool,
|
||||
}
|
||||
|
||||
async fn edit_request(
|
||||
Extension(data): Extension<State>,
|
||||
Path(slug): Path<String>,
|
||||
Json(req): Json<EditEntry>,
|
||||
) -> impl IntoResponse {
|
||||
let (ref data, _) = *data.read().await;
|
||||
|
||||
let (id, mut entry) = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => (r.id, serde_json::from_str::<Entry>(&r.value).unwrap()),
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(e) => return Json(e.into()),
|
||||
};
|
||||
|
||||
// check edit code
|
||||
if hash(req.edit_code.clone() + &entry.salt) != entry.edit_code {
|
||||
return Json(Error::NotAllowed.into());
|
||||
}
|
||||
|
||||
// handle delete
|
||||
if req.delete {
|
||||
let views_id = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries.views('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => r.id,
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(e) => return Json(e.into()),
|
||||
};
|
||||
|
||||
return match data.remove(id).await {
|
||||
Ok(_) => match data.remove(views_id).await {
|
||||
Ok(_) => Json(ApiReturn {
|
||||
ok: true,
|
||||
message: "Success".to_string(),
|
||||
payload: None,
|
||||
}),
|
||||
Err(e) => Json(e.into()),
|
||||
},
|
||||
Err(e) => Json(e.into()),
|
||||
};
|
||||
}
|
||||
|
||||
// check edited slug and edit code
|
||||
if let Some(new_slug) = req.new_slug {
|
||||
if new_slug.len() < 2 {
|
||||
return Json(Error::DataTooShort("slug".to_string()).into());
|
||||
}
|
||||
|
||||
if new_slug.len() > 32 {
|
||||
return Json(Error::DataTooLong("slug".to_string()).into());
|
||||
}
|
||||
|
||||
// check for existing
|
||||
if data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries('{}')", new_slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
.is_ok()
|
||||
{
|
||||
return Json(Error::MiscError("Slug already in use".to_string()).into());
|
||||
}
|
||||
|
||||
let views_id = match data
|
||||
.query(&SimplifiedQuery {
|
||||
query: AppDataSelectQuery::KeyIs(format!("entries.views('{}')", slug)),
|
||||
mode: AppDataSelectMode::One(0),
|
||||
})
|
||||
.await
|
||||
{
|
||||
Ok(r) => match r {
|
||||
AppDataQueryResult::One(r) => r.id,
|
||||
AppDataQueryResult::Many(_) => unreachable!(),
|
||||
},
|
||||
Err(e) => return Json(e.into()),
|
||||
};
|
||||
|
||||
// rename
|
||||
if let Err(e) = data.rename(id, format!("entries('{}')", new_slug)).await {
|
||||
return Json(e.into());
|
||||
}
|
||||
|
||||
if let Err(e) = data
|
||||
.rename(views_id, format!("entries.views('{}')", new_slug))
|
||||
.await
|
||||
{
|
||||
return Json(e.into());
|
||||
}
|
||||
|
||||
entry.slug = new_slug;
|
||||
}
|
||||
|
||||
if let Some(new_edit_code) = req.new_edit_code {
|
||||
entry.salt = salt();
|
||||
entry.edit_code = hash(new_edit_code + &entry.salt);
|
||||
}
|
||||
|
||||
// update
|
||||
entry.content = req.content;
|
||||
entry.edited = unix_epoch_timestamp();
|
||||
|
||||
if let Err(e) = data
|
||||
.update(id, serde_json::to_string(&entry).unwrap())
|
||||
.await
|
||||
{
|
||||
return Json(e.into());
|
||||
}
|
||||
|
||||
// return
|
||||
Json(ApiReturn {
|
||||
ok: true,
|
||||
message: "Success".to_string(),
|
||||
payload: Some(entry.slug),
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue