import ssl
import discord
from discord import app_commands
import os, sys
import aiohttp
from datetime import datetime
from discord.ext import tasks
from player import *

#Скрыть сообщение если надо ephemeral=True


class NeedDiscordAuthOfSteam(Exception):
	pass


class DiscordClient(discord.Client):
	ANY_INPUT = "ссылка на стим | имя игрока | стимид"
	discord2steam_cache = {}
	backend_url = ""
	secret_key = ""
	stats = {}
	show_stats_prev = 0

	def __init__(self, backend_url, secret_key):
		self.backend_url = backend_url
		self.secret_key = secret_key
		###################################################
		super().__init__(intents=discord.Intents.default())
		self.tree = app_commands.CommandTree(self)
		self.load_extensions(['user_ext', 'admin_ext', 'other_ext'])

	def load_extensions(self, extensions_path):
		if type(extensions_path) == str:
			extensions_path = [extensions_path]
		for path in extensions_path:
			print(f"Load extensions from: {path}")
			sys.path.insert(0, path)
			for extension in os.listdir(path):
				extension, ext = os.path.splitext(extension)
				if ext != ".py":
					continue
				print(f"Loading: {extension}")
				__import__(extension).Extension(self)
			sys.path.pop(0)

	async def setup_hook(self):
		self.stats = await self.GetStats()
		print("sync tree")
		if os.getenv("MAIN_DISCORD_SERVER_ID",""):
			await self.tree.sync(guild=discord.Object(int(os.getenv("MAIN_DISCORD_SERVER_ID"))))
		await self.tree.sync()

	def setup_events(self):
		@self.event
		async def on_ready():
			print(f'Logged in as {self.user} (ID: {self.user.id})')

	async def GetSteam64OfDiscord(self, user, no_cache = False):
		if user.id in self.discord2steam_cache and not no_cache:
			return self.discord2steam_cache[user.id]
		
		async with aiohttp.ClientSession(cookies={"secretkey":self.secret_key}) as session:
			async with session.get(f"{self.backend_url}/api/discord?discord_id={user.id}", ssl=False) as response:
				steamid_response = await response.json()
				if steamid_response != None:
					self.discord2steam_cache[user.id] = steamid_response["steam64"]
				else:
					raise NeedDiscordAuthOfSteam
		
		return self.discord2steam_cache[user.id]
		
	async def GetPlayer(self, profile, requester_steam64, load_profile = True):
		player = Player(profile, requester_steam64, self.stats)
		await player.GetSteamID()
		if load_profile:
			await player.LoadProfile()
		return player

	async def GetStats(self):
		async with aiohttp.ClientSession() as session:
			async with session.get(f"{os.getenv('BACKEND_URL')}/api/stats", ssl=False) as response:
				return await response.json()

if __name__ == "__main__":
	DiscordClient(
		os.getenv("BACKEND_URL"),
		os.getenv("BACKEND_SECRETKEY")
	).run(os.getenv("DISCORD_TOKEN"))