from dotenv import load_dotenv
from os import getenv, listdir

from interactions import Client, Status
from pymongo import MongoClient  # Used for database management

from gwendolyn.utils import log
from gwendolyn.exceptions import NoToken, CannotConnectToService
from gwendolyn.funcs import Other, BetterNetflix, Sonarr, Radarr, TMDb, QBittorrent, Money, Games

class Gwendolyn(Client):
  def __init__(self):
    """Initialize the bot."""
    initiation_parameters = {
      "status": Status.DND,
      "delete_unused_application_cmds": True,
      "sync_interactions": True
    }
    super().__init__(**initiation_parameters)

    self._add_clients_and_options()
    self._add_functions()
    self._add_extensions()

  def _add_clients_and_options(self):
    """Add all the client, option and credentials objects."""
    load_dotenv()
    self.testing = getenv("GWENDOLYN_TESTING", "False").lower() in ('true', '1', 't')
    self.enable_nework_services = getenv("NETWORK_SERVICES", "False").lower() in ('true', '1', 't')

    self.bot_token = getenv("DISCORD_TOKEN")

    if self.bot_token == "TOKEN":
      raise NoToken()

    self.admins = getenv("ADMINS").split(",")

    mongo_user = getenv("MONGODB_USER")
    mongo_password = getenv("MONGODB_PASSWORD")
    mongo_url = f"mongodb+srv://{mongo_user}:{mongo_password}@gwendolyn"
    mongo_url += ".qkwfy.mongodb.net/Gwendolyn?retryWrites=true&w=majority"
    database_client = MongoClient(mongo_url)

    try:
      database_client.admin.command("ping")
      self.log("Connected to Mango Client")
    except:
      raise CannotConnectToService("Mango Client")

    if self.testing:
      self.log("Testing mode")
      self.database = database_client["Gwendolyn-Test"]
    else:
      self.database = database_client["Gwendolyn"]

  def _add_functions(self):
    self.other = Other(self)
    self.money = Money(self)
    if self.enable_nework_services:
      self.better_netflix = BetterNetflix(
        self,
        radarr=Radarr(getenv("RADARR_IP"),getenv("RADARR_PORT"),getenv("RADARR_API_KEY")),
        sonarr=Sonarr(getenv("SONARR_IP"),getenv("SONARR_PORT"),getenv("SONARR_API_KEY")),
        tmdb=TMDb(getenv("TMDB_API_ACCESS_TOKEN")),
        qbittorrent=QBittorrent(
          getenv("QBITTORRENT_IP"),
          getenv("QBITTORRENT_PORT"),
          getenv("QBITTORRENT_USERNAME"),
          getenv("QBITTORRENT_PASSWORD")
        )
      )

  def _add_extensions(self):
    """Load cogs."""
    for filename in listdir("./gwendolyn/ext"):
      if filename.endswith(".py"):
        if filename == "better_netflix.py" and not self.enable_nework_services:
          continue

        self.load_extension(f"gwendolyn.ext.{filename[:-3]}")

  def log(self, messages, channel: str = "", level: int = 20): # pylint:disable=no-self-use
    """Log a message. Described in utils/util_functions.py."""
    log(messages, channel, level)

  def start(self):
    super().start(self.bot_token)