import os
import sys
import shutil
from argparse import ArgumentParser
from tempfile import TemporaryDirectory

from cyan import tbhutils, tbhtypes


def main(parser: ArgumentParser) -> None:
  args = parser.parse_args()
  args.i = os.path.normpath(args.input)

  if args.output is not None:
    args.o = os.path.normpath(args.output)
    if not (args.o.endswith(".app") or args.o.endswith(".ipa")):
      print("[?] output's file extension not specified; will create ipa")
      args.o += ".ipa"
  else:
    args.o = args.i

  # this also modifies some args, like -f,
  # to ensure there are no duplicates, etc
  arg_err = tbhutils.validate_inputs(args)
  if arg_err is not None:
    parser.error(arg_err)

  INPUT_IS_IPA = True if args.i.endswith(".ipa") else False
  OUTPUT_IS_IPA = True if args.o.endswith(".ipa") else False

  with TemporaryDirectory() as tmpdir, tbhtypes.LeavingCM():
    app_path, plist_path = tbhutils.get_app(args.i, tmpdir, INPUT_IS_IPA)
    app = tbhtypes.AppBundle(app_path, plist_path)

    if app.executable.is_encrypted():
      if args.ignore_encrypted:
        print("[?] main binary is encrypted, ignoring")
      else:
        sys.exit("[!] main binary is encrypted; exiting")

    if args.f is not None:
      app.executable.inject(args.f, tmpdir)

    if args.remove_supported_devices:
      app.plist.remove_uisd()
    if args.no_watch:
      app.remove_watch_apps()
    if args.enable_documents:
      app.plist.enable_documents()
    if args.fakesign:
      app.fakesign_all()
    if args.thin:
      app.thin_all()

    # create subdirectories if necessary
    if "/" in args.o:
        os.makedirs(os.path.dirname(args.o), exist_ok=True)

    # done !
    if OUTPUT_IS_IPA:
      print(f"[*] generating ipa with compression level {args.compress}..")
      tbhutils.make_ipa(tmpdir, args.o, args.compress)
      print(f"[*] generated ipa at {args.o}")
    else:
      if os.path.isdir(args.o):
        shutil.rmtree(args.o)

      shutil.move(app_path, args.o)
      print(f"[*] generated app at {args.o}")