From 20103bdef335be863bed4be066cbe2df9de00bcb Mon Sep 17 00:00:00 2001
From: Jeroen van der Heijden <jeroen@cesbit.com>
Date: Sat, 4 Jan 2025 11:16:36 +0100
Subject: [PATCH] Restore env var

---
 lib/state.py   | 19 +++++++++++++++++++
 lib/version.py |  2 +-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/lib/state.py b/lib/state.py
index c76614d..75f1132 100644
--- 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
index 7a1472e..0076ad8 100644
--- 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__