src/litestorepkg/lib/cli.nim
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
import
parseopt,
strutils,
json,
os,
strtabs
import
logger,
config,
types,
utils
const favicon = "../../admin/favicon.ico".slurp
var
operation = opRun
directory:string = ""
readonly = false
logLevel = "warn"
mount = false
auth = newJNull()
exOperation:string = ""
customResources = newStringTable()
exFile:string = ""
exBody:string = ""
exType:string = ""
exUri:string = ""
let
usage* = appname & " v" & pkgVersion & " - Lightweight REST Document Store" & """
(c) 2015-2018 Fabio Cevasco
Usage:
litestore [command] [option1 option2 ...]
Commands:
run Start LiteStore server (default if no command specified).
delete Delete a previously-imported specified directory (requires -d).
execute Execute an operation on data stored in the datastore (requires -o, -u, and in certain cases -f or -b and -t).
import Import the specified directory into the datastore (requires -d).
export Export the previously-imported specified directory to the current directory (requires -d).
optimize Optimize search indexes.
vacuum Vacuum datastore.
Options:
-a, --address Specify server address (default: 127.0.0.1).
--auth Specify an authentication/authorization configuration file.
-b, --body Specify a string containing input data for an operation to be executed.
-c, --custom Specify a path containing custom resource definitions.
-d, --directory Specify a directory to serve, import, export, delete, or mount.
-f, --file Specify a file containing input data for an operation to be executed.
-h, --help Display this message.
-l, --log Specify the log level: debug, info, warn, error, none (default: info)
-m, --mount Mirror database changes to the specified directory on the filesystem.
-o, --operation Specify an operation to execute via the execute command: get, put, delete, patch, post, head, options.
-p, --port Specify server port number (default: 9500).
-r, --readonly Allow only data retrieval operations.
-s, --store Specify a datastore file (default: data.db)
-t, --type Specify a content type for the body an operation to be executed via the execute command.
-u, --uri Specify an uri to execute an operation through the execute command.
-v, --version Display the program version.
"""
for kind, key, val in getOpt():
case kind:
of cmdArgument:
case key:
of "run":
operation = opRun
of "import":
operation = opImport
of "execute":
operation = opExecute
of "export":
operation = opExport
of "delete":
operation = opDelete
of "optimize":
operation = opOptimize
of "vacuum":
operation = opVacuum
else:
discard
of cmdLongOption, cmdShortOption:
case key:
of "address", "a":
if val == "":
fail(100, "Address not specified.")
address = val
of "port", "p":
if val == "":
fail(101, "Port not specified.")
port = val.parseInt
of "store", "s":
file = val
of "log", "l":
if val == "":
fail(102, "Log level not specified.")
case val:
of "info":
LOG.level = lvInfo
of "warn":
LOG.level = lvWarn
of "debug":
LOG.level = lvDebug
of "error":
LOG.level = lvError
of "none":
LOG.level = lvNone
else:
fail(103, "Invalid log level '$1'" % val)
loglevel = val
of "directory", "d":
if val == "":
fail(104, "Directory not specified.")
directory = val
of "custom", "c":
if val == "":
fail(115, "Custom resources path not specified.")
if not val.existsDir():
fail(116, "Custom resources directory does not exist.")
for file in val.walkDir():
if file.kind == pcFile or file.kind == pcLinkToFile:
customResources[file.path.splitFile[1]] = file.path.readFile()
of "operation", "o":
if val == "":
fail(106, "Operation not specified.")
exOperation = val
of "file", "f":
if val == "":
fail(107, "File not specified.")
exFile = val
of "uri", "u":
if val == "":
fail(108, "URI not specified.")
exUri = val
of "body", "b":
if val == "":
fail(112, "Body not specified.")
exBody = val
of "type", "t":
if val == "":
fail(113, "Content type not specified.")
exType = val
of "auth":
if val == "":
fail(114, "Authentication/Authorization configuration file not specified.")
auth = val.parseFile
of "mount", "m":
mount = true
of "version", "v":
echo pkgVersion
quit(0)
of "help", "h":
echo usage
quit(0)
of "readonly", "r":
readonly = true
else:
discard
else:
discard
# Validation
if directory == "" and (operation in [opDelete, opImport, opExport] or mount):
fail(105, "--directory option not specified.")
if exFile == "" and (exOperation in ["put", "post", "patch"]):
fail(109, "--file option not specified")
if exUri == "" and operation == opExecute:
fail(110, "--uri option not specified")
if exOperation == "" and operation == opExecute:
fail(111, "--operation option not specified")
LS.operation = operation
LS.address = address
LS.port = port
LS.file = file
LS.directory = directory
LS.readonly = readonly
LS.favicon = favicon
LS.loglevel = loglevel
LS.auth = auth
LS.customResources = customResources
LS.mount = mount
LS.execution.file = exFile
LS.execution.body = exBody
LS.execution.ctype = exType
LS.execution.uri = exUri
LS.execution.operation = exOperation
|