From c39c54a0d81dfe9067be84759e064c3c6b0eec1a Mon Sep 17 00:00:00 2001
From: mike8643 <98910897+mike8643@users.noreply.github.com>
Date: Thu, 7 Mar 2024 13:57:41 -0500
Subject: [PATCH] otisserv: `set_destination` and `locations` endpoints for
 non-prime users (#243)

ios shortcut support
---
 selfdrive/navd/otisserv.py | 61 +++++++++++++++++++++++++++++++-------
 1 file changed, 50 insertions(+), 11 deletions(-)

diff --git a/selfdrive/navd/otisserv.py b/selfdrive/navd/otisserv.py
index 1dbf2a53350c22..d8fca7fc44804f 100644
--- a/selfdrive/navd/otisserv.py
+++ b/selfdrive/navd/otisserv.py
@@ -50,7 +50,10 @@ def do_GET(self):
       return
     if self.path == '/?reset=1':
       params.put("NavDestination", "")
-    if use_amap:
+    if self.path == '/locations':
+      self.get_locations()
+      return
+    elif use_amap:
       if self.path == '/style.css':
         self.send_response(200)
         self.send_header("Content-type", "text/css")
@@ -108,16 +111,25 @@ def do_GET(self):
       if self.get_app_token() is None:
         self.display_page_app_token()
         return
-      self.display_page_addr_input()
+      if self.path != '/locations':
+        self.display_page_addr_input()
 
   def do_POST(self):
     use_amap = params.get_bool("EnableAmap")
     use_gmap = not use_amap and params.get_bool("EnableGmap")
 
     postvars = self.parse_POST()
-    self.send_response(200)
-    self.send_header("Content-type", "text/html")
-    self.end_headers()
+    # set_destination endpoint
+    if self.path == '/set_destination':
+      self.send_response(200)
+      self.send_header("Content-type", "application/json")
+      self.end_headers()
+      response_data = {'success': True}
+      self.wfile.write(json.dumps(response_data).encode('utf-8'))
+    else:
+      self.send_response(200)
+      self.send_header("Content-type", "text/html")
+      self.end_headers()
 
     if use_amap:
       # amap token
@@ -172,6 +184,16 @@ def do_POST(self):
           lng, lat = self.gcj02towgs84(lng, lat)
         params.put('NavDestination', "{\"latitude\": %f, \"longitude\": %f, \"place_name\": \"%s\"}" % (lat, lng, name))
         self.to_json(lat, lng, save_type, name)
+    if postvars is not None:
+      latitude_value = postvars.get("latitude")
+      longitude_value = postvars.get("longitude")
+      if latitude_value is not None and latitude_value != "" and longitude_value is not None and longitude_value != "":
+        lat = float(latitude_value)
+        lng = float(longitude_value)
+        save_type = "recent"
+        name = postvars.get("place_name", [""])
+        params.put('NavDestination', "{\"latitude\": %f, \"longitude\": %f, \"place_name\": \"%s\"}" % (lat, lng, name))
+        self.to_json(lat, lng, save_type, name)
       # favorites
       if not use_gmap and "fav_val" in postvars:
         addr = postvars.get("fav_val")[0]
@@ -208,12 +230,13 @@ def do_POST(self):
           else:
             self.display_page_addr_input("Place Not Found")
             return
-    if use_amap:
-      self.display_page_amap()
-    elif use_gmap:
-      self.display_page_gmap()
-    else:
-      self.display_page_addr_input()
+    if self.path != '/set_destination':
+      if use_amap:
+        self.display_page_amap()
+      elif use_gmap:
+        self.display_page_gmap()
+      else:
+        self.display_page_addr_input()
 
   def get_logo(self):
     self.send_response(200)
@@ -223,6 +246,14 @@ def get_logo(self):
     self.wfile.write(f.read())
     f.close()
 
+  def get_locations(self):
+    self.send_response(200)
+    self.send_header('Content-type','application/json')
+    self.end_headers()
+    val = params.get("ApiCache_NavDestinations", encoding='utf-8')
+    if val is not None:
+      self.wfile.write(val.encode('utf-8'))
+
   def get_gmap_css(self):
     self.wfile.write(bytes(self.get_parsed_template("gmap/style.css"), "utf-8"))
 
@@ -335,6 +366,14 @@ def parse_POST(self):
       postvars = parse_qs(
         self.rfile.read(length).decode('utf-8'),
         keep_blank_values=1)
+    elif ctype == 'application/json':
+      length = int(self.headers['content-length'])
+      post_data = self.rfile.read(length).decode('utf-8')
+      try:
+        postvars = json.loads(post_data)
+      except json.JSONDecodeError:
+        self.send_error(400, 'Invalid JSON data')
+        return None
     else:
       postvars = {}
     return postvars