Added Device List Query

utils/clientfunctions.py

@@ -17,9 +17,10 @@
 import utils.pretty as ct
 
 #Standard Libary Imports:
-import datetime
+from datetime import datetime
 import json
 import os
+import re
 
 #3rd Party Imports:
 import pandas as pd
@@ -184,3 +185,62 @@ def devicehistory(url, outputjson: bool):
 	else:
 		print(ct.colorText("No execution history found or data is not in expected format.", "red"))
 		
+def findAllAgents(url):        
+    endpoint = url + '/v1/agent/find'
+    payload = {}
+    headers = {"X-APIKey": os.getenv('APIKEY')}
+    payload = json.dumps(payload)
+    response = requests.post(endpoint, headers=headers, data=payload, verify=False)
+    result = json.loads(response.text)
+    data = pd.DataFrame(result["response"]["agents"])
+    return(data)
+
+def findAgents(url, device_input_str):
+
+    os.makedirs("device_search", exist_ok=True)
+    # Step 1: Get the DataFrame from your function
+    df = findAllAgents(url)
+
+    # Step 2: Parse the input string into device names (newline-separated only)
+    device_names = device_input_str.strip().split('\n')
+    device_names = [name.strip() for name in device_names if name.strip()]
+
+    # Step 3: Build a regex pattern for case-insensitive matching
+    pattern = '|'.join([re.escape(name) for name in device_names])
+    regex = re.compile(pattern, re.IGNORECASE)
+
+    # Step 4: Filter the DataFrame using regex
+    matched_df = df[df['hostname'].apply(lambda x: bool(regex.search(str(x))))]
+
+    # Step 5: Export to CSV with timestamp
+    timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    filename = f"agentsearch_{timestamp}.csv"
+    matched_df.to_csv(f"device_search\\{filename}", index=False)
+
+    print(ct.colorText(f"\n✅ Matched devices exported to: device_search\\{filename}","green"))
+
+def promptForDevices():
+    
+    print(ct.colorText("🔍 Device Search", "cyan"))
+    print(ct.colorText("Enter the device hostnames you'd like to search for, one per line.", "cyan"))
+    print(ct.colorText("When you're done, press Enter twice.\n", "cyan"))
+    print(ct.colorText("Example:", "cyan"))
+    print(ct.colorText("H00000", "cyan"))
+    print(ct.colorText("UTN00000", "cyan"))
+    print(ct.colorText("i-hSuperSecretServer", "cyan"))
+    print(ct.colorText("u-hVenderBroke\n", "cyan"))
+
+
+    # Collect multiline input from user
+    print(ct.colorText("Paste or type your device names below:", "white"))
+    device_input_lines = []
+    while True:
+        line = input()
+        if line == "":
+            break
+        device_input_lines.append(line)
+
+    device_input_str = "\n".join(device_input_lines)
+
+    return device_input_str
+