from flask import Flask, request, jsonify
import numpy as np
import random
import time

app = Flask(__name__)

# Sample database simulation
shared_data = {
    "users": [
        # Sample user data structure
        # {
        #     "id": 1,
        #     "timestamp": int(time.time()),
        #     "udid": "sample_udid_123",
        #     "platform": "android",
        #     "credit": 100
        # }
    ]
}

# Account model structure
account_model = {
    "id": None,  # int (auto-incremented)
    "timestamp": int(time.time()),  # int (current timestamp)
    "udid": None,  # string
    "platform": None,  # string
    "credit": 100  # int (default 100)
}

# Helper function to generate a new ID
def generate_new_id():
    if not shared_data["users"]:
        return 1
    return max(user["id"] for user in shared_data["users"]) + 1

# Ana sayfa
@app.route('/')
def home():
    return "stablediffusionapi test"

# GET Endpoint'leri
@app.route('/users', methods=['GET'])
def get_users():
    return jsonify({"users": shared_data["users"]})

# GET user by UDID
@app.route('/user/<udid>', methods=['GET'])
def get_user_by_udid(udid):
    # Search for user in the database
    user = next((u for u in shared_data["users"] if u["udid"] == udid), None)
    
    if user:
        # User found, return the user data
        return jsonify({
            "id": user["id"],
            "timestamp": user["timestamp"],
            "udid": user["udid"],
            "platform": user["platform"],
            "credit": user["credit"]
        })
    else:
        # User not found, create new user with 100 credits
        new_user = {
            "id": generate_new_id(),
            "timestamp": int(time.time()),
            "udid": udid,
            "platform": request.headers.get('User-Agent', 'unknown'),  # Get platform from User-Agent
            "credit": 100  # Default credit
        }
        
        # Add to database
        shared_data["users"].append(new_user)
        
        # Return the newly created user
        return jsonify({
            "id": new_user["id"],
            "timestamp": new_user["timestamp"],
            "udid": new_user["udid"],
            "platform": new_user["platform"],
            "credit": new_user["credit"]
        }), 201  # 201 Created status code

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)