#!/usr/bin/env python3

from subprocess import run
from time import sleep


while True:
  try:
    from pyotp import TOTP;break
  except:
    run("pip3 install pyotp", shell=True)

editor="geany"
dctr_of_ks={ "tradeogre":"8XM7TRQLZNUK0K5QJZGOPAMYVN"}

    
def main():
  choose()
 

def choose():
  while True:
    ch=input(f"""What is that you want to do? Specify with a number.
1. Display a TOTP code for a service of your choice.
2. Edit this program, for example to add a service requiring TOTP or to change a text editor.\n""")
    if ch=='1':display_current_code(display_menu(dctr_of_ks));break
    elif ch=='2':
      run(f"nohup {editor} {__file__}>/dev/null 2>&1 &", shell=True);break

def display_menu(dctr_of_ks):
  while True:
    print("Select a service:")
    for index, key in enumerate(dctr_of_ks.keys(), start=1):
      print(f"{index}. {key}")
    choice = input("Select the the service with a number of your choice or press Q to quit) ")
    if choice.lower() == 'q':
      print("Exiting...");break
    try:
      index = int(choice)
      if 1 <= index <= len(dctr_of_ks):
        selected_key = list(dctr_of_ks.keys())[index - 1]
        print(f"You selected '{selected_key}'.")
        return dctr_of_ks[selected_key]
      else:
        print("Invalid choice. Please enter a valid number.")
    except ValueError:print("Invalid input. Please enter a number or Q to quit.")
  
def select_name_of_service():
  for k in dctr_of_ks:
    print(k)
  choice=input
  
def display_current_code(key):
  totp = TOTP(key)
  while True:
    print(totp.now())
    sleep(5)
 

if __name__ == "__main__":
  main()