nostr/gossip
1
0
Fork 0
mirror of https://github.com/mikedilger/gossip.git synced 2024-09-19 11:43:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Settings: added PublicKey and EncryptedPrivateKey

Browse Source
This commit is contained in:
Mike Dilger 2022-12-27 20:35:54 +13:00
parent ef7849e659
commit 2802fe01a0
1 changed files with 40 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/settings.rs
View File

@ -1,6 +1,8 @@
use crate::error::Error;
use crate::globals::GLOBALS;
use nostr_types::{EncryptedPrivateKey, PublicKey};
use serde::{Deserialize, Serialize};
use tracing::error;
pub const DEFAULT_FEED_CHUNK: u64 = 43200; // 12 hours
pub const DEFAULT_OVERLAP: u64 = 600; // 10 minutes
@ -13,8 +15,6 @@ pub const DEFAULT_MAX_RELAYS: u8 = 15;
#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct Settings {
// user_private_key
// version
pub feed_chunk: u64,
pub overlap: u64,
pub autofollow: bool,
@ -23,6 +23,8 @@ pub struct Settings {
pub view_threaded: bool,
pub num_relays_per_person: u8,
pub max_relays: u8,
pub public_key: Option<PublicKey>,
pub encrypted_private_key: Option<EncryptedPrivateKey>,
}
impl Default for Settings {
@ -36,6 +38,8 @@ impl Default for Settings {
view_threaded: DEFAULT_VIEW_THREADED,
num_relays_per_person: DEFAULT_NUM_RELAYS_PER_PERSON,
max_relays: DEFAULT_MAX_RELAYS,
public_key: None,
encrypted_private_key: None,
}
}
}
@ -73,6 +77,18 @@ impl Settings {
"max_relays" => {
settings.max_relays = row.1.parse::<u8>().unwrap_or(DEFAULT_MAX_RELAYS)
}
"encrypted_private_key" => {
settings.encrypted_private_key = Some(EncryptedPrivateKey(row.1))
}
"public_key" => {
settings.public_key = match PublicKey::try_from_hex_string(&row.1) {
Ok(pk) => Some(pk),
Err(e) => {
error!("Public key in database is invalid or corrupt: {}", e);
None
}
}
}
_ => {}
}
}
@ -110,6 +126,28 @@ impl Settings {
self.max_relays,
))?;
// Save private key identity
if let Some(ref epk) = self.encrypted_private_key {
let mut stmt = db.prepare(
"REPLACE INTO SETTINGS (key, value) VALUES ('encrypted_private_key', ?)",
)?;
stmt.execute((&epk.0,))?;
} else {
// Otherwise delete any such setting
let mut stmt = db.prepare("DELETE FROM settings WHERE key='encrypted_private_key'")?;
stmt.execute(())?;
}
// Save public key
if let Some(ref pk) = self.public_key {
let mut stmt =
db.prepare("REPLACE INTO SETTINGS (key, value) VALUES ('public_key', ?)")?;
stmt.execute((pk.as_hex_string(),))?;
} else {
// Otherwise delete any such setting
let mut stmt = db.prepare("DELETE FROM settings WHERE key='public_key'")?;
stmt.execute(())?;
}
Ok(())
}
}