qualys-mcp 2.1.1__tar.gz → 2.1.2__tar.gz

{qualys_mcp-2.1.1 → qualys_mcp-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qualys-mcp
-Version: 2.1.1
+Version: 2.1.2
 Summary: MCP server for Qualys security APIs - natural language interaction with vulnerability, asset, and cloud security data
 Project-URL: Homepage, https://github.com/nelssec/qualys-mcp
 Project-URL: Repository, https://github.com/nelssec/qualys-mcp

{qualys_mcp-2.1.1 → qualys_mcp-2.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "qualys-mcp"
-version = "2.1.1"
+version = "2.1.2"
 description = "MCP server for Qualys security APIs - natural language interaction with vulnerability, asset, and cloud security data"
 readme = "README.md"
 license = "MIT"

{qualys_mcp-2.1.1 → qualys_mcp-2.1.2}/qualys_mcp.py

@@ -112,6 +112,46 @@ def get_assets(limit=100, qql=None):
         return []
 
 
+def get_eol_assets(stage_filter="EOL,EOL/EOS", limit=500):
+    url = f"{GATEWAY_URL}/rest/2.0/search/am/asset?pageSize={limit}"
+    token = get_bearer_token()
+
+    filter_body = json.dumps({
+        "filters": [
+            {"field": "operatingSystem.lifecycle.stage", "operator": "IN", "value": stage_filter}
+        ]
+    })
+
+    req = Request(url, data=filter_body.encode(), method='POST')
+    req.add_header('Authorization', f'Bearer {token}' if token else f'Basic {BASIC_AUTH}')
+    req.add_header('Content-Type', 'application/json')
+    req.add_header('X-Requested-With', 'qualys-mcp')
+
+    try:
+        with urlopen(req, timeout=60) as resp:
+            data = json.loads(resp.read())
+            assets = []
+            for a in data.get('assetListData', {}).get('asset', []):
+                os_info = a.get('operatingSystem', {})
+                lifecycle = os_info.get('lifecycle', {})
+                assets.append({
+                    'assetId': a.get('assetId'),
+                    'address': a.get('address', ''),
+                    'dnsName': a.get('dnsHostName', ''),
+                    'operatingSystem': {
+                        'osName': f"{os_info.get('name', '')} {os_info.get('version', '')}".strip(),
+                        'lifecycle': {
+                            'stage': lifecycle.get('stage', ''),
+                            'eolDate': lifecycle.get('eolDate', ''),
+                            'eosDate': lifecycle.get('eosDate', '')
+                        }
+                    }
+                })
+            return assets
+    except Exception as e:
+        return []
+
+
 def get_images(limit=100, severity=None):
     url = f"{GATEWAY_URL}/csapi/v1.3/images?pageSize={limit}"
     if severity:
@@ -484,7 +524,7 @@ def get_tech_debt(days_until_eol: int = 0) -> dict:
         'byOS': []
     }
 
-    assets = get_assets(500)
+    assets = get_eol_assets("EOL,EOL/EOS,EOS", 500)
     result['stats']['total'] = len(assets)
 
     today = datetime.utcnow().date()
@@ -523,7 +563,7 @@ def get_tech_debt(days_until_eol: int = 0) -> dict:
             os_data[os_name]['eol'] += 1
             if len(result['currentEOL']) < 20:
                 result['currentEOL'].append(asset_info)
-        elif stage == 'EOS':
+        elif stage in ('EOS', 'EOL/EOS'):
             result['stats']['currentEOS'] += 1
             os_data[os_name]['eos'] += 1
             if len(result['currentEOS']) < 20: