* Moving Keystore related methods into its own module

+ VideoStore Trait designed out
+ comon error types yey
This commit is contained in:
shockrah 2021-09-27 15:51:19 -07:00
parent ae3dd475c1
commit 32bf68361d
4 changed files with 97 additions and 63 deletions

4
dblib/src/err.rs Normal file
View File

@ -0,0 +1,4 @@
use std::error::Error;
pub type Result<T> = ::std::result::Result<T, Box<dyn Error>>;

67
dblib/src/keystore.rs Normal file
View File

@ -0,0 +1,67 @@
use std::collections::HashMap;
use std::fs::File;
use std::io::BufReader;
use std::io::BufWriter;
use crate::{err, DB, dev_urandom, KeyStore};
use crate::ApiKey;
impl KeyStore for DB {
fn new_entry(filename: String, name: String) -> err::Result<String> {
//! Create a new API key entry in the data-store given by <filename>
//! Key value is generated via /dev/urandom on unix systems
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let mut data: DB = serde_json::from_reader(reader)?;
// Generate some new key data
let key = dev_urandom().unwrap(); // create a new key
let entry = ApiKey { key: key.clone(), name };
// Finally write back the data to the structure and disk
data.keys.push(entry);
let file = File::open(filename.as_str())?;
let writer = BufWriter::new(file);
serde_json::to_writer(writer, &data)?;
Ok(key)
}
fn remove_entry(filename: String, name: String) -> err::Result<bool> {
//! Removes the API key entry based on the given name
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let mut data: DB = serde_json::from_reader(reader)?;
let idx = data.keys.iter().position(|item| item.name == name);
if let Some(idx) = idx {
data.keys.remove(idx);
Ok(true)
} else {
Ok(false)
}
}
fn get_entry(filename: String, name: String) -> err::Result<Option<String>> {
//! Get the API key based on name
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let data: DB = serde_json::from_reader(reader)?;
// WARNING: extreme autism
match data.keys.iter().find(|item| item.name == name) {
Some(item) => Ok(Some(item.key.clone())),
None => Ok(None)
}
}
fn new_store(filename: String) -> err::Result<()> {
//! Generate a new empty data store with the given filename
//! NOTE: The filename must include the extension(programmatically)
let file = File::create(filename)?;
let empty = DB { keys: vec![] , videos: HashMap::new() };
let writer = BufWriter::new(file);
serde_json::to_writer(writer, &empty)?;
Ok(())
}
}

View File

@ -1,12 +1,12 @@
use std::error::Error;
use std::collections::HashMap;
use std::fs::File;
use std::io::BufReader;
use std::io::BufWriter;
use std::io::prelude::*;
use serde::{Serialize, Deserialize};
pub mod request;
pub mod keystore;
pub mod err;
use request::ApiKey;
fn dev_urandom() -> std::io::Result<String> {
@ -23,69 +23,31 @@ fn dev_urandom() -> std::io::Result<String> {
return Ok(output);
}
#[derive(Serialize, Deserialize)]
pub struct VideoMeta {
pub id: u64,
pub string: String,
pub desc: Option<String>
}
#[derive(Serialize, Deserialize)]
pub struct DB {
keys: Vec<ApiKey>
keys: Vec<ApiKey>,
videos: HashMap<String, VideoMeta>
}
impl DB {
pub fn new_entry(filename: String, name: String) -> Result<String, Box<dyn Error>> {
//! Create a new API key entry in the data-store given by <filename>
//! Key value is generated via /dev/urandom on unix systems
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let mut data: DB = serde_json::from_reader(reader)?;
// Generate some new key data
let key = dev_urandom().unwrap(); // create a new key
let entry = ApiKey { key: key.clone(), name };
// Finally write back the data to the structure and disk
data.keys.push(entry);
let file = File::open(filename.as_str())?;
let writer = BufWriter::new(file);
serde_json::to_writer(writer, &data)?;
Ok(key)
}
pub fn remove_entry(filename: String, name: String) -> Result<bool, Box<dyn Error>> {
//! Removes the API key entry based on the given name
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let mut data: DB = serde_json::from_reader(reader)?;
let idx = data.keys.iter().position(|item| item.name == name);
if let Some(idx) = idx {
data.keys.remove(idx);
Ok(true)
} else {
Ok(false)
}
}
pub fn get_entry(filename: String, name: String) -> Result<Option<String>, Box<dyn Error>> {
//! Get the API key based on name
let file = File::open(filename.as_str())?;
let reader = BufReader::new(file);
let data: DB = serde_json::from_reader(reader)?;
// WARNING: extreme autism
match data.keys.iter().find(|item| item.name == name) {
Some(item) => Ok(Some(item.key.clone())),
None => Ok(None)
}
}
pub fn new_store(filename: String) -> Result<(), Box<dyn Error>> {
//! Generate a new empty data store with the given filename
//! NOTE: The filename must include the extension(programmatically)
let file = File::create(filename)?;
let empty = DB { keys: vec![] };
let writer = BufWriter::new(file);
serde_json::to_writer(writer, &empty)?;
Ok(())
}
// TODO: add some proper lifetime management here and some docs
pub trait KeyStore {
fn new_entry(filename: String, name: String) -> err::Result<String>;
fn remove_entry(filename: String, name: String) -> err::Result<bool>;
fn get_entry(filename: String, name: String) -> err::Result<Option<String>>;
fn new_store(filename: String) -> err::Result<()>;
}
pub trait VideoStore {
fn new_video(name: &str, desc: &str) -> err::Result<()>;
fn del_video(id: u64) -> err::Result<()>;
fn get_video(id: Option<u64>, name: Option<&str>) -> err::Result<VideoMeta>;
fn rename_video(id: u64, new: &str) -> err::Result<()>;
fn redescribe_video(id: u64, new: &str) -> err::Result<()>;
}

View File

@ -1,5 +1,6 @@
use std::env;
use crate::DB;
use crate::KeyStore;
use rocket::request::{self, Request, FromRequest};
use rocket::http::Status;