1 files changed, 61 insertions, 0 deletions
diff --git a/src/spdk/scripts/spdkcli.py b/src/spdk/scripts/spdkcli.py
new file mode 100755
index 00000000..71d8857f
--- /dev/null
+++ b/src/spdk/scripts/spdkcli.py
@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+import sys
+import argparse
+import configshell_fb
+from os import getuid
+from configshell_fb import ConfigShell, shell
+from spdkcli import UIRoot
+from pyparsing import (alphanums, Optional, Suppress, Word, Regex,
+                       removeQuotes, dblQuotedString, OneOrMore)
+
+
+def add_quotes_to_shell(spdk_shell):
+    command = shell.locatedExpr(Word(alphanums + '_'))('command')
+    value = dblQuotedString.addParseAction(removeQuotes)
+    value_word = Word(alphanums + ';,=_\+/.<>()~@:-%[]')
+    keyword = Word(alphanums + '_\-')
+    kparam = shell.locatedExpr(keyword + Suppress('=') +
+                               Optional(value | value_word, default=''))('kparams*')
+    pparam = shell.locatedExpr(value | value_word)('pparams*')
+    parameters = OneOrMore(kparam | pparam)
+    bookmark = Regex('@([A-Za-z0-9:_.]|-)+')
+    pathstd = Regex('([A-Za-z0-9:_.\[\]]|-)*' + '/' + '([A-Za-z0-9:_.\[\]/]|-)*') \
+        | '..' | '.'
+    path = shell.locatedExpr(bookmark | pathstd | '*')('path')
+    spdk_shell._parser = Optional(path) + Optional(command) + Optional(parameters)
+
+
+def main():
+    """
+    Start SPDK CLI
+    :return:
+    """
+    spdk_shell = ConfigShell("~/.scripts")
+    add_quotes_to_shell(spdk_shell)
+
+    parser = argparse.ArgumentParser(description="SPDK command line interface")
+    parser.add_argument("-s", dest="socket", help="RPC socket path", default="/var/tmp/spdk.sock")
+    parser.add_argument("-v", dest="verbose", help="Print request/response JSON for configuration calls",
+                        default=False, action="store_true")
+    parser.add_argument("commands", metavar="command", type=str, nargs="*", default="",
+                        help="commands to execute by SPDKCli as one-line command")
+    args = parser.parse_args()
+
+    root_node = UIRoot(args.socket, spdk_shell)
+    root_node.verbose = args.verbose
+    try:
+        root_node.refresh()
+    except BaseException:
+        pass
+
+    if len(args.commands) > 0:
+        spdk_shell.run_cmdline(" ".join(args.commands))
+        sys.exit(0)
+
+    spdk_shell.con.display("SPDK CLI v0.1")
+    spdk_shell.con.display("")
+    spdk_shell.run_interactive()
+
+
+if __name__ == "__main__":
+    main()