import os
import tempfile
import shutil

if os.path.exists("/tmp/submit_private"):
  shutil.rmtree("/tmp/submit_private")

with tempfile.TemporaryDirectory() as tempdir:
  config_path = os.path.join(tempdir, "config.json")
  
  assert not os.system(f"ssh-keygen -f {tempdir}/ssh-key-ecdsa -t ecdsa -b 521 -q -N \"\"")

  config = """
  model: whisper
  model_key: whisper
  config.json:
      name: 'faster-whisper-server:latest'
      support_devices:
      - cpu
      model_path: ''
      port: 8080
      other_ports: []
      other_ports_count: 1
      entrypoint: start.bat
      MIN_CHUNK: 2.5
      MIN_ADD_CHUNK: 2.5
      COMPUTE_TYPE: int8
      NUM_WORKERS: 1
      CPU_THREADS: 2
      BEAM_SIZE: 5
      BATCH: 1
      LANG: auto
      DEVICE: cpu
      CHUNK_LENGTH: 5
      CLASS_MODEL: ./models/faster-whisper-base
      EN_MODEL: ./models/faster-whisper-base
      ZH_MODEL: ./models/faster-whisper-base
      RU_MODEL: ./models/faster-whisper-base
      PT_MODEL: ./models/faster-whisper-base
      AR_MODEL: ./models/faster-whisper-base
      NEW_VERSION: 1
      NEED_RESET: 0
  leaderboard_options:
    nfs:
      - name: whisper
        srcRelativePath: leaderboard/pc_asr/en.tar.gz
        mountPoint: /tmp
        source: ceph_customer
  """

  with open(config_path, "w") as f:
    f.write(config)

  os.environ["SSH_KEY_DIR"] = tempdir
  os.environ["SUBMIT_CONFIG_FILEPATH"] = config_path
  os.environ["MODEL_MAPPING"] = '{"whisper": "edge-ml.tar.gz"}'

  from run_async_a10 import get_sut_url_windows


  print(get_sut_url_windows())
  
  import time
  time.sleep(3600)