Restore env var

infrasonar · Jan 4, 2025 · 20103bd · 20103bd
1 parent d225b40
commit 20103bd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/lib/state.py b/lib/state.py
@@ -577,6 +577,25 @@ def set(cls, state: dict):
         for name in configs_to_delete:
             del cls.config_data[name]
 
+        # socat (API forwarder)
+        socat_target_addr = state.get('socat_target_addr')
+        if socat_target_addr:
+            services['socat'] = _SOCAT
+        else:
+            try:
+                del services['socat']
+            except KeyError:
+                pass
+
+        # update environment variable (all verified with sanity check)
+        for key in (
+            'agentcore_token',
+            'agent_token',
+            'agentcore_zone_id',
+            'socat_target_addr',
+        ):
+            cls.env_data[key.upper()] = state[key]
+
         cls.write()
         asyncio.ensure_future(cls.update())
 

diff --git a/lib/version.py b/lib/version.py
@@ -1,3 +1,3 @@
-__version__ = '1.0.0'
+__version__ = '1.0.1'
 
 IS_RELEASE_VERSION = '-' not in __version__