UI Improvements
This commit is contained in:
+51
-52
@@ -20,8 +20,8 @@ import re
|
||||
import dotenv
|
||||
import pandas as pd
|
||||
|
||||
import flows.localApproval as la
|
||||
import services.policyhandler as policyh
|
||||
from flows.otp import generate, otp_activities_by_agent, revoke
|
||||
from flows.prepPolicy import (
|
||||
buildPathsandPublishers,
|
||||
buildPreflights,
|
||||
@@ -30,32 +30,26 @@ from flows.prepPolicy import (
|
||||
sortHashes,
|
||||
)
|
||||
from flows.quietAgent import findQuietAgents
|
||||
|
||||
from flows.otp import (
|
||||
generate,
|
||||
otp_activities_by_agent,
|
||||
revoke
|
||||
)
|
||||
from utils.selector import Selector
|
||||
from services.agenthandler import findAgents, selectAgents, moveAgentToRelatedPolicy
|
||||
from services.agenthandler import findAgents, moveAgentToRelatedPolicy, selectAgents
|
||||
from services.API import AirlockAPIWrapper
|
||||
from utils.configmanager import load_env
|
||||
from utils.selector import Selector
|
||||
from utils.utils import (
|
||||
areYouSure,
|
||||
colorText,
|
||||
displayIntro,
|
||||
get_sanitized_input,
|
||||
open_directory,
|
||||
printEnforceChecklist,
|
||||
)
|
||||
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
dotenv.load_dotenv()
|
||||
|
||||
def menu_main(api: AirlockAPIWrapper):
|
||||
working_dir = load_env("WORKING_DIR")
|
||||
extras = load_env("EXTRAS")
|
||||
while True:
|
||||
displayIntro()
|
||||
# Add Settings, and give option to change working dir
|
||||
@@ -63,31 +57,41 @@ def menu_main(api: AirlockAPIWrapper):
|
||||
print(colorText("2. 🎫 - OTP", "yellow"))
|
||||
print(colorText("3. 🔄 - Move to Audit/Enforcement", "yellow"))
|
||||
print(colorText("4. 🔍 - Device Search", "yellow"))
|
||||
print(colorText("5. 🛡️ - Policy Enforcement Tools", "yellow"))
|
||||
print(colorText("5. 🔇 - Find Quiet Hosts", "yellow"))
|
||||
if extras == "POLICYPREP" : print(colorText("6. 🛡️ - Policy Enforcement Tools", "yellow"))
|
||||
print(colorText("F. 📂 - Open Working Directory", "yellow"))
|
||||
print(colorText("S. 🛠️ - Settings", "yellow"))
|
||||
print(colorText("Q. 🔚 - Quit", "yellow"))
|
||||
|
||||
|
||||
choice = input(colorText("\nEnter Menu Item: ", "white"))
|
||||
choice = get_sanitized_input("\nEnter Menu Item: ")
|
||||
if choice == "1":
|
||||
print("This Feature is still in development")
|
||||
input("Press enter to continue")
|
||||
get_sanitized_input("Press enter to continue")
|
||||
elif choice == "2":
|
||||
menu_otp(api)
|
||||
elif choice == "3":
|
||||
choices = ["audit", "enforcement"]
|
||||
print(colorText("Move devices to which state?:", "yellow"))
|
||||
direction = Selector.select_string(choices, False, False)
|
||||
devices = selectAgents(api)
|
||||
if direction and devices:
|
||||
print(colorText("Would you like to continue with these devices?","white"))
|
||||
for device in devices:
|
||||
print(device.hostname)
|
||||
confirm = Selector.confirm()
|
||||
if direction and devices and confirm:
|
||||
for device in devices:
|
||||
moveAgentToRelatedPolicy(api,device, direction[0])
|
||||
elif choice == "4":
|
||||
findAgents(api,False)
|
||||
elif choice == "5":
|
||||
menu_policymanagment(api)
|
||||
elif choice == "F":
|
||||
findQuietAgents(api)
|
||||
elif choice == "6":
|
||||
if extras == "POLICYPREP": menu_policymanagment(api)
|
||||
elif choice.upper() == "F":
|
||||
open_directory(working_dir)
|
||||
elif choice == "S":
|
||||
elif choice.upper() == "S":
|
||||
menu_settings()
|
||||
elif choice == "Q":
|
||||
elif choice.upper() == "Q":
|
||||
break
|
||||
else:
|
||||
print(colorText("Invalid choice. Please try again.", "red"))
|
||||
@@ -106,8 +110,7 @@ def menu_policy_enforce(api: AirlockAPIWrapper):
|
||||
|
||||
while True:
|
||||
printEnforceChecklist(selected_policies, destination_policy, destination_allowlist)
|
||||
|
||||
choice = input(colorText("\nEnter your choice: ", "white"))
|
||||
choice = get_sanitized_input("\nEnter your choice: ")
|
||||
|
||||
if choice == "1":
|
||||
selected_policies = selectPolicies(api,True)
|
||||
@@ -206,13 +209,12 @@ def menu_policy_enforce(api: AirlockAPIWrapper):
|
||||
|
||||
elif choice == "7":
|
||||
areYouSure()
|
||||
confirmation = input(colorText("Type 'I AGREE' to continue: ", "white"))
|
||||
|
||||
confirmation = get_sanitized_input("Type 'I AGREE' to continue: ")
|
||||
if (
|
||||
tested
|
||||
and destination_policy
|
||||
and destination_allowlist
|
||||
and confirmation.strip().upper() == "I AGREE"
|
||||
and confirmation.strip() == "I AGREE"
|
||||
):
|
||||
print(colorText("Proceeding with the code...", "yellow"))
|
||||
api.hash_add_to_allowlist(destination_allowlist[0].applicationid, processed_hashes)
|
||||
@@ -227,14 +229,14 @@ def menu_policy_enforce(api: AirlockAPIWrapper):
|
||||
logger.error(" - `destination_policy` is missing or invalid.")
|
||||
if not destination_allowlist:
|
||||
logger.error(" - `destination_allowlist` is missing or invalid.")
|
||||
if confirmation.strip().upper() != "I AGREE":
|
||||
if confirmation.strip() != "I AGREE":
|
||||
logger.error(" - User did not confirm with 'I AGREE'. Received: '%s'", confirmation.strip())
|
||||
|
||||
elif choice == "F":
|
||||
elif choice.upper() == "F":
|
||||
open_directory(working_dir)
|
||||
elif choice == "S":
|
||||
elif choice.upper() == "S":
|
||||
menu_settings()
|
||||
elif choice == "Q":
|
||||
elif choice.upper == "B":
|
||||
break
|
||||
|
||||
|
||||
@@ -252,9 +254,9 @@ def menu_otp(api: AirlockAPIWrapper):
|
||||
print(colorText("3. ❌ -Revoke OTPs", "cyan"))
|
||||
print(colorText("F. 📂 - Open Working Directory", "yellow"))
|
||||
print(colorText("S. 🛠️ - Settings", "yellow"))
|
||||
print(colorText("Q. 🔚 - Quit", "yellow"))
|
||||
print(colorText("B. 🔙 - Back", "yellow"))
|
||||
|
||||
choice = input("Enter your choice: ")
|
||||
choice = get_sanitized_input("Enter your choice: ")
|
||||
|
||||
if choice == "1":
|
||||
otp_list = generate(api)
|
||||
@@ -263,54 +265,51 @@ def menu_otp(api: AirlockAPIWrapper):
|
||||
otp_activities_by_agent(api)
|
||||
elif choice == "3":
|
||||
revoke(api)
|
||||
elif choice == "F":
|
||||
elif choice.upper() == "F":
|
||||
open_directory(working_dir)
|
||||
elif choice == "S":
|
||||
elif choice.upper() == "S":
|
||||
menu_settings()
|
||||
elif choice == "Q":
|
||||
elif choice.upper() == "B":
|
||||
break
|
||||
|
||||
def menu_policymanagment(api: AirlockAPIWrapper):
|
||||
working_dir = load_env("WORKING_DIR")
|
||||
while True:
|
||||
print(colorText("1. 🔇 - Find Quiet Hosts", "yellow"))
|
||||
print(colorText("2. 🔒 - Prepare Policy For Enforcement", "yellow"))
|
||||
print(colorText("3. 🔄 - Update Audit Policies from Enforcement Policies", "yellow"))
|
||||
print(colorText("1. 🔒 - Prepare Policy For Enforcement", "yellow"))
|
||||
print(colorText("2. 🔄 - Update Audit Policies from Enforcement Policies", "yellow"))
|
||||
print(colorText("F. 📂 - Open Working Directory", "yellow"))
|
||||
print(colorText("S. 🛠️ - Settings", "yellow"))
|
||||
print(colorText("Q. 🔚 - Quit", "yellow"))
|
||||
|
||||
choice = input(colorText("\nEnter Menu Item: ", "white"))
|
||||
print(colorText("B. 🔙 - Back", "yellow"))
|
||||
choice = get_sanitized_input("\n Enter Menu Item: ")
|
||||
|
||||
if choice == "1":
|
||||
findQuietAgents(api)
|
||||
elif choice == "2":
|
||||
menu_policy_enforce(api)
|
||||
elif choice == "3":
|
||||
elif choice == "2":
|
||||
areYouSure()
|
||||
confirmation = input(colorText("Type 'I AGREE' to continue: ", "white"))
|
||||
if confirmation.strip().upper() == "I AGREE":
|
||||
confirmation = get_sanitized_input("Type 'I AGREE' to continue: ")
|
||||
if confirmation.strip() == "I AGREE":
|
||||
policyh.updateAuditPoliciesFromEnforcementPolices(api)
|
||||
|
||||
elif choice == "F":
|
||||
elif choice.upper() == "F":
|
||||
open_directory(working_dir)
|
||||
elif choice == "S":
|
||||
elif choice.upper() == "S":
|
||||
menu_settings()
|
||||
elif choice == "Q":
|
||||
elif choice.upper() == "B":
|
||||
break
|
||||
else:
|
||||
print(colorText("Invalid choice. Please try again.", "red"))
|
||||
def menu_settings():
|
||||
while True:
|
||||
print(colorText("\n--- 🛠️ Settings Submenu 🛠️ ---", "cyan"))
|
||||
print(colorText("1. Change Working Dir", "cyan"))
|
||||
print(colorText("This Feature is still in development", "cyan"))
|
||||
# print(colorText("2. Sub-option B","cyan"))
|
||||
print(colorText("Q. Return to Main Menu", "cyan"))
|
||||
choice = input("Enter your choice: ")
|
||||
print(colorText("B. 🔙 - Back", "yellow"))
|
||||
choice = get_sanitized_input("Enter your choice: ")
|
||||
|
||||
if choice == "1":
|
||||
pass #TODO ADD CHANGE WORKDIR CODE
|
||||
|
||||
elif choice == "Q":
|
||||
elif choice.upper() == "B":
|
||||
print("Returning to Main Menu...")
|
||||
break
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user