Better stuff

src/account.rs

@@ -0,0 +1,56 @@
+use std::{collections::HashMap, path::PathBuf};
+
+use crate::executor::pump_fun::PumpDevAccount;
+use anyhow::{Context, anyhow};
+use serde::{Deserialize, Serialize};
+
+pub fn get_accounts_path() -> anyhow::Result<PathBuf> {
+    Ok(std::env::home_dir()
+        .context("Failed to get home dir")?
+        .join(".config")
+        .join("sol-hunter")
+        .join("accounts.json"))
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+#[serde(tag = "type")]
+pub enum Account {
+    PumpDev(PumpDevAccount),
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct AccountManger {
+    pub active: String,
+    pub accounts: HashMap<String, Account>,
+}
+
+impl AccountManger {
+    pub async fn get() -> anyhow::Result<Self> {
+        let path = get_accounts_path()?;
+
+        if !path.exists() {
+            let default = Self::new().await?;
+
+            tokio::fs::create_dir_all(path.parent().unwrap()).await?;
+            tokio::fs::write(&path, default.to_string()?).await?;
+
+            return Ok(default);
+        }
+
+        let output = tokio::fs::read_to_string(path).await?;
+
+        Self::from_str(&output)
+    }
+
+    pub async fn new() -> anyhow::Result<Self> {
+        Err(anyhow!("Not implemented yet"))
+    }
+
+    pub fn from_str(s: &str) -> anyhow::Result<Self> {
+        serde_json::from_str(s).context("Failed to parse accounts")
+    }
+
+    pub fn to_string(&self) -> anyhow::Result<String> {
+        serde_json::to_string(self).context("Failed to serialize accounts")
+    }
+}