from locale import currency
from glQiwiApi import QiwiWallet
from glQiwiApi.core.event_fetching import executor
from glQiwiApi.core.event_fetching.dispatcher import QiwiDispatcher
from glQiwiApi.core.event_fetching.executor import Context
from glQiwiApi.core.event_fetching.filters import ExceptionFilter
from glQiwiApi.qiwi.clients.wallet.types import Transaction, TransactionType, TransactionStatus
from glQiwiApi.qiwi.exceptions import QiwiAPIError

from backend_integration import BackendClient
from colors import *
import os, sys

prices = {
	20.0: 1*24*60*60,
	75.0: 7*24*60*60,
	150.0: 30*24*60*60
}

def price_checker(amount):
	seconds2give = 0
	for rub, sec in prices.items():
		if amount >= rub:
			seconds2give = sec
	return seconds2give

class Client:
	wallet: QiwiWallet
	qiwi_dp: QiwiDispatcher
	backend: BackendClient

	def __init__(self):
		if not os.getenv("BACKEND_URL", "") or not os.getenv("SECRET_KEY", ""):
			error("BACKEND_URL or SECRET_KEY is not setted!")
			sys.exit(1)
		if not os.getenv("QIWI_APIKEY", "") or not os.getenv("QIWI_PHONENUMBER", ""):
			error("QIWI_APIKEY or QIWI_PHONENUMBER in not setted!")
			sys.exit(2)

		self.backend = BackendClient()
		self.qiwi_dp = QiwiDispatcher()
		self.wallet = QiwiWallet(api_access_token=os.getenv("QIWI_APIKEY"), phone_number=os.getenv("QIWI_PHONENUMBER"))

	def run(self):
		executor.start_polling(self.wallet, self.qiwi_dp, on_startup=self.on_startup)

	async def on_startup(self, ctx: Context):
		print(await self.wallet.get_balance())
		await self.backend.pulse()

	def build_handler(self):
		@self.qiwi_dp.transaction_handler()
		async def handle_transaction(t: Transaction, ctx: Context):
			if t.type != Transaction.type == TransactionType.IN:
				return
			if t.status != TransactionStatus.SUCCESS:
				return
			########################################
			currency = t.sum.currency
			if str(currency) != "643":#643-rub code
				warning(f"[{t.id}] currency not be rub, drop")
				return
			########################################
			money = t.sum.amount
			seconds = price_checker(money)
			if seconds == 0:
				warning(f"[{t.id}] so smol donate {money} RUB")
				return
			########################################
			splitted = t.comment.split()
			steam2 = None
			for maybe_steam in splitted:
				if "STEAM_" in maybe_steam.upper():
					steam2 = maybe_steam.upper()
					break
			if steam2 is None:
				warning(f"[{t.id}] comment not contain steam {t.comment} RUB")
				return
			########################################
			# add vip
			info(f"Add vip {steam2}, amount {seconds}")
			#await self.backend.vip(steam2, seconds, f"rub={int(money)}")

if __name__ == "__main__":
	print("Build date: "+os.getenv("BUILDDATE", "not setted"))
	client = Client()
	client.run()