44 lines
1.4 KiB
Nim
44 lines
1.4 KiB
Nim
|
import parsetoml, strutils
|
||
|
|
import ./[types, utils]
|
||
|
|
|
||
|
|
proc findKey(profile: Profile, path: string): TomlValueRef =
|
||
|
|
let keys = path.split(".")
|
||
|
|
let target = keys[keys.high]
|
||
|
|
|
||
|
|
var current = profile
|
||
|
|
for i in 0 ..< keys.high:
|
||
|
|
let temp = current.getOrDefault(keys[i])
|
||
|
|
if temp == nil:
|
||
|
|
return nil
|
||
|
|
current = temp
|
||
|
|
|
||
|
|
return current.getOrDefault(target)
|
||
|
|
|
||
|
|
# Takes a specific "."-separated path as input and returns a default value if the key does not exits
|
||
|
|
# Example: cq.profile.getString("http-get.agent.heartbeat.prefix", "not found") returns the string value of the
|
||
|
|
# prefix key, or "not found" if the target key or any sub-tables don't exist
|
||
|
|
|
||
|
|
proc getString*(profile: Profile, path: string, default: string = ""): string =
|
||
|
|
let key = profile.findKey(path)
|
||
|
|
if key == nil:
|
||
|
|
return default
|
||
|
|
return key.getStr(default)
|
||
|
|
|
||
|
|
proc getBool*(profile: Profile, path: string, default: bool = false): bool =
|
||
|
|
let key = profile.findKey(path)
|
||
|
|
if key == nil:
|
||
|
|
return default
|
||
|
|
return key.getBool(default)
|
||
|
|
|
||
|
|
proc getInt*(profile: Profile, path: string, default = 0): int =
|
||
|
|
let key = profile.findKey(path)
|
||
|
|
if key == nil:
|
||
|
|
return default
|
||
|
|
return key.getInt(default)
|
||
|
|
|
||
|
|
proc getTable*(profile: Profile, path: string): TomlTableRef =
|
||
|
|
let key = profile.findKey(path)
|
||
|
|
if key == nil:
|
||
|
|
return new TomlTableRef
|
||
|
|
return key.getTable()
|