Merge pull request #59 from mbrg/feature/whoiam

add gui for copilot
mbrg · Aug 6, 2024 · 7993579 · 7993579
2 parents 931e63a + 9e90fd5
commit 7993579
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/powerpwn/cli/arguments.py b/src/powerpwn/cli/arguments.py
@@ -161,6 +161,11 @@ def module_copilot(command_subparsers: argparse.ArgumentParser):
     dump.add_argument("-d", "--directory", type=str, required=True, help="Path to whoami output directory")
     dump.add_argument("-g", "--gui", action="store_true", help="Run local server for gui.")
 
+    gui = copilot_subparsers.add_parser(
+        "gui", description="Browse data in a local server", help="Browse collected data in a simple gui on a local server"
+    )
+    gui.add_argument("-d", "--directory", type=str, required=True, help="Data directory")
+
 
 def copilot_modules(parser):
     parser.add_argument("-u", "--user", required=True, type=str, help="User email to connect.")

diff --git a/src/powerpwn/cli/runners.py b/src/powerpwn/cli/runners.py
@@ -178,6 +178,10 @@ def run_phishing_command(args):
 
 
 def run_copilot_chat_command(args):
+    if args.copilot_subcommand == "gui":
+        CopilotGui().run(args.directory)
+        return
+
     parsed_args = ChatArguments(
         user=args.user,
         password=args.password,