98b7c07171
* Update `add_license` script * run script * add `remove duplicate lines script` and run it * Revert changes `English spaces` * strip whitespaces * Revert `GPL` in files with `apache/mit license` * don't append `gpl license` in files with other lic * Don't append `gpl header` in files with other lic. * re-ran script * include c and cpp files too * remove duplicate header * rebase nit
82 lines
2.0 KiB
Rust
82 lines
2.0 KiB
Rust
// Copyright 2015-2018 Parity Technologies (UK) Ltd.
|
|
// This file is part of Parity.
|
|
|
|
// Parity is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Parity is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Parity. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
use std::path::PathBuf;
|
|
use std::{env, fs};
|
|
use rand::{Rng, OsRng};
|
|
use ethstore::accounts_dir::{KeyDirectory, RootDiskDirectory};
|
|
use ethstore::{Error, SafeAccount};
|
|
|
|
pub fn random_dir() -> PathBuf {
|
|
let mut rng = OsRng::new().unwrap();
|
|
let mut dir = env::temp_dir();
|
|
dir.push(format!("{:x}-{:x}", rng.next_u64(), rng.next_u64()));
|
|
dir
|
|
}
|
|
|
|
pub struct TransientDir {
|
|
dir: RootDiskDirectory,
|
|
path: PathBuf,
|
|
}
|
|
|
|
impl TransientDir {
|
|
pub fn create() -> Result<Self, Error> {
|
|
let path = random_dir();
|
|
let result = TransientDir {
|
|
dir: RootDiskDirectory::create(&path)?,
|
|
path: path,
|
|
};
|
|
|
|
Ok(result)
|
|
}
|
|
|
|
pub fn open() -> Self {
|
|
let path = random_dir();
|
|
TransientDir {
|
|
dir: RootDiskDirectory::at(&path),
|
|
path: path,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Drop for TransientDir {
|
|
fn drop(&mut self) {
|
|
fs::remove_dir_all(&self.path).expect("Expected to remove temp dir");
|
|
}
|
|
}
|
|
|
|
impl KeyDirectory for TransientDir {
|
|
fn load(&self) -> Result<Vec<SafeAccount>, Error> {
|
|
self.dir.load()
|
|
}
|
|
|
|
fn update(&self, account: SafeAccount) -> Result<SafeAccount, Error> {
|
|
self.dir.update(account)
|
|
}
|
|
|
|
fn insert(&self, account: SafeAccount) -> Result<SafeAccount, Error> {
|
|
self.dir.insert(account)
|
|
}
|
|
|
|
fn remove(&self, account: &SafeAccount) -> Result<(), Error> {
|
|
self.dir.remove(account)
|
|
}
|
|
|
|
fn unique_repr(&self) -> Result<u64, Error> {
|
|
self.dir.unique_repr()
|
|
}
|
|
}
|