import sqlite3

class DB:
    def __init__(self, conn):
        self.conn = conn
        pass

    def get_subforums(self):
        return self._db().execute('select forum_id, name, description from subforums').fetchall()

    def get_subforum(self, subforum):
        return self._db().execute('select name, description from subforums where forum_id = ?', (subforum,)).fetchone()

    def get_threads(self, subforum):
        return self._db().execute('select thread_id, title from threads where forum_id = ?', (subforum,)).fetchall()

    def get_thread(self, thread):
        db = self._db()
        title, text, author, author_id = db.execute('''
            select title, text, name, author_id
            from threads, users
            where thread_id = ? and author_id = user_id
            ''',
            (thread,)
        ).fetchone()
        comments = db.execute('''
            select comment_id, parent_id, name, text
            from comments, users
            where thread_id = ? and author_id = user_id
            ''',
            (thread,)
        ).fetchall()
        return title, text, author, author_id, comments

    def get_thread_title(self, thread_id):
        return self._db().execute('''
            select title
            from threads
            where thread_id = ?
            ''',
            (thread_id,)
        ).fetchone()

    def get_comments(self, thread):
        return self._db().execute('''
            select text
            from comments
            where thread_id = ?
            ''',
            (thread,)
        ).fetchall()

    def get_comment_tree(self, comment):
        db = self._db()
        parent = db.execute('select text from comments where comment_id = ?', (comment,)).fetchall()
        children = db.execute('select text from comments where parent_id = ?', (comment,)).fetchall()
        print(parent, children)
        return str(parent) + str(children)
        return parent

    def get_user_password(self, username):
        return self._db().execute('''
            select user_id, password
            from users
            where name = ?
            ''',
            (username,)
        ).fetchone()

    def get_user_public_info(self, user_id):
        return self._db().execute('''
            select name, about
            from users
            where user_id = ?
            ''',
            (user_id,)
        ).fetchone()

    def get_user_private_info(self, user_id):
        return self._db().execute('''
            select about
            from users
            where user_id = ?
            ''',
            (user_id,)
        ).fetchone()

    def set_user_private_info(self, user_id, about):
        db = self._db()
        db.execute('''
            update users
            set about = ?
            where user_id = ?
            ''',
            (about, user_id)
        )
        db.commit()

    def add_thread(self, author_id, forum_id, title, text, time):
        db = self._db()
        c = db.cursor()
        c.execute('''
            insert into threads (author_id, forum_id, title, text,
                create_time, modify_time, update_time)
            values (?, ?, ?, ?, ?, ?, ?)
            ''',
            (author_id, forum_id, title, text, time, time, time)
        )
        rowid = c.lastrowid
        db.commit()
        return db.execute('''
            select thread_id
            from threads
            where rowid = ?
            ''',
            (rowid,)
        ).fetchone()

    def delete_thread(self, thread_id):
        db = self._db()
        db.execute('''
            delete
            from threads
            where thread_id = ?
            ''',
            (thread_id,)
        )
        db.commit()

    def _db(self):
        return sqlite3.connect(self.conn)