From a67a1bc0a97afabcd4fe3d0f175cb742e3dd63ea Mon Sep 17 00:00:00 2001
From: Ken Van Hoeylandt <git@kenvanhoeylandt.net>
Date: Thu, 5 Jun 2025 22:46:07 +0200
Subject: [PATCH] Build tool can now download SDKs

---
 ExternalApps/HelloWorld/.gitignore |   1 +
 ExternalApps/HelloWorld/ttbuild.py | 111 +++++++++++++++++++++--------
 2 files changed, 83 insertions(+), 29 deletions(-)

diff --git a/ExternalApps/HelloWorld/.gitignore b/ExternalApps/HelloWorld/.gitignore
index a5309e6b..a21183d4 100644
--- a/ExternalApps/HelloWorld/.gitignore
+++ b/ExternalApps/HelloWorld/.gitignore
@@ -1 +1,2 @@
 build*/
+.ttbuild/
diff --git a/ExternalApps/HelloWorld/ttbuild.py b/ExternalApps/HelloWorld/ttbuild.py
index 6846405b..5f7c9a15 100644
--- a/ExternalApps/HelloWorld/ttbuild.py
+++ b/ExternalApps/HelloWorld/ttbuild.py
@@ -1,58 +1,111 @@
 import sys
 import os
+import urllib.request
+import zipfile
 
-def printWarning(message):
+esp_platforms = ["esp32", "esp32s3"]
+ttbuild_path = '.ttbuild'
+ttbuild_version = '0.1.0'
+
+def print_warning(message):
     if sys.platform == 'win32':
         print(f"ERROR: {message}")
     else:
         print(f"\033[93mWARNING: {message}\033[m")
 
-def printError(message):
+def print_error(message):
     if sys.platform == 'win32':
         print(f"ERROR: {message}")
     else:
         print(f"\033[91mERROR: {message}\033[m")
 
-def exitWithError(message):
-    printError(message)
+def exit_with_error(message):
+    print_error(message)
     sys.exit(1)
 
-def isValidPlatformName(name):
+def is_valid_platform_name(name):
     return name == "all" or name == "esp32" or name == "esp32s3"
 
-def build(platformName):
-    print(f"Platform: {platformName}")
-    os.system(f"cp sdkconfig.{platformName} sdkconfig")
-    os.system(f"idf.py -B build-{platformName} build")
+def build(version, platforms):
+    for platform in platforms:
+        print(f"Platform: {platform}")
+        sdk_dir = get_sdk_dir(version, platform)
+        print(f"Using SDK at {sdk_dir}")
+        os.environ['TACTILITY_SDK_PATH'] = sdk_dir
+        os.system(f"cp sdkconfig.{platform} sdkconfig")
+        # os.system(f"idf.py -B build-{platform} build")
 
-def buildAll():
-    build("esp32")
-    build("esp32s3")
-
-def printHelp():
+def print_help():
     print("Usage:")
     print("\tpython ttbuild.py [platformName]")
     print("\tplatformName:   all|esp32|esp32s3")
 
+def validate_environment():
+    if os.environ.get('IDF_PATH') is None:
+        exit_with_error("IDF is not installed or activated. Ensure you installed the toolset and ran the export command.")
+    if os.environ.get('TACTILITY_SDK_PATH') is not None:
+        print_warning("TACTILITY_SDK_PATH is set, but will be ignored by this command")
+
+def setup_environment():
+    os.makedirs(ttbuild_path, exist_ok=True)
+
+def get_sdk_dir(version, platform):
+    return os.path.join(ttbuild_path, f"{version}-{platform}", 'TactilitySDK')
+
+def get_sdk_root_dir(version, platform):
+    return os.path.join(ttbuild_path, f"{version}-{platform}")
+
+def get_sdk_url(version, platform):
+    return f"https://cdn.tactility.one/TactilitySDK-{version}-{platform}.zip"
+
+def sdk_exists(version, platform):
+    sdk_dir = get_sdk_dir(version, platform)
+    return os.path.isdir(sdk_dir)
+
+def sdk_download(version, platform):
+    sdk_root_dir = get_sdk_root_dir(version, platform)
+    os.makedirs(sdk_root_dir, exist_ok=True)
+    sdk_url = get_sdk_url(version, platform)
+    print(f"Downloading SDK version {version} for {platform} from {sdk_url}")
+    request = urllib.request.Request(
+        sdk_url,
+        data=None,
+        headers={
+            'User-Agent': f"Tactility Build Tool {ttbuild_version}"
+        }
+    )
+    response = urllib.request.urlopen(request)
+    filepath = os.path.join(sdk_root_dir, f"{version}-{platform}.zip")
+    file = open(filepath, mode='wb')
+    file.write(response.read())
+    file.close()
+    with zipfile.ZipFile(filepath, 'r') as zip_ref:
+        zip_ref.extractall(sdk_root_dir)
+    return True
+
+def sdk_download_all(version, platforms):
+    for platform in platforms:
+        if not sdk_exists(version, platform):
+            sdk_download(version, platform)
+        else:
+            print(f"Using cached download for SDK version {version} and platform {platform}")
+
 if __name__ == "__main__":
-    print("Tactility Build System v0.1.0")
+    print(f"Tactility Build System v{ttbuild_version}")
     # Environment
-    if os.environ.get('IDF_PATH') == None:
-        exitWithError("IDF is not installed or activated. Ensure you installed the toolset and ran the export command.")
-    if os.environ.get('TACTILITY_SDK_PATH') != None:
-        printWarning("TACTILITY_SDK_PATH is set, but will be ignored by this command")
-    os.environ['TACTILITY_SDK_PATH'] = '../../release/TactilitySDK'
+    validate_environment()
+    setup_environment()
     # Argument validation
     if len(sys.argv) == 1:
-        printHelp()
+        print_help()
         sys.exit()
-    platformName = sys.argv[1]
-    if not isValidPlatformName(platformName):
-        printHelp()
-        exitWithError("Invalid platform name")
+    platform_arg = sys.argv[1]
+    if not is_valid_platform_name(platform_arg):
+        print_help()
+        exit_with_error("Invalid platform name")
     # Build
-    if platformName == "all":
-        buildAll()
-    else:
-        build(sys.argv[1])
+    platforms_to_build = esp_platforms if platform_arg == "all" else [platform_arg]
+    sdk_version = '0.4.0-dev'
+    sdk_download_all(sdk_version, platforms_to_build)
+    build(sdk_version, platforms_to_build)