#!/usr/bin/env python3
"""Load packages into a SQLite database with the tables of sql/doc.sql and sql/store.sql.

Non-normative tooling.
Usage: python3 tools/load.py --sqlite lab.db PACKAGE [PACKAGE ...]
       python3 tools/load.py --sqlite lab.db --bundle --built-from REV PACKAGE [PACKAGE ...]
With --bundle the new file is a scienting-bundle/1: it also gets the tables of sql/bundle-sqlite.sql,
the full-text search index and bundle_meta (spec/13 §13.5).
Each package is validated first, and nothing is loaded if any package fails.
Files without tables of their own (assets, datasets, derived files) are copied by SHA-256 into an
asset store folder beside the database, named <database>.assets/.
"""

import argparse
import os
import sqlite3
import sys

ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ROOT)

from tools.oracle import db, package  # noqa: E402
from tools.oracle.schema import Store  # noqa: E402


def main():
    ap = argparse.ArgumentParser(description=__doc__.splitlines()[0])
    ap.add_argument("--sqlite", required=True, help="the database file; created with the tables if new")
    ap.add_argument("--bundle", action="store_true", help="build a new scienting-bundle/1 file with search")
    ap.add_argument("--built-from", help="with --bundle: the store revision the packages come from")
    ap.add_argument("--store", help="with --bundle: the store's name (default: the packages' common store)")
    ap.add_argument("packages", nargs="+", help="package folders")
    a = ap.parse_args()
    if a.bundle and (os.path.exists(a.sqlite) or not a.built_from):
        ap.error("--bundle needs --built-from and a database file that does not exist yet")
    store, bad = Store(os.path.join(ROOT, "schemas")), False
    for p in a.packages:
        for e in package.validate(p, store).errors:
            print("%s: %s: %s: %s" % (p, e["where"], e["rule"], e["message"]), file=sys.stderr)
            bad = True
    if bad:
        sys.exit("Nothing was loaded.")
    new = not os.path.exists(a.sqlite)
    conn = sqlite3.connect(a.sqlite)
    if a.bundle:
        db.create_bundle(conn, os.path.join(ROOT, "sql"))
    elif new:
        db.create(conn, os.path.join(ROOT, "sql"))
    for p in a.packages:
        key = db.load(conn, p, a.sqlite + ".assets")
        n = conn.execute("SELECT COUNT(*) FROM nodes WHERE doc_key = ?", (key,)).fetchone()[0]
        if a.bundle:
            print("%s: %d nodes, %d search rows" % (key, n, db.index(conn, key)))
        else:
            print("%s: %d nodes" % (key, n))
    if a.bundle:
        stores = {r[0] for r in conn.execute("SELECT DISTINCT store FROM documents WHERE store IS NOT NULL")}
        store = a.store or (stores.pop() if len(stores) == 1 else None)
        if store is None:
            sys.exit("The packages name no single store; give one with --store.")
        db.write_meta(conn, store, a.built_from)
    conn.commit()


if __name__ == "__main__":
    main()