NijiHolo_EN_ID_Bot/src/main.py

import sys
import asyncio
import argparse
from argparse import RawTextHelpFormatter
import code

import nest_asyncio

import talent_lists
import ttweetqueue as ttq
import api_secrets
import catchup
import listen
from twapi import TwAPI

PROGRAM_ARGS = None

MODES_HELP_STR = '''mode to run the bot at:
l,listen:      listen for new tweets from all accounts; will not terminate unless error occurs
c,catchup:     scan all tweets from all accounts; will terminate when done
d,delete-all:  delete all tweets on account provided by secrets.ini; make sure the function is uncommented in twapi.py'''

def init_argparse():
    p = argparse.ArgumentParser(description='Twitter bot that follows interactions between Nijisanji EN/ID and hololive EN/ID members.', formatter_class=RawTextHelpFormatter)
    p.add_argument('mode', nargs='?', \
        help=MODES_HELP_STR)
    p.add_argument('--show-tokens', action='store_true', help='[DO NOT USE IN PUBLIC SETTING] print stored tokens from secrets.ini')
    p.add_argument('--announce-catchup', action='store_true', help='In catch-up mode, post a tweet announcing catch-up mode.')
    p.add_argument('--auto-listen', action='store_true', help='In catch-up mode, transition to listen mode after successfuly catching up.')
    p.add_argument('--no-delay', action='store_true', help='In self-destruct mode, clear tweets without safety waiting.')
    return p

def command_line():
    # TODO (extra): implement command line mode for manually controlling the bot
    print('Shell coming soon. For now, here\'s a Python interpretor.')
    code.interact(local=globals())
    pass

async def self_destruct():
    if not PROGRAM_ARGS.no_delay:
        print('\033[31;6m-----DELETING ALL TWEETS IN 10 SECONDS!! PRESS CTRL+C TO CANCEL.-----\033[0m')
        await asyncio.sleep(10)
    await TwAPI.instance.nuke_tweets()

async def async_main():
    global PROGRAM_ARGS

    ## Determine running mode
    # match PROGRAM_ARGS.mode.lower():
        # case 'l' | 'listen':
        #     print('RUNNING IN LISTEN MODE\n')
        #     await listen.run()
        # case 'c' | 'catchup':
        #     print('RUNNING IN CATCH-UP MODE\n')
        #     if await catchup.run(PROGRAM_ARGS) and PROGRAM_ARGS.auto_listen:
        #         print('CATCH-UP MODE DONE, GOING INTO LISTEN MODE')
        #         await listen.run()
        # case 'd' | 'delete-all':
        #     print('WARNING: SELF-DESTRUCT MODE')
        #     await self_destruct()
        # case 'cmd':
        #     command_line()
        # case _:
        #     print('\ninvalid mode. run with no arguments or "-h" for help page, including mode list.')
        #     return
    mode = PROGRAM_ARGS.mode.lower()
    if mode in ['l', 'listen']:
        print('RUNNING IN LISTEN MODE')
        await listen.run()
    elif mode in ['c', 'catchup']:
        print('RUNNING IN CATCH UP MODE')
        if await catchup.run(PROGRAM_ARGS) and PROGRAM_ARGS.auto_listen:
            print('CATCH UP MODE DONE, GOING INTO LISTEN MODE')
            listen.run()
    elif mode in ['d', 'delete-all']:
        print('WARNING: SELF-DESTRUCT MODE')
        await self_destruct()
    elif mode == 'cmd':
        command_line()
    else:
        print('\ninvalid mode. run with no arguments or -h for help and modes')

def main():
    global PROGRAM_ARGS

    parser = init_argparse()
    if len(sys.argv) < 2:
        parser.print_help()
        return

    PROGRAM_ARGS = parser.parse_args()

    if PROGRAM_ARGS.show_tokens:
        print(api_secrets.get_all_secrets())

    if PROGRAM_ARGS.mode is None: return

    ## We expect to run in some mode now.

    # Initialize shared API instance
    TwAPI()

    # Initialize talent account lists
    talent_lists.init()

    # Initialize queue files system
    ttq.TalentTweetQueue()

    ## Asynchronous execution
    nest_asyncio.apply()
    asyncio.run(async_main())
    

if __name__ == "__main__":
    main()