105 lines
2.5 KiB
Python
105 lines
2.5 KiB
Python
# Copyright 2019 miruka
|
|
# This file is part of harmonyqml, licensed under LGPLv3.
|
|
|
|
import asyncio
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
import aiofiles
|
|
from dataclasses import dataclass, field
|
|
|
|
from .backend import Backend
|
|
|
|
JsonData = Dict[str, Any]
|
|
|
|
WRITE_LOCK = asyncio.Lock()
|
|
|
|
|
|
@dataclass
|
|
class ConfigFile:
|
|
backend: Backend = field(repr=False)
|
|
filename: str = field()
|
|
use_data_dir: bool = False
|
|
|
|
@property
|
|
def path(self) -> Path:
|
|
# pylint: disable=no-member
|
|
dirs = self.backend.app.appdirs
|
|
to = dirs.user_data_dir if self.use_data_dir else dirs.user_config_dir
|
|
return Path(to) / self.filename
|
|
|
|
|
|
@dataclass
|
|
class JSONConfigFile(ConfigFile):
|
|
async def default_data(self) -> JsonData:
|
|
return {}
|
|
|
|
|
|
async def read(self) -> JsonData:
|
|
try:
|
|
return json.loads(self.path.read_text())
|
|
except (json.JSONDecodeError, FileNotFoundError):
|
|
return await self.default_data()
|
|
|
|
|
|
async def write(self, data: JsonData) -> None:
|
|
js = json.dumps(data, indent=4, ensure_ascii=False, sort_keys=True)
|
|
|
|
async with WRITE_LOCK:
|
|
self.path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
async with aiofiles.open(self.path, "w") as new:
|
|
await new.write(js)
|
|
|
|
|
|
@dataclass
|
|
class Accounts(JSONConfigFile):
|
|
filename: str = "accounts.json"
|
|
|
|
async def any_saved(self) -> bool:
|
|
return bool(await self.read())
|
|
|
|
|
|
async def add(self, user_id: str) -> None:
|
|
# pylint: disable=no-member
|
|
client = self.backend.clients[user_id]
|
|
|
|
await self.write({
|
|
**await self.read(),
|
|
client.user_id: {
|
|
"homeserver": client.homeserver,
|
|
"token": client.access_token,
|
|
"device_id": client.device_id,
|
|
}
|
|
})
|
|
|
|
|
|
async def delete(self, user_id: str) -> None:
|
|
await self.write({
|
|
uid: info
|
|
for uid, info in (await self.read()).items() if uid != user_id
|
|
})
|
|
|
|
|
|
@dataclass
|
|
class UISettings(JSONConfigFile):
|
|
filename: str = "ui-settings.json"
|
|
|
|
async def default_data(self) -> JsonData:
|
|
return {
|
|
"writeAliases": {}
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class UIState(JSONConfigFile):
|
|
filename: str = "ui-state.json"
|
|
use_data_dir: bool = True
|
|
|
|
async def default_data(self) -> JsonData:
|
|
return {
|
|
"page": "Pages/Default.qml",
|
|
"pageProperties": {},
|
|
}
|