import telebot
from call2api import *

bot = telebot.TeleBot('YOUR_BOT_TOKEN')

@bot.message_handler(commands=['help'])
def send_help(message):
    help_text = '''
    ```
/set_creds <nick> <token> - Установка ника, токена.
/check_token              - Проверка валидности токена.
/check_bal                - Проверка баланса пользователя.
/get_time2cdm             - Получения времени CDM.
/get_stat [date]          - Получение статистики. Опционально: указание даты.
/transfer_coins <dst_nick> <amount> - Перевести CDM другому пользователю.
/gen_fp <amount>          - Генерация чека для оплаты.
/list_fp                  - Список чеков.
/del_fp <fp_id>           - Удалить чек.
```'''
    bot.reply_to(message, help_text, parse_mode = 'Markdown')

def parse_command_data(message, num_params, set_creds=False):
    try:
        if creds_setted() or set_creds:
            command_parts = message.text.split()
            if len(command_parts) < num_params + 1:
                bot.reply_to(message, 'Недопустимые параметры.')
                raise ValueError
            if num_params == 1:
                return command_parts[1]
            return command_parts[1:num_params + 1]
        else:
            bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')
    except ValueError:
        bot.reply_to(message, 'Недопустимые параметры')
        return None

# Задание ника и токена для удобства
global NICK, TOKEN
NICK = None
TOKEN = None
def creds_setted():
    global NICK, TOKEN
    if NICK and TOKEN:
        return True
    else:
        return False
def get_nick():
    global NICK
    return NICK
def get_token():
    global TOKEN
    return TOKEN


@bot.message_handler(commands=['set_creds'])
def set_creds(message):
    params = parse_command_data(message, 2, set_creds=True)
    if params:
        nick, token = params
        global NICK, TOKEN
        NICK = nick; TOKEN = token
        bot.reply_to(message, f'Данные установлены')

@bot.message_handler(commands=['check_token'])
def handle_check_token(message):
    if creds_setted():
        response = check_token(get_nick(), get_token())
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')
    else:
        bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')

@bot.message_handler(commands=['check_bal'])
def handle_check_bal(message):
    if creds_setted():
        response = check_bal(get_nick(), get_token())
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')
    else:
        bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')

@bot.message_handler(commands=['get_time2cdm'])
def handle_get_time2cdm(message):
    if creds_setted():
        response = get_time2cdm(get_nick(), get_token())
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')
    else:
        bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')

@bot.message_handler(commands=['get_stat'])
def handle_get_stat(message):
    if creds_setted():
        date = message.text.split()[1] if len(message.text.split()) > 1 else None
        response = get_stat(get_nick(), get_token(), date)
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')
    else:
        bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')

@bot.message_handler(commands=['transfer_coins'])
def handle_transfer_coins(message):
    params = parse_command_data(message, 2)
    if params:
        dst_nick, amount = params
        response = transfer_coins(get_nick(), get_token(), dst_nick, amount)
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')

@bot.message_handler(commands=['gen_fp'])
def handle_gen_fp(message):
    params = parse_command_data(message, 1)
    if params:
        amount = params
        response = gen_fp(get_nick(), get_token(), str(amount), str(message.from_user.id))
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')

@bot.message_handler(commands=['list_fp'])
def handle_list_fp(message):
    if creds_setted():
        response = list_fp(get_nick(), get_token())
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')
    else:
        bot.reply_to(message, f'`/set_creds <nick> <token> - Установка ника, токена.`', parse_mode='Markdown')

@bot.message_handler(commands=['del_fp'])
def handle_del_fp(message):
    params = parse_command_data(message, 1)
    if params:
        fp_id = params
        response = del_fp(get_nick(), get_token(), fp_id)
        bot.reply_to(message, f'Результат: ```\n{response}```', parse_mode='Markdown')


bot.polling()