mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-11-26 11:21:02 +00:00
Refactoring websocket code, adding create community
- Adding create community
This commit is contained in:
parent
e570c70701
commit
e1cb805cfc
7 changed files with 440 additions and 220 deletions
|
@ -4,7 +4,7 @@ use diesel::result::Error;
|
||||||
use schema::user_::dsl::*;
|
use schema::user_::dsl::*;
|
||||||
use serde::{Serialize, Deserialize};
|
use serde::{Serialize, Deserialize};
|
||||||
use {Crud,is_email_regex};
|
use {Crud,is_email_regex};
|
||||||
use jsonwebtoken::{encode, decode, Header, Validation};
|
use jsonwebtoken::{encode, decode, Header, Validation, TokenData};
|
||||||
use bcrypt::{DEFAULT_COST, hash};
|
use bcrypt::{DEFAULT_COST, hash};
|
||||||
|
|
||||||
#[derive(Queryable, Identifiable, PartialEq, Debug)]
|
#[derive(Queryable, Identifiable, PartialEq, Debug)]
|
||||||
|
@ -60,9 +60,20 @@ impl Crud<UserForm> for User_ {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
struct Claims {
|
pub struct Claims {
|
||||||
id: i32,
|
pub id: i32,
|
||||||
username: String
|
pub username: String,
|
||||||
|
pub iss: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Claims {
|
||||||
|
pub fn decode(jwt: &str) -> Result<TokenData<Claims>, jsonwebtoken::errors::Error> {
|
||||||
|
let v = Validation {
|
||||||
|
validate_exp: false,
|
||||||
|
..Validation::default()
|
||||||
|
};
|
||||||
|
decode::<Claims>(&jwt, "secret".as_ref(), &v)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type Jwt = String;
|
type Jwt = String;
|
||||||
|
@ -70,7 +81,8 @@ impl User_ {
|
||||||
pub fn jwt(&self) -> Jwt {
|
pub fn jwt(&self) -> Jwt {
|
||||||
let my_claims = Claims {
|
let my_claims = Claims {
|
||||||
id: self.id,
|
id: self.id,
|
||||||
username: self.name.to_owned()
|
username: self.name.to_owned(),
|
||||||
|
iss: "rrf".to_string() // TODO this should come from config file
|
||||||
};
|
};
|
||||||
encode(&Header::default(), &my_claims, "secret".as_ref()).unwrap()
|
encode(&Header::default(), &my_claims, "secret".as_ref()).unwrap()
|
||||||
}
|
}
|
||||||
|
@ -86,12 +98,13 @@ impl User_ {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn find_by_jwt(conn: &PgConnection, jwt: &str) -> Result<Self, Error> {
|
pub fn find_by_jwt(conn: &PgConnection, jwt: &str) -> Result<Self, Error> {
|
||||||
let token = decode::<Claims>(&jwt, "secret".as_ref(), &Validation::default())
|
let claims: Claims = Claims::decode(&jwt).expect("Invalid token").claims;
|
||||||
.expect("Couldn't decode jwt");
|
Self::read(&conn, claims.id)
|
||||||
Self::read(&conn, token.claims.id)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use establish_connection;
|
use establish_connection;
|
||||||
|
|
|
@ -4,7 +4,6 @@ use std::time::{Instant, Duration};
|
||||||
use server::actix::*;
|
use server::actix::*;
|
||||||
use server::actix_web::server::HttpServer;
|
use server::actix_web::server::HttpServer;
|
||||||
use server::actix_web::{fs, http, ws, App, Error, HttpRequest, HttpResponse};
|
use server::actix_web::{fs, http, ws, App, Error, HttpRequest, HttpResponse};
|
||||||
use std::str::FromStr;
|
|
||||||
|
|
||||||
use server::websocket_server::server::*;
|
use server::websocket_server::server::*;
|
||||||
|
|
||||||
|
@ -82,15 +81,16 @@ impl Handler<WSMessage> for WSSession {
|
||||||
type Result = ();
|
type Result = ();
|
||||||
|
|
||||||
fn handle(&mut self, msg: WSMessage, ctx: &mut Self::Context) {
|
fn handle(&mut self, msg: WSMessage, ctx: &mut Self::Context) {
|
||||||
|
println!("id: {} msg: {}", self.id, msg.0);
|
||||||
ctx.text(msg.0);
|
ctx.text(msg.0);
|
||||||
|
ctx.text("NO");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
use server::serde_json::Value;
|
|
||||||
/// WebSocket message handler
|
/// WebSocket message handler
|
||||||
impl StreamHandler<ws::Message, ws::ProtocolError> for WSSession {
|
impl StreamHandler<ws::Message, ws::ProtocolError> for WSSession {
|
||||||
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
||||||
println!("WEBSOCKET MESSAGE: {:?}", msg);
|
println!("WEBSOCKET MESSAGE: {:?} from id: {}", msg, self.id);
|
||||||
match msg {
|
match msg {
|
||||||
ws::Message::Ping(msg) => {
|
ws::Message::Ping(msg) => {
|
||||||
self.hb = Instant::now();
|
self.hb = Instant::now();
|
||||||
|
@ -100,86 +100,29 @@ impl StreamHandler<ws::Message, ws::ProtocolError> for WSSession {
|
||||||
self.hb = Instant::now();
|
self.hb = Instant::now();
|
||||||
}
|
}
|
||||||
ws::Message::Text(text) => {
|
ws::Message::Text(text) => {
|
||||||
let m = text.trim();
|
let m = text.trim().to_owned();
|
||||||
let json: Value = serde_json::from_str(m).unwrap();
|
|
||||||
|
ctx.state()
|
||||||
// Get the OP command, and its data
|
.addr
|
||||||
let op: &str = &json["op"].as_str().unwrap();
|
.send(StandardMessage {
|
||||||
let data: &Value = &json["data"];
|
id: self.id,
|
||||||
|
msg: m
|
||||||
let user_operation: UserOperation = UserOperation::from_str(op).unwrap();
|
})
|
||||||
|
.into_actor(self)
|
||||||
match user_operation {
|
.then(|res, _, ctx| {
|
||||||
UserOperation::Login => {
|
match res {
|
||||||
let login: Login = serde_json::from_str(&data.to_string()).unwrap();
|
Ok(res) => ctx.text(res),
|
||||||
ctx.state()
|
Err(e) => {
|
||||||
.addr
|
eprintln!("{}", &e);
|
||||||
.send(login)
|
// ctx.text(e);
|
||||||
.into_actor(self)
|
}
|
||||||
.then(|res, _, ctx| {
|
}
|
||||||
match res {
|
// Ok(res) => ctx.text(res),
|
||||||
Ok(response) => match response {
|
// // something is wrong with chat server
|
||||||
Ok(t) => ctx.text(serde_json::to_string(&t).unwrap()),
|
// _ => ctx.stop(),
|
||||||
Err(e) => {
|
fut::ok(())
|
||||||
let error_message_str: String = serde_json::to_string(&e).unwrap();
|
})
|
||||||
eprintln!("{}", &error_message_str);
|
.wait(ctx);
|
||||||
ctx.text(&error_message_str);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
_ => println!("Something is wrong"),
|
|
||||||
}
|
|
||||||
fut::ok(())
|
|
||||||
})
|
|
||||||
.wait(ctx)
|
|
||||||
},
|
|
||||||
UserOperation::Register => {
|
|
||||||
let register: Register = serde_json::from_str(&data.to_string()).unwrap();
|
|
||||||
ctx.state()
|
|
||||||
.addr
|
|
||||||
.send(register)
|
|
||||||
.into_actor(self)
|
|
||||||
.then(|res, _, ctx| {
|
|
||||||
match res {
|
|
||||||
Ok(response) => match response {
|
|
||||||
Ok(t) => ctx.text(serde_json::to_string(&t).unwrap()),
|
|
||||||
Err(e) => {
|
|
||||||
let error_message_str: String = serde_json::to_string(&e).unwrap();
|
|
||||||
eprintln!("{}", &error_message_str);
|
|
||||||
ctx.text(&error_message_str);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
_ => println!("Something is wrong"),
|
|
||||||
}
|
|
||||||
fut::ok(())
|
|
||||||
})
|
|
||||||
.wait(ctx)
|
|
||||||
},
|
|
||||||
UserOperation::CreateCommunity => {
|
|
||||||
use server::actions::community::CommunityForm;
|
|
||||||
let auth: &str = &json["auth"].as_str().unwrap();
|
|
||||||
let community_form: CommunityForm = serde_json::from_str(&data.to_string()).unwrap();
|
|
||||||
ctx.state()
|
|
||||||
.addr
|
|
||||||
.send(community_form)
|
|
||||||
.into_actor(self)
|
|
||||||
.then(|res, _, ctx| {
|
|
||||||
match res {
|
|
||||||
Ok(response) => match response {
|
|
||||||
Ok(t) => ctx.text(serde_json::to_string(&t).unwrap()),
|
|
||||||
Err(e) => {
|
|
||||||
let error_message_str: String = serde_json::to_string(&e).unwrap();
|
|
||||||
eprintln!("{}", &error_message_str);
|
|
||||||
ctx.text(&error_message_str);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
_ => println!("Something is wrong"),
|
|
||||||
}
|
|
||||||
fut::ok(())
|
|
||||||
})
|
|
||||||
.wait(ctx)
|
|
||||||
},
|
|
||||||
_ => ctx.text(format!("!!! unknown command: {:?}", m)),
|
|
||||||
}
|
|
||||||
|
|
||||||
// we check for /sss type of messages
|
// we check for /sss type of messages
|
||||||
// if m.starts_with('/') {
|
// if m.starts_with('/') {
|
||||||
|
|
|
@ -6,10 +6,13 @@ use actix::prelude::*;
|
||||||
use rand::{rngs::ThreadRng, Rng};
|
use rand::{rngs::ThreadRng, Rng};
|
||||||
use std::collections::{HashMap, HashSet};
|
use std::collections::{HashMap, HashSet};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serde_json::{Result, Value};
|
||||||
use bcrypt::{verify};
|
use bcrypt::{verify};
|
||||||
|
use std::str::FromStr;
|
||||||
|
|
||||||
use {Crud,establish_connection};
|
use {Crud, Joinable, establish_connection};
|
||||||
use actions::community::*;
|
use actions::community::*;
|
||||||
|
use actions::user::*;
|
||||||
|
|
||||||
#[derive(EnumString,ToString,Debug)]
|
#[derive(EnumString,ToString,Debug)]
|
||||||
pub enum UserOperation {
|
pub enum UserOperation {
|
||||||
|
@ -58,6 +61,24 @@ pub struct ClientMessage {
|
||||||
pub room: String,
|
pub room: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize)]
|
||||||
|
pub struct StandardMessage {
|
||||||
|
/// Id of the client session
|
||||||
|
pub id: usize,
|
||||||
|
/// Peer message
|
||||||
|
pub msg: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl actix::Message for StandardMessage {
|
||||||
|
type Result = String;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize)]
|
||||||
|
pub struct StandardResponse<T> {
|
||||||
|
op: String,
|
||||||
|
response: T
|
||||||
|
}
|
||||||
|
|
||||||
/// List of available rooms
|
/// List of available rooms
|
||||||
pub struct ListRooms;
|
pub struct ListRooms;
|
||||||
|
|
||||||
|
@ -80,10 +101,6 @@ pub struct Login {
|
||||||
pub password: String
|
pub password: String
|
||||||
}
|
}
|
||||||
|
|
||||||
impl actix::Message for Login {
|
|
||||||
type Result = Result<LoginResponse, ErrorMessage>;
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
pub struct Register {
|
pub struct Register {
|
||||||
username: String,
|
username: String,
|
||||||
|
@ -98,23 +115,15 @@ pub struct LoginResponse {
|
||||||
jwt: String
|
jwt: String
|
||||||
}
|
}
|
||||||
|
|
||||||
impl actix::Message for Register {
|
#[derive(Serialize, Deserialize)]
|
||||||
type Result = Result<LoginResponse, ErrorMessage>;
|
pub struct CreateCommunity {
|
||||||
|
name: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
// #[derive(Serialize, Deserialize)]
|
|
||||||
// pub struct CreateCommunity {
|
|
||||||
// name: String
|
|
||||||
// }
|
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
pub struct CreateCommunityResponse {
|
pub struct CreateCommunityResponse {
|
||||||
op: String,
|
op: String,
|
||||||
community: Community
|
data: Community
|
||||||
}
|
|
||||||
|
|
||||||
impl actix::Message for CommunityForm {
|
|
||||||
type Result = Result<CreateCommunityResponse, ErrorMessage>;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// `ChatServer` manages chat rooms and responsible for coordinating chat
|
/// `ChatServer` manages chat rooms and responsible for coordinating chat
|
||||||
|
@ -152,6 +161,16 @@ impl ChatServer {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Send message only to self
|
||||||
|
fn send(&self, message: &str, id: &usize) {
|
||||||
|
// println!("{:?}", self.sessions);
|
||||||
|
if let Some(addr) = self.sessions.get(id) {
|
||||||
|
println!("msg: {}", message);
|
||||||
|
// println!("{:?}", addr.connected());
|
||||||
|
let _ = addr.do_send(WSMessage(message.to_owned()));
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Make actor from `ChatServer`
|
/// Make actor from `ChatServer`
|
||||||
|
@ -219,121 +238,176 @@ impl Handler<ClientMessage> for ChatServer {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Handler for `ListRooms` message.
|
/// Handler for Message message.
|
||||||
impl Handler<ListRooms> for ChatServer {
|
impl Handler<StandardMessage> for ChatServer {
|
||||||
type Result = MessageResult<ListRooms>;
|
type Result = MessageResult<StandardMessage>;
|
||||||
|
|
||||||
fn handle(&mut self, _: ListRooms, _: &mut Context<Self>) -> Self::Result {
|
fn handle(&mut self, msg: StandardMessage, _: &mut Context<Self>) -> Self::Result {
|
||||||
let mut rooms = Vec::new();
|
|
||||||
|
|
||||||
for key in self.rooms.keys() {
|
let json: Value = serde_json::from_str(&msg.msg)
|
||||||
rooms.push(key.to_owned())
|
.expect("Couldn't parse message");
|
||||||
}
|
|
||||||
|
|
||||||
MessageResult(rooms)
|
let data: &Value = &json["data"];
|
||||||
}
|
let op = &json["op"].as_str().unwrap();
|
||||||
}
|
let auth = &json["auth"].as_str();
|
||||||
|
let user_operation: UserOperation = UserOperation::from_str(&op).unwrap();
|
||||||
|
|
||||||
/// Join room, send disconnect message to old room
|
let res: String = match user_operation {
|
||||||
/// send join message to new room
|
UserOperation::Login => {
|
||||||
impl Handler<Join> for ChatServer {
|
let login: Login = serde_json::from_str(&data.to_string()).unwrap();
|
||||||
type Result = ();
|
login.perform()
|
||||||
|
},
|
||||||
fn handle(&mut self, msg: Join, _: &mut Context<Self>) {
|
UserOperation::Register => {
|
||||||
let Join { id, name } = msg;
|
let register: Register = serde_json::from_str(&data.to_string()).unwrap();
|
||||||
let mut rooms = Vec::new();
|
register.perform()
|
||||||
|
},
|
||||||
// remove session from all rooms
|
UserOperation::CreateCommunity => {
|
||||||
for (n, sessions) in &mut self.rooms {
|
let create_community: CreateCommunity = serde_json::from_str(&data.to_string()).unwrap();
|
||||||
if sessions.remove(&id) {
|
match auth {
|
||||||
rooms.push(n.to_owned());
|
Some(auth) => {
|
||||||
|
create_community.perform(auth)
|
||||||
|
},
|
||||||
|
None => serde_json::to_string(
|
||||||
|
&ErrorMessage {
|
||||||
|
op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
error: "Not logged in.".to_string()
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {
|
||||||
|
let e = ErrorMessage {
|
||||||
|
op: "Unknown".to_string(),
|
||||||
|
error: "Unknown User Operation".to_string()
|
||||||
|
};
|
||||||
|
serde_json::to_string(&e).unwrap()
|
||||||
}
|
}
|
||||||
}
|
// _ => "no".to_string()
|
||||||
// send message to other users
|
};
|
||||||
for room in rooms {
|
|
||||||
self.send_room_message(&room, "Someone disconnected", 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
if self.rooms.get_mut(&name).is_none() {
|
|
||||||
self.rooms.insert(name.clone(), HashSet::new());
|
|
||||||
}
|
// let data: &Value = &json["data"];
|
||||||
self.send_room_message(&name, "Someone connected", id);
|
// let res = StandardResponse {op: "nope".to_string(), response: "hi".to_string()};
|
||||||
self.rooms.get_mut(&name).unwrap().insert(id);
|
// let out = serde_json::to_string(&res).unwrap();
|
||||||
|
MessageResult(res)
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Handler<Login> for ChatServer {
|
// /// Handler for `ListRooms` message.
|
||||||
|
// impl Handler<ListRooms> for ChatServer {
|
||||||
|
// type Result = MessageResult<ListRooms>;
|
||||||
|
|
||||||
type Result = MessageResult<Login>;
|
// fn handle(&mut self, _: ListRooms, _: &mut Context<Self>) -> Self::Result {
|
||||||
fn handle(&mut self, msg: Login, _: &mut Context<Self>) -> Self::Result {
|
// let mut rooms = Vec::new();
|
||||||
|
|
||||||
|
// for key in self.rooms.keys() {
|
||||||
|
// rooms.push(key.to_owned())
|
||||||
|
// }
|
||||||
|
|
||||||
|
// MessageResult(rooms)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// /// Join room, send disconnect message to old room
|
||||||
|
// /// send join message to new room
|
||||||
|
// impl Handler<Join> for ChatServer {
|
||||||
|
// type Result = ();
|
||||||
|
|
||||||
|
// fn handle(&mut self, msg: Join, _: &mut Context<Self>) {
|
||||||
|
// let Join { id, name } = msg;
|
||||||
|
// let mut rooms = Vec::new();
|
||||||
|
|
||||||
|
// // remove session from all rooms
|
||||||
|
// for (n, sessions) in &mut self.rooms {
|
||||||
|
// if sessions.remove(&id) {
|
||||||
|
// rooms.push(n.to_owned());
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// // send message to other users
|
||||||
|
// for room in rooms {
|
||||||
|
// self.send_room_message(&room, "Someone disconnected", 0);
|
||||||
|
// }
|
||||||
|
|
||||||
|
// if self.rooms.get_mut(&name).is_none() {
|
||||||
|
// self.rooms.insert(name.clone(), HashSet::new());
|
||||||
|
// }
|
||||||
|
// self.send_room_message(&name, "Someone connected", id);
|
||||||
|
// self.rooms.get_mut(&name).unwrap().insert(id);
|
||||||
|
// }
|
||||||
|
|
||||||
|
// }
|
||||||
|
|
||||||
|
pub trait Perform {
|
||||||
|
fn perform(&self) -> String;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait PerformAuth {
|
||||||
|
fn perform(&self, auth: &str) -> String;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Perform for Login {
|
||||||
|
fn perform(&self) -> String {
|
||||||
|
|
||||||
use actions::user::*;
|
|
||||||
let conn = establish_connection();
|
let conn = establish_connection();
|
||||||
|
|
||||||
// Fetch that username / email
|
// Fetch that username / email
|
||||||
let user: User_ = match User_::find_by_email_or_username(&conn, &msg.username_or_email) {
|
let user: User_ = match User_::find_by_email_or_username(&conn, &self.username_or_email) {
|
||||||
Ok(user) => user,
|
Ok(user) => user,
|
||||||
Err(e) => return MessageResult(
|
Err(e) => return serde_json::to_string(
|
||||||
Err(
|
&ErrorMessage {
|
||||||
ErrorMessage {
|
op: UserOperation::Login.to_string(),
|
||||||
op: UserOperation::Login.to_string(),
|
error: "Couldn't find that username or email".to_string()
|
||||||
error: "Couldn't find that username or email".to_string()
|
}
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
.unwrap()
|
||||||
};
|
};
|
||||||
|
|
||||||
// Verify the password
|
// Verify the password
|
||||||
let valid: bool = verify(&msg.password, &user.password_encrypted).unwrap_or(false);
|
let valid: bool = verify(&self.password, &user.password_encrypted).unwrap_or(false);
|
||||||
if !valid {
|
if !valid {
|
||||||
return MessageResult(
|
return serde_json::to_string(
|
||||||
Err(
|
&ErrorMessage {
|
||||||
ErrorMessage {
|
op: UserOperation::Login.to_string(),
|
||||||
op: UserOperation::Login.to_string(),
|
error: "Password incorrect".to_string()
|
||||||
error: "Password incorrect".to_string()
|
}
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return the jwt
|
// Return the jwt
|
||||||
MessageResult(
|
serde_json::to_string(
|
||||||
Ok(
|
&LoginResponse {
|
||||||
LoginResponse {
|
op: UserOperation::Login.to_string(),
|
||||||
op: UserOperation::Login.to_string(),
|
jwt: user.jwt()
|
||||||
jwt: user.jwt()
|
}
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
.unwrap()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Handler<Register> for ChatServer {
|
impl Perform for Register {
|
||||||
|
fn perform(&self) -> String {
|
||||||
|
|
||||||
type Result = MessageResult<Register>;
|
|
||||||
fn handle(&mut self, msg: Register, _: &mut Context<Self>) -> Self::Result {
|
|
||||||
|
|
||||||
use actions::user::*;
|
|
||||||
let conn = establish_connection();
|
let conn = establish_connection();
|
||||||
|
|
||||||
// Make sure passwords match
|
// Make sure passwords match
|
||||||
if msg.password != msg.password_verify {
|
if &self.password != &self.password_verify {
|
||||||
return MessageResult(
|
return serde_json::to_string(
|
||||||
Err(
|
&ErrorMessage {
|
||||||
ErrorMessage {
|
op: UserOperation::Register.to_string(),
|
||||||
op: UserOperation::Register.to_string(),
|
error: "Passwords do not match.".to_string()
|
||||||
error: "Passwords do not match.".to_string()
|
}
|
||||||
}
|
)
|
||||||
)
|
.unwrap();
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Register the new user
|
// Register the new user
|
||||||
let user_form = UserForm {
|
let user_form = UserForm {
|
||||||
name: msg.username,
|
name: self.username.to_owned(),
|
||||||
email: msg.email,
|
email: self.email.to_owned(),
|
||||||
password_encrypted: msg.password,
|
password_encrypted: self.password.to_owned(),
|
||||||
preferred_username: None,
|
preferred_username: None,
|
||||||
updated: None
|
updated: None
|
||||||
};
|
};
|
||||||
|
@ -341,55 +415,232 @@ impl Handler<Register> for ChatServer {
|
||||||
// Create the user
|
// Create the user
|
||||||
let inserted_user = match User_::create(&conn, &user_form) {
|
let inserted_user = match User_::create(&conn, &user_form) {
|
||||||
Ok(user) => user,
|
Ok(user) => user,
|
||||||
Err(e) => return MessageResult(
|
Err(e) => {
|
||||||
Err(
|
return serde_json::to_string(
|
||||||
ErrorMessage {
|
&ErrorMessage {
|
||||||
op: UserOperation::Register.to_string(),
|
op: UserOperation::Register.to_string(),
|
||||||
error: "User already exists.".to_string() // overwrite the diesel error
|
error: "User already exists.".to_string() // overwrite the diesel error
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
)
|
.unwrap()
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Return the jwt
|
// Return the jwt
|
||||||
MessageResult(
|
serde_json::to_string(
|
||||||
Ok(
|
&LoginResponse {
|
||||||
LoginResponse {
|
op: UserOperation::Register.to_string(),
|
||||||
op: UserOperation::Register.to_string(),
|
jwt: inserted_user.jwt()
|
||||||
jwt: inserted_user.jwt()
|
}
|
||||||
}
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
.unwrap()
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl PerformAuth for CreateCommunity {
|
||||||
|
fn perform(&self, auth: &str) -> String {
|
||||||
|
|
||||||
impl Handler<CommunityForm> for ChatServer {
|
|
||||||
|
|
||||||
type Result = MessageResult<CommunityForm>;
|
|
||||||
|
|
||||||
fn handle(&mut self, form: CommunityForm, _: &mut Context<Self>) -> Self::Result {
|
|
||||||
let conn = establish_connection();
|
let conn = establish_connection();
|
||||||
let community = match Community::create(&conn, &form) {
|
|
||||||
|
let claims = match Claims::decode(&auth) {
|
||||||
|
Ok(claims) => claims.claims,
|
||||||
|
Err(e) => {
|
||||||
|
return serde_json::to_string(
|
||||||
|
&ErrorMessage {
|
||||||
|
op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
error: "Community user already exists.".to_string() // overwrite the diesel error
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let user_id = claims.id;
|
||||||
|
let iss = claims.iss;
|
||||||
|
|
||||||
|
// Register the new user
|
||||||
|
let community_form = CommunityForm {
|
||||||
|
name: self.name.to_owned(),
|
||||||
|
updated: None
|
||||||
|
};
|
||||||
|
|
||||||
|
let inserted_community = match Community::create(&conn, &community_form) {
|
||||||
Ok(community) => community,
|
Ok(community) => community,
|
||||||
Err(e) => return MessageResult(
|
Err(e) => {
|
||||||
Err(
|
return serde_json::to_string(
|
||||||
ErrorMessage {
|
&ErrorMessage {
|
||||||
op: UserOperation::CreateCommunity.to_string(),
|
op: UserOperation::CreateCommunity.to_string(),
|
||||||
error: "Community already exists.".to_string() // overwrite the diesel error
|
error: "Community already exists.".to_string() // overwrite the diesel error
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
)
|
.unwrap()
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
MessageResult(
|
let community_user_form = CommunityUserForm {
|
||||||
Ok(
|
community_id: inserted_community.id,
|
||||||
CreateCommunityResponse {
|
fedi_user_id: format!("{}/{}", iss, user_id)
|
||||||
op: UserOperation::CreateCommunity.to_string(),
|
};
|
||||||
community: community
|
|
||||||
}
|
let inserted_community_user = match CommunityUser::join(&conn, &community_user_form) {
|
||||||
)
|
Ok(user) => user,
|
||||||
|
Err(e) => {
|
||||||
|
return serde_json::to_string(
|
||||||
|
&ErrorMessage {
|
||||||
|
op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
error: "Community user already exists.".to_string() // overwrite the diesel error
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
// Return the jwt
|
||||||
|
serde_json::to_string(
|
||||||
|
&CreateCommunityResponse {
|
||||||
|
op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
data: inserted_community
|
||||||
|
}
|
||||||
)
|
)
|
||||||
|
.unwrap()
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// impl Handler<Login> for ChatServer {
|
||||||
|
|
||||||
|
// type Result = MessageResult<Login>;
|
||||||
|
// fn handle(&mut self, msg: Login, _: &mut Context<Self>) -> Self::Result {
|
||||||
|
|
||||||
|
// let conn = establish_connection();
|
||||||
|
|
||||||
|
// // Fetch that username / email
|
||||||
|
// let user: User_ = match User_::find_by_email_or_username(&conn, &msg.username_or_email) {
|
||||||
|
// Ok(user) => user,
|
||||||
|
// Err(e) => return MessageResult(
|
||||||
|
// Err(
|
||||||
|
// ErrorMessage {
|
||||||
|
// op: UserOperation::Login.to_string(),
|
||||||
|
// error: "Couldn't find that username or email".to_string()
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// };
|
||||||
|
|
||||||
|
// // Verify the password
|
||||||
|
// let valid: bool = verify(&msg.password, &user.password_encrypted).unwrap_or(false);
|
||||||
|
// if !valid {
|
||||||
|
// return MessageResult(
|
||||||
|
// Err(
|
||||||
|
// ErrorMessage {
|
||||||
|
// op: UserOperation::Login.to_string(),
|
||||||
|
// error: "Password incorrect".to_string()
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// }
|
||||||
|
|
||||||
|
// // Return the jwt
|
||||||
|
// MessageResult(
|
||||||
|
// Ok(
|
||||||
|
// LoginResponse {
|
||||||
|
// op: UserOperation::Login.to_string(),
|
||||||
|
// jwt: user.jwt()
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// impl Handler<Register> for ChatServer {
|
||||||
|
|
||||||
|
// type Result = MessageResult<Register>;
|
||||||
|
// fn handle(&mut self, msg: Register, _: &mut Context<Self>) -> Self::Result {
|
||||||
|
|
||||||
|
// let conn = establish_connection();
|
||||||
|
|
||||||
|
// // Make sure passwords match
|
||||||
|
// if msg.password != msg.password_verify {
|
||||||
|
// return MessageResult(
|
||||||
|
// Err(
|
||||||
|
// ErrorMessage {
|
||||||
|
// op: UserOperation::Register.to_string(),
|
||||||
|
// error: "Passwords do not match.".to_string()
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// );
|
||||||
|
// }
|
||||||
|
|
||||||
|
// // Register the new user
|
||||||
|
// let user_form = UserForm {
|
||||||
|
// name: msg.username,
|
||||||
|
// email: msg.email,
|
||||||
|
// password_encrypted: msg.password,
|
||||||
|
// preferred_username: None,
|
||||||
|
// updated: None
|
||||||
|
// };
|
||||||
|
|
||||||
|
// // Create the user
|
||||||
|
// let inserted_user = match User_::create(&conn, &user_form) {
|
||||||
|
// Ok(user) => user,
|
||||||
|
// Err(e) => return MessageResult(
|
||||||
|
// Err(
|
||||||
|
// ErrorMessage {
|
||||||
|
// op: UserOperation::Register.to_string(),
|
||||||
|
// error: "User already exists.".to_string() // overwrite the diesel error
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// };
|
||||||
|
|
||||||
|
// // Return the jwt
|
||||||
|
// MessageResult(
|
||||||
|
// Ok(
|
||||||
|
// LoginResponse {
|
||||||
|
// op: UserOperation::Register.to_string(),
|
||||||
|
// jwt: inserted_user.jwt()
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
|
||||||
|
// impl Handler<CreateCommunity> for ChatServer {
|
||||||
|
|
||||||
|
// type Result = MessageResult<CreateCommunity>;
|
||||||
|
|
||||||
|
// fn handle(&mut self, msg: CreateCommunity, _: &mut Context<Self>) -> Self::Result {
|
||||||
|
// let conn = establish_connection();
|
||||||
|
|
||||||
|
// let user_id = Claims::decode(&msg.auth).id;
|
||||||
|
|
||||||
|
// let community_form = CommunityForm {
|
||||||
|
// name: msg.name,
|
||||||
|
// updated: None
|
||||||
|
// };
|
||||||
|
|
||||||
|
// let community = match Community::create(&conn, &community_form) {
|
||||||
|
// Ok(community) => community,
|
||||||
|
// Err(e) => return MessageResult(
|
||||||
|
// Err(
|
||||||
|
// ErrorMessage {
|
||||||
|
// op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
// error: "Community already exists.".to_string() // overwrite the diesel error
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// };
|
||||||
|
|
||||||
|
// MessageResult(
|
||||||
|
// Ok(
|
||||||
|
// CreateCommunityResponse {
|
||||||
|
// op: UserOperation::CreateCommunity.to_string(),
|
||||||
|
// community: community
|
||||||
|
// }
|
||||||
|
// )
|
||||||
|
// )
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
|
@ -5,6 +5,8 @@ import { CommunityForm, UserOperation } from '../interfaces';
|
||||||
import { WebSocketService, UserService } from '../services';
|
import { WebSocketService, UserService } from '../services';
|
||||||
import { msgOp } from '../utils';
|
import { msgOp } from '../utils';
|
||||||
|
|
||||||
|
import { Community } from '../interfaces';
|
||||||
|
|
||||||
interface State {
|
interface State {
|
||||||
communityForm: CommunityForm;
|
communityForm: CommunityForm;
|
||||||
}
|
}
|
||||||
|
@ -28,6 +30,7 @@ export class CreateCommunity extends Component<any, State> {
|
||||||
.subscribe(
|
.subscribe(
|
||||||
(msg) => this.parseMessage(msg),
|
(msg) => this.parseMessage(msg),
|
||||||
(err) => console.error(err),
|
(err) => console.error(err),
|
||||||
|
() => console.log("complete")
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -80,10 +83,14 @@ export class CreateCommunity extends Component<any, State> {
|
||||||
|
|
||||||
parseMessage(msg: any) {
|
parseMessage(msg: any) {
|
||||||
let op: UserOperation = msgOp(msg);
|
let op: UserOperation = msgOp(msg);
|
||||||
|
console.log(msg);
|
||||||
if (msg.error) {
|
if (msg.error) {
|
||||||
alert(msg.error);
|
alert(msg.error);
|
||||||
return;
|
return;
|
||||||
} else {
|
} else {
|
||||||
|
if (op == UserOperation.CreateCommunity) {
|
||||||
|
let community: Community = msg.data;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -35,6 +35,7 @@ export class Login extends Component<any, State> {
|
||||||
.subscribe(
|
.subscribe(
|
||||||
(msg) => this.parseMessage(msg),
|
(msg) => this.parseMessage(msg),
|
||||||
(err) => console.error(err),
|
(err) => console.error(err),
|
||||||
|
() => console.log("complete")
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -3,10 +3,17 @@ export enum UserOperation {
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface User {
|
export interface User {
|
||||||
id: number
|
id: number;
|
||||||
username: string;
|
username: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export interface Community {
|
||||||
|
id: number;
|
||||||
|
name: string;
|
||||||
|
published: Date;
|
||||||
|
updated?: Date;
|
||||||
|
}
|
||||||
|
|
||||||
export interface LoginForm {
|
export interface LoginForm {
|
||||||
username_or_email: string;
|
username_or_email: string;
|
||||||
password: string;
|
password: string;
|
||||||
|
@ -21,7 +28,6 @@ export interface RegisterForm {
|
||||||
|
|
||||||
export interface CommunityForm {
|
export interface CommunityForm {
|
||||||
name: string;
|
name: string;
|
||||||
updated?: number
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface PostForm {
|
export interface PostForm {
|
||||||
|
|
|
@ -19,9 +19,9 @@ export class UserService {
|
||||||
}
|
}
|
||||||
|
|
||||||
public login(jwt: string) {
|
public login(jwt: string) {
|
||||||
|
this.setUser(jwt);
|
||||||
Cookies.set("jwt", jwt);
|
Cookies.set("jwt", jwt);
|
||||||
console.log("jwt cookie set");
|
console.log("jwt cookie set");
|
||||||
this.setUser(jwt);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public logout() {
|
public logout() {
|
||||||
|
@ -42,7 +42,6 @@ export class UserService {
|
||||||
private setUser(jwt: string) {
|
private setUser(jwt: string) {
|
||||||
this.user = jwt_decode(jwt);
|
this.user = jwt_decode(jwt);
|
||||||
this.sub.next(this.user);
|
this.sub.next(this.user);
|
||||||
console.log(this.user.username);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public static get Instance(){
|
public static get Instance(){
|
||||||
|
|
Loading…
Reference in a new issue