#!/usr/bin/env python

# $Id$
#
# Copyright (C) 2015-2016  Parsons Government Services ("PARSONS")
# Portions copyright (C) 2014  Dragon Research Labs ("DRL")
# Portions copyright (C) 2010-2013  Internet Systems Consortium ("ISC")
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notices and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND PARSONS, DRL, AND ISC DISCLAIM
# ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL
# PARSONS, DRL, OR ISC BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR
# CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
# OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
# NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
# WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

"""
Back up data from SQL databases, looking at config file to figure out
which databases and what credentials to use with them, and eliminating
duplicates in cases where we've configured multiple applications to
share a single database.
"""

import os
import sys
import time
import argparse
import subprocess
import rpki.config

os.environ["TZ"] = "UTC"
time.tzset()

cfg = rpki.config.argparser(doc = __doc__, section = "myrpki")
cfg.argparser.add_argument("-o", "--output", type = argparse.FileType("wb"), default = sys.stdout,
                           help = "destination for SQL dump (default: stdout)")
cfg.argparser.add_argument("-v", "--verbose", action = "store_true",
                           help = "whistle while you work")
args = cfg.argparser.parse_args()

templates = dict(mysql      = "mysqldump --add-drop-database -u{username} -p{password} -B{database}",
                 sqlite3    = "sqlite3 {database} .dump",
                 postgresql = "sudo -u {username} pg_dump {database}")

cmds = []

for name in ("rpkid", "irdbd", "pubd"):
    if cfg.getboolean("start_" + name, False):
        cmd = templates[cfg.get("sql-engine", section = name)]
        cmd = cmd.format(database = cfg.get("sql-database", section = name),
                         username = cfg.get("sql-username", section = name),
                         password = cfg.get("sql-password", section = name))
        if cmd not in cmds:
            cmds.append(cmd)

for cmd in cmds:
    if args.verbose:
        sys.stderr.write("[Running \"{}\"]\n".format(cmd))
    subprocess.check_call(cmd.split(), stdout = args.output)