# PFEIFER - MAPD - Modified by FrogAi for FrogPilot to automatically update
import os
import socket
import stat
import subprocess
import time
import urllib.error
import urllib.request

from openpilot.common.realtime import Ratekeeper

VERSION = 'v1'
GITHUB_VERSION_URL = f"https://github.com/FrogAi/FrogPilot-Resources/raw/Versions/mapd_version_{VERSION}.txt"
GITLAB_VERSION_URL = f"https://gitlab.com/FrogAi/FrogPilot-Resources/-/raw/Versions/mapd_version_{VERSION}.txt"

MAPD_PATH = '/data/media/0/osm/mapd'
VERSION_PATH = '/data/media/0/osm/mapd_version'

def ping_url(url, timeout=5):
  no_internet = 0
  while True:
    if no_internet > 5:
      break
    try:
      urllib.request.urlopen(url, timeout=timeout)
      return True
    except (urllib.error.URLError, socket.timeout):
      no_internet += 1
      time.sleep(10)
  return False

def get_latest_version():
  urls = [GITHUB_VERSION_URL, GITLAB_VERSION_URL]
  for url in urls:
    try:
      with urllib.request.urlopen(url) as response:
        return response.read().decode('utf-8').strip()
    except Exception as e:
      print(f"Failed to get latest version from {url}. Error: {e}")
  print("Failed to get latest version from both sources.")
  return None

def download(current_version):
  urls = [
    f"https://github.com/pfeiferj/openpilot-mapd/releases/download/{current_version}/mapd",
    f"https://gitlab.com/FrogAi/openpilot-mapd/-/releases/download/{current_version}/mapd"
  ]

  mapd_dir = os.path.dirname(MAPD_PATH)
  if not os.path.exists(mapd_dir):
    os.makedirs(mapd_dir)

  for url in urls:
    try:
      with urllib.request.urlopen(url) as f:
        with open(MAPD_PATH, 'wb') as output:
          output.write(f.read())
          os.fsync(output)
          current_permissions = stat.S_IMODE(os.lstat(MAPD_PATH).st_mode)
          os.chmod(MAPD_PATH, current_permissions | stat.S_IEXEC)
        with open(VERSION_PATH, 'w') as output:
          output.write(current_version)
          os.fsync(output)
      print(f"Successfully downloaded mapd from {url}")
      return
    except Exception as e:
      print(f"Failed to download from {url}. Error: {e}")

  print(f"Failed to download mapd for version {current_version} from both sources.")

def mapd_thread(sm=None, pm=None):
  rk = Ratekeeper(0.05, print_delay_threshold=None)

  while True:
    try:
      if ping_url("https://github.com"):
        current_version = get_latest_version()
        if current_version:
          if not os.path.exists(MAPD_PATH):
            download(current_version)
            continue
          if not os.path.exists(VERSION_PATH):
            download(current_version)
            continue
          with open(VERSION_PATH) as f:
            content = f.read()
            if content != current_version:
              download(current_version)
              continue

      process = subprocess.Popen(MAPD_PATH)
      process.wait()
    except Exception as e:
      print(e)

    rk.keep_time()

def main(sm=None, pm=None):
  mapd_thread(sm, pm)

if __name__ == "__main__":
  main()