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 = %*(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) |