#!/usr/bin/python
"""This tool reads a yaml file as generated by importpkg.py on stdin and
updates the database with the contents."""

import optparse
import sqlite3
import sys

from debian.debian_support import version_compare
import yaml

def readyaml(db, stream):
    cur = db.cursor()
    cur.execute("PRAGMA foreign_keys = ON;")
    gen = yaml.safe_load_all(stream)
    metadata = next(gen)
    package = metadata["package"]
    cur.execute("SELECT id, version FROM package WHERE name = ?;",
                    (package,))
    row = cur.fetchone()
    if row:
        pid, version = row
        if version_compare(version, metadata["version"]) > 0:
            return
    else:
        pid = None

    cur.execute("BEGIN;")
    cur.execute("SELECT name, id FROM function;")
    funcmapping = dict(cur.fetchall())
    if pid is not None:
        cur.execute("DELETE FROM content WHERE pid = ?;", (pid,))
        cur.execute("DELETE FROM dependency WHERE pid = ?;", (pid,))
        cur.execute("UPDATE package SET version = ?, architecture = ?, source = ? WHERE id = ?;",
                    (metadata["version"], metadata["architecture"], metadata["source"], pid))
    else:
        cur.execute("INSERT INTO package (name, version, architecture, source) VALUES (?, ?, ?, ?);",
                    (package, metadata["version"], metadata["architecture"],
                     metadata["source"]))
        pid = cur.lastrowid
    cur.executemany("INSERT INTO dependency (pid, required) VALUES (?, ?);",
                    ((pid, dep) for dep in metadata["depends"]))
    for entry in gen:
        if entry == "commit":
            db.commit()
            return

        cur.execute("INSERT INTO content (pid, filename, size) VALUES (?, ?, ?);",
                    (pid, entry["name"], entry["size"]))
        cid = cur.lastrowid
        cur.executemany("INSERT INTO hash (cid, fid, hash) VALUES (?, ?, ?);",
                        ((cid, funcmapping[func], hexhash)
                         for func, hexhash in entry["hashes"].items()))
    raise ValueError("missing commit block")

def main():
    parser = optparse.OptionParser()
    parser.add_option("-d", "--database", action="store",
                      default="test.sqlite3",
                      help="path to the sqlite3 database file")
    options, args = parser.parse_args()
    db = sqlite3.connect(options.database)
    readyaml(db, sys.stdin)

if __name__ == "__main__":
    main()