Improve version detection and dark mode icon

Author: rsousa88

app/main.py

@@ -37,6 +37,8 @@
 CONTAINER_STATS_CACHE_TTL_SECONDS = 10
 DISK_USAGE_CACHE = {}
 DISK_USAGE_CACHE_TTL_SECONDS = 300
+IMAGE_VERSION_CACHE = {}
+IMAGE_VERSION_CACHE_TTL_SECONDS = 3600
 DISK_USAGE_INFLIGHT = set()
 DISK_USAGE_INFLIGHT_LOCK = threading.Lock()
 
@@ -80,6 +82,22 @@ def set_cached_disk_usage(host_id, data):
     DISK_USAGE_CACHE[host_id] = {'timestamp': time.time(), 'data': data}
 
 
+def get_cached_image_version(cache_key):
+    entry = IMAGE_VERSION_CACHE.get(cache_key)
+    if not entry:
+        return None
+    if time.time() - entry['timestamp'] > IMAGE_VERSION_CACHE_TTL_SECONDS:
+        return None
+    return entry['data']
+
+
+def set_cached_image_version(cache_key, version, source):
+    IMAGE_VERSION_CACHE[cache_key] = {
+        'timestamp': time.time(),
+        'data': (version, source)
+    }
+
+
 def refresh_disk_usage_async(host_id, docker_client):
     def run():
         try:
@@ -204,6 +222,18 @@ def strip_image_tag(image_name):
 
     return f"{prefix}/{last}" if prefix else last
 
+def _extract_version_from_labels(labels, source):
+    for key, value in (labels or {}).items():
+        if not value:
+            continue
+        key_lower = key.lower()
+        if key_lower in VERSION_LABEL_KEYS or 'version' in key_lower:
+            match = VERSION_IN_TEXT_RE.search(str(value))
+            if match:
+                return match.group(0), source
+    return None, None
+
+
 def get_image_version(container, image_name, docker_client=None):
     """Best-effort version from image labels, falling back to version-like tags."""
     labels = {}
@@ -214,10 +244,9 @@ def get_image_version(container, image_name, docker_client=None):
     except Exception:
         labels = {}
 
-    for key in VERSION_LABEL_KEYS:
-        value = labels.get(key)
-        if value:
-            return value.strip(), 'label'
+    version, source = _extract_version_from_labels(labels, 'label')
+    if version:
+        return version, source
 
     ref_name = labels.get('org.opencontainers.image.ref.name', '')
     if ref_name:
@@ -244,10 +273,9 @@ def get_image_version(container, image_name, docker_client=None):
             image_id = container.image.id if container.image else None
             image_obj = docker_client.images.get(image_id) if image_id else None
             image_labels = image_obj.attrs.get('Config', {}).get('Labels', {}) if image_obj else {}
-            for key in VERSION_LABEL_KEYS:
-                value = (image_labels or {}).get(key)
-                if value:
-                    return value.strip(), 'label'
+            version, source = _extract_version_from_labels(image_labels, 'label')
+            if version:
+                return version, source
             ref_name = (image_labels or {}).get('org.opencontainers.image.ref.name', '')
             if ref_name:
                 match = VERSION_IN_TEXT_RE.search(ref_name)
@@ -278,6 +306,20 @@ def get_image_version(container, image_name, docker_client=None):
         if match:
             return match.group(0), 'tag'
 
+    if docker_client and image_name and ':' in image_name:
+        cache_key = image_name
+        cached = get_cached_image_version(cache_key)
+        if cached:
+            return cached
+        try:
+            registry_data = docker_client.images.get_registry_data(image_name)
+            registry_version = get_registry_version(registry_data)
+            if registry_version and registry_version != 'unknown':
+                set_cached_image_version(cache_key, registry_version, 'registry')
+                return registry_version, 'registry'
+        except Exception:
+            pass
+
     return 'unknown', 'unknown'
 
 def get_registry_version(registry_data):

templates/index.html

@@ -256,6 +256,14 @@
             stroke: currentColor;
         }
 
+        .github-icon {
+            color: #2b2b2b;
+        }
+
+        body.dark-mode .github-icon {
+            color: #d6e2f0;
+        }
+
         .update-status {
             font-size: 11px;
             color: var(--text-secondary);
@@ -1026,7 +1034,7 @@ <h2>Docker Containers</h2>
                                 {% endif %}
                                 {% if container.image_links.github %}
                                 <span class="tooltip-container">
-                                    <a href="{{ container.image_links.github }}" target="_blank" style="color: #333; text-decoration: none; font-size: 14px;">
+                                    <a href="{{ container.image_links.github }}" target="_blank" class="github-icon" style="text-decoration: none; font-size: 14px;">
                                         <svg width="14" height="14" viewBox="0 0 16 16" fill="currentColor"><path d="M8 0C3.58 0 0 3.58 0 8c0 3.54 2.29 6.53 5.47 7.59.4.07.55-.17.55-.38 0-.19-.01-.82-.01-1.49-2.01.37-2.53-.49-2.69-.94-.09-.23-.48-.94-.82-1.13-.28-.15-.68-.52-.01-.53.63-.01 1.08.58 1.23.82.72 1.21 1.87.87 2.33.66.07-.52.28-.87.51-1.07-1.78-.2-3.64-.89-3.64-3.95 0-.87.31-1.59.82-2.15-.08-.2-.36-1.02.08-2.12 0 0 .67-.21 2.2.82.64-.18 1.32-.27 2-.27.68 0 1.36.09 2 .27 1.53-1.04 2.2-.82 2.2-.82.44 1.1.16 1.92.08 2.12.51.56.82 1.27.82 2.15 0 3.07-1.87 3.75-3.65 3.95.29.25.54.73.54 1.48 0 1.07-.01 1.93-.01 2.2 0 .21.15.46.55.38A8.013 8.013 0 0016 8c0-4.42-3.58-8-8-8z"></path></svg>
                                     </a>
                                     <span class="tooltip-text">GitHub</span>