Extended functionality

src/entity/account.rs

@@ -1,6 +1,10 @@
 //! `SeaORM` Entity. Generated by sea-orm-codegen 0.11.2
 
-use sea_orm::entity::prelude::*;
+use chacha20poly1305::{aead::Aead, AeadCore, ChaCha20Poly1305, KeyInit};
+use pbkdf2::pbkdf2_hmac_array;
+use rand::{rngs::OsRng, RngCore};
+use sea_orm::{prelude::*, ActiveValue::Set, QuerySelect};
+use sha2::Sha256;
 
 #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
 #[sea_orm(table_name = "account")]
@@ -21,3 +25,82 @@ pub struct Model {
 pub enum Relation {}
 
 impl ActiveModelBehavior for ActiveModel {}
+
+struct Cipher {
+    chacha: ChaCha20Poly1305,
+}
+
+impl Cipher {
+    fn new(password: &[u8], salt: &[u8]) -> Self {
+        let key = pbkdf2_hmac_array::<Sha256, 32>(password, salt, 480000);
+
+        Self {
+            chacha: ChaCha20Poly1305::new(&key.into()),
+        }
+    }
+
+    pub fn encrypt(&self, value: &[u8]) -> crate::Result<Vec<u8>> {
+        let nonce = ChaCha20Poly1305::generate_nonce(&mut OsRng);
+        let mut result = self.chacha.encrypt(&nonce, value).unwrap();
+        result.extend(nonce);
+        Ok(result)
+    }
+
+    fn decrypt(&self, value: &[u8]) -> crate::Result<Vec<u8>> {
+        let (data, nonce) = value.split_at(value.len() - 12);
+        self.chacha
+            .decrypt(nonce.into(), data)
+            .map_err(|err| err.into())
+    }
+}
+
+impl ActiveModel {
+    pub fn from_unencrypted(
+        user_id: u64,
+        name: String,
+        login: &str,
+        password: &str,
+        master_pass: &str,
+    ) -> crate::Result<Self> {
+        let mut salt = vec![0; 64];
+        OsRng.fill_bytes(&mut salt);
+        let cipher = Cipher::new(master_pass.as_ref(), &salt);
+        let enc_login = Set(cipher.encrypt(login.as_ref())?);
+        let enc_password = Set(cipher.encrypt(password.as_ref())?);
+        Ok(Self {
+            name: Set(name),
+            user_id: Set(user_id),
+            salt: Set(salt),
+            enc_login,
+            enc_password,
+        })
+    }
+}
+
+impl Model {
+    pub fn decrypt(&self, master_pass: &str) -> crate::Result<(String, String)> {
+        let cipher = Cipher::new(master_pass.as_ref(), self.salt.as_ref());
+        let login = String::from_utf8(cipher.decrypt(self.enc_login.as_ref())?)?;
+        let password = String::from_utf8(cipher.decrypt(self.enc_password.as_ref())?)?;
+        Ok((login, password))
+    }
+}
+
+#[derive(Copy, Clone, EnumIter, DeriveColumn, Debug)]
+enum GetNamesQuery {
+    AccountName,
+}
+
+impl Entity {
+    /// Gets a list of account names of a user
+    pub async fn get_names(user_id: u64, db: &DatabaseConnection) -> crate::Result<Vec<String>> {
+        Self::find()
+            .select_only()
+            .column_as(Column::Name, GetNamesQuery::AccountName)
+            .filter(Column::UserId.eq(user_id))
+            .into_values::<_, GetNamesQuery>()
+            .all(db)
+            .await
+            .map_err(|err| err.into())
+    }
+}

src/entity/master_pass.rs

@@ -1,6 +1,8 @@
 //! `SeaORM` Entity. Generated by sea-orm-codegen 0.11.2
 
-use sea_orm::entity::prelude::*;
+use rand::{rngs::OsRng, RngCore};
+use scrypt::{scrypt, Params};
+use sea_orm::{entity::prelude::*, ActiveValue::Set};
 
 #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
 #[sea_orm(table_name = "master_pass")]
@@ -17,3 +19,23 @@ pub struct Model {
 pub enum Relation {}
 
 impl ActiveModelBehavior for ActiveModel {}
+
+impl ActiveModel {
+    pub fn from_unencrypted(user_id: u64, password: &str) -> crate::Result<Self> {
+        let mut salt = vec![0; 64];
+        OsRng.fill_bytes(&mut salt);
+        let params = Params::new(
+            Params::RECOMMENDED_LOG_N,
+            Params::RECOMMENDED_R,
+            Params::RECOMMENDED_P,
+            128,
+        )?;
+        let mut password_hash = vec![0; 128];
+        scrypt(password.as_ref(), &salt, &params, &mut password_hash)?;
+        Ok(Self {
+            user_id: Set(user_id),
+            salt: Set(salt),
+            password_hash: Set(password_hash),
+        })
+    }
+}