all repos — pls @ 4b069128d910c3155f8d6cb454def7e8754726d2

A polite but determined task runner.

src/plspkg/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
 143
 144
 145
 146
 147
 148
 149
 150
 151
 152
 153
 154
import
  os,
  json,
  logging,
  strutils,
  sequtils,
  pegs

type
  PlsProject* = object
    version*: int
    dir*: string
    tasks*: JsonNode
    targets*: JsonNode


type PlsError = ref object of ValueError 

const plsTpl* = "pls.json".slurp
const systemHelp = "help.json".slurp

let systemProps = @["$$os:$1" % hostOS, "$$cpu:$1" % hostCPU]
let placeholder = peg"'{{' {[^}]+} '}}'"

proc newPlsProject*(dir: string): PlsProject =
  result.dir = dir

proc configFile*(prj: PlsProject): string = 
  return prj.dir/"pls.json"

proc configured*(prj: PlsProject): bool =
  return fileExists(prj.configFile)

proc init*(prj: var PlsProject) =
  var o = parseJson(plsTpl)
  prj.configFile.writeFile(o.pretty)

proc load*(prj: var PlsProject) =
  if not prj.configFile.fileExists:
    fatal "Project not initialized - configuration file not found."
    quit(10)
  let cfg = prj.configFile.parseFile
  prj.version = cfg["version"].getInt
  prj.tasks = cfg["tasks"]
  prj.targets = cfg["targets"]

proc help*(prj: var PlsProject): JsonNode =
  result = newJObject()
  if prj.configured:
    prj.load
    for k, v in systemHelp.parseJson.pairs:
      result[k] = v
    for k, v in prj.tasks.pairs:
      if v.hasKey("$syntax") and v.hasKey("$description"):
        result[k] = ("""
          {
            "$$syntax": "$1",
            "$$description": "$2"
          }
        """ % [v["$syntax"].getStr, v["$description"].getStr]).parseJson

proc save*(prj: PlsProject) = 
  var o = newJObject()
  o["version"] = %prj.version
  o["tasks"] = %prj.tasks
  o["targets"] = %prj.targets
  prj.configFile.writeFile(o.pretty)

proc defTarget*(prj: var PlsProject, alias: string, props: var JsonNode) =
  for k, v in props.mpairs:
    if v == newJNull():
      props.delete(k)
  if not prj.targets.hasKey alias:
    notice "Adding target '$1'..." % alias
    prj.targets[alias] = newJObject()
    prj.targets[alias]["name"] = %alias
  else:
    notice "Updating target '$1'..." % alias
    prj.targets[alias] = newJObject()
  for key, val in props.pairs:
    prj.targets[alias][key] = val
    notice "  $1: $2" % [key, $val]
  prj.save
  notice "Target '$1' saved." % alias

proc undefTarget*(prj: var PlsProject, alias: string) =
  prj.targets.delete(alias)
  prj.save
  notice "Target '$1' removed." % alias

proc defTask*(prj: var PlsProject, alias: string, props: var JsonNode) =
  for k, v in props.mpairs:
    if v == newJNull():
      props.delete(k):
    elif v.kind == JObject:
      for kk, vv in v.pairs:
        if vv == newJNull():
          v.delete(kk)
  if not prj.tasks.hasKey alias:
    notice "Adding task '$1'..." % alias
    prj.tasks[alias] = newJObject()
  else:
    notice "Updating task '$1'..." % alias
    prj.tasks[alias] = newJObject()
  for key, val in props.pairs:
    prj.tasks[alias][key] = val
    notice "  $1: $2" % [key, $val]
  prj.save
  notice "Task '$1' saved." % alias

proc undefTask*(prj: var PlsProject, alias: string) =
  prj.load
  prj.tasks.delete(alias)
  prj.save
  notice "Task '$1' removed." % alias

proc lookupTask(prj: PlsProject, task: string, ps: seq[string], cmd: var JsonNode): bool =
  let props = ps.concat(systemProps);
  if not prj.tasks.hasKey task:
    warn "Task '$1' not found" % task
    return
  var cmds = prj.tasks[task]
  var score = 0
  # Cycle through task definitions
  for key, val in cmds:
    if key == "$syntax" or key == "$description":
      continue
    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 PlsProject, task, alias: string): int {.discardable.} =
  prj.load
  if not prj.targets.hasKey alias:
    raise PlsError(msg: "Target definition '$1' not found. Nothing to do." % [alias])
  let target = prj.targets[alias]
  var keys = newSeq[string](0)
  for key, val in target.pairs:
    keys.add key
  var res: JsonNode
  var cmd: string
  if prj.lookupTask(task, keys, res):
    cmd = res["cmd"].getStr.replace(placeholder) do (m: int, n: int, c: openArray[string]) -> string:
      return target[c[0]].getStr
    notice "Executing: $1" % cmd
    result = execShellCmd cmd
  else:
    debug "Task '$1' not available for target '$2'" % [task, alias]
  setCurrentDir(prj.dir)