import sqlite3, json, time

db_path = '/root/.9router/db/data.sqlite'
try:
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    rows = cursor.execute("SELECT id, data FROM providerConnections WHERE provider='antigravity'").fetchall()
    unlocked_count = 0
    for cid, d in rows:
        if d:
            data = json.loads(d)
            if data.get('testStatus') == 'unavailable' or data.get('errorCode') in [404, 429]:
                data['testStatus'] = 'active'
                data['errorCode'] = None
                data['backoffLevel'] = 0
                cursor.execute("UPDATE providerConnections SET data=? WHERE id=?", (json.dumps(data), cid))
                unlocked_count += 1
    if unlocked_count > 0:
        conn.commit()
        print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Auto-unlocked {unlocked_count} AG account(s)")
    conn.close()
except Exception as e:
    print(f"Error in auto_unlock_ag: {e}")