2021-03-29 08:44:26 +00:00
|
|
|
use super::response::User;
|
2021-04-20 19:23:04 +00:00
|
|
|
use super::response::{AppStatus, ResponseType};
|
2021-04-12 14:05:11 +00:00
|
|
|
use crate::api::request::Request;
|
2021-04-20 19:23:04 +00:00
|
|
|
use crate::chat::init_config;
|
2021-03-29 08:44:26 +00:00
|
|
|
use crate::chat::types::PackedMessage;
|
2021-03-26 19:04:39 +00:00
|
|
|
use crate::chat::types::SP;
|
2021-03-31 04:12:26 +00:00
|
|
|
use crate::db::{self, messages, types, users};
|
2021-03-26 19:04:39 +00:00
|
|
|
use async_std::io::Result;
|
2021-04-20 19:23:04 +00:00
|
|
|
use fcpv2::client::fcp_types::{ClientHello, ClientPut};
|
2021-04-01 12:44:13 +00:00
|
|
|
use fcpv2::types::{
|
|
|
|
traits::{FcpParser, FcpRequest},
|
|
|
|
SSK,
|
|
|
|
};
|
2021-03-26 19:04:39 +00:00
|
|
|
use rusqlite::Connection;
|
2021-03-29 08:44:26 +00:00
|
|
|
use serde_json::json;
|
2021-04-20 19:23:04 +00:00
|
|
|
use std::fs::File;
|
|
|
|
use std::path::Path;
|
2021-04-06 10:03:57 +00:00
|
|
|
use std::time::SystemTime;
|
2021-03-31 04:12:26 +00:00
|
|
|
use uuid::Uuid;
|
2021-04-01 12:44:13 +00:00
|
|
|
|
2021-04-05 16:39:36 +00:00
|
|
|
use crate::db::types::Id;
|
|
|
|
|
2021-04-12 14:05:11 +00:00
|
|
|
pub fn start_app(server_sender: SP) -> Result<()> {
|
2021-04-20 19:23:04 +00:00
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToFreenet(
|
|
|
|
ClientHello::new("start_app_request".to_string(), 2.0).convert(),
|
|
|
|
))
|
|
|
|
.unwrap();
|
|
|
|
let config_path = Path::new(".hole.toml");
|
2021-04-21 17:11:57 +00:00
|
|
|
match File::open(&config_path) {
|
|
|
|
Err(e) => {
|
|
|
|
log::debug!("creating new config file...");
|
|
|
|
// std::fs::File::create(&config_path).unwrap();
|
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToFreenet(
|
|
|
|
fcpv2::client::fcp_types::GenerateSSK {
|
|
|
|
identifier: Some("config-SSK".to_string()),
|
|
|
|
}
|
|
|
|
.convert(),
|
|
|
|
))
|
|
|
|
.unwrap()
|
|
|
|
}
|
|
|
|
Ok(res) => {
|
|
|
|
let conf = std::fs::read_to_string(&config_path).unwrap();
|
|
|
|
log::debug!("Responsing to start_app: {}", &conf);
|
2021-04-21 20:07:28 +00:00
|
|
|
let toml: crate::chat::Config = toml::from_str(&conf[..]).unwrap();
|
2021-04-21 17:11:57 +00:00
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToClient(
|
2021-04-21 20:07:28 +00:00
|
|
|
serde_json::to_string(&crate::api::response::ResponseType::InitialConfig {
|
|
|
|
id: toml.id.clone(),
|
|
|
|
public_key: toml.public_key.clone(),
|
|
|
|
private_key: toml.private_key.clone(),
|
|
|
|
})
|
|
|
|
.unwrap(),
|
2021-04-21 17:11:57 +00:00
|
|
|
))
|
|
|
|
.unwrap();
|
|
|
|
log::debug!("Responsing to start_app");
|
|
|
|
} // TODO converting file from TOML to JSON and sending it to frontend
|
2021-04-20 19:23:04 +00:00
|
|
|
};
|
|
|
|
|
2021-03-26 19:04:39 +00:00
|
|
|
Ok(())
|
2021-03-29 08:44:26 +00:00
|
|
|
//sending *JSON*, what everything is OK
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|
|
|
|
|
2021-04-12 14:05:11 +00:00
|
|
|
pub fn stop_app(conn: &Connection, server_sender: SP) -> Result<()> {
|
2021-03-26 20:25:30 +00:00
|
|
|
std::process::exit(0)
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|
|
|
|
|
2021-04-20 19:23:04 +00:00
|
|
|
pub fn load_users(conn: &Connection, server_sender: SP) -> Result<()> {
|
2021-03-29 08:44:26 +00:00
|
|
|
let jsoned_users: Vec<_> = users::load_all_users(conn)
|
|
|
|
.unwrap()
|
|
|
|
.into_iter()
|
|
|
|
.map(|x| x.to_jsonable())
|
|
|
|
.collect();
|
2021-04-21 20:07:28 +00:00
|
|
|
let users: String = serde_json::to_string(&crate::api::response::ResponseType::UserList {
|
2021-03-29 08:44:26 +00:00
|
|
|
users: jsoned_users,
|
2021-03-31 04:12:26 +00:00
|
|
|
})
|
|
|
|
.unwrap();
|
|
|
|
let _ = server_sender.send(PackedMessage::ToClient(users)).unwrap();
|
2021-03-29 08:44:26 +00:00
|
|
|
Ok(())
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|
2021-04-20 19:23:04 +00:00
|
|
|
pub fn send_message(
|
|
|
|
user_id: Id,
|
|
|
|
message: String,
|
|
|
|
conn: &Connection,
|
|
|
|
server_sender: SP,
|
|
|
|
) -> Result<()> {
|
2021-04-22 02:02:20 +00:00
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
|
|
|
log::debug!("CUM CUM");
|
2021-04-12 14:05:11 +00:00
|
|
|
if let Ok(user_data) = db::users::get_user_by_id(user_id, conn) {
|
2021-04-05 16:39:36 +00:00
|
|
|
// Add message to DB
|
2021-04-01 12:44:13 +00:00
|
|
|
let key = user_data.insert_key;
|
|
|
|
let identifier = &user_data.id.0.to_string()[..];
|
2021-04-22 15:28:07 +00:00
|
|
|
let message_id: u32 = user_data.my_messages_count;
|
|
|
|
let id = Id(uuid::Uuid::parse_str(identifier).expect("failed to parse user ID"));
|
|
|
|
let _ = db::users::increase_my_messages_count(id.clone(), conn);
|
2021-04-06 10:03:57 +00:00
|
|
|
let db_message = db::types::Message {
|
2021-04-05 16:39:36 +00:00
|
|
|
id: message_id,
|
|
|
|
date: chrono::offset::Local::now(),
|
2021-04-22 15:28:07 +00:00
|
|
|
user_id: id.clone(),
|
2021-04-12 14:05:11 +00:00
|
|
|
message: message.clone(),
|
2021-04-22 02:02:20 +00:00
|
|
|
from_me: true,
|
2021-04-05 16:39:36 +00:00
|
|
|
};
|
|
|
|
let _ = db::messages::add_my_message(db_message, conn).unwrap();
|
2021-04-22 15:28:07 +00:00
|
|
|
log::debug!("sending new message to freent");
|
2021-04-01 12:44:13 +00:00
|
|
|
let fcp_req: String =
|
2021-04-22 15:28:07 +00:00
|
|
|
ClientPut::new_default_direct(fcpv2::types::USK{ ssk: key, path: format!("{}/{}", &identifier, message_id)}, &format!("{}/{}", &identifier, &message_id )[..], &message[..]).convert();
|
2021-04-01 12:44:13 +00:00
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToFreenet(fcp_req))
|
|
|
|
.unwrap();
|
2021-04-22 15:28:07 +00:00
|
|
|
log::debug!("Increasing messages count");
|
|
|
|
let _ = db::users::increase_my_messages_count(id.clone(), &conn);
|
2021-04-01 12:44:13 +00:00
|
|
|
Ok(())
|
|
|
|
} else {
|
|
|
|
// create error types
|
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToClient(
|
|
|
|
json!(super::response::AppError {
|
|
|
|
res_type: super::response::ErrorType::WrongUserId
|
|
|
|
})
|
|
|
|
.to_string(),
|
|
|
|
))
|
|
|
|
.unwrap();
|
|
|
|
Ok(())
|
|
|
|
}
|
2021-03-29 08:44:26 +00:00
|
|
|
//sending FCP request
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|
|
|
|
|
2021-04-20 19:23:04 +00:00
|
|
|
pub fn load_messages(
|
|
|
|
user_id: Id,
|
|
|
|
start_index: u32,
|
|
|
|
count: u32,
|
|
|
|
conn: &Connection,
|
|
|
|
server_sender: SP,
|
|
|
|
) -> Result<()> {
|
|
|
|
let messages = db::messages::select_n_last_messages(user_id, start_index, count, conn).unwrap();
|
2021-04-01 12:44:13 +00:00
|
|
|
let jsoned = json!(messages);
|
2021-04-22 15:28:07 +00:00
|
|
|
let _ = server_sender.send(PackedMessage::ToClient(jsoned.to_string())).unwrap();
|
2021-04-01 12:44:13 +00:00
|
|
|
Ok(())
|
|
|
|
|
2021-03-29 08:44:26 +00:00
|
|
|
//sending *JSON*
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|
2021-04-01 12:44:13 +00:00
|
|
|
// Adding user to DB
|
2021-04-20 19:23:04 +00:00
|
|
|
pub fn add_user(
|
|
|
|
name: String,
|
|
|
|
insert_key: String,
|
|
|
|
sign_key: String,
|
|
|
|
conn: &Connection,
|
|
|
|
server_sender: SP,
|
|
|
|
) -> Result<()> {
|
2021-04-21 20:07:28 +00:00
|
|
|
let new_id = Uuid::new_v4();
|
2021-03-31 04:12:26 +00:00
|
|
|
let user = db::types::User {
|
2021-04-21 20:07:28 +00:00
|
|
|
id: db::types::Id(new_id.clone()),
|
|
|
|
name: name.clone(),
|
|
|
|
sign_key: sign_key.clone(),
|
2021-04-12 14:05:11 +00:00
|
|
|
insert_key: SSK::parse(&insert_key[..]).unwrap(),
|
2021-03-31 04:12:26 +00:00
|
|
|
messages_count: 0,
|
2021-04-22 15:28:07 +00:00
|
|
|
my_messages_count: 0,
|
2021-03-31 04:12:26 +00:00
|
|
|
};
|
2021-04-21 20:07:28 +00:00
|
|
|
let user_jsoned = crate::api::response::User{
|
|
|
|
id: new_id.clone().to_string(),
|
|
|
|
name: name.clone(),
|
|
|
|
sign_key: sign_key.clone(),
|
|
|
|
insert_key: insert_key,
|
2021-04-22 15:28:07 +00:00
|
|
|
messages_count: 0,
|
|
|
|
my_messages_count: 0
|
2021-04-21 20:07:28 +00:00
|
|
|
};
|
2021-03-31 04:12:26 +00:00
|
|
|
db::users::add_user(user, &conn).unwrap();
|
2021-04-01 12:44:13 +00:00
|
|
|
// Sending "Ok" response to client
|
2021-04-21 20:07:28 +00:00
|
|
|
//
|
2021-04-22 15:28:07 +00:00
|
|
|
//loading all users to frontend
|
2021-04-21 20:07:28 +00:00
|
|
|
load_users(conn, server_sender).unwrap();
|
|
|
|
|
|
|
|
// TODO senging only one user to client{
|
|
|
|
/*
|
2021-04-01 12:44:13 +00:00
|
|
|
server_sender
|
|
|
|
.send(PackedMessage::ToClient(
|
2021-04-21 20:07:28 +00:00
|
|
|
json!(ResponseType::UserAdded(user_jsoned))
|
2021-04-01 12:44:13 +00:00
|
|
|
.to_string(),
|
|
|
|
))
|
|
|
|
.unwrap();
|
2021-04-21 20:07:28 +00:00
|
|
|
*/
|
2021-03-31 04:12:26 +00:00
|
|
|
Ok(())
|
2021-03-26 19:04:39 +00:00
|
|
|
}
|