Daniel García
5 years ago
20 changed files with 812 additions and 2 deletions
@ -0,0 +1 @@ |
|||
DROP TABLE sends; |
|||
@ -0,0 +1,25 @@ |
|||
CREATE TABLE sends ( |
|||
uuid CHAR(36) NOT NULL PRIMARY KEY, |
|||
user_uuid CHAR(36) REFERENCES users (uuid), |
|||
organization_uuid CHAR(36) REFERENCES organizations (uuid), |
|||
|
|||
name TEXT NOT NULL, |
|||
notes TEXT, |
|||
|
|||
atype INTEGER NOT NULL, |
|||
data TEXT NOT NULL, |
|||
key TEXT NOT NULL, |
|||
password_hash BLOB, |
|||
password_salt BLOB, |
|||
password_iter INTEGER, |
|||
|
|||
max_access_count INTEGER, |
|||
access_count INTEGER NOT NULL, |
|||
|
|||
creation_date DATETIME NOT NULL, |
|||
revision_date DATETIME NOT NULL, |
|||
expiration_date DATETIME, |
|||
deletion_date DATETIME NOT NULL, |
|||
|
|||
disabled BOOLEAN NOT NULL |
|||
); |
|||
@ -0,0 +1 @@ |
|||
DROP TABLE sends; |
|||
@ -0,0 +1,25 @@ |
|||
CREATE TABLE sends ( |
|||
uuid CHAR(36) NOT NULL PRIMARY KEY, |
|||
user_uuid CHAR(36) REFERENCES users (uuid), |
|||
organization_uuid CHAR(36) REFERENCES organizations (uuid), |
|||
|
|||
name TEXT NOT NULL, |
|||
notes TEXT, |
|||
|
|||
atype INTEGER NOT NULL, |
|||
data TEXT NOT NULL, |
|||
key TEXT NOT NULL, |
|||
password_hash BYTEA, |
|||
password_salt BYTEA, |
|||
password_iter INTEGER, |
|||
|
|||
max_access_count INTEGER, |
|||
access_count INTEGER NOT NULL, |
|||
|
|||
creation_date TIMESTAMP NOT NULL, |
|||
revision_date TIMESTAMP NOT NULL, |
|||
expiration_date TIMESTAMP, |
|||
deletion_date TIMESTAMP NOT NULL, |
|||
|
|||
disabled BOOLEAN NOT NULL |
|||
); |
|||
@ -0,0 +1 @@ |
|||
DROP TABLE sends; |
|||
@ -0,0 +1,25 @@ |
|||
CREATE TABLE sends ( |
|||
uuid TEXT NOT NULL PRIMARY KEY, |
|||
user_uuid TEXT REFERENCES users (uuid), |
|||
organization_uuid TEXT REFERENCES organizations (uuid), |
|||
|
|||
name TEXT NOT NULL, |
|||
notes TEXT, |
|||
|
|||
atype INTEGER NOT NULL, |
|||
data TEXT NOT NULL, |
|||
key TEXT NOT NULL, |
|||
password_hash BLOB, |
|||
password_salt BLOB, |
|||
password_iter INTEGER, |
|||
|
|||
max_access_count INTEGER, |
|||
access_count INTEGER NOT NULL, |
|||
|
|||
creation_date DATETIME NOT NULL, |
|||
revision_date DATETIME NOT NULL, |
|||
expiration_date DATETIME, |
|||
deletion_date DATETIME NOT NULL, |
|||
|
|||
disabled BOOLEAN NOT NULL |
|||
); |
|||
@ -0,0 +1,383 @@ |
|||
use std::{io::Read, path::Path}; |
|||
|
|||
use chrono::{DateTime, Duration, Utc}; |
|||
use multipart::server::{save::SavedData, Multipart, SaveResult}; |
|||
use rocket::{http::ContentType, Data}; |
|||
use rocket_contrib::json::Json; |
|||
use serde_json::Value; |
|||
|
|||
use crate::{ |
|||
api::{ApiResult, EmptyResult, JsonResult, JsonUpcase, Notify, UpdateType}, |
|||
auth::{Headers, Host}, |
|||
db::{models::*, DbConn}, |
|||
CONFIG, |
|||
}; |
|||
|
|||
pub fn routes() -> Vec<rocket::Route> { |
|||
routes![ |
|||
post_send, |
|||
post_send_file, |
|||
post_access, |
|||
post_access_file, |
|||
put_send, |
|||
delete_send, |
|||
put_remove_password |
|||
] |
|||
} |
|||
|
|||
#[derive(Deserialize)] |
|||
#[allow(non_snake_case)] |
|||
pub struct SendData { |
|||
pub Type: i32, |
|||
pub Key: String, |
|||
pub Password: Option<String>, |
|||
pub MaxAccessCount: Option<i32>, |
|||
pub ExpirationDate: Option<DateTime<Utc>>, |
|||
pub DeletionDate: DateTime<Utc>, |
|||
pub Disabled: bool, |
|||
|
|||
// Data field
|
|||
pub Name: String, |
|||
pub Notes: Option<String>, |
|||
pub Text: Option<Value>, |
|||
pub File: Option<Value>, |
|||
} |
|||
|
|||
fn create_send(data: SendData, user_uuid: String) -> ApiResult<Send> { |
|||
let data_val = if data.Type == SendType::Text as i32 { |
|||
data.Text |
|||
} else if data.Type == SendType::File as i32 { |
|||
data.File |
|||
} else { |
|||
err!("Invalid Send type") |
|||
}; |
|||
|
|||
let data_str = if let Some(mut d) = data_val { |
|||
d.as_object_mut().and_then(|o| o.remove("Response")); |
|||
serde_json::to_string(&d)? |
|||
} else { |
|||
err!("Send data not provided"); |
|||
}; |
|||
|
|||
if data.DeletionDate > Utc::now() + Duration::days(31) { |
|||
err!( |
|||
"You cannot have a Send with a deletion date that far into the future. Adjust the Deletion Date to a value less than 31 days from now and try again." |
|||
); |
|||
} |
|||
|
|||
let mut send = Send::new(data.Type, data.Name, data_str, data.Key, data.DeletionDate.naive_utc()); |
|||
send.user_uuid = Some(user_uuid); |
|||
send.notes = data.Notes; |
|||
send.max_access_count = data.MaxAccessCount; |
|||
send.expiration_date = data.ExpirationDate.map(|d| d.naive_utc()); |
|||
send.disabled = data.Disabled; |
|||
send.atype = data.Type; |
|||
|
|||
send.set_password(data.Password.as_deref()); |
|||
|
|||
Ok(send) |
|||
} |
|||
|
|||
#[post("/sends", data = "<data>")] |
|||
fn post_send(data: JsonUpcase<SendData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult { |
|||
let data: SendData = data.into_inner().data; |
|||
|
|||
if data.Type == SendType::File as i32 { |
|||
err!("File sends should use /api/sends/file") |
|||
} |
|||
|
|||
let mut send = create_send(data, headers.user.uuid.clone())?; |
|||
send.save(&conn)?; |
|||
nt.send_user_update(UpdateType::SyncSendCreate, &headers.user); |
|||
|
|||
Ok(Json(send.to_json())) |
|||
} |
|||
|
|||
#[post("/sends/file", format = "multipart/form-data", data = "<data>")] |
|||
fn post_send_file(data: Data, content_type: &ContentType, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult { |
|||
let boundary = content_type.params().next().expect("No boundary provided").1; |
|||
|
|||
let mut mpart = Multipart::with_body(data.open(), boundary); |
|||
|
|||
// First entry is the SendData JSON
|
|||
let mut model_entry = match mpart.read_entry()? { |
|||
Some(e) if &*e.headers.name == "model" => e, |
|||
Some(_) => err!("Invalid entry name"), |
|||
None => err!("No model entry present"), |
|||
}; |
|||
|
|||
let mut buf = String::new(); |
|||
model_entry.data.read_to_string(&mut buf)?; |
|||
let data = serde_json::from_str::<crate::util::UpCase<SendData>>(&buf)?; |
|||
|
|||
// Get the file length and add an extra 10% to avoid issues
|
|||
const SIZE_110_MB: u64 = 115_343_360; |
|||
|
|||
let size_limit = match CONFIG.user_attachment_limit() { |
|||
Some(0) => err!("File uploads are disabled"), |
|||
Some(limit_kb) => { |
|||
let left = (limit_kb * 1024) - Attachment::size_by_user(&headers.user.uuid, &conn); |
|||
if left <= 0 { |
|||
err!("Attachment size limit reached! Delete some files to open space") |
|||
} |
|||
std::cmp::Ord::max(left as u64, SIZE_110_MB) |
|||
} |
|||
None => SIZE_110_MB, |
|||
}; |
|||
|
|||
// Create the Send
|
|||
let mut send = create_send(data.data, headers.user.uuid.clone())?; |
|||
let file_id: String = data_encoding::HEXLOWER.encode(&crate::crypto::get_random(vec![0; 32])); |
|||
|
|||
if send.atype != SendType::File as i32 { |
|||
err!("Send content is not a file"); |
|||
} |
|||
|
|||
let file_path = Path::new(&CONFIG.sends_folder()).join(&send.uuid).join(&file_id); |
|||
|
|||
// Read the data entry and save the file
|
|||
let mut data_entry = match mpart.read_entry()? { |
|||
Some(e) if &*e.headers.name == "data" => e, |
|||
Some(_) => err!("Invalid entry name"), |
|||
None => err!("No model entry present"), |
|||
}; |
|||
|
|||
let size = match data_entry |
|||
.data |
|||
.save() |
|||
.memory_threshold(0) |
|||
.size_limit(size_limit) |
|||
.with_path(&file_path) |
|||
{ |
|||
SaveResult::Full(SavedData::File(_, size)) => size as i32, |
|||
SaveResult::Full(other) => { |
|||
std::fs::remove_file(&file_path).ok(); |
|||
err!(format!("Attachment is not a file: {:?}", other)); |
|||
} |
|||
SaveResult::Partial(_, reason) => { |
|||
std::fs::remove_file(&file_path).ok(); |
|||
err!(format!("Attachment size limit exceeded with this file: {:?}", reason)); |
|||
} |
|||
SaveResult::Error(e) => { |
|||
std::fs::remove_file(&file_path).ok(); |
|||
err!(format!("Error: {:?}", e)); |
|||
} |
|||
}; |
|||
|
|||
// Set ID and sizes
|
|||
let mut data_value: Value = serde_json::from_str(&send.data)?; |
|||
if let Some(o) = data_value.as_object_mut() { |
|||
o.insert(String::from("Id"), Value::String(file_id)); |
|||
o.insert(String::from("Size"), Value::Number(size.into())); |
|||
o.insert( |
|||
String::from("SizeName"), |
|||
Value::String(crate::util::get_display_size(size)), |
|||
); |
|||
} |
|||
send.data = serde_json::to_string(&data_value)?; |
|||
|
|||
// Save the changes in the database
|
|||
send.save(&conn)?; |
|||
nt.send_user_update(UpdateType::SyncSendCreate, &headers.user); |
|||
|
|||
Ok(Json(send.to_json())) |
|||
} |
|||
|
|||
#[derive(Deserialize)] |
|||
#[allow(non_snake_case)] |
|||
pub struct SendAccessData { |
|||
pub Password: Option<String>, |
|||
} |
|||
|
|||
#[post("/sends/access/<access_id>", data = "<data>")] |
|||
fn post_access(access_id: String, data: JsonUpcase<SendAccessData>, conn: DbConn) -> JsonResult { |
|||
let mut send = match Send::find_by_access_id(&access_id, &conn) { |
|||
Some(s) => s, |
|||
None => err_code!("Send not found", 404), |
|||
}; |
|||
|
|||
if let Some(max_access_count) = send.max_access_count { |
|||
if send.access_count > max_access_count { |
|||
err_code!("Max access count reached", 404); |
|||
} |
|||
} |
|||
|
|||
if let Some(expiration) = send.expiration_date { |
|||
if Utc::now().naive_utc() > expiration { |
|||
err_code!("Send has expired", 404) |
|||
} |
|||
} |
|||
|
|||
if Utc::now().naive_utc() > send.deletion_date { |
|||
err_code!("Send has been deleted", 404) |
|||
} |
|||
|
|||
if send.disabled { |
|||
err_code!("Send has been disabled", 404) |
|||
} |
|||
|
|||
if send.password_hash.is_some() { |
|||
match data.into_inner().data.Password { |
|||
Some(ref p) if send.check_password(p) => { /* Nothing to do here */ } |
|||
Some(_) => err!("Invalid password."), |
|||
None => err_code!("Password not provided", 401), |
|||
} |
|||
} |
|||
|
|||
// Files are incremented during the download
|
|||
if send.atype == SendType::Text as i32 { |
|||
send.access_count += 1; |
|||
} |
|||
|
|||
send.save(&conn)?; |
|||
|
|||
Ok(Json(send.to_json())) |
|||
} |
|||
|
|||
#[post("/sends/<send_id>/access/file/<file_id>", data = "<data>")] |
|||
fn post_access_file( |
|||
send_id: String, |
|||
file_id: String, |
|||
data: JsonUpcase<SendAccessData>, |
|||
host: Host, |
|||
conn: DbConn, |
|||
) -> JsonResult { |
|||
let mut send = match Send::find_by_uuid(&send_id, &conn) { |
|||
Some(s) => s, |
|||
None => err_code!("Send not found", 404), |
|||
}; |
|||
|
|||
if let Some(max_access_count) = send.max_access_count { |
|||
if send.access_count > max_access_count { |
|||
err_code!("Max access count reached", 404); |
|||
} |
|||
} |
|||
|
|||
if let Some(expiration) = send.expiration_date { |
|||
if Utc::now().naive_utc() > expiration { |
|||
err_code!("Send has expired", 404) |
|||
} |
|||
} |
|||
|
|||
if Utc::now().naive_utc() > send.deletion_date { |
|||
err_code!("Send has been deleted", 404) |
|||
} |
|||
|
|||
if send.disabled { |
|||
err_code!("Send has been disabled", 404) |
|||
} |
|||
|
|||
if send.password_hash.is_some() { |
|||
match data.into_inner().data.Password { |
|||
Some(ref p) if send.check_password(p) => { /* Nothing to do here */ } |
|||
Some(_) => err!("Invalid password."), |
|||
None => err_code!("Password not provided", 401), |
|||
} |
|||
} |
|||
|
|||
send.access_count += 1; |
|||
|
|||
send.save(&conn)?; |
|||
|
|||
Ok(Json(json!({ |
|||
"Object": "send-fileDownload", |
|||
"Id": file_id, |
|||
"Url": format!("{}/sends/{}/{}", &host.host, send_id, file_id) |
|||
}))) |
|||
} |
|||
|
|||
#[put("/sends/<id>", data = "<data>")] |
|||
fn put_send(id: String, data: JsonUpcase<SendData>, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult { |
|||
let data: SendData = data.into_inner().data; |
|||
|
|||
let mut send = match Send::find_by_uuid(&id, &conn) { |
|||
Some(s) => s, |
|||
None => err!("Send not found"), |
|||
}; |
|||
|
|||
if send.user_uuid.as_ref() != Some(&headers.user.uuid) { |
|||
err!("Send is not owned by user") |
|||
} |
|||
|
|||
if send.atype != data.Type { |
|||
err!("Sends can't change type") |
|||
} |
|||
|
|||
let data_val = if data.Type == SendType::Text as i32 { |
|||
data.Text |
|||
} else if data.Type == SendType::File as i32 { |
|||
data.File |
|||
} else { |
|||
err!("Invalid Send type") |
|||
}; |
|||
|
|||
let data_str = if let Some(mut d) = data_val { |
|||
d.as_object_mut().and_then(|d| d.remove("Response")); |
|||
serde_json::to_string(&d)? |
|||
} else { |
|||
err!("Send data not provided"); |
|||
}; |
|||
|
|||
if data.DeletionDate > Utc::now() + Duration::days(31) { |
|||
err!( |
|||
"You cannot have a Send with a deletion date that far into the future. Adjust the Deletion Date to a value less than 31 days from now and try again." |
|||
); |
|||
} |
|||
send.data = data_str; |
|||
send.name = data.Name; |
|||
send.key = data.Key; |
|||
send.deletion_date = data.DeletionDate.naive_utc(); |
|||
send.notes = data.Notes; |
|||
send.max_access_count = data.MaxAccessCount; |
|||
send.expiration_date = data.ExpirationDate.map(|d| d.naive_utc()); |
|||
send.disabled = data.Disabled; |
|||
|
|||
// Only change the value if it's present
|
|||
if let Some(password) = data.Password { |
|||
send.set_password(Some(&password)); |
|||
} |
|||
|
|||
send.save(&conn)?; |
|||
nt.send_user_update(UpdateType::SyncSendUpdate, &headers.user); |
|||
|
|||
Ok(Json(send.to_json())) |
|||
} |
|||
|
|||
#[delete("/sends/<id>")] |
|||
fn delete_send(id: String, headers: Headers, conn: DbConn, nt: Notify) -> EmptyResult { |
|||
let send = match Send::find_by_uuid(&id, &conn) { |
|||
Some(s) => s, |
|||
None => err!("Send not found"), |
|||
}; |
|||
|
|||
if send.user_uuid.as_ref() != Some(&headers.user.uuid) { |
|||
err!("Send is not owned by user") |
|||
} |
|||
|
|||
if send.atype == SendType::File as i32 { |
|||
std::fs::remove_dir_all(Path::new(&CONFIG.sends_folder()).join(&send.uuid)).ok(); |
|||
} |
|||
|
|||
send.delete(&conn)?; |
|||
nt.send_user_update(UpdateType::SyncSendDelete, &headers.user); |
|||
|
|||
Ok(()) |
|||
} |
|||
|
|||
#[put("/sends/<id>/remove-password")] |
|||
fn put_remove_password(id: String, headers: Headers, conn: DbConn, nt: Notify) -> JsonResult { |
|||
let mut send = match Send::find_by_uuid(&id, &conn) { |
|||
Some(s) => s, |
|||
None => err!("Send not found"), |
|||
}; |
|||
|
|||
if send.user_uuid.as_ref() != Some(&headers.user.uuid) { |
|||
err!("Send is not owned by user") |
|||
} |
|||
|
|||
send.set_password(None); |
|||
send.save(&conn)?; |
|||
nt.send_user_update(UpdateType::SyncSendUpdate, &headers.user); |
|||
|
|||
Ok(Json(send.to_json())) |
|||
} |
|||
@ -0,0 +1,235 @@ |
|||
use chrono::{NaiveDateTime, Utc}; |
|||
use serde_json::Value; |
|||
|
|||
use super::{Organization, User}; |
|||
|
|||
db_object! { |
|||
#[derive(Identifiable, Queryable, Insertable, Associations, AsChangeset)] |
|||
#[table_name = "sends"] |
|||
#[changeset_options(treat_none_as_null="true")] |
|||
#[belongs_to(User, foreign_key = "user_uuid")] |
|||
#[belongs_to(Organization, foreign_key = "organization_uuid")] |
|||
#[primary_key(uuid)] |
|||
pub struct Send { |
|||
pub uuid: String, |
|||
|
|||
pub user_uuid: Option<String>, |
|||
pub organization_uuid: Option<String>, |
|||
|
|||
|
|||
pub name: String, |
|||
pub notes: Option<String>, |
|||
|
|||
pub atype: i32, |
|||
pub data: String, |
|||
pub key: String, |
|||
pub password_hash: Option<Vec<u8>>, |
|||
password_salt: Option<Vec<u8>>, |
|||
password_iter: Option<i32>, |
|||
|
|||
pub max_access_count: Option<i32>, |
|||
pub access_count: i32, |
|||
|
|||
pub creation_date: NaiveDateTime, |
|||
pub revision_date: NaiveDateTime, |
|||
pub expiration_date: Option<NaiveDateTime>, |
|||
pub deletion_date: NaiveDateTime, |
|||
|
|||
pub disabled: bool, |
|||
} |
|||
} |
|||
|
|||
#[derive(Copy, Clone, PartialEq, Eq, num_derive::FromPrimitive)] |
|||
pub enum SendType { |
|||
Text = 0, |
|||
File = 1, |
|||
} |
|||
|
|||
impl Send { |
|||
pub fn new(atype: i32, name: String, data: String, key: String, deletion_date: NaiveDateTime) -> Self { |
|||
let now = Utc::now().naive_utc(); |
|||
|
|||
Self { |
|||
uuid: crate::util::get_uuid(), |
|||
user_uuid: None, |
|||
organization_uuid: None, |
|||
|
|||
name, |
|||
notes: None, |
|||
|
|||
atype, |
|||
data, |
|||
key, |
|||
password_hash: None, |
|||
password_salt: None, |
|||
password_iter: None, |
|||
|
|||
max_access_count: None, |
|||
access_count: 0, |
|||
|
|||
creation_date: now, |
|||
revision_date: now, |
|||
expiration_date: None, |
|||
deletion_date, |
|||
|
|||
disabled: false, |
|||
} |
|||
} |
|||
|
|||
pub fn set_password(&mut self, password: Option<&str>) { |
|||
const PASSWORD_ITER: i32 = 100_000; |
|||
|
|||
if let Some(password) = password { |
|||
self.password_iter = Some(PASSWORD_ITER); |
|||
let salt = crate::crypto::get_random_64(); |
|||
let hash = crate::crypto::hash_password(password.as_bytes(), &salt, PASSWORD_ITER as u32); |
|||
self.password_salt = Some(salt); |
|||
self.password_hash = Some(hash); |
|||
} else { |
|||
self.password_iter = None; |
|||
self.password_salt = None; |
|||
self.password_hash = None; |
|||
} |
|||
} |
|||
|
|||
pub fn check_password(&self, password: &str) -> bool { |
|||
match (&self.password_hash, &self.password_salt, self.password_iter) { |
|||
(Some(hash), Some(salt), Some(iter)) => { |
|||
crate::crypto::verify_password_hash(password.as_bytes(), salt, hash, iter as u32) |
|||
} |
|||
_ => false, |
|||
} |
|||
} |
|||
|
|||
pub fn to_json(&self) -> Value { |
|||
use crate::util::format_date; |
|||
use data_encoding::BASE64URL_NOPAD; |
|||
use uuid::Uuid; |
|||
|
|||
let data: Value = serde_json::from_str(&self.data).unwrap_or_default(); |
|||
|
|||
json!({ |
|||
"Id": self.uuid, |
|||
"AccessId": BASE64URL_NOPAD.encode(Uuid::parse_str(&self.uuid).unwrap_or_default().as_bytes()), |
|||
"Type": self.atype, |
|||
|
|||
"Name": self.name, |
|||
"Notes": self.notes, |
|||
"Text": if self.atype == SendType::Text as i32 { Some(&data) } else { None }, |
|||
"File": if self.atype == SendType::File as i32 { Some(&data) } else { None }, |
|||
|
|||
"Key": self.key, |
|||
"MaxAccessCount": self.max_access_count, |
|||
"AccessCount": self.access_count, |
|||
"Password": self.password_hash.as_deref().map(|h| BASE64URL_NOPAD.encode(h)), |
|||
"Disabled": self.disabled, |
|||
|
|||
"RevisionDate": format_date(&self.revision_date), |
|||
"ExpirationDate": self.expiration_date.as_ref().map(format_date), |
|||
"DeletionDate": format_date(&self.deletion_date), |
|||
"Object": "send", |
|||
}) |
|||
} |
|||
} |
|||
|
|||
use crate::db::DbConn; |
|||
|
|||
use crate::api::EmptyResult; |
|||
use crate::error::MapResult; |
|||
|
|||
impl Send { |
|||
pub fn save(&mut self, conn: &DbConn) -> EmptyResult { |
|||
// self.update_users_revision(conn);
|
|||
self.revision_date = Utc::now().naive_utc(); |
|||
|
|||
db_run! { conn: |
|||
sqlite, mysql { |
|||
match diesel::replace_into(sends::table) |
|||
.values(SendDb::to_db(self)) |
|||
.execute(conn) |
|||
{ |
|||
Ok(_) => Ok(()), |
|||
// Record already exists and causes a Foreign Key Violation because replace_into() wants to delete the record first.
|
|||
Err(diesel::result::Error::DatabaseError(diesel::result::DatabaseErrorKind::ForeignKeyViolation, _)) => { |
|||
diesel::update(sends::table) |
|||
.filter(sends::uuid.eq(&self.uuid)) |
|||
.set(SendDb::to_db(self)) |
|||
.execute(conn) |
|||
.map_res("Error saving send") |
|||
} |
|||
Err(e) => Err(e.into()), |
|||
}.map_res("Error saving send") |
|||
} |
|||
postgresql { |
|||
let value = SendDb::to_db(self); |
|||
diesel::insert_into(sends::table) |
|||
.values(&value) |
|||
.on_conflict(sends::uuid) |
|||
.do_update() |
|||
.set(&value) |
|||
.execute(conn) |
|||
.map_res("Error saving send") |
|||
} |
|||
} |
|||
} |
|||
|
|||
pub fn delete(&self, conn: &DbConn) -> EmptyResult { |
|||
// self.update_users_revision(conn);
|
|||
|
|||
db_run! { conn: { |
|||
diesel::delete(sends::table.filter(sends::uuid.eq(&self.uuid))) |
|||
.execute(conn) |
|||
.map_res("Error deleting send") |
|||
}} |
|||
} |
|||
|
|||
pub fn delete_all_by_user(user_uuid: &str, conn: &DbConn) -> EmptyResult { |
|||
for send in Self::find_by_user(user_uuid, &conn) { |
|||
send.delete(&conn)?; |
|||
} |
|||
Ok(()) |
|||
} |
|||
|
|||
pub fn find_by_access_id(access_id: &str, conn: &DbConn) -> Option<Self> { |
|||
use data_encoding::BASE64URL_NOPAD; |
|||
use uuid::Uuid; |
|||
|
|||
let uuid_vec = match BASE64URL_NOPAD.decode(access_id.as_bytes()) { |
|||
Ok(v) => v, |
|||
Err(_) => return None, |
|||
}; |
|||
|
|||
let uuid = match Uuid::from_slice(&uuid_vec) { |
|||
Ok(u) => u.to_string(), |
|||
Err(_) => return None, |
|||
}; |
|||
|
|||
Self::find_by_uuid(&uuid, conn) |
|||
} |
|||
|
|||
pub fn find_by_uuid(uuid: &str, conn: &DbConn) -> Option<Self> { |
|||
db_run! {conn: { |
|||
sends::table |
|||
.filter(sends::uuid.eq(uuid)) |
|||
.first::<SendDb>(conn) |
|||
.ok() |
|||
.from_db() |
|||
}} |
|||
} |
|||
|
|||
pub fn find_by_user(user_uuid: &str, conn: &DbConn) -> Vec<Self> { |
|||
db_run! {conn: { |
|||
sends::table |
|||
.filter(sends::user_uuid.eq(user_uuid)) |
|||
.load::<SendDb>(conn).expect("Error loading sends").from_db() |
|||
}} |
|||
} |
|||
|
|||
pub fn find_by_org(org_uuid: &str, conn: &DbConn) -> Vec<Self> { |
|||
db_run! {conn: { |
|||
sends::table |
|||
.filter(sends::organization_uuid.eq(org_uuid)) |
|||
.load::<SendDb>(conn).expect("Error loading sends").from_db() |
|||
}} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue