lib/project.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 |
import
os,
json,
logging,
strutils,
sequtils,
pegs
type
NiftyProject* = object
dir*: string
storage*: string
commands*: JsonNode
packages*: JsonNode
const niftyTpl* = "nifty.json".slurp
const systemHelp = "help.json".slurp
let placeholder = peg"'{{' {[^}]+} '}}'"
proc newNiftyProject*(dir: string): NiftyProject =
result.dir = dir
proc configFile*(prj: NiftyProject): string =
return prj.dir/"nifty.json"
proc configured*(prj: NiftyProject): bool =
return fileExists(prj.configFile)
proc init*(prj: var NiftyProject, storage: string) =
prj.storage = storage
createDir(prj.dir/prj.storage)
var o = parseJson(niftyTpl % [prj.storage])
prj.configFile.writeFile(o.pretty)
proc load*(prj: var NiftyProject) =
if not prj.configFile.fileExists:
fatal "Project not initialized - configuration file not found."
quit(10)
let cfg = prj.configFile.parseFile
prj.storage = cfg["storage"].getStr
prj.storage.createDir()
prj.commands = cfg["commands"]
prj.packages = cfg["packages"]
proc help*(prj: NiftyProject): JsonNode =
result = systemHelp.parseJson
for k, v in prj.commands.pairs:
if v.hasKey("_syntax") and v.hasKey("_description"):
result[k] = %*("""
{
"_syntax": "$1",
"_description": "$2"
}
""" % [v["_syntax"].getStr, v["_description"].getStr])
proc save*(prj: NiftyProject) =
var o = newJObject()
o["storage"] = %prj.storage
o["commands"] = %prj.commands
o["packages"] = %prj.packages
prj.configFile.writeFile(o.pretty)
proc map*(prj: var NiftyProject, alias: string, props: JsonNode) =
prj.load
if not prj.packages.hasKey alias:
notice "Adding package definition '$1'..." % alias
prj.packages[alias] = newJObject()
prj.packages[alias]["name"] = %alias
else:
notice "Updating package definition '$1'..." % alias
for key, val in props.pairs:
prj.packages[alias][key] = val
notice " $1: $2" % [key, $val]
prj.save
notice "Package definition '$1' saved." % alias
proc unmap*(prj: var NiftyProject, alias: string) =
prj.load
if not prj.packages.hasKey alias:
warn "Package definition '$1' not found. Nothing to do." % alias
return
prj.packages.delete(alias)
prj.save
notice "Package definition '$1' removed." % alias
proc lookupCommand(prj: NiftyProject, command: string, props: seq[string], cmd: var JsonNode): bool =
if not prj.commands.hasKey command:
warn "Command '$1' not found" % command
return
var cmds = prj.commands[command]
var score = 0
# Cycle through command definitions
for key, val in cmds:
var params = key.split("+")
# Check if all params are available
var match = params.all do (x: string) -> bool:
props.contains(x)
if match and params.len > score:
score = params.len
cmd = val
return score > 0
proc execute*(prj: var NiftyProject, command, alias: string): int =
prj.load
if not prj.packages.hasKey alias:
warn "Package definition '$1' not found within $2. Nothing to do." % [alias, prj.dir]
return
notice "$1: $2" % [command, alias]
let package = prj.packages[alias]
var keys = newSeq[string](0)
for key, val in package.pairs:
keys.add key
var res: JsonNode
var cmd: string
var pwd = prj.storage
if prj.lookupCommand(command, keys, res):
cmd = res["cmd"].getStr.replace(placeholder) do (m: int, n: int, c: openArray[string]) -> string:
return package[c[0]].getStr
if res.hasKey("pwd"):
pwd = res["pwd"].getStr.replace(placeholder) do (m: int, n: int, c: openArray[string]) -> string:
return package[c[0]].getStr
pwd = prj.storage/pwd
notice "Executing: $1" % cmd
pwd.createDir()
pwd.setCurrentDir()
result = execShellCmd cmd
else:
warn "Command '$1' not available for package '$2'" % [command, alias]
setCurrentDir(prj.dir)
proc executeRec*(prj: var NiftyProject, command, alias: string) =
let pwd = getCurrentDir();
if (execute(prj, command, alias) != 0):
return
var childProj = newNiftyProject(pwd/prj.storage/alias)
if childProj.configured:
childProj.load()
setCurrentDir(childProj.dir)
for key, val in childProj.packages.pairs:
childProj.executeRec(command, key)
setCurrentDir(pwd)
|