upub/src/main.rs

pub mod model;
pub mod migrations;
pub mod activitystream;
pub mod activitypub;
pub mod server;

use clap::{Parser, Subcommand};
use sea_orm::{ConnectOptions, Database, EntityTrait, IntoActiveModel};
use sea_orm_migration::MigratorTrait;

use crate::activitystream::{BaseType, ObjectType};

#[derive(Parser)]
/// all names were taken
struct CliArgs {
	#[clap(subcommand)]
	/// command to run
	command: CliCommand,

	#[arg(short, long, default_value = "sqlite://./upub.db")]
	/// database connection uri
	database: String,

	#[arg(long, default_value_t=false)]
	/// run with debug level tracing
	debug: bool,
}

#[derive(Clone, Subcommand)]
enum CliCommand {
	/// run fediverse server
	Serve ,

	/// apply database migrations
	Migrate,

	/// generate fake user, note and activity
	Faker,

	/// 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,
	},
}

#[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();

	let mut opts = ConnectOptions::new(&args.database);
	opts
		.max_connections(1);

	let db = Database::connect(opts)
		.await.expect("error connecting to db");

	match args.command {
		CliCommand::Serve => server::serve(db)
			.await,

		CliCommand::Migrate => migrations::Migrator::up(&db, None)
			.await.expect("error applying migrations"),

		CliCommand::Faker => model::faker(&db)
			.await.expect("error creating fake entities"),

		CliCommand::Fetch { uri, save } => fetch(&db, &uri, save)
			.await.expect("error fetching object"),
	}
}



async fn fetch(db: &sea_orm::DatabaseConnection, uri: &str, save: bool) -> reqwest::Result<()> {
	use crate::activitystream::{Base, Object};

	let mut node = activitystream::Node::from(uri);
	tracing::info!("fetching object");
	node.fetch().await?;
	tracing::info!("fetched node");

	let obj = node.get().expect("node still empty after fetch?");

	tracing::info!("fetched object:{}, name:{}", obj.id().unwrap_or(""), obj.name().unwrap_or(""));
	
	if save {
		match obj.base_type() {
			Some(BaseType::Object(ObjectType::Actor(_))) => {
				model::user::Entity::insert(
					model::user::Model::new(obj).unwrap().into_active_model()
				).exec(db).await.unwrap();
			},
			Some(BaseType::Object(ObjectType::Activity(_))) => {
				model::activity::Entity::insert(
					model::activity::Model::new(obj).unwrap().into_active_model()
				).exec(db).await.unwrap();
			},
			Some(BaseType::Object(ObjectType::Note)) => {
				model::object::Entity::insert(
					model::object::Model::new(obj).unwrap().into_active_model()
				).exec(db).await.unwrap();
			},
			Some(BaseType::Object(t)) => tracing::warn!("not implemented: {:?}", t),
			Some(BaseType::Link(_)) => tracing::error!("fetched another link?"),
			None => tracing::error!("no type on object"),
		}
	}

	Ok(())
}