summaryrefslogtreecommitdiffstats
path: root/adenosine-pds/src/db.rs
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2022-11-06 23:08:33 -0800
committerBryan Newbold <bnewbold@robocracy.org>2022-11-06 23:08:33 -0800
commit388d4c004146865161e2661fb63c705e7f097f3a (patch)
tree40ad5f05fd805b779b95fc27780cb143fafe88bf /adenosine-pds/src/db.rs
parent31d427f7342a58308501ba7e1f74751393299cf1 (diff)
downloadadenosine-388d4c004146865161e2661fb63c705e7f097f3a.tar.gz
adenosine-388d4c004146865161e2661fb63c705e7f097f3a.zip
pds: refactor to use typed identifiers a lot more
Diffstat (limited to 'adenosine-pds/src/db.rs')
-rw-r--r--adenosine-pds/src/db.rs30
1 files changed, 16 insertions, 14 deletions
diff --git a/adenosine-pds/src/db.rs b/adenosine-pds/src/db.rs
index 0fae769..5f185c8 100644
--- a/adenosine-pds/src/db.rs
+++ b/adenosine-pds/src/db.rs
@@ -1,5 +1,5 @@
/// ATP database (as distinct from blockstore)
-use crate::{AtpSession, KeyPair};
+use crate::{AtpSession, Did, KeyPair};
use anyhow::{anyhow, Result};
use lazy_static::lazy_static;
use log::debug;
@@ -74,7 +74,7 @@ impl AtpDatabase {
pub fn create_account(
&mut self,
- did: &str,
+ did: &Did,
username: &str,
password: &str,
email: &str,
@@ -89,8 +89,8 @@ impl AtpDatabase {
username,
password_bcrypt,
email,
- did,
- recovery_pubkey
+ did.to_string(),
+ recovery_pubkey,
))?;
Ok(())
}
@@ -105,18 +105,19 @@ impl AtpDatabase {
let mut stmt = self
.conn
.prepare_cached("SELECT did, password_bcrypt FROM account WHERE username = ?1")?;
- let (did, password_bcrypt): (String, String) =
+ let (did_col, password_bcrypt): (String, String) =
stmt.query_row(params!(username), |row| Ok((row.get(0)?, row.get(1)?)))?;
if !bcrypt::verify(password, &password_bcrypt)? {
return Err(anyhow!("password did not match"));
}
+ let did = Did::from_str(&did_col)?;
let jwt = keypair.ucan(&did)?;
let mut stmt = self
.conn
.prepare_cached("INSERT INTO session (did, jwt) VALUES (?1, ?2)")?;
- stmt.execute(params!(did, jwt))?;
+ stmt.execute(params!(did.to_string(), jwt))?;
Ok(AtpSession {
- did,
+ did: did.to_string(),
name: username.to_string(),
accessJwt: jwt.to_string(),
refreshJwt: jwt,
@@ -124,12 +125,13 @@ impl AtpDatabase {
}
/// Returns the DID that a token is valid for, or None if session not found
- pub fn check_auth_token(&mut self, jwt: &str) -> Result<Option<String>> {
+ pub fn check_auth_token(&mut self, jwt: &str) -> Result<Option<Did>> {
let mut stmt = self
.conn
.prepare_cached("SELECT did FROM session WHERE jwt = $1")?;
- let did_maybe = stmt.query_row(params!(jwt), |row| row.get(0)).optional()?;
- Ok(did_maybe)
+ let did_maybe: Option<String> =
+ stmt.query_row(params!(jwt), |row| row.get(0)).optional()?;
+ Ok(did_maybe.map(|v| Did::from_str(&v).expect("valid DID in database")))
}
pub fn delete_session(&mut self, jwt: &str) -> Result<bool> {
@@ -140,18 +142,18 @@ impl AtpDatabase {
Ok(count >= 1)
}
- pub fn put_did_doc(&mut self, did: &str, did_doc: &Value) -> Result<()> {
+ pub fn put_did_doc(&mut self, did: &Did, did_doc: &Value) -> Result<()> {
let mut stmt = self
.conn
.prepare_cached("INSERT INTO did_doc (did, doc_json) VALUES (?1, ?2)")?;
- stmt.execute(params!(did, did_doc.to_string()))?;
+ stmt.execute(params!(did.to_string(), did_doc.to_string()))?;
Ok(())
}
- pub fn get_did_doc(&mut self, did: &str) -> Result<Value> {
+ pub fn get_did_doc(&mut self, did: &Did) -> Result<Value> {
let mut stmt = self
.conn
.prepare_cached("SELECT doc_json FROM did_doc WHERE did = $1")?;
- let doc_json: String = stmt.query_row(params!(did), |row| row.get(0))?;
+ let doc_json: String = stmt.query_row(params!(did.to_string()), |row| row.get(0))?;
Ok(Value::from_str(&doc_json)?)
}
}