import sqlite3
import json


class NovelCiwei:
    def __init__(self, db_path: str):
        self._db = sqlite3.connect(db_path, check_same_thread=False)

    def get_book_in_shelf(self, book_id: int):
        cur = self._db.execute(
            'SELECT book_info FROM shelf_book_info WHERE book_id = ?;',
            [str(book_id)])
        for i in cur:
            return json.loads(i[0])

    def get_book_in_readhistory(self, book_id: int):
        cur = self._db.execute(
            'SELECT book_info FROM read_history WHERE book_id = ?;',
            [str(book_id)])
        for i in cur:
            return json.loads(i[0])

    def get_all_books(self):
        cur = self._db.execute('SELECT book_id, COUNT(*) FROM catalog1 WHERE is_download = 1 GROUP BY book_id;')  # noqa: E501
        return cur.fetchall()

    def get_books(self):
        cur = self._db.execute('SELECT * FROM shelf_book_info;')
        cur.row_factory = sqlite3.Row
        return cur.fetchall()

    def get_chapter_with_bookid(self, book_id: int):
        cur = self._db.execute(
            'SELECT * FROM catalog1 WHERE book_id = ? ORDER BY chapter_index;',
            [str(book_id)])
        cur.row_factory = sqlite3.Row
        return cur.fetchall()

    def get_chapter_with_bookid_division(self, book_id: int, division_id: int):
        cur = self._db.execute(
            'SELECT * FROM catalog1 WHERE book_id = ? AND division_id = ? AND is_download = 1 ORDER BY chapter_index;',  # noqa: E501
            [str(book_id), str(division_id)])
        cur.row_factory = sqlite3.Row
        return cur.fetchall()

    def get_divisions_with_bookid(self, book_id: int):
        cur = self._db.execute('SELECT * FROM division WHERE book_id = ? ORDER BY division_index;', [str(book_id)])  # noqa: E501
        cur.row_factory = sqlite3.Row
        return cur.fetchall()

    def get_chapter(self, chapter_id: int):
        cur = self._db.execute(
            'SELECT * FROM catalog1 WHERE chapter_id = ?;', [str(chapter_id)])
        cur.row_factory = sqlite3.Row
        return cur.fetchone()

    def get_read_history(self):
        cur = self._db.execute(
            'SELECT * FROM read_history ORDER BY readtime DESC;')
        cur.row_factory = sqlite3.Row
        return cur.fetchall()