battle_royale_sim/bot.py

import os as _os

from utils import logs as _log

from telegram.ext import Application
from telegram.ext import CommandHandler
from telegram.ext import MessageHandler
from telegram.ext import filters
from telegram import ReplyKeyboardMarkup
from telegram import ReplyKeyboardRemove

from entities import arena as _arena
from bot_libs import player_handling as _bot_player
from bot_libs import simulation as _bot_sim
from bot_libs import repeating as _bot_repeat
from bot_libs import syms as _botsyms


async def bot_start(update, context):
  await update.message.reply_text(_botsyms.START_MSG)

  keyboard = [
      ['Init/Restart'],
      ['Add Player', 'Add random Players', 'Add random color Players'],
      ['Get Players', 'Get Alive Players', 'Get Death Players', 'Get Ranking Players',],
      ['Simulate Day', 'Run Periodically']
  ]
  reply_markup= ReplyKeyboardMarkup(keyboard, one_time_keyboard=False, resize_keyboard=True)

  chat_id = update.effective_chat.id
  _log.log_debug(f'{chat_id}: I\'m building the world\'s game...')
  Arena= _arena.BrSimArena()

  await update.message.reply_text('Ho creato il mondo di gioco', reply_markup=reply_markup)
  context.application.bot_data['arena'] = Arena

async def update_bot(update, context):
  update.message.reply_text('Sto aggiornando il Bot...')
  chat_id = update.effective_chat.id
  if update.message.chat.id not in _botsyms.SUPER_USERS: return _log.log_warning(f'update_bot: user {chat_id} not allowed')
  _log.log_debug(f'update_bot: {chat_id} bot is updating...')
  err= _os.system(f'cd {_botsyms.BOT_PATH}; git pull')
  if err:
    _log.log_error(f'update_bot: {chat_id} error {err} while trying to update the app')
    return await update.message.reply_text('Errore durante l\'aggiornamento del Bot')
  err= _os.system('systemctl --user start battle_royale.service')
  if err:
    _log.log_error(f'update_bot: {chat_id} error {err} while trying to upstart the app')
    return await update.message.reply_text('Errore durante il riavvio del Bot')
  _log.log_debug(f'update_bot: {chat_id} bot successfully updated')
  await update.message.reply_text('Bot aggiornato e riavviato!')

async def bot_commands(update, context):
  text= update.message.text
  chat_id = update.effective_chat.id

  if text == 'upstart': return await update_bot(update, context)

  if text == 'Init/Restart':
    _log.log_info(f'bot_command: {chat_id} Init/Restart')
    return await bot_start(update, context)

  if text == 'Add Player':
    _log.log_info(f'bot_command: {chat_id} Add Player')
    context.application.bot_data['ask_name'] = 1
    if 'ask_seconds' in context.application.bot_data:
      del(context.application.bot_data['ask_seconds'])
    return await update.message.reply_text('Inserisci il Nome del giocatore (o piu\' nomi separati da virgola)')
  if text == 'Get Players':
    _log.log_info(f'bot_command: {chat_id} Get Players')
    return await _bot_player.get_players(update, context)
  if text == 'Get Alive Players':
    _log.log_info(f'bot_command: {chat_id} Get Alive Players')
    return await _bot_player.get_alive_players(update, context)
  if text == 'Get Death Players':
    _log.log_info(f'bot_command: {chat_id} Get Death Players')
    return await _bot_player.get_death_players(update, context)
  if text == 'Get Ranking Players':
    _log.log_info(f'bot_command: {chat_id} Get Ranking Players')
    return await _bot_player.get_ranking_players(update, context)
  if text == 'Simulate Day':
    _log.log_info(f'bot_command: {chat_id} Simulate Day')
    return await _bot_sim.simulate_day(context, chat_id)
  if text == 'Run Periodically':
    _log.log_info(f'bot_command: {chat_id} Run Periodically')
    context.application.bot_data['ask_seconds'] = 1
    if 'ask_name' in context.application.bot_data:
      del(context.application.bot_data['ask_name'])
    return await update.message.reply_text('Inserisci il numero di secondi, ad esempio \n(60 = 1 minuto)(600 = 10 minuti)\n(3600 = 1 ora)\n(86400 = 1 giorno)')

  waiting_for_name= context.application.bot_data.get('ask_name')
  if waiting_for_name or text in ['Add random Players', 'Add random color Players']:
    _log.log_debug(f'bot_command: {chat_id} Collected Player Name {text}')
    if 'ask_name' in context.application.bot_data:
      del(context.application.bot_data['ask_name'])
      players= text.split(',')
      for player in players:
        await _bot_player.add_player(update, context, player.strip())
    elif text == 'Add random Players':
      await _bot_player.add_random_players(update, context, colors_names= False)
    elif text == 'Add random color Players':
      await _bot_player.add_random_players(update, context, colors_names= True)

    Arena= context.application.bot_data['arena']
    players= [p.get_name_and_stats() for p in Arena.get_players()]
    players_str= '\n'.join(players)
    return await update.message.reply_text(f'Ecco i {len(players)} giocatori presenti nel mondo do gioco: \n{players_str}')

  waiting_for_seconds= context.application.bot_data.get('ask_seconds')
  if waiting_for_seconds:
    _log.log_debug(f'bot_command: {chat_id} User Wants to auto-run the game every {text} seconds')
    try: text= int(text)
    except: return
    seconds= max(1, text)
    return await _bot_repeat.start_loop_game(update, context, seconds)

  _log.log_debug(f'bot_command: {chat_id} sent this text: {text}')
  await update.message.reply_text(_botsyms.WIP_MSG)

def main():
  application = Application.builder().token(_botsyms.TOKEN).build()

  application.add_handler(CommandHandler('start', bot_start))
  application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, bot_commands))

  _log.log_debug('Bot is running...')
  application.run_polling()

if __name__ == '__main__':
  main()