nero/tsuki
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

feat: database

Browse source
This commit is contained in:
timokoesters 2020-02-20 10:12:13 +01:00
parent cd777af41c
commit c2c18b4651
3 changed files with 311 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

74
src/main.rs
View file

@ -2,16 +2,19 @@
mod ruma_wrapper;
use {
rocket::{get, post, put, routes},
directories::ProjectDirs,
rocket::{get, post, put, routes, State},
ruma_client_api::{
error::{Error, ErrorKind},
r0::{
account::register, alias::get_alias, membership::join_room_by_id,
message::create_message_event,
message::create_message_event, session::login,
},
unversioned::get_supported_versions,
},
ruma_identifiers::UserId,
ruma_wrapper::{MatrixResult, Ruma},
sled::Db,
std::convert::TryInto,
};
@ -23,8 +26,13 @@ fn get_supported_versions_route() -> MatrixResult<get_supported_versions::Respon
}
#[post("/_matrix/client/r0/register", data = "<body>")]
fn register_route(body: Ruma<register::Request>) -> MatrixResult<register::Response> {
let user_id = match (*format!(
fn register_route(
db: State<Db>,
body: Ruma<register::Request>,
) -> MatrixResult<register::Response> {
let users = db.open_tree("users").unwrap();
let user_id: UserId = match (*format!(
"@{}:localhost",
body.username.clone().unwrap_or("randomname".to_owned())
))
@ -33,13 +41,28 @@ fn register_route(body: Ruma<register::Request>) -> MatrixResult<register::Respo
Err(_) => {
return MatrixResult(Err(Error {
kind: ErrorKind::InvalidUsername,
message: "Username was invalid. ".to_owned(),
message: "Username was invalid.".to_owned(),
status_code: http::StatusCode::BAD_REQUEST,
}))
}
Ok(user_id) => user_id,
};
if users.contains_key(user_id.to_string()).unwrap() {
return MatrixResult(Err(Error {
kind: ErrorKind::UserInUse,
message: "Desired user ID is already taken.".to_owned(),
status_code: http::StatusCode::BAD_REQUEST,
}));
}
users
.insert(
user_id.to_string(),
&*body.password.clone().unwrap_or_default(),
)
.unwrap();
MatrixResult(Ok(register::Response {
access_token: "randomtoken".to_owned(),
home_server: "localhost".to_owned(),
@ -48,6 +71,38 @@ fn register_route(body: Ruma<register::Request>) -> MatrixResult<register::Respo
}))
}
#[post("/_matrix/client/r0/login", data = "<body>")]
fn login_route(db: State<Db>, body: Ruma<login::Request>) -> MatrixResult<login::Response> {
let user_id = if let login::UserInfo::MatrixId(username) = &body.user {
let user_id = format!("@{}:localhost", username);
let users = db.open_tree("users").unwrap();
if !users.contains_key(user_id.clone()).unwrap() {
dbg!();
return MatrixResult(Err(Error {
kind: ErrorKind::Forbidden,
message: "UserId not found.".to_owned(),
status_code: http::StatusCode::BAD_REQUEST,
}));
}
user_id
} else {
dbg!();
return MatrixResult(Err(Error {
kind: ErrorKind::Unknown,
message: "Bad login type.".to_owned(),
status_code: http::StatusCode::BAD_REQUEST,
}));
};
return MatrixResult(Ok(login::Response {
user_id: (*user_id).try_into().unwrap(), // User id is correct because the user is already registered
access_token: "randomtoken".to_owned(),
home_server: Some("localhost".to_owned()),
device_id: body.device_id.clone().unwrap_or("randomid".to_owned()),
well_known: None,
}));
}
#[get("/_matrix/client/r0/directory/room/<room_alias>")]
fn get_alias_route(room_alias: String) -> MatrixResult<get_alias::Response> {
let room_id = match &*room_alias {
@ -97,16 +152,25 @@ fn create_message_event_route(
fn main() {
pretty_env_logger::init();
let db = sled::open(
ProjectDirs::from("xyz", "koesters", "matrixserver")
.unwrap()
.data_dir(),
)
.unwrap();
rocket::ignite()
.mount(
"/",
routes![
get_supported_versions_route,
register_route,
login_route,
get_alias_route,
join_room_by_id_route,
create_message_event_route,
],
)
.manage(db)
.launch();
}