forked from alemi/upub
136 lines
3.5 KiB
Rust
136 lines
3.5 KiB
Rust
use std::{str::Utf8Error, sync::Arc};
|
|
|
|
use openssl::rsa::Rsa;
|
|
use sea_orm::{DatabaseConnection, DbErr, EntityTrait};
|
|
|
|
use crate::{dispatcher::Dispatcher, fetcher::Fetcher, model};
|
|
|
|
#[derive(Clone)]
|
|
pub struct Context(Arc<ContextInner>);
|
|
struct ContextInner {
|
|
db: DatabaseConnection,
|
|
domain: String,
|
|
protocol: String,
|
|
fetcher: Fetcher,
|
|
// TODO keep these pre-parsed
|
|
app: model::application::Model,
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! url {
|
|
($ctx:expr, $($args: tt)*) => {
|
|
format!("{}{}{}", $ctx.protocol(), $ctx.base(), format!($($args)*))
|
|
};
|
|
}
|
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
pub enum ContextError {
|
|
#[error("database error: {0}")]
|
|
Db(#[from] DbErr),
|
|
|
|
#[error("openssl error: {0}")]
|
|
OpenSSL(#[from] openssl::error::ErrorStack),
|
|
|
|
#[error("invalid UTF8 PEM key: {0}")]
|
|
UTF8Error(#[from] Utf8Error)
|
|
}
|
|
|
|
impl Context {
|
|
pub async fn new(db: DatabaseConnection, mut domain: String) -> Result<Self, ContextError> {
|
|
let protocol = if domain.starts_with("http://")
|
|
{ "http://" } else { "https://" }.to_string();
|
|
if domain.ends_with('/') {
|
|
domain.replace_range(domain.len()-1.., "");
|
|
}
|
|
if domain.starts_with("http") {
|
|
domain = domain.replace("https://", "").replace("http://", "");
|
|
}
|
|
for _ in 0..1 { // TODO customize delivery workers amount
|
|
Dispatcher::spawn(db.clone(), domain.clone(), 30); // TODO ew don't do it this deep and secretly!!
|
|
}
|
|
let app = match model::application::Entity::find().one(&db).await? {
|
|
Some(model) => model,
|
|
None => {
|
|
tracing::info!("generating application keys");
|
|
let rsa = Rsa::generate(2048)?;
|
|
let privk = std::str::from_utf8(&rsa.private_key_to_pem()?)?.to_string();
|
|
let pubk = std::str::from_utf8(&rsa.public_key_to_pem()?)?.to_string();
|
|
let system = model::application::ActiveModel {
|
|
id: sea_orm::ActiveValue::NotSet,
|
|
private_key: sea_orm::ActiveValue::Set(privk.clone()),
|
|
public_key: sea_orm::ActiveValue::Set(pubk.clone()),
|
|
created: sea_orm::ActiveValue::Set(chrono::Utc::now()),
|
|
};
|
|
model::application::Entity::insert(system).exec(&db).await?;
|
|
// sqlite doesn't resurn last inserted id so we're better off just querying again, it's just one time
|
|
model::application::Entity::find().one(&db).await?.expect("could not find app config just inserted")
|
|
}
|
|
};
|
|
|
|
let fetcher = Fetcher::new(db.clone(), domain.clone(), app.private_key.clone());
|
|
|
|
Ok(Context(Arc::new(ContextInner {
|
|
db, domain, protocol, app, fetcher,
|
|
})))
|
|
}
|
|
|
|
pub fn app(&self) -> &model::application::Model {
|
|
&self.0.app
|
|
}
|
|
|
|
pub fn db(&self) -> &DatabaseConnection {
|
|
&self.0.db
|
|
}
|
|
|
|
pub fn base(&self) -> &str {
|
|
&self.0.domain
|
|
}
|
|
|
|
pub fn protocol(&self) -> &str {
|
|
&self.0.protocol
|
|
}
|
|
|
|
pub fn uri(&self, entity: &str, id: String) -> String {
|
|
if id.starts_with("http") { id } else {
|
|
format!("{}{}/{}/{}", self.0.protocol, self.0.domain, entity, id)
|
|
}
|
|
}
|
|
|
|
pub fn fetch(&self) -> &Fetcher {
|
|
&self.0.fetcher
|
|
}
|
|
|
|
/// get full user id uri
|
|
pub fn uid(&self, id: String) -> String {
|
|
self.uri("users", id)
|
|
}
|
|
|
|
/// get full object id uri
|
|
pub fn oid(&self, id: String) -> String {
|
|
self.uri("objects", id)
|
|
}
|
|
|
|
/// get full activity id uri
|
|
pub fn aid(&self, id: String) -> String {
|
|
self.uri("activities", id)
|
|
}
|
|
|
|
/// get bare id, usually an uuid but unspecified
|
|
pub fn id(&self, id: String) -> String {
|
|
if id.starts_with(&self.0.domain) {
|
|
id.split('/').last().unwrap_or("").to_string()
|
|
} else {
|
|
id
|
|
}
|
|
}
|
|
|
|
pub fn server(id: &str) -> String {
|
|
id
|
|
.replace("https://", "")
|
|
.replace("http://", "")
|
|
.split('/')
|
|
.next()
|
|
.unwrap_or("")
|
|
.to_string()
|
|
}
|
|
}
|