OTP Generator Added

utils/clientfunctions.py

@@ -5,6 +5,25 @@ import pandas as pd
 import utils.pretty as ct
 
 
+def findAgentID(url):
+	   
+    print(ct.colorText("WARNING: Device Name is Case Sensitive", "red"))
+    hostname = input(ct.colorText("Enter Device Name: ", "white"))
+    
+    endpoint = url + '/v1/agent/find'
+    payload = {
+            "hostname" : f"{hostname}"
+    }
+
+    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.loc[0, "agentid"]
+
 def getDestAllowlistFromClientID(url, clientid):
 
     #This is dependant on their being an allowlist in the policy containing the name "localapproval"
@@ -76,4 +95,71 @@ def getPolicyName(url, groupid):
     result = json.loads(response.text)
     data = pd.DataFrame(result["response"]["groups"])
     name = data.loc[data['groupid'] == f"{groupid}", 'name'].values[0]
-    return name
+    return name
+import datetime
+import requests
+import json
+import os
+import utils.pretty as ct
+
+def devicehistory(url, outputjson: bool):
+	endpoint = url + '/v1/getexechistory'
+	print("\n")
+	print(ct.colorText("1. Today", "yellow"))
+	print(ct.colorText("2. Last 24 Hours", "yellow"))
+	print(ct.colorText("3. Past 7 Days", "yellow"))
+	print(ct.colorText("4. Past 30 Days", "yellow"))
+	print(ct.colorText("5. Custom Date Range","yellow"))
+	choice = input(ct.colorText("\nSelect Date Range: ", "white"))
+	today = datetime.date.today()
+	today = today.strftime("%Y-%m-%d")
+	if choice == '1':
+		date_selected = today
+	elif choice == '2':
+		date_selected = datetime.date.today() - datetime.timedelta(days=1)
+		date_selected = date_selected.strftime('%Y-%m-%d')
+	elif choice == '3':
+		date_selected = datetime.date.today() - datetime.timedelta(days=7)
+		date_selected = date_selected.strftime('%Y-%m-%d')
+	elif choice == '4':
+		date_selected = datetime.date.today() - datetime.timedelta(days=30)
+		date_selected = date_selected.strftime('%Y-%m-%d')
+	elif choice == "5":
+		print(ct.colorText("Please Input Dates as YYYY-MM-DD", "cyan"))
+		date_selected = input(ct.colorText("From: ", "white"))
+		today = input(ct.colorText("Date To: ", "white"))
+	print(ct.colorText("WARNING: Device Name is Case Sensitive", "red"))
+	device = input(ct.colorText("Enter Device Name: ", "white"))
+	payload_dict = {
+		"datefrom": date_selected,
+		"dateto": today,
+		"hostname": device
+	}
+	payload = json.dumps(payload_dict)
+	print(ct.colorText(payload, "green"))
+	headers = {
+		"X-APIKey": os.getenv('APIKEY')
+	}
+
+	response = requests.request("POST", endpoint, headers=headers, data=payload, verify=False)
+
+	if outputjson:
+		return response
+
+	parse_text = json.loads(response.text)
+
+	# Safely get exechistory
+	exechistory = parse_text.get('response', {}).get('exechistory')
+
+	if isinstance(exechistory, list):
+		for block in exechistory:
+			print(ct.colorText(f"Command: {block.get('commandline', 'N/A')}", "green"))
+			print(ct.colorText(f"Date: {block.get('datetime', 'N/A')}", "green"))
+			print(ct.colorText(f"Filename: {block.get('filename', 'N/A')}", "green"))
+			print(ct.colorText(f"Policy Name: {block.get('policyname', 'N/A')}", "green"))
+			print(ct.colorText(f"Hostname: {block.get('hostname', 'N/A')}", "green"))
+			print(ct.colorText(f"Hash: {block.get('sha256', 'N/A')}", "green"))
+			print("\n")
+	else:
+		print(ct.colorText("No execution history found or data is not in expected format.", "red"))
+