fixes

src/accounts/manager.rs

@@ -0,0 +1,209 @@
+//! Account manager: loading, saving, migration, and resolution.
+//!
+//! Handles `accounts.toml` lifecycle and legacy `./tdlib_data/` migration
+//! to XDG data directory.
+
+use std::fs;
+use std::path::PathBuf;
+
+use super::profile::{account_db_path, validate_account_name, AccountsConfig};
+
+/// Returns the path to `accounts.toml` in the config directory.
+///
+/// `~/.config/tele-tui/accounts.toml`
+pub fn accounts_config_path() -> Option<PathBuf> {
+    dirs::config_dir().map(|mut path| {
+        path.push("tele-tui");
+        path.push("accounts.toml");
+        path
+    })
+}
+
+/// Loads `accounts.toml` or creates it with default values.
+///
+/// On first run, also attempts to migrate legacy `./tdlib_data/` directory
+/// to the XDG data location.
+pub fn load_or_create() -> AccountsConfig {
+    let config_path = match accounts_config_path() {
+        Some(path) => path,
+        None => {
+            tracing::warn!("Could not determine config directory for accounts, using defaults");
+            return AccountsConfig::default_single();
+        }
+    };
+
+    if config_path.exists() {
+        // Load existing config
+        match fs::read_to_string(&config_path) {
+            Ok(content) => match toml::from_str::<AccountsConfig>(&content) {
+                Ok(config) => return config,
+                Err(e) => {
+                    tracing::warn!("Could not parse accounts.toml: {}", e);
+                    return AccountsConfig::default_single();
+                }
+            },
+            Err(e) => {
+                tracing::warn!("Could not read accounts.toml: {}", e);
+                return AccountsConfig::default_single();
+            }
+        }
+    }
+
+    // First run: migrate legacy data if present, then create default config
+    migrate_legacy();
+
+    let config = AccountsConfig::default_single();
+    if let Err(e) = save(&config) {
+        tracing::warn!("Could not save initial accounts.toml: {}", e);
+    }
+    config
+}
+
+/// Saves `AccountsConfig` to `accounts.toml`.
+pub fn save(config: &AccountsConfig) -> Result<(), String> {
+    let config_path = accounts_config_path()
+        .ok_or_else(|| "Could not determine config directory".to_string())?;
+
+    // Ensure parent directory exists
+    if let Some(parent) = config_path.parent() {
+        fs::create_dir_all(parent)
+            .map_err(|e| format!("Could not create config directory: {}", e))?;
+    }
+
+    let toml_string = toml::to_string_pretty(config)
+        .map_err(|e| format!("Could not serialize accounts config: {}", e))?;
+
+    fs::write(&config_path, toml_string)
+        .map_err(|e| format!("Could not write accounts.toml: {}", e))?;
+
+    Ok(())
+}
+
+/// Migrates legacy `./tdlib_data/` from CWD to XDG data dir.
+///
+/// If `./tdlib_data/` exists in the current working directory, moves it to
+/// `~/.local/share/tele-tui/accounts/default/tdlib_data/`.
+fn migrate_legacy() {
+    let legacy_path = PathBuf::from("tdlib_data");
+    if !legacy_path.exists() || !legacy_path.is_dir() {
+        return;
+    }
+
+    let target = account_db_path("default");
+
+    // Don't overwrite if target already exists
+    if target.exists() {
+        tracing::info!(
+            "Legacy ./tdlib_data/ found but target already exists at {}, skipping migration",
+            target.display()
+        );
+        return;
+    }
+
+    // Create parent directories
+    if let Some(parent) = target.parent() {
+        if let Err(e) = fs::create_dir_all(parent) {
+            tracing::error!("Could not create target directory for migration: {}", e);
+            return;
+        }
+    }
+
+    // Move (rename) the directory
+    match fs::rename(&legacy_path, &target) {
+        Ok(()) => {
+            tracing::info!(
+                "Migrated ./tdlib_data/ -> {}",
+                target.display()
+            );
+        }
+        Err(e) => {
+            tracing::error!(
+                "Could not migrate ./tdlib_data/ to {}: {}",
+                target.display(),
+                e
+            );
+        }
+    }
+}
+
+/// Resolves which account to use from CLI arg or default.
+///
+/// # Arguments
+///
+/// * `config` - The loaded accounts configuration
+/// * `account_arg` - Optional account name from `--account` CLI flag
+///
+/// # Returns
+///
+/// The resolved account name and its db_path.
+///
+/// # Errors
+///
+/// Returns an error if the specified account is not found or the name is invalid.
+pub fn resolve_account(
+    config: &AccountsConfig,
+    account_arg: Option<&str>,
+) -> Result<(String, PathBuf), String> {
+    let account_name = account_arg.unwrap_or(&config.default_account);
+
+    // Validate name
+    validate_account_name(account_name)?;
+
+    // Find account in config
+    let _account = config.find_account(account_name).ok_or_else(|| {
+        let available: Vec<&str> = config.accounts.iter().map(|a| a.name.as_str()).collect();
+        format!(
+            "Account '{}' not found. Available accounts: {}",
+            account_name,
+            available.join(", ")
+        )
+    })?;
+
+    let db_path = account_db_path(account_name);
+    Ok((account_name.to_string(), db_path))
+}
+
+/// Adds a new account to `accounts.toml` and creates its data directory.
+///
+/// Validates the name, checks for duplicates, adds the profile to config,
+/// saves the config, and creates the data directory.
+///
+/// # Returns
+///
+/// The db_path for the new account.
+///
+/// # Errors
+///
+/// Returns an error if the name is invalid, already exists, or I/O fails.
+pub fn add_account(name: &str, display_name: &str) -> Result<std::path::PathBuf, String> {
+    validate_account_name(name)?;
+
+    let mut config = load_or_create();
+
+    // Check for duplicate
+    if config.find_account(name).is_some() {
+        return Err(format!("Account '{}' already exists", name));
+    }
+
+    // Add new profile
+    config.accounts.push(super::profile::AccountProfile {
+        name: name.to_string(),
+        display_name: display_name.to_string(),
+    });
+
+    // Save config
+    save(&config)?;
+
+    // Create data directory
+    ensure_account_dir(name)
+}
+
+/// Ensures the account data directory exists.
+///
+/// Creates `~/.local/share/tele-tui/accounts/{name}/tdlib_data/` if needed.
+pub fn ensure_account_dir(account_name: &str) -> Result<PathBuf, String> {
+    let db_path = account_db_path(account_name);
+    fs::create_dir_all(&db_path)
+        .map_err(|e| format!("Could not create account directory: {}", e))?;
+    Ok(db_path)
+}

src/accounts/mod.rs

@@ -0,0 +1,11 @@
+//! Account profiles module for multi-account support.
+//!
+//! Manages account profiles stored in `~/.config/tele-tui/accounts.toml`.
+//! Each account has its own TDLib database directory under
+//! `~/.local/share/tele-tui/accounts/{name}/tdlib_data/`.
+
+pub mod manager;
+pub mod profile;
+
+pub use manager::{add_account, ensure_account_dir, load_or_create, resolve_account, save};
+pub use profile::{account_db_path, validate_account_name, AccountProfile, AccountsConfig};

src/accounts/profile.rs

@@ -0,0 +1,147 @@
+//! Account profile data structures and validation.
+//!
+//! Defines `AccountProfile` and `AccountsConfig` for multi-account support.
+//! Account names are validated to contain only alphanumeric characters, hyphens, and underscores.
+
+use serde::{Deserialize, Serialize};
+use std::path::PathBuf;
+
+/// Configuration for all accounts, stored in `~/.config/tele-tui/accounts.toml`.
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct AccountsConfig {
+    /// Name of the default account to use when no `--account` flag is provided.
+    pub default_account: String,
+
+    /// List of configured accounts.
+    pub accounts: Vec<AccountProfile>,
+}
+
+/// A single account profile.
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct AccountProfile {
+    /// Unique identifier (used in directory names and CLI flag).
+    pub name: String,
+
+    /// Human-readable display name.
+    pub display_name: String,
+}
+
+impl AccountsConfig {
+    /// Creates a default config with a single "default" account.
+    pub fn default_single() -> Self {
+        Self {
+            default_account: "default".to_string(),
+            accounts: vec![AccountProfile {
+                name: "default".to_string(),
+                display_name: "Default".to_string(),
+            }],
+        }
+    }
+
+    /// Finds an account by name.
+    pub fn find_account(&self, name: &str) -> Option<&AccountProfile> {
+        self.accounts.iter().find(|a| a.name == name)
+    }
+}
+
+impl AccountProfile {
+    /// Computes the TDLib database directory path for this account.
+    ///
+    /// Returns `~/.local/share/tele-tui/accounts/{name}/tdlib_data`
+    /// (or platform equivalent via `dirs::data_dir()`).
+    pub fn db_path(&self) -> PathBuf {
+        account_db_path(&self.name)
+    }
+}
+
+/// Computes the TDLib database directory path for a given account name.
+///
+/// Returns `{data_dir}/tele-tui/accounts/{name}/tdlib_data`.
+pub fn account_db_path(account_name: &str) -> PathBuf {
+    let mut path = dirs::data_dir().unwrap_or_else(|| PathBuf::from("."));
+    path.push("tele-tui");
+    path.push("accounts");
+    path.push(account_name);
+    path.push("tdlib_data");
+    path
+}
+
+/// Validates an account name.
+///
+/// Valid names contain only lowercase alphanumeric characters, hyphens, and underscores.
+/// Must be 1-32 characters long.
+///
+/// # Errors
+///
+/// Returns a descriptive error message if the name is invalid.
+pub fn validate_account_name(name: &str) -> Result<(), String> {
+    if name.is_empty() {
+        return Err("Account name cannot be empty".to_string());
+    }
+    if name.len() > 32 {
+        return Err("Account name cannot be longer than 32 characters".to_string());
+    }
+    if !name
+        .chars()
+        .all(|c| c.is_ascii_lowercase() || c.is_ascii_digit() || c == '-' || c == '_')
+    {
+        return Err(
+            "Account name can only contain lowercase letters, digits, hyphens, and underscores"
+                .to_string(),
+        );
+    }
+    if name.starts_with('-') || name.starts_with('_') {
+        return Err("Account name cannot start with a hyphen or underscore".to_string());
+    }
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_validate_account_name_valid() {
+        assert!(validate_account_name("default").is_ok());
+        assert!(validate_account_name("work").is_ok());
+        assert!(validate_account_name("my-account").is_ok());
+        assert!(validate_account_name("account_2").is_ok());
+        assert!(validate_account_name("a").is_ok());
+    }
+
+    #[test]
+    fn test_validate_account_name_invalid() {
+        assert!(validate_account_name("").is_err());
+        assert!(validate_account_name("My Account").is_err());
+        assert!(validate_account_name("UPPER").is_err());
+        assert!(validate_account_name("with spaces").is_err());
+        assert!(validate_account_name("-starts-with-dash").is_err());
+        assert!(validate_account_name("_starts-with-underscore").is_err());
+        assert!(validate_account_name(&"a".repeat(33)).is_err());
+    }
+
+    #[test]
+    fn test_default_single_config() {
+        let config = AccountsConfig::default_single();
+        assert_eq!(config.default_account, "default");
+        assert_eq!(config.accounts.len(), 1);
+        assert_eq!(config.accounts[0].name, "default");
+    }
+
+    #[test]
+    fn test_find_account() {
+        let config = AccountsConfig::default_single();
+        assert!(config.find_account("default").is_some());
+        assert!(config.find_account("nonexistent").is_none());
+    }
+
+    #[test]
+    fn test_db_path_contains_account_name() {
+        let path = account_db_path("work");
+        let path_str = path.to_string_lossy();
+        assert!(path_str.contains("tele-tui"));
+        assert!(path_str.contains("accounts"));
+        assert!(path_str.contains("work"));
+        assert!(path_str.ends_with("tdlib_data"));
+    }
+}