160 lines
3.7 KiB
Rust
160 lines
3.7 KiB
Rust
pub mod server;
|
|
pub mod model;
|
|
pub mod routes;
|
|
pub mod cli;
|
|
|
|
pub mod errors;
|
|
|
|
#[cfg(feature = "migrations")]
|
|
mod migrations;
|
|
|
|
#[cfg(feature = "migrations")]
|
|
use sea_orm_migration::MigratorTrait;
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use sea_orm::{ConnectOptions, Database};
|
|
|
|
pub use errors::UpubResult as Result;
|
|
use tower_http::{cors::CorsLayer, trace::TraceLayer};
|
|
|
|
pub const VERSION: &str = env!("CARGO_PKG_VERSION");
|
|
|
|
#[derive(Parser)]
|
|
/// all names were taken
|
|
struct CliArgs {
|
|
#[clap(subcommand)]
|
|
/// command to run
|
|
command: CliCommand,
|
|
|
|
#[arg(short = 'd', long = "db", default_value = "sqlite://./upub.db")]
|
|
/// database connection uri
|
|
database: String,
|
|
|
|
#[arg(short = 'D', long, default_value = "http://localhost:3000")]
|
|
/// instance base domain, for AP ids
|
|
domain: String,
|
|
|
|
#[arg(long, default_value_t=false)]
|
|
/// run with debug level tracing
|
|
debug: bool,
|
|
}
|
|
|
|
#[derive(Clone, Subcommand)]
|
|
enum CliCommand {
|
|
/// run fediverse server
|
|
Serve ,
|
|
|
|
#[cfg(feature = "migrations")]
|
|
/// apply database migrations
|
|
Migrate,
|
|
|
|
#[cfg(feature = "faker")]
|
|
/// generate fake user, note and activity
|
|
Faker{
|
|
/// how many fake statuses to insert for root user
|
|
count: u64,
|
|
},
|
|
|
|
/// fetch a single AP object
|
|
Fetch {
|
|
/// object id, or uri, to fetch
|
|
uri: String,
|
|
|
|
#[arg(long, default_value_t = false)]
|
|
/// store fetched object in local db
|
|
save: bool,
|
|
},
|
|
|
|
/// follow a remote relay
|
|
Relay {
|
|
/// actor url, same as with pleroma
|
|
actor: String,
|
|
|
|
#[arg(long, default_value_t = false)]
|
|
/// instead of sending a follow request, send an accept
|
|
accept: bool
|
|
},
|
|
|
|
/// run db maintenance tasks
|
|
Fix {
|
|
#[arg(long, default_value_t = false)]
|
|
/// fix likes counts for posts
|
|
likes: bool,
|
|
|
|
#[arg(long, default_value_t = false)]
|
|
/// fix shares counts for posts
|
|
shares: bool,
|
|
|
|
#[arg(long, default_value_t = false)]
|
|
/// fix replies counts for posts
|
|
replies: bool,
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
|
|
let args = CliArgs::parse();
|
|
|
|
tracing_subscriber::fmt()
|
|
.compact()
|
|
.with_max_level(if args.debug { tracing::Level::DEBUG } else { tracing::Level::INFO })
|
|
.init();
|
|
|
|
// TODO can i do connectoptions.into() or .connect() and skip these ugly bindings?
|
|
let mut opts = ConnectOptions::new(&args.database);
|
|
|
|
opts
|
|
.sqlx_logging_level(tracing::log::LevelFilter::Debug);
|
|
|
|
let db = Database::connect(opts)
|
|
.await.expect("error connecting to db");
|
|
|
|
match args.command {
|
|
#[cfg(feature = "migrations")]
|
|
CliCommand::Migrate =>
|
|
migrations::Migrator::up(&db, None)
|
|
.await.expect("error applying migrations"),
|
|
|
|
#[cfg(feature = "faker")]
|
|
CliCommand::Faker { count } =>
|
|
cli::faker(&db, args.domain, count)
|
|
.await.expect("error creating fake entities"),
|
|
|
|
CliCommand::Fetch { uri, save } =>
|
|
cli::fetch(db, args.domain, uri, save)
|
|
.await.expect("error fetching object"),
|
|
|
|
CliCommand::Relay { actor, accept } =>
|
|
cli::relay(db, args.domain, actor, accept)
|
|
.await.expect("error registering/accepting relay"),
|
|
|
|
CliCommand::Fix { likes, shares, replies } =>
|
|
cli::fix(db, likes, shares, replies)
|
|
.await
|
|
.expect("failed running fix task"),
|
|
|
|
CliCommand::Serve => {
|
|
let ctx = server::Context::new(db, args.domain)
|
|
.await.expect("failed creating server context");
|
|
|
|
use routes::activitypub::ActivityPubRouter;
|
|
use routes::mastodon::MastodonRouter;
|
|
|
|
let router = axum::Router::new()
|
|
.ap_routes()
|
|
.mastodon_routes() // no-op if mastodon feature is disabled
|
|
.layer(CorsLayer::permissive())
|
|
.layer(TraceLayer::new_for_http())
|
|
.with_state(ctx);
|
|
|
|
// run our app with hyper, listening locally on port 3000
|
|
let listener = tokio::net::TcpListener::bind("127.0.0.1:3000")
|
|
.await.expect("could not bind tcp socket");
|
|
|
|
axum::serve(listener, router)
|
|
.await
|
|
.expect("failed serving application")
|
|
},
|
|
}
|
|
}
|