Conduit is a simple, fast and reliable chat server powered by Matrix https://conduit.rs
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
333 lines
10 KiB
333 lines
10 KiB
#![feature(proc_macro_hygiene, decl_macro)] |
|
mod data; |
|
mod database; |
|
mod pdu; |
|
mod ruma_wrapper; |
|
mod utils; |
|
|
|
pub use data::Data; |
|
pub use database::Database; |
|
pub use pdu::PduEvent; |
|
|
|
use log::{debug, error}; |
|
use rocket::{get, options, post, put, routes, State}; |
|
use ruma_client_api::{ |
|
error::{Error, ErrorKind}, |
|
r0::{ |
|
account::register, alias::get_alias, membership::join_room_by_id, |
|
message::create_message_event, session::login, sync::sync_events, |
|
}, |
|
unversioned::get_supported_versions, |
|
}; |
|
use ruma_events::{collections::all::RoomEvent, room::message::MessageEvent, EventResult}; |
|
use ruma_identifiers::{EventId, UserId}; |
|
use ruma_wrapper::{MatrixResult, Ruma}; |
|
use serde_json::map::Map; |
|
use std::{ |
|
collections::HashMap, |
|
convert::{TryFrom, TryInto}, |
|
path::PathBuf, |
|
}; |
|
|
|
#[get("/_matrix/client/versions")] |
|
fn get_supported_versions_route() -> MatrixResult<get_supported_versions::Response> { |
|
MatrixResult(Ok(get_supported_versions::Response { |
|
versions: vec!["r0.6.0".to_owned()], |
|
unstable_features: HashMap::new(), |
|
})) |
|
} |
|
|
|
#[post("/_matrix/client/r0/register", data = "<body>")] |
|
fn register_route( |
|
data: State<Data>, |
|
body: Ruma<register::Request>, |
|
) -> MatrixResult<register::Response> { |
|
// Validate user id |
|
let user_id: UserId = match (*format!( |
|
"@{}:{}", |
|
body.username.clone().unwrap_or("randomname".to_owned()), |
|
data.hostname() |
|
)) |
|
.try_into() |
|
{ |
|
Err(_) => { |
|
debug!("Username invalid"); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::InvalidUsername, |
|
message: "Username was invalid.".to_owned(), |
|
status_code: http::StatusCode::BAD_REQUEST, |
|
})); |
|
} |
|
Ok(user_id) => user_id, |
|
}; |
|
|
|
// Check if username is creative enough |
|
if data.user_exists(&user_id) { |
|
debug!("ID already taken"); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::UserInUse, |
|
message: "Desired user ID is already taken.".to_owned(), |
|
status_code: http::StatusCode::BAD_REQUEST, |
|
})); |
|
} |
|
|
|
// Create user |
|
data.user_add(&user_id, body.password.clone()); |
|
|
|
// Generate new device id if the user didn't specify one |
|
let device_id = body |
|
.device_id |
|
.clone() |
|
.unwrap_or_else(|| "TODO:randomdeviceid".to_owned()); |
|
|
|
// Add device |
|
data.device_add(&user_id, &device_id); |
|
|
|
// Generate new token for the device |
|
let token = "TODO:randomtoken".to_owned(); |
|
data.token_replace(&user_id, &device_id, token.clone()); |
|
|
|
MatrixResult(Ok(register::Response { |
|
access_token: token, |
|
home_server: data.hostname().to_owned(), |
|
user_id, |
|
device_id, |
|
})) |
|
} |
|
|
|
#[post("/_matrix/client/r0/login", data = "<body>")] |
|
fn login_route(data: State<Data>, body: Ruma<login::Request>) -> MatrixResult<login::Response> { |
|
// Validate login method |
|
let user_id = |
|
if let (login::UserInfo::MatrixId(mut username), login::LoginInfo::Password { password }) = |
|
(body.user.clone(), body.login_info.clone()) |
|
{ |
|
if !username.contains(':') { |
|
username = format!("@{}:{}", username, data.hostname()); |
|
} |
|
if let Ok(user_id) = (*username).try_into() { |
|
if !data.user_exists(&user_id) {} |
|
|
|
// Check password |
|
if let Some(correct_password) = data.password_get(&user_id) { |
|
if password == correct_password { |
|
// Success! |
|
user_id |
|
} else { |
|
debug!("Invalid password."); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::Unknown, |
|
message: "".to_owned(), |
|
status_code: http::StatusCode::FORBIDDEN, |
|
})); |
|
} |
|
} else { |
|
debug!("UserId does not exist (has no assigned password). Can't log in."); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::Forbidden, |
|
message: "".to_owned(), |
|
status_code: http::StatusCode::FORBIDDEN, |
|
})); |
|
} |
|
} else { |
|
debug!("Invalid UserId."); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::Unknown, |
|
message: "Bad login type.".to_owned(), |
|
status_code: http::StatusCode::BAD_REQUEST, |
|
})); |
|
} |
|
} else { |
|
debug!("Bad login type"); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::Unknown, |
|
message: "Bad login type.".to_owned(), |
|
status_code: http::StatusCode::BAD_REQUEST, |
|
})); |
|
}; |
|
|
|
// Generate new device id if the user didn't specify one |
|
let device_id = body |
|
.device_id |
|
.clone() |
|
.unwrap_or("TODO:randomdeviceid".to_owned()); |
|
|
|
// Add device |
|
data.device_add(&user_id, &device_id); |
|
|
|
// Generate a new token for the device |
|
let token = "TODO:randomtoken".to_owned(); |
|
data.token_replace(&user_id, &device_id, token.clone()); |
|
|
|
return MatrixResult(Ok(login::Response { |
|
user_id, |
|
access_token: token, |
|
home_server: Some(data.hostname().to_owned()), |
|
device_id, |
|
well_known: None, |
|
})); |
|
} |
|
|
|
#[get("/_matrix/client/r0/directory/room/<room_alias>")] |
|
fn get_alias_route(room_alias: String) -> MatrixResult<get_alias::Response> { |
|
// TODO |
|
let room_id = match &*room_alias { |
|
"#room:localhost" => "!xclkjvdlfj:localhost", |
|
_ => { |
|
debug!("Room not found."); |
|
return MatrixResult(Err(Error { |
|
kind: ErrorKind::NotFound, |
|
message: "Room not found.".to_owned(), |
|
status_code: http::StatusCode::NOT_FOUND, |
|
})); |
|
} |
|
} |
|
.try_into() |
|
.unwrap(); |
|
|
|
MatrixResult(Ok(get_alias::Response { |
|
room_id, |
|
servers: vec!["localhost".to_owned()], |
|
})) |
|
} |
|
|
|
#[post("/_matrix/client/r0/rooms/<_room_id>/join", data = "<body>")] |
|
fn join_room_by_id_route( |
|
_room_id: String, |
|
body: Ruma<join_room_by_id::Request>, |
|
) -> MatrixResult<join_room_by_id::Response> { |
|
// TODO |
|
MatrixResult(Ok(join_room_by_id::Response { |
|
room_id: body.room_id.clone(), |
|
})) |
|
} |
|
|
|
#[put( |
|
"/_matrix/client/r0/rooms/<_room_id>/send/<_event_type>/<_txn_id>", |
|
data = "<body>" |
|
)] |
|
fn create_message_event_route( |
|
data: State<Data>, |
|
_room_id: String, |
|
_event_type: String, |
|
_txn_id: String, |
|
body: Ruma<create_message_event::Request>, |
|
) -> MatrixResult<create_message_event::Response> { |
|
// Construct event |
|
let mut event = RoomEvent::RoomMessage(MessageEvent { |
|
content: body.data.clone().into_result().unwrap(), |
|
event_id: EventId::try_from("$thiswillbefilledinlater").unwrap(), |
|
origin_server_ts: utils::millis_since_unix_epoch(), |
|
room_id: Some(body.room_id.clone()), |
|
sender: body.user_id.clone().expect("user is authenticated"), |
|
unsigned: Map::default(), |
|
}); |
|
|
|
// Generate event id |
|
let event_id = EventId::try_from(&*format!( |
|
"${}", |
|
ruma_signatures::reference_hash(&serde_json::to_value(&event).unwrap()) |
|
.expect("ruma can calculate reference hashes") |
|
)) |
|
.expect("ruma's reference hashes are correct"); |
|
|
|
// Insert event id |
|
if let RoomEvent::RoomMessage(message) = &mut event { |
|
message.event_id = event_id.clone(); |
|
data.pdu_append_message(&event_id, &body.room_id, message.clone()); |
|
} else { |
|
error!("only roommessages are handled currently"); |
|
} |
|
|
|
MatrixResult(Ok(create_message_event::Response { event_id })) |
|
} |
|
|
|
#[get("/_matrix/client/r0/sync", data = "<body>")] |
|
fn sync_route( |
|
data: State<Data>, |
|
body: Ruma<sync_events::Request>, |
|
) -> MatrixResult<sync_events::Response> { |
|
let mut joined_rooms = HashMap::new(); |
|
{ |
|
let pdus = data.pdus_all(); |
|
let mut room_events = Vec::new(); |
|
|
|
for pdu in pdus { |
|
room_events.push(pdu.to_room_event()); |
|
} |
|
|
|
joined_rooms.insert( |
|
"!roomid:localhost".try_into().unwrap(), |
|
sync_events::JoinedRoom { |
|
account_data: sync_events::AccountData { events: Vec::new() }, |
|
summary: sync_events::RoomSummary { |
|
heroes: Vec::new(), |
|
joined_member_count: None, |
|
invited_member_count: None, |
|
}, |
|
unread_notifications: sync_events::UnreadNotificationsCount { |
|
highlight_count: None, |
|
notification_count: None, |
|
}, |
|
timeline: sync_events::Timeline { |
|
limited: Some(false), |
|
prev_batch: Some("".to_owned()), |
|
events: room_events, |
|
}, |
|
state: sync_events::State { events: Vec::new() }, |
|
ephemeral: sync_events::Ephemeral { events: Vec::new() }, |
|
}, |
|
); |
|
} |
|
|
|
MatrixResult(Ok(sync_events::Response { |
|
next_batch: String::new(), |
|
rooms: sync_events::Rooms { |
|
leave: Default::default(), |
|
join: joined_rooms, |
|
invite: Default::default(), |
|
}, |
|
presence: sync_events::Presence { events: Vec::new() }, |
|
device_lists: Default::default(), |
|
device_one_time_keys_count: Default::default(), |
|
to_device: sync_events::ToDevice { events: Vec::new() }, |
|
})) |
|
} |
|
|
|
#[options("/<_segments..>")] |
|
fn options_route(_segments: PathBuf) -> MatrixResult<create_message_event::Response> { |
|
MatrixResult(Err(Error { |
|
kind: ErrorKind::NotFound, |
|
message: "Room not found.".to_owned(), |
|
status_code: http::StatusCode::NOT_FOUND, |
|
})) |
|
} |
|
|
|
fn main() { |
|
// Log info by default |
|
if let Err(_) = std::env::var("RUST_LOG") { |
|
std::env::set_var("RUST_LOG", "matrixserver=debug,info"); |
|
} |
|
pretty_env_logger::init(); |
|
|
|
let data = Data::load_or_create("localhost"); |
|
data.debug(); |
|
|
|
rocket::ignite() |
|
.mount( |
|
"/", |
|
routes![ |
|
get_supported_versions_route, |
|
register_route, |
|
login_route, |
|
get_alias_route, |
|
join_room_by_id_route, |
|
create_message_event_route, |
|
sync_route, |
|
options_route, |
|
], |
|
) |
|
.manage(data) |
|
.launch() |
|
.unwrap(); |
|
}
|
|
|