From 2862e368bc238d82ed5107634fe3c00bc51a3d03 Mon Sep 17 00:00:00 2001
From: Keith Wiley <keith.wiley@alleninstitute.org>
Date: Tue, 20 Aug 2024 13:41:23 -0700
Subject: [PATCH] Skeleton service work.

---
 caveclient/endpoints.py       |  9 +++------
 caveclient/skeletonservice.py | 31 +++++++++++++++++++++++++------
 2 files changed, 28 insertions(+), 12 deletions(-)

diff --git a/caveclient/endpoints.py b/caveclient/endpoints.py
index 1af01538..c3595b16 100644
--- a/caveclient/endpoints.py
+++ b/caveclient/endpoints.py
@@ -294,17 +294,14 @@
 # ------ Skeleton endpoints
 # -------------------------------
 
-# Correct:         https://minnie.microns-daf.com/skeletoncache/api/v1/minnie65_phase3_v1/precomputed/skeleton/info
-# What I'm seeing: https://minniev1.microns-daf.com/            api/v1/minnie65_phase3_v1/precomputed/skeleton/info
-
 skeletonservice_common = {}
 
 skeleton_v1 = "{skeleton_server_address}/api/v1"
 skeletonservice_endpoints_v1 = {
-    "skeleton_info": skeleton_v1 + "/{datastack_name}/precomputed/skeleton/info",
-    "skeleton_by_rid": skeleton_v1 + "/{datastack_name}/precomputed/skeleton/{root_id}",
+    "skeleton_info":             skeleton_v1 + "/{datastack_name}/precomputed/skeleton/info",
+    "skeleton_by_rid":           skeleton_v1 + "/{datastack_name}/precomputed/skeleton/{root_id}",
+    "get_skeleton_by_full_desc": skeleton_v1 + "/{datastack_name}/precomputed/skeleton/{skeleton_version}/{root_id}/{output_format}",
 }
 skeletonservice_api_versions = {
     1: skeletonservice_endpoints_v1
 }
-
diff --git a/caveclient/skeletonservice.py b/caveclient/skeletonservice.py
index f7eb6ca5..35fa484f 100644
--- a/caveclient/skeletonservice.py
+++ b/caveclient/skeletonservice.py
@@ -107,6 +107,19 @@ def __init__(
             over_client=over_client,
         )
     
+    def get_skeleton_info(
+        self,
+        datastack_name,
+    ):
+        endpoint_mapping = self.default_url_mapping
+        endpoint_mapping["datastack_name"] = datastack_name
+
+        url = self._endpoints["skeleton_info"].format_map(endpoint_mapping)
+        print(f"get_skeleton_info() url: {url}")
+
+        response = self.session.get(url)
+        return handle_response(response, False)
+    
     def get_skeleton_by_datastack_and_rid(
         self,
         datastack_name,
@@ -114,10 +127,9 @@ def get_skeleton_by_datastack_and_rid(
     ):
         endpoint_mapping = self.default_url_mapping
         endpoint_mapping["datastack_name"] = datastack_name
-        endpoint_mapping["rid"] = rid
+        endpoint_mapping["root_id"] = rid
 
-        # url = self._endpoints["version_metadata"].format_map(endpoint_mapping)
-        url = self._endpoints["skeleton_info"].format_map(endpoint_mapping)
+        url = self._endpoints["skeleton_by_rid"].format_map(endpoint_mapping)
         print(f"get_skeleton_by_datastack_and_rid() url: {url}")
 
         response = self.session.get(url)
@@ -126,18 +138,25 @@ def get_skeleton_by_datastack_and_rid(
     def get_skeleton_by_full_desc(
         self,
         datastack_name,
+        skeleton_version,
         rid,
         output_format,
+        # bucket,
+        # verbose_level,
     ):
         endpoint_mapping = self.default_url_mapping
         endpoint_mapping["datastack"] = datastack_name
-        endpoint_mapping["rid"] = rid
+        endpoint_mapping["skeleton_version"] = skeleton_version
+        endpoint_mapping["root_id"] = rid
         endpoint_mapping["output_format"] = output_format
+        # endpoint_mapping["bucket"] = bucket
+        # endpoint_mapping["verbose_level_"] = verbose_level
 
-        url = self._endpoints["version_metadata"].format_map(endpoint_mapping)
+        url = self._endpoints["get_skeleton_by_full_desc"].format_map(endpoint_mapping)
+        print(f"get_skeleton_by_full_desc() url: {url}")
 
         response = self.session.get(url)
-        return handle_response(response)
+        return handle_response(response, False)
     
 
 client_mapping = {