#
# Project: Healthcare Assistant Bot
# Author: Molla Samser
# Author's Email: info@rskworld.com
# Author's Phone: +91 93305 39277
# Author's Website: https://rskworld.in
# Year: 2026
#

import sqlite3

def init_db():
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    
    # Create appointments table
    c.execute('''
        CREATE TABLE IF NOT EXISTS appointments (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            date TEXT NOT NULL,
            time TEXT NOT NULL,
            description TEXT
        )
    ''')
    
    # Create reminders table
    c.execute('''
        CREATE TABLE IF NOT EXISTS reminders (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            medication TEXT NOT NULL,
            time TEXT NOT NULL
        )
    ''')
    
    # Create users table
    c.execute('''
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            username TEXT NOT NULL UNIQUE,
            email TEXT NOT NULL UNIQUE,
            password TEXT NOT NULL,
            role TEXT DEFAULT 'user'
        )
    ''')
    
    conn.commit()
    conn.close()

def add_appointment(date, time, description):
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("INSERT INTO appointments (date, time, description) VALUES (?, ?, ?)", (date, time, description))
    conn.commit()
    conn.close()

def get_all_appointments():
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("SELECT * FROM appointments")
    appointments = c.fetchall()
    conn.close()
    return appointments

def add_reminder(medication, time):
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("INSERT INTO reminders (medication, time) VALUES (?, ?)", (medication, time))
    conn.commit()
    conn.close()

def get_all_reminders():
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("SELECT * FROM reminders")
    reminders = c.fetchall()
    conn.close()
    return reminders

def add_user(username, email, password, role='user'):
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("INSERT INTO users (username, email, password, role) VALUES (?, ?, ?, ?)", (username, email, password, role))
    conn.commit()
    conn.close()

def get_user_by_username(username):
    conn = sqlite3.connect('healthcare.db')
    c = conn.cursor()
    c.execute("SELECT * FROM users WHERE username = ?", (username,))
    user = c.fetchone()
    conn.close()
    return user