from io import TextIOWrapper
import asyncpraw as praw
import os
from typing import Final
from macros import *
import toml


RE_DATA_POSTS: Final[str] = "posts"
CFG_DATA_RE:   Final[str] = "reddit"


class Bot:
  args:     dict = {"NO_RUST": True, "dev": True, "py": True, "port": 2920}
  r_id:     str = os.environ.get("ASSISTANT_R_ID")
  secret:   str = os.environ.get("ASSISTANT_R_TOKEN")
  username: str = os.environ.get("ASSISTANT_R_NAME")
  password: str = os.environ.get("ASSISTANT_R_PASS")

  fetch_limit = 0

  useragent: str =\
    f"{username} by u/RandomPersonDotExe aka u/Byte_Dice"\
      if r_id == "YmZjr4zLr2qtHdpQXtj0sBOOdJzrXQ"\
    else f"{username} (Original program by u/RandomPersonDotExe aka u/Byte_Dice)"

  if password is None:
    py_error("Environment variable \"ASSISTANT_R_PASS\" is null!")
  if secret is None:
    py_error("Environment variable \"ASSISTANT_R_TOKEN\" is null!")

  def __init__(self):
    self.r: praw.Reddit = praw.Reddit(
      client_id =     self.r_id,
      client_secret = self.secret,
      username =      self.username,
      password =      self.password,
      user_agent =    self.useragent
    )
    self.sr_list: list[str] = ["bytedicetesting"]
    self.sr = None
    self.data_f: TextIOWrapper = None
    self.data: dict = {}

  async def initialize(self):
    self.sr = await self.r.subreddit("+".join(self.sr_list))

  async def set_args(self, args: dict):
    self.args = args

  async def stop(self) -> bool:
    if self.r:
      await self.r.close()
      py_print("Stopped Reddit bot.")
      return True
    
    return False
  
  async def update_cfg_str(self, new_cfg: str) -> bool:
    json_cfg = toml.loads(new_cfg)
    self.sr_list = json_cfg[CFG_DATA_RE]["subreddits"].split("+")
    self.sr = await self.r.subreddit("+".join(self.sr_list))
    self.fetch_limit = json_cfg[CFG_DATA_RE]["fetch_limit"]
    return True
  
  async def update_cfg(self, new_cfg: dict) -> bool:
    self.sr_list = new_cfg[CFG_DATA_RE]["subreddits"].split("+")
    self.sr = await self.r.subreddit("+".join(self.sr_list))
    self.fetch_limit = new_cfg[CFG_DATA_RE]["fetch_limit"]
    return True