initial

src/dutylog/infrastructure/utils/config.py

@@ -1,15 +1,70 @@
 from dataclasses import dataclass
-from os import getenv
+from pathlib import Path
+
+import tomllib
+
+
+@dataclass
+class BotConfig:
+    token: str
+    creator_id: int
+
+
+@dataclass
+class DatabaseConfig:
+    host: str
+    port: int
+    user: str
+    password: str
+    database: str
+    
+    @property
+    def url(self) -> str:
+        return f"postgresql+asyncpg://{self.user}:{self.password}@{self.host}:{self.port}/{self.database}"
 
 
 @dataclass
 class Config:
-    bot_token: str
-    database_url: str
+    bot: BotConfig
+    database: DatabaseConfig
 
 
 def load_config() -> Config:
+    config_path = Path("config.toml")
+    
+    if not config_path.exists():
+        raise FileNotFoundError(f"Config file not found: {config_path}")
+    
+    with open(config_path, "rb") as f:
+        data = tomllib.load(f)
+    
+    if "bot" not in data:
+        raise KeyError("Missing required section: bot")
+    if "database" not in data:
+        raise KeyError("Missing required section: database")
+    
+    bot_data = data["bot"]
+    if "token" not in bot_data:
+        raise KeyError("Missing required field: bot.token")
+    if "creator_id" not in bot_data:
+        raise KeyError("Missing required field: bot.creator_id")
+    
+    database_data = data["database"]
+    required_db_fields = ["host", "port", "user", "password", "database"]
+    for field in required_db_fields:
+        if field not in database_data:
+            raise KeyError(f"Missing required field: database.{field}")
+    
     return Config(
-        bot_token=getenv("BOT_TOKEN", ""),
-        database_url=getenv("DATABASE_URL", "sqlite+aiosqlite:///./db.sqlite3"),
+        bot=BotConfig(
+            token=bot_data["token"],
+            creator_id=bot_data["creator_id"],
+        ),
+        database=DatabaseConfig(
+            host=database_data["host"],
+            port=database_data["port"],
+            user=database_data["user"],
+            password=database_data["password"],
+            database=database_data["database"],
+        ),
     )