norgbackup/norg/config/init.nim

113 lines
4 KiB
Nim

import parsetoml
import streams
import os
import ../model/config_type
import ../model/command_type
import ../model/encryption_type
import ../model/args_type
import notifier_config
import actions_config
import maintenance_config
export args_type
export config_type
proc parseSourceDirectories*(in_conf: TomlValueRef): seq[string] =
var src_dirs: seq[string] = @[]
for dir in in_conf{"source_directories"}.getElems():
src_dirs.add(dir.getStr())
# this is here for backwards compatibiliy
for dir in in_conf{"source_dirs"}.getElems():
norg_config.source_directories.add(dir.getStr())
return src_dirs
proc parseRepositories*(rep_conf: TomlValueRef): seq[Repository] =
var repos: seq[Repository] = @[]
for r in rep_conf.getElems():
let rtable = r.getTable()
var repo = Repository()
repo.path = rtable["path"].getStr()
repo.label = rtable["label"].getStr()
if rtable.hasKey("append_only"):
repo.append_only = rtable["append_only"].getBool()
if rtable.hasKey("tool"):
repo.tool = rtable["tool"].getStr("borg").toBackupTool()
repos.add(repo)
return repos
proc parseEncryption*(enc_conf: TomlValueRef) =
setEncryptionPassphrase(enc_conf{"encryption_passphrase"}.getStr(""))
setEncryptionPassphraseFD(enc_conf{"encryption_passphrase_fd"}.getStr(""))
setEncryptionPassphraseFile(enc_conf{"encryption_passphrase_file"}.getStr(""))
setEncryptionPassCommand(enc_conf{"encryption_passcommand"}.getStr(""))
proc parseLogging(nc: var NorgConfig, log_conf: TomlValueRef) =
let ll = log_conf{"log_level"}.getStr()
if ll != "":
nc.log_info.level = ll.toLogLevel()
nc.log_info.file = log_conf{"log_file"}.getStr()
discard
proc addConfigBinaries*(current: var Binaries, bin_conf: TomlValueRef) =
let
borg = bin_conf{"borg_bin"}.getStr()
restic = bin_conf{"restic_bin"}.getStr()
if borg != "":
current.borg.insert(borg,0)
if restic != "":
current.restic.insert(restic,0)
proc addExclusions*(current: var Exclusions, exc_conf: TomlValueRef) =
current.exclude_file = exc_conf{"exclude_file"}.getStr()
current.excludes = @[]
current.patterns_file = exc_conf{"patterns_file"}.getStr()
current.patterns = @[]
for e in exc_conf{"excludes"}.getElems():
current.excludes.add(e.getStr())
for p in exc_conf{"patterns"}.getElems():
current.patterns.add(p.getStr())
proc parseConfigFile*(file: string): NorgConfig =
if not fileExists(file):
echo "No Config File provided or file does not exist."
return NorgConfig()
let in_conf = parsetoml.parseFile(file)
norg_config.source_directories = parseSourceDirectories(in_conf)
# Add config file to backup
norg_config.source_directories.add(file)
norg_config.repositories = parseRepositories(in_conf{"repositories"})
if norg_config.args.repository != "":
echo "Filtering repo list to selected: ", norg_config.args.repository
norg_config.repositories = norg_config.repositories.findRepository(norg_config.args.repository)
parseEncryption(in_conf{"encryption"})
norg_config.notifiers = parseNotifiers(in_conf)
norg_config.actions = parseActions(in_conf{"actions"})
norg_config.maintenance = parseMaintenance(in_conf{"maintenance"})
norg_config.bins.addConfigBinaries(in_conf{"binaries"})
norg_config.bins.setBinaryLocations()
norg_config.exclusions.addExclusions(in_conf{"exclusions"})
norg_config.parseLogging(in_conf{"logging"})
return norg_config
const SAMPLE_CONFIG = staticRead("../../docs/static/downloads/norg.toml.sample")
proc generateConfigFile*(file: string) =
if fileExists(file):
echo "Sensibly not overwriting existing file: " & file
echo "Exiting"
return
echo "Generating Configuration to " & file
var r_strm = newStringStream(SAMPLE_CONFIG)
var w_strm = newFileStream(file, fmWrite)
var line = ""
if r_strm != nil and w_strm != nil:
while r_strm.readLine(line):
w_strm.writeLine("#" & line)
r_strm.close()
w_strm.close()
echo "Done. Now edit " & file & " to your requirements and run:"
echo "norg -c " & file & " init"
else:
echo "Error writing sample config file"