use crate::cnf::SERVER_NAME; use crate::dbs::Auth; use crate::dbs::Session; use crate::err::Error; use crate::iam::token::{Claims, HEADER}; use crate::kvs::Datastore; use crate::sql::Object; use crate::sql::Value; use chrono::{Duration, Utc}; use jsonwebtoken::{encode, EncodingKey}; use std::sync::Arc; pub async fn signup( kvs: &Datastore, strict: bool, session: &mut Session, vars: Object, ) -> Result, Error> { // Parse the specified variables let ns = vars.get("NS").or_else(|| vars.get("ns")); let db = vars.get("DB").or_else(|| vars.get("db")); let sc = vars.get("SC").or_else(|| vars.get("sc")); // Check if the parameters exist match (ns, db, sc) { (Some(ns), Some(db), Some(sc)) => { // Process the provided values let ns = ns.to_raw_string(); let db = db.to_raw_string(); let sc = sc.to_raw_string(); // Attempt to signin to specified scope super::signup::sc(kvs, strict, session, ns, db, sc, vars).await } _ => Err(Error::InvalidAuth), } } pub async fn sc( kvs: &Datastore, strict: bool, session: &mut Session, ns: String, db: String, sc: String, vars: Object, ) -> Result, Error> { // Create a new readonly transaction let mut tx = kvs.transaction(false, false).await?; // Check if the supplied NS Login exists match tx.get_sc(&ns, &db, &sc).await { Ok(sv) => { match sv.signup { // This scope allows signin Some(val) => { // Setup the query params let vars = Some(vars.0); // Setup the query session let sess = Session::for_db(&ns, &db); // Compute the value with the params match kvs.compute(val, &sess, vars, strict).await { // The signin value succeeded Ok(val) => match val.record() { // There is a record returned Some(rid) => { // Create the authentication key let key = EncodingKey::from_secret(sv.code.as_ref()); // Create the authentication claim let val = Claims { iss: Some(SERVER_NAME.to_owned()), iat: Some(Utc::now().timestamp()), nbf: Some(Utc::now().timestamp()), exp: Some( match sv.session { Some(v) => { Utc::now() + Duration::from_std(v.0).unwrap() } _ => Utc::now() + Duration::hours(1), } .timestamp(), ), ns: Some(ns.to_owned()), db: Some(db.to_owned()), sc: Some(sc.to_owned()), id: Some(rid.to_raw()), ..Claims::default() }; // Create the authentication token let enc = encode(&HEADER, &val, &key); // Set the authentication on the session session.tk = Some(val.into()); session.ns = Some(ns.to_owned()); session.db = Some(db.to_owned()); session.sc = Some(sc.to_owned()); session.sd = Some(Value::from(rid)); session.au = Arc::new(Auth::Sc(ns, db, sc)); // Create the authentication token match enc { // The auth token was created successfully Ok(tk) => Ok(Some(tk)), // There was an error creating the token _ => Err(Error::InvalidAuth), } } // No record was returned _ => Err(Error::InvalidAuth), }, // The signin query failed _ => Err(Error::InvalidAuth), } } // This scope does not allow signin _ => Err(Error::InvalidAuth), } } // The scope does not exists _ => Err(Error::InvalidAuth), } }