* Moving Keystore related methods into its own module
+ VideoStore Trait designed out + comon error types yey
This commit is contained in:
parent
ae3dd475c1
commit
32bf68361d
4
dblib/src/err.rs
Normal file
4
dblib/src/err.rs
Normal 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
67
dblib/src/keystore.rs
Normal 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(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -1,12 +1,12 @@
|
|||||||
use std::error::Error;
|
use std::collections::HashMap;
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
use std::io::BufReader;
|
|
||||||
use std::io::BufWriter;
|
|
||||||
use std::io::prelude::*;
|
use std::io::prelude::*;
|
||||||
|
|
||||||
use serde::{Serialize, Deserialize};
|
use serde::{Serialize, Deserialize};
|
||||||
|
|
||||||
pub mod request;
|
pub mod request;
|
||||||
|
pub mod keystore;
|
||||||
|
pub mod err;
|
||||||
use request::ApiKey;
|
use request::ApiKey;
|
||||||
|
|
||||||
fn dev_urandom() -> std::io::Result<String> {
|
fn dev_urandom() -> std::io::Result<String> {
|
||||||
@ -23,69 +23,31 @@ fn dev_urandom() -> std::io::Result<String> {
|
|||||||
return Ok(output);
|
return Ok(output);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize)]
|
||||||
|
pub struct VideoMeta {
|
||||||
|
pub id: u64,
|
||||||
|
pub string: String,
|
||||||
|
pub desc: Option<String>
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
pub struct DB {
|
pub struct DB {
|
||||||
keys: Vec<ApiKey>
|
keys: Vec<ApiKey>,
|
||||||
|
videos: HashMap<String, VideoMeta>
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: add some proper lifetime management here and some docs
|
||||||
impl DB {
|
pub trait KeyStore {
|
||||||
pub fn new_entry(filename: String, name: String) -> Result<String, Box<dyn Error>> {
|
fn new_entry(filename: String, name: String) -> err::Result<String>;
|
||||||
//! Create a new API key entry in the data-store given by <filename>
|
fn remove_entry(filename: String, name: String) -> err::Result<bool>;
|
||||||
//! Key value is generated via /dev/urandom on unix systems
|
fn get_entry(filename: String, name: String) -> err::Result<Option<String>>;
|
||||||
let file = File::open(filename.as_str())?;
|
fn new_store(filename: String) -> err::Result<()>;
|
||||||
let reader = BufReader::new(file);
|
}
|
||||||
|
|
||||||
let mut data: DB = serde_json::from_reader(reader)?;
|
pub trait VideoStore {
|
||||||
// Generate some new key data
|
fn new_video(name: &str, desc: &str) -> err::Result<()>;
|
||||||
let key = dev_urandom().unwrap(); // create a new key
|
fn del_video(id: u64) -> err::Result<()>;
|
||||||
let entry = ApiKey { key: key.clone(), name };
|
fn get_video(id: Option<u64>, name: Option<&str>) -> err::Result<VideoMeta>;
|
||||||
|
fn rename_video(id: u64, new: &str) -> err::Result<()>;
|
||||||
// Finally write back the data to the structure and disk
|
fn redescribe_video(id: u64, new: &str) -> err::Result<()>;
|
||||||
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(())
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
use std::env;
|
use std::env;
|
||||||
use crate::DB;
|
use crate::DB;
|
||||||
|
use crate::KeyStore;
|
||||||
|
|
||||||
use rocket::request::{self, Request, FromRequest};
|
use rocket::request::{self, Request, FromRequest};
|
||||||
use rocket::http::Status;
|
use rocket::http::Status;
|
||||||
|
Loading…
Reference in New Issue
Block a user