From df8bc1c203492110ef1aa66b8fd2dc40fbd425a5 Mon Sep 17 00:00:00 2001
From: Alexander Zaft <a.zaft@fz-juelich.de>
Date: Thu, 30 Mar 2023 09:45:38 +0200
Subject: [PATCH] cli: add argparse and inlcudes before repl

+ add argparse to cli
+ add option for files that are executed after connection to the modules
  but before repl starts
+ add option to skip interactive mode after executing files

Change-Id: I8f01db84b2c91d4bf1a7b397e8fa1bf0c87ddf0d
Reviewed-on: https://forge.frm2.tum.de/review/c/secop/frappy/+/30823
Tested-by: Jenkins Automated Tests <pedersen+jenkins@frm2.tum.de>
Tested-by: Markus Zolliker <markus.zolliker@psi.ch>
Reviewed-by: Alexander Zaft <a.zaft@fz-juelich.de>
---
 bin/frappy-cli | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/bin/frappy-cli b/bin/frappy-cli
index c929ff2..0c10f55 100755
--- a/bin/frappy-cli
+++ b/bin/frappy-cli
@@ -26,6 +26,7 @@ from __future__ import print_function
 
 import sys
 import code
+import argparse
 from os import path
 
 # Add import path for inplace usage
@@ -33,6 +34,19 @@ sys.path.insert(0, path.abspath(path.join(path.dirname(__file__), '..')))
 
 from frappy.client.interactive import Client, watch
 
+def parseArgv(argv):
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-i', '--include',
+                        help='file to execute after connecting to the clients', metavar='file',
+                        type=str, action='append', default=[])
+    parser.add_argument('-o', '--only-execute',
+                        help='Do not go into interactive mode after executing files. \
+                        Has no effect without --include.', action='store_true')
+    parser.add_argument('node',
+                        help='Nodes the client should connect to.\n', metavar='host:port',
+                        nargs='*', type=str, default=[])
+    return parser.parse_args(argv)
+
 _USAGE = """
 Usage:
 %s
@@ -55,14 +69,16 @@ c = Client('localhost:5000')
 
 Client.show_usage = False
 
-if len(sys.argv) < 2:
+
+args = parseArgv(sys.argv[1:])
+if not args.node:
     _usage_args = ("\ncli = Client('localhost:5000')\n", 'cli')
     success = True
 else:
     _usage_args = ('', '_c0')
     success = False
 
-for _idx, _node in enumerate(sys.argv[1:]):
+for _idx, _node in enumerate(args.node):
     _client_name = '_c%d' % _idx
     try:
         setattr(sys.modules['__main__'], _client_name, Client(_node, name=_client_name))
@@ -70,6 +86,19 @@ for _idx, _node in enumerate(sys.argv[1:]):
     except Exception as e:
         print(repr(e))
 
+
+file_success = False
+try:
+    for file in args.include:
+        with open(file, 'r') as f:
+            exec(f.read())
+    file_success = True
+except Exception as e:
+    print('Error while executing %s: %s' % (file, e))
+
 if success:
+    if args.include and file_success and args.only_execute:
+        print('skipping interactive mode')
+        exit()
     print(_USAGE % _usage_args)
     code.interact(banner='', local=sys.modules['__main__'].__dict__)