from flask import current_app
from flask import g
import sqlite3

from book import Book

def get_db():
    """Connect to the application's configured database. The connection
    is unique for each request and will be reused if this is called
    again.
    """
    if "db" not in g:
        g.db = sqlite3.connect(
            current_app.config["DATABASE"], detect_types=sqlite3.PARSE_DECLTYPES
        )
        g.db.row_factory = sqlite3.Row

    return g.db

def get_book(isbn):
    db = get_db()

    data = db.execute(
        """
        SELECT * FROM book WHERE isbn=?
        """,
        (isbn,)
    ).fetchone()

    if data is None:
        raise IndexError("Book not in db")

    book = Book(isbn)
    book._manual_load(
        data["title"],
        publisher=data["publisher"],
        publish_date=data["publish_date"],
        author=data["author"],
        count=data["count"]
    )
    return book

def increment_count(book):
    if book.count == -1:
        book = get_book(book.isbn)

    db = get_db()
    db.execute(
        """
        UPDATE book SET count=?
        WHERE isbn=?
        """,
        (book.count+1, book.isbn)
    )
    db.commit()

def add_book(book):
    try:
        book = get_book(book.isbn)
        increment_count(book)
        return "duplicate"
    except IndexError:
        if book.count == -1:
            book.count = 1

        db = get_db()
        db.execute(
            """
            INSERT INTO book (isbn, count, title, author, publisher, publish_date)
            VALUES (?, ?, ?, ?, ?, ?)
            """,
            (book.isbn, book.count, book.title, book.author, book.publisher, book.publish_date)
        )
        db.commit()
        return "added"