mirror of
https://github.com/YGGverse/Yoda.git
synced 2025-01-15 17:20:08 +00:00
implement to_string method, prevent memory index overwrite on validation step
This commit is contained in:
parent
cda94cba2e
commit
c86dca53bd
@ -119,18 +119,24 @@ impl Gemini {
|
|||||||
Value::ProfileIdentityGeminiId(value) => Some(value),
|
Value::ProfileIdentityGeminiId(value) => Some(value),
|
||||||
Value::UseGuestSession => None,
|
Value::UseGuestSession => None,
|
||||||
Value::GenerateNewAuth => Some(
|
Value::GenerateNewAuth => Some(
|
||||||
profile
|
match profile
|
||||||
.identity
|
.identity
|
||||||
.gemini
|
.gemini
|
||||||
.make(None, &widget.form.name.value().unwrap())
|
.make(None, &widget.form.name.value().unwrap())
|
||||||
.unwrap(),
|
{
|
||||||
|
Ok(profile_identity_gemini_id) => profile_identity_gemini_id,
|
||||||
|
Err(reason) => todo!("{}", reason.to_string()),
|
||||||
|
},
|
||||||
),
|
),
|
||||||
Value::ImportPem => Some(
|
Value::ImportPem => Some(
|
||||||
profile
|
match profile
|
||||||
.identity
|
.identity
|
||||||
.gemini
|
.gemini
|
||||||
.add(&widget.form.file.pem.take().unwrap())
|
.add(&widget.form.file.pem.take().unwrap())
|
||||||
.unwrap(),
|
{
|
||||||
|
Ok(profile_identity_gemini_id) => profile_identity_gemini_id,
|
||||||
|
Err(reason) => todo!("{}", reason.to_string()),
|
||||||
|
},
|
||||||
),
|
),
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -98,7 +98,7 @@ impl Profile {
|
|||||||
// Init identity component
|
// Init identity component
|
||||||
let identity = Rc::new(match Identity::new(connection, profile_id) {
|
let identity = Rc::new(match Identity::new(connection, profile_id) {
|
||||||
Ok(result) => result,
|
Ok(result) => result,
|
||||||
Err(reason) => todo!("{:?}", reason),
|
Err(reason) => todo!("{:?}", reason.to_string()),
|
||||||
});
|
});
|
||||||
|
|
||||||
// Result
|
// Result
|
||||||
|
@ -28,12 +28,12 @@ impl Bookmark {
|
|||||||
match database.records(None) {
|
match database.records(None) {
|
||||||
Ok(records) => {
|
Ok(records) => {
|
||||||
for record in records {
|
for record in records {
|
||||||
if memory.add(record.request, record.id).is_err() {
|
if let Err(reason) = memory.add(record.request, record.id) {
|
||||||
todo!()
|
todo!("{}", reason.to_string())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(reason) => todo!("{reason}"),
|
Err(reason) => todo!("{}", reason.to_string()),
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return new `Self`
|
// Return new `Self`
|
||||||
|
@ -23,8 +23,17 @@ impl Memory {
|
|||||||
/// Add new record with `request` as key and `id` as value
|
/// Add new record with `request` as key and `id` as value
|
||||||
/// * validate record with same key does not exist yet
|
/// * validate record with same key does not exist yet
|
||||||
pub fn add(&self, request: String, id: i64) -> Result<(), Error> {
|
pub fn add(&self, request: String, id: i64) -> Result<(), Error> {
|
||||||
match self.index.borrow_mut().insert(request, id) {
|
// Borrow shared index access
|
||||||
Some(key) => Err(Error::Overwrite(key)), // @TODO prevent?
|
let mut index = self.index.borrow_mut();
|
||||||
|
|
||||||
|
// Prevent existing key overwrite
|
||||||
|
if index.contains_key(&request) {
|
||||||
|
return Err(Error::Overwrite(request));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Slot should be free, let check it twice
|
||||||
|
match index.insert(request, id) {
|
||||||
|
Some(_) => return Err(Error::Unexpected),
|
||||||
None => Ok(()),
|
None => Ok(()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -34,15 +43,15 @@ impl Memory {
|
|||||||
pub fn delete(&self, request: &str) -> Result<(), Error> {
|
pub fn delete(&self, request: &str) -> Result<(), Error> {
|
||||||
match self.index.borrow_mut().remove(request) {
|
match self.index.borrow_mut().remove(request) {
|
||||||
Some(_) => Ok(()),
|
Some(_) => Ok(()),
|
||||||
None => Err(Error::NotFound),
|
None => Err(Error::Unexpected), // @TODO
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get `id` by `request` from memory index
|
/// Get `id` by `request` from memory index
|
||||||
pub fn get(&self, request: &str) -> Result<i64, Error> {
|
pub fn get(&self, request: &str) -> Result<i64, Error> {
|
||||||
match self.index.borrow().get(request) {
|
match self.index.borrow().get(request) {
|
||||||
Some(&id) => Ok(id),
|
Some(&value) => Ok(value),
|
||||||
None => Err(Error::NotFound),
|
None => Err(Error::Unexpected), // @TODO
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,16 @@
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Error {
|
pub enum Error {
|
||||||
NotFound,
|
Overwrite(String),
|
||||||
Overwrite(i64),
|
Unexpected,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Error {
|
||||||
|
pub fn to_string(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::Overwrite(key) => {
|
||||||
|
format!("Overwrite attempt for existing record `{key}`")
|
||||||
|
}
|
||||||
|
Self::Unexpected => format!("Unexpected error"),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -55,7 +55,7 @@ impl Identity {
|
|||||||
if let Some(id) = self.gemini.auth.memory.match_priority(request) {
|
if let Some(id) = self.gemini.auth.memory.match_priority(request) {
|
||||||
match self.gemini.memory.get(id) {
|
match self.gemini.memory.get(id) {
|
||||||
Ok(pem) => return Some(pem),
|
Ok(pem) => return Some(pem),
|
||||||
Err(reason) => todo!("{:?}", reason),
|
Err(reason) => todo!("{:?}", reason.to_string()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
None
|
None
|
||||||
|
@ -3,3 +3,16 @@ pub enum Error {
|
|||||||
Database(sqlite::Error),
|
Database(sqlite::Error),
|
||||||
Gemini(super::gemini::Error),
|
Gemini(super::gemini::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Error {
|
||||||
|
pub fn to_string(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::Database(reason) => {
|
||||||
|
format!("Database error: {}", reason.to_string())
|
||||||
|
}
|
||||||
|
Self::Gemini(reason) => {
|
||||||
|
format!("Could not init Gemini identity: {}", reason.to_string())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -1,7 +1,22 @@
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Error {
|
pub enum Error {
|
||||||
Auth(super::auth::Error),
|
Auth(super::auth::Error),
|
||||||
Certificate(Box<dyn std::error::Error>), // @TODO
|
Certificate(Box<dyn std::error::Error>),
|
||||||
Database(sqlite::Error),
|
Database(sqlite::Error),
|
||||||
Memory(super::memory::Error),
|
Memory(super::memory::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Error {
|
||||||
|
pub fn to_string(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::Auth(reason) => format!("Could not create auth: {}", reason.to_string()),
|
||||||
|
Self::Certificate(reason) => {
|
||||||
|
format!("Could not create certificate: {}", reason.to_string())
|
||||||
|
}
|
||||||
|
Self::Database(reason) => {
|
||||||
|
format!("Database error: {}", reason.to_string())
|
||||||
|
}
|
||||||
|
Self::Memory(reason) => format!("Memory error: {}", reason.to_string()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -22,9 +22,18 @@ impl Memory {
|
|||||||
|
|
||||||
/// Add new record with `id` as key and `pem` as value
|
/// Add new record with `id` as key and `pem` as value
|
||||||
/// * validate record with same key does not exist yet
|
/// * validate record with same key does not exist yet
|
||||||
pub fn add(&self, id: i64, pem: String) -> Result<(), Error> {
|
pub fn add(&self, profile_identity_gemini_id: i64, pem: String) -> Result<(), Error> {
|
||||||
match self.index.borrow_mut().insert(id, pem) {
|
// Borrow shared index access
|
||||||
Some(key) => Err(Error::Overwrite(key)), // @TODO prevent?
|
let mut index = self.index.borrow_mut();
|
||||||
|
|
||||||
|
// Prevent existing key overwrite
|
||||||
|
if index.contains_key(&profile_identity_gemini_id) {
|
||||||
|
return Err(Error::Overwrite(profile_identity_gemini_id));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Slot should be free, let check it twice
|
||||||
|
match index.insert(profile_identity_gemini_id, pem) {
|
||||||
|
Some(_) => return Err(Error::Unexpected),
|
||||||
None => Ok(()),
|
None => Ok(()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -2,5 +2,21 @@
|
|||||||
pub enum Error {
|
pub enum Error {
|
||||||
Clear,
|
Clear,
|
||||||
NotFound(i64),
|
NotFound(i64),
|
||||||
Overwrite(String),
|
Overwrite(i64),
|
||||||
|
Unexpected,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Error {
|
||||||
|
pub fn to_string(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::Clear => format!("Could not cleanup memory index"),
|
||||||
|
Self::NotFound(key) => {
|
||||||
|
format!("Record `{key}` not found in memory index")
|
||||||
|
}
|
||||||
|
Self::Overwrite(key) => {
|
||||||
|
format!("Overwrite attempt for existing record `{key}`")
|
||||||
|
}
|
||||||
|
Self::Unexpected => format!("Unexpected error"),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user