import sqlite3
import os
from flask import g

def get_db_path():
    if os.path.exists('/data'):
        return '/data/udids.db'
    elif os.path.exists('/home/user_data'):
        return '/home/user_data/udids.db'
    return 'udids.db'

def get_db():
    if 'db' not in g:
        db_path = get_db_path()
        os.makedirs(os.path.dirname(db_path), exist_ok=True)
        g.db = sqlite3.connect(db_path)
        g.db.row_factory = sqlite3.Row
    return g.db

def close_db(e=None):
    db = g.pop('db', None)
    if db is not None:
        db.close()

def init_db(app):
    with app.app_context():
        db = get_db()
        cursor = db.cursor()
        
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS udids (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                udid TEXT NOT NULL UNIQUE
            )
        ''')
        
        # Örnek veri ekleme
        cursor.execute('SELECT COUNT(*) FROM udids')
        if cursor.fetchone()[0] == 0:
            cursor.execute('INSERT INTO udids (udid) VALUES (?)', ('deneme-udid',))
        
        db.commit()