Misc changes.

src/main.rs

@@ -10,6 +10,7 @@ pub mod email;
 
 use dotenv;
 use std::thread;
+use rocket_cors::AllowedOrigins;
 
 fn main() {
 	dotenv::dotenv().ok();
@@ -19,5 +20,12 @@ fn main() {
 		websocket::launch_server();
 	});
 
-	routes::mount(rocket::ignite()).launch();
+    let cors = rocket_cors::CorsOptions {
+        allowed_origins: AllowedOrigins::All,
+        ..Default::default()
+    }.to_cors().unwrap();
+
+	routes::mount(rocket::ignite())
+		.attach(cors)
+		.launch();
 }

src/routes/account.rs

@@ -267,7 +267,8 @@ pub fn login(info: Json<Login>) -> JsonValue {
 
 						json!({
 							"success": true,
-							"access_token": token
+							"access_token": token,
+							"id": user.id
 						})
 					},
 					false => json!({
@@ -282,3 +283,28 @@ pub fn login(info: Json<Login>) -> JsonValue {
 			})
 		}
 }
+
+
+#[derive(Serialize, Deserialize)]
+pub struct Token {
+	token: String,
+}
+
+/// login to a Revolt account via token
+#[post("/token", data = "<info>")]
+pub fn token(info: Json<Token>) -> JsonValue {
+	let col = database::get_collection("users");
+
+	if let Some(u) =
+		col.find_one(doc! { "access_token": info.token.clone() }, None).expect("Failed user lookup") {
+			json!({
+				"success": true,
+				"id": u.get_str("_id").unwrap(),
+			})
+		} else {
+			json!({
+				"success": false,
+				"error": "Invalid token!",
+			})
+		}
+}

src/routes/channel.rs

@@ -53,7 +53,8 @@ pub fn channel(user: User, target: Channel) -> Option<JsonValue> {
 	Some(
 		json!({
 			"id": target.id,
-			"type": target.channel_type
+			"type": target.channel_type,
+			"recipients": get_recipients(&target),
 		}
 	))
 }
@@ -158,10 +159,12 @@ pub fn send_message(user: User, target: Channel, message: Json<SendMessage>) ->
 				get_recipients(&target),
 				json!({
 					"type": "message",
-					"id": id.clone(),
-					"channel": target.id,
-					"author": user.id,
-					"content": message.content.clone(),
+					"data": {
+						"id": id.clone(),
+						"channel": target.id,
+						"author": user.id,
+						"content": message.content.clone(),
+					},
 				}).to_string()
 			);
 
@@ -178,6 +181,23 @@ pub fn send_message(user: User, target: Channel, message: Json<SendMessage>) ->
 		})
 }
 
+/// get a message
+#[get("/<target>/messages/<message>")]
+pub fn get_message(user: User, target: Channel, message: Message) -> Option<JsonValue> {
+	if !has_permission(&user, &target) {
+		return None
+	}
+
+	Some(
+		json!({
+			"id": message.id,
+			"author": message.author,
+			"content": message.content,
+			"edited": if let Some(t) = message.edited { Some(t.timestamp()) } else { None }
+		})
+	)
+}
+
 #[derive(Serialize, Deserialize)]
 pub struct EditMessage {
 	content: String,
@@ -215,10 +235,12 @@ pub fn edit_message(user: User, target: Channel, message: Message, edit: Json<Se
 						get_recipients(&target),
 						json!({
 							"type": "message_update",
-							"id": message.id,
-							"channel": target.id,
-							"content": message.content.clone(),
-							"edited": edited.timestamp()
+							"data": {
+								"id": message.id,
+								"channel": target.id,
+								"content": edit.content.clone(),
+								"edited": edited.timestamp()
+							},
 						}).to_string()
 					);
 
@@ -261,8 +283,10 @@ pub fn delete_message(user: User, target: Channel, message: Message) -> Option<J
 						get_recipients(&target),
 						json!({
 							"type": "message_delete",
-							"id": message.id,
-							"channel": target.id
+							"data": {
+								"id": message.id,
+								"channel": target.id
+							},
 						}).to_string()
 					);
 

src/routes/mod.rs

@@ -6,7 +6,7 @@ pub mod channel;
 
 pub fn mount(rocket: Rocket) -> Rocket {
 	rocket
-		.mount("/api/account", routes![ account::create, account::verify_email, account::resend_email, account::login ])
+		.mount("/api/account", routes![ account::create, account::verify_email, account::resend_email, account::login, account::token ])
 		.mount("/api/users", routes![ user::me, user::user, user::lookup, user::dms, user::dm, user::get_friends, user::get_friend, user::add_friend, user::remove_friend ])
 		.mount("/api/channels", routes![ channel::channel, channel::delete, channel::messages, channel::send_message, channel::edit_message, channel::delete_message ])
 }

src/routes/user.rs

@@ -15,14 +15,17 @@ pub fn me(user: User) -> JsonValue {
 		"username": user.username,
 		"email": user.email,
 		"verified": user.email_verification.verified,
-		"created_timestamp": Ulid::from_string(&user.id).unwrap().datetime().timestamp(),
 	})
 }
 
 /// retrieve another user's information
-#[get("/<_target>")]
-pub fn user(_user: User, _target: User) -> JsonValue {
-	json!([])
+#[get("/<target>")]
+pub fn user(user: User, target: User) -> JsonValue {
+	json!({
+		"id": target.id,
+		"username": target.username,
+		"relationship": get_relationship(&user, &target) as u8
+	})
 }
 
 #[derive(Serialize, Deserialize)]
@@ -33,7 +36,7 @@ pub struct Query {
 /// lookup a user on Revolt
 /// currently only supports exact username searches
 #[post("/lookup", data = "<query>")]
-pub fn lookup(_user: User, query: Json<Query>) -> JsonValue {
+pub fn lookup(user: User, query: Json<Query>) -> JsonValue {
 	let col = database::get_collection("users");
 
 	let users = col.find(
@@ -42,12 +45,13 @@ pub fn lookup(_user: User, query: Json<Query>) -> JsonValue {
 	).expect("Failed user lookup");
 
 	let mut results = Vec::new();
-	for user in users {
-		let u: User = from_bson(bson::Bson::Document(user.unwrap())).expect("Failed to unwrap user.");
+	for item in users {
+		let u: User = from_bson(bson::Bson::Document(item.unwrap())).expect("Failed to unwrap user.");
 		results.push(
 			json!({
 				"id": u.id,
-				"username": u.username
+				"username": u.username,
+				"relationship": get_relationship(&user, &u) as u8
 			})
 		);
 	}
@@ -98,11 +102,12 @@ pub fn dm(user: User, target: User) -> JsonValue {
 	let col = database::get_collection("channels");
 
 	match col.find_one(
-		doc! { "type": channel::ChannelType::DM as i32, "recipients": [ user.id.clone(), target.id.clone() ] },
+		doc! { "type": channel::ChannelType::DM as i32, "recipients": { "$all": [ user.id.clone(), target.id.clone() ] } },
 		None
 	).expect("Failed channel lookup") {
 		Some(channel) =>
 			json!({
+				"success": true,
 				"id": channel.get_str("_id").unwrap()
 			}),
 		None => {
@@ -244,7 +249,8 @@ pub fn add_friend(user: User, target: User) -> JsonValue {
 			).expect("Failed update query.");
 
 			json!({
-				"success": true
+				"success": true,
+				"status": Relationship::FRIEND as u8,
 			})
 		},
 		Relationship::BLOCKED =>
@@ -289,7 +295,8 @@ pub fn add_friend(user: User, target: User) -> JsonValue {
 			).expect("Failed update query.");
 
 			json!({
-				"success": true
+				"success": true,
+				"status": Relationship::OUTGOING as u8,
 			})
 		},
 		Relationship::SELF =>

src/websocket/mod.rs

@@ -40,6 +40,8 @@ impl Handler for Server {
 						if let Some(_) = self.id {
 							self.out.send(
 								json!({
+									"type": "authenticate",
+									"success": false,
 									"error": "Already authenticated!"
 								})
 								.to_string()
@@ -71,6 +73,7 @@ impl Handler for Server {
 									self.id = Some(id.to_string());
 									self.out.send(
 										json!({
+											"type": "authenticate",
 											"success": true
 										})
 										.to_string()
@@ -79,6 +82,8 @@ impl Handler for Server {
 								None =>
 									self.out.send(
 										json!({
+											"type": "authenticate",
+											"success": false,
 											"error": "Invalid authentication token."
 										})
 										.to_string()
@@ -87,6 +92,8 @@ impl Handler for Server {
 						} else {
 							self.out.send(
 								json!({
+									"type": "authenticate",
+									"success": false,
 									"error": "Missing authentication token."
 								})
 								.to_string()
@@ -141,7 +148,7 @@ pub fn launch_server() {
 		}
 	}
 
-	listen("127.0.0.1:3012", |out| { Server { out: out, id: None, internal: Ulid::new().to_string() } }).unwrap()
+	listen("192.168.0.10:9999", |out| { Server { out: out, id: None, internal: Ulid::new().to_string() } }).unwrap()
 }
 
 pub fn send_message(id: String, message: String) -> std::result::Result<(), ()> {