Merge pull request #38 from balenablocks/support-local-mode

Add support for local mode

README.md

@@ -26,6 +26,7 @@ services:
     restart: always
     labels:
       io.balena.features.balena-api: '1' # necessary to discover services
+      io.balena.features.supervisor-api: 1  # necessary to discover services in local mode
     privileged: true # necessary to change container hostname
     ports:
       - "8080" # only necessary if using ExternalHttpListener (see below)
@@ -47,6 +48,7 @@ services:
     restart: always
     labels:
       io.balena.features.balena-api: '1' # necessary to discover services
+      io.balena.features.supervisor-api: 1  # necessary to discover services in local mode
     privileged: true # necessary to change container hostname
     ports:
       - "8080" # only necessary if using ExternalHttpListener (see below)

autowire.py

@@ -4,6 +4,7 @@
 from pluginbase import PluginBase
 from balena import Balena
 import toml
+import requests
 
 class AutoWire():
   balena = ""
@@ -15,15 +16,25 @@ def __init__(self, balena):
    ### Gets the services running on the device from the release definition ###
   def GetServices(self):
     if self.services is None:
-      # Use the device UUID to get the device model
+      # Use the device UUID to get the device model and check device mode
       device_id = os.environ.get('BALENA_DEVICE_UUID')
-      device = self.balena.models.device.get_with_service_details(device_id, False)
-      # get the commit the device is on
-      commit = device["is_on__commit"]
-      # use the commit to get the release the device is on
-      release = self.balena.models.release.get(commit)
-      # use the release to find the services configured
-      self.services = release["composition"]["services"]
+      local_mode = balena.models.device.is_in_local_mode(device_id)
+
+      if local_mode:
+        supervisor_address = os.environ.get('BALENA_SUPERVISOR_ADDRESS')
+        r = requests.get(supervisor_address + '/v2/local/target-state')
+        services = r.json()["state"]["local"]["apps"]["1"]["services"]
+        self.services = {}
+        for service in services:
+          self.services[service.get("serviceName")] = service.get("config")
+      else:
+        device = self.balena.models.device.get_with_service_details(device_id, False)
+        # get the commit the device is on
+        commit = device["is_on__commit"]
+        # use the commit to get the release the device is on
+        release = self.balena.models.release.get(commit)
+        # use the release to find the services configured
+        self.services = release["composition"]["services"]
 
     return self.services
 

docker-compose.yml

@@ -5,7 +5,8 @@ services:
     image: balenablocks/connector:raspberrypi3
     restart: always
     labels:
-      io.balena.features.balena-api: '1' # necessary to discover services
+      io.balena.features.balena-api: '1'    # necessary to discover services
+      io.balena.features.supervisor-api: 1  # necessary to discover services in local mode
     privileged: true # necessary to change container hostname
     ports:
       - "8080" # only necessary if using ExternalHttpListener (see below)

plugins/internalHttpPullInput.py

@@ -9,7 +9,7 @@ def FindPullDataSources(services):
       if "expose" in details.keys():
         portNumber = details["expose"]
         if len(portNumber) == 1:
-            if portNumber[0] == pullSourcePort:
+            if portNumber[0].split('/')[0] == pullSourcePort:
                 outputDict[service] = portNumber
 
     return outputDict
@@ -31,7 +31,7 @@ def invoke(services):
     timeout = "{timeout}s"
     data_format = "json"
     name_override = "{service}"
-""".format(service=service, port=port[0], timeout=timeout)
+""".format(service=service, port=port[0].split('/')[0], timeout=timeout)
             output = (output + sourceConf)
 
     return output

requirements.txt

@@ -1,3 +1,5 @@
 balena-sdk==9.5.0
 PluginBase
-toml
+toml
+requests
+PyJWT==2.1.0