2020-05-03 15:25:31 +00:00
|
|
|
#![warn(rust_2018_idioms)]
|
2020-04-09 18:47:03 +00:00
|
|
|
|
2020-12-08 09:33:44 +00:00
|
|
|
pub mod appservice_server;
|
2020-08-14 09:31:31 +00:00
|
|
|
pub mod client_server;
|
2020-08-14 09:34:15 +00:00
|
|
|
pub mod server_server;
|
2020-06-12 11:18:25 +00:00
|
|
|
|
2020-03-30 11:46:18 +00:00
|
|
|
mod database;
|
2020-05-03 15:25:31 +00:00
|
|
|
mod error;
|
2020-04-04 09:53:37 +00:00
|
|
|
mod pdu;
|
2020-08-14 09:34:15 +00:00
|
|
|
mod push_rules;
|
2020-02-15 21:42:21 +00:00
|
|
|
mod ruma_wrapper;
|
2020-03-29 11:48:44 +00:00
|
|
|
mod utils;
|
2020-02-15 21:42:21 +00:00
|
|
|
|
2020-03-30 11:46:18 +00:00
|
|
|
pub use database::Database;
|
2020-11-14 22:13:06 +00:00
|
|
|
pub use error::{ConduitLogger, Error, Result};
|
|
|
|
use log::LevelFilter;
|
2020-04-04 09:53:37 +00:00
|
|
|
pub use pdu::PduEvent;
|
2020-07-26 03:08:00 +00:00
|
|
|
pub use rocket::State;
|
2020-06-09 13:13:17 +00:00
|
|
|
pub use ruma_wrapper::{ConduitResult, Ruma, RumaResponse};
|
2020-03-30 11:46:18 +00:00
|
|
|
|
2020-12-19 15:00:11 +00:00
|
|
|
use rocket::{catch, catchers, fairing::AdHoc, routes, Request};
|
2020-04-03 15:27:08 +00:00
|
|
|
|
2020-04-29 06:48:56 +00:00
|
|
|
fn setup_rocket() -> rocket::Rocket {
|
2020-11-14 22:13:06 +00:00
|
|
|
// Force log level off, so we can use our own logger
|
2020-12-05 20:03:43 +00:00
|
|
|
std::env::set_var("ROCKET_LOG_LEVEL", "off");
|
2020-11-14 22:13:06 +00:00
|
|
|
|
2020-02-15 21:42:21 +00:00
|
|
|
rocket::ignite()
|
2020-02-18 21:07:57 +00:00
|
|
|
.mount(
|
|
|
|
"/",
|
|
|
|
routes![
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_supported_versions_route,
|
2020-05-01 21:17:25 +00:00
|
|
|
client_server::get_register_available_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::register_route,
|
2020-07-31 12:40:28 +00:00
|
|
|
client_server::get_login_types_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::login_route,
|
2020-07-27 15:38:00 +00:00
|
|
|
client_server::whoami_route,
|
2020-05-24 20:10:09 +00:00
|
|
|
client_server::logout_route,
|
2020-07-05 06:35:19 +00:00
|
|
|
client_server::logout_all_route,
|
2020-07-02 18:38:25 +00:00
|
|
|
client_server::change_password_route,
|
2020-07-05 05:48:19 +00:00
|
|
|
client_server::deactivate_route,
|
2020-04-19 12:14:47 +00:00
|
|
|
client_server::get_capabilities_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_pushrules_all_route,
|
2020-05-01 18:26:57 +00:00
|
|
|
client_server::set_pushrule_route,
|
|
|
|
client_server::set_pushrule_enabled_route,
|
2020-07-11 12:08:37 +00:00
|
|
|
client_server::get_room_event_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_filter_route,
|
|
|
|
client_server::create_filter_route,
|
|
|
|
client_server::set_global_account_data_route,
|
|
|
|
client_server::get_global_account_data_route,
|
|
|
|
client_server::set_displayname_route,
|
|
|
|
client_server::get_displayname_route,
|
|
|
|
client_server::set_avatar_url_route,
|
|
|
|
client_server::get_avatar_url_route,
|
|
|
|
client_server::get_profile_route,
|
|
|
|
client_server::set_presence_route,
|
|
|
|
client_server::upload_keys_route,
|
2020-05-17 17:56:40 +00:00
|
|
|
client_server::get_keys_route,
|
|
|
|
client_server::claim_keys_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::create_backup_route,
|
|
|
|
client_server::update_backup_route,
|
2020-08-27 12:48:20 +00:00
|
|
|
client_server::delete_backup_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::get_latest_backup_route,
|
|
|
|
client_server::get_backup_route,
|
2020-08-27 12:48:20 +00:00
|
|
|
client_server::add_backup_key_sessions_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::add_backup_keys_route,
|
2020-08-27 12:48:20 +00:00
|
|
|
client_server::delete_backup_key_session_route,
|
|
|
|
client_server::delete_backup_key_sessions_route,
|
|
|
|
client_server::delete_backup_keys_route,
|
|
|
|
client_server::get_backup_key_session_route,
|
|
|
|
client_server::get_backup_key_sessions_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::get_backup_keys_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::set_read_marker_route,
|
2020-12-19 15:00:11 +00:00
|
|
|
client_server::set_receipt_route,
|
2020-04-12 19:12:50 +00:00
|
|
|
client_server::create_typing_event_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::create_room_route,
|
2020-05-25 21:24:13 +00:00
|
|
|
client_server::redact_event_route,
|
|
|
|
client_server::create_alias_route,
|
|
|
|
client_server::delete_alias_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_alias_route,
|
|
|
|
client_server::join_room_by_id_route,
|
|
|
|
client_server::join_room_by_id_or_alias_route,
|
2020-07-10 06:49:05 +00:00
|
|
|
client_server::joined_members_route,
|
2020-04-19 12:14:47 +00:00
|
|
|
client_server::leave_room_route,
|
2020-04-28 17:56:34 +00:00
|
|
|
client_server::forget_room_route,
|
2020-07-25 16:35:22 +00:00
|
|
|
client_server::joined_rooms_route,
|
2020-06-04 13:02:27 +00:00
|
|
|
client_server::kick_user_route,
|
|
|
|
client_server::ban_user_route,
|
|
|
|
client_server::unban_user_route,
|
2020-04-14 11:54:32 +00:00
|
|
|
client_server::invite_user_route,
|
2020-05-25 21:24:13 +00:00
|
|
|
client_server::set_room_visibility_route,
|
|
|
|
client_server::get_room_visibility_route,
|
2020-05-19 14:28:03 +00:00
|
|
|
client_server::get_public_rooms_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_public_rooms_filtered_route,
|
2020-04-14 11:54:32 +00:00
|
|
|
client_server::search_users_route,
|
2020-04-19 12:14:47 +00:00
|
|
|
client_server::get_member_events_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::get_protocols_route,
|
2020-08-12 21:32:39 +00:00
|
|
|
client_server::send_message_event_route,
|
|
|
|
client_server::send_state_event_for_key_route,
|
|
|
|
client_server::send_state_event_for_empty_key_route,
|
2020-05-18 07:22:07 +00:00
|
|
|
client_server::get_state_events_route,
|
|
|
|
client_server::get_state_events_for_key_route,
|
|
|
|
client_server::get_state_events_for_empty_key_route,
|
2020-07-27 15:36:54 +00:00
|
|
|
client_server::sync_events_route,
|
2020-06-04 11:58:55 +00:00
|
|
|
client_server::get_context_route,
|
2020-04-28 17:56:34 +00:00
|
|
|
client_server::get_message_events_route,
|
2020-08-18 10:15:27 +00:00
|
|
|
client_server::search_events_route,
|
2020-04-14 11:54:32 +00:00
|
|
|
client_server::turn_server_route,
|
2020-05-01 18:26:57 +00:00
|
|
|
client_server::send_event_to_device_route,
|
|
|
|
client_server::get_media_config_route,
|
2020-05-18 15:53:34 +00:00
|
|
|
client_server::create_content_route,
|
|
|
|
client_server::get_content_route,
|
|
|
|
client_server::get_content_thumbnail_route,
|
2020-06-03 11:41:30 +00:00
|
|
|
client_server::get_devices_route,
|
|
|
|
client_server::get_device_route,
|
|
|
|
client_server::update_device_route,
|
|
|
|
client_server::delete_device_route,
|
|
|
|
client_server::delete_devices_route,
|
2020-07-26 20:33:20 +00:00
|
|
|
client_server::get_tags_route,
|
|
|
|
client_server::update_tag_route,
|
|
|
|
client_server::delete_tag_route,
|
2020-04-11 07:30:11 +00:00
|
|
|
client_server::options_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::upload_signing_keys_route,
|
|
|
|
client_server::upload_signatures_route,
|
2020-07-29 15:37:26 +00:00
|
|
|
client_server::get_key_changes_route,
|
2020-07-30 16:14:47 +00:00
|
|
|
client_server::get_pushers_route,
|
2020-06-16 10:11:38 +00:00
|
|
|
client_server::set_pushers_route,
|
2020-08-06 11:21:53 +00:00
|
|
|
client_server::upgrade_room_route,
|
2020-12-05 20:03:43 +00:00
|
|
|
server_server::get_server_version_route,
|
|
|
|
server_server::get_server_keys_route,
|
|
|
|
server_server::get_server_keys_deprecated_route,
|
2020-08-14 09:29:32 +00:00
|
|
|
server_server::get_public_rooms_route,
|
2020-09-17 12:44:47 +00:00
|
|
|
server_server::get_public_rooms_filtered_route,
|
2020-08-14 09:29:32 +00:00
|
|
|
server_server::send_transaction_message_route,
|
2020-09-25 10:26:29 +00:00
|
|
|
server_server::get_missing_events_route,
|
2020-10-05 20:19:22 +00:00
|
|
|
server_server::get_profile_information_route,
|
2020-02-18 21:07:57 +00:00
|
|
|
],
|
|
|
|
)
|
2020-12-19 15:00:11 +00:00
|
|
|
.register(catchers![not_found_catcher])
|
2020-12-05 20:03:43 +00:00
|
|
|
.attach(AdHoc::on_attach("Config", |rocket| async {
|
|
|
|
let data =
|
|
|
|
Database::load_or_create(rocket.figment().extract().expect("config is valid"))
|
2020-12-06 10:05:51 +00:00
|
|
|
.await
|
2020-12-05 20:03:43 +00:00
|
|
|
.expect("config is valid");
|
2020-04-29 06:48:56 +00:00
|
|
|
|
2020-12-08 09:33:44 +00:00
|
|
|
data.sending
|
|
|
|
.start_handler(&data.globals, &data.rooms, &data.appservice);
|
2020-11-14 22:13:06 +00:00
|
|
|
log::set_boxed_logger(Box::new(ConduitLogger {
|
|
|
|
db: data.clone(),
|
|
|
|
last_logs: Default::default(),
|
|
|
|
}))
|
|
|
|
.unwrap();
|
|
|
|
log::set_max_level(LevelFilter::Info);
|
2020-09-15 14:13:54 +00:00
|
|
|
|
2020-04-29 06:48:56 +00:00
|
|
|
Ok(rocket.manage(data))
|
|
|
|
}))
|
2020-04-10 11:36:57 +00:00
|
|
|
}
|
|
|
|
|
2020-06-16 11:53:15 +00:00
|
|
|
#[rocket::main]
|
|
|
|
async fn main() {
|
|
|
|
setup_rocket().launch().await.unwrap();
|
2020-02-15 21:42:21 +00:00
|
|
|
}
|
2020-12-19 15:00:11 +00:00
|
|
|
|
|
|
|
#[catch(404)]
|
|
|
|
fn not_found_catcher(_req: &'_ Request<'_>) -> String {
|
|
|
|
"404 Not Found".to_owned()
|
|
|
|
}
|