111 lines
3.7 KiB
Python
111 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
_CREDENTIALS_DIR = ".yuxi"
|
|
_CREDENTIALS_SUBDIR = "credentials"
|
|
_CREDENTIALS_FILE = "matrix_credentials.json"
|
|
|
|
|
|
@dataclass
|
|
class MatrixCredentials:
|
|
user_id: str = ""
|
|
homeserver: str = ""
|
|
access_token: str = ""
|
|
device_id: str = ""
|
|
display_name: str = ""
|
|
|
|
def to_dict(self) -> dict[str, str]:
|
|
return {
|
|
"user_id": self.user_id,
|
|
"homeserver": self.homeserver,
|
|
"access_token": self.access_token,
|
|
"device_id": self.device_id,
|
|
"display_name": self.display_name,
|
|
}
|
|
|
|
@classmethod
|
|
def from_dict(cls, data: dict[str, Any]) -> MatrixCredentials:
|
|
return cls(
|
|
user_id=data.get("user_id", ""),
|
|
homeserver=data.get("homeserver", ""),
|
|
access_token=data.get("access_token", ""),
|
|
device_id=data.get("device_id", ""),
|
|
display_name=data.get("display_name", ""),
|
|
)
|
|
|
|
@property
|
|
def is_valid(self) -> bool:
|
|
return bool(self.user_id and self.homeserver and self.access_token)
|
|
|
|
|
|
class CredentialStore:
|
|
def __init__(self, base_dir: str | None = None):
|
|
if base_dir:
|
|
self._base_dir = base_dir
|
|
else:
|
|
home = os.path.expanduser("~")
|
|
self._base_dir = os.path.join(home, _CREDENTIALS_DIR)
|
|
self._store_dir = os.path.join(self._base_dir, _CREDENTIALS_SUBDIR, "matrix")
|
|
|
|
def _get_path(self, account_id: str) -> str:
|
|
safe_id = account_id.replace("@", "_").replace(":", "_").replace("/", "_")
|
|
return os.path.join(self._store_dir, f"{safe_id}.json")
|
|
|
|
def save(self, credentials: MatrixCredentials) -> bool:
|
|
os.makedirs(self._store_dir, exist_ok=True)
|
|
filepath = self._get_path(credentials.user_id or "default")
|
|
try:
|
|
with open(filepath, "w", encoding="utf-8") as f:
|
|
json.dump(credentials.to_dict(), f, indent=2)
|
|
logger.info(f"Matrix credentials saved: {filepath}")
|
|
return True
|
|
except OSError as e:
|
|
logger.error(f"Matrix credentials save failed: {e}")
|
|
return False
|
|
|
|
def load(self, account_id: str = "default") -> MatrixCredentials | None:
|
|
filepath = self._get_path(account_id)
|
|
if not os.path.exists(filepath):
|
|
return None
|
|
try:
|
|
with open(filepath, encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
return MatrixCredentials.from_dict(data)
|
|
except (json.JSONDecodeError, OSError) as e:
|
|
logger.error(f"Matrix credentials load failed: {e}")
|
|
return None
|
|
|
|
def delete(self, account_id: str) -> bool:
|
|
filepath = self._get_path(account_id)
|
|
try:
|
|
if os.path.exists(filepath):
|
|
os.remove(filepath)
|
|
logger.info(f"Matrix credentials deleted: {filepath}")
|
|
return True
|
|
except OSError as e:
|
|
logger.error(f"Matrix credentials delete failed: {e}")
|
|
return False
|
|
|
|
def list_accounts(self) -> list[str]:
|
|
if not os.path.isdir(self._store_dir):
|
|
return []
|
|
accounts = []
|
|
for filename in os.listdir(self._store_dir):
|
|
if filename.endswith(".json"):
|
|
accounts.append(filename.rsplit(".", 1)[0])
|
|
return accounts
|
|
|
|
def load_all(self) -> list[MatrixCredentials]:
|
|
accounts = []
|
|
for account_id in self.list_accounts():
|
|
creds = self.load(account_id)
|
|
if creds and creds.is_valid:
|
|
accounts.append(creds)
|
|
return accounts
|