import database


class TgDatabase(database.BasicDatabase):
    def __init__(self):
        super().__init__('tg.db')

        self.cursor.execute("""
        CREATE TABLE IF NOT EXISTS chats (
            "id"	    INTEGER,
            "active"	INTEGER NOT NULL DEFAULT 0,
            "rules"             TEXT,
            "greeting_join"	    TEXT,
            "ai_prompt"         TEXT,
            PRIMARY KEY("id"))
        """)

        self.cursor.execute("""
        CREATE TABLE IF NOT EXISTS users (
            "chat_id"	INTEGER,
            "user_id"	INTEGER,
            "last_message"	INTEGER NOT NULL DEFAULT 0,
            "messages_today"	INTEGER NOT NULL DEFAULT 0,
            "messages_month"	INTEGER NOT NULL DEFAULT 0,
            "warnings"	INTEGER NOT NULL DEFAULT 0,
            PRIMARY KEY("chat_id","user_id"))
        """)

        self.conn.commit()


DB = TgDatabase()