server.py

from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse, HTMLResponse
from pydantic import BaseModel
import pyodbc
from fastapi.middleware.cors import CORSMiddleware
import os

app = FastAPI()

# CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# ✅ FIX: Kiểm tra thư mục static tồn tại
if not os.path.exists("static"):
    os.makedirs("static")

# Serve static files
app.mount("/static", StaticFiles(directory="static"), name="static")

# Kết nối SQL Server
conn_str = (
    "DRIVER={ODBC Driver 17 for SQL Server};"
    "SERVER=HYHONCUTE;"
    "DATABASE=RescueDB;"
    "Trusted_Connection=yes;"
    "TrustServerCertificate=yes;"
)

def get_conn():
    try:
        return pyodbc.connect(conn_str)
    except Exception as e:
        print(f"❌ Lỗi kết nối DB: {e}")
        raise

# Models
class RescueReport(BaseModel):
    name: str
    phone: str
    message: str
    lat: float
    lng: float
    address: str = ""

# Root -> Form gửi cứu hộ
@app.get("/")
def read_root():
    return FileResponse("static/index.html")

# ✅ FIX: Bản đồ với error handling
@app.get("/map")
def read_map():
    try:
        if not os.path.exists("static/map.html"):
            return HTMLResponse(
                content="<h1>❌ File map.html không tồn tại!</h1><p>Vui lòng tạo file static/map.html</p>",
                status_code=404
            )
        return FileResponse("static/map.html")
    except Exception as e:
        return HTMLResponse(content=f"<h1>❌ Lỗi: {str(e)}</h1>", status_code=500)

# API test kết nối
@app.get("/api/test")
def test():
    return {"status": "ok", "message": "Server đang chạy!"}

# API test DB
@app.get("/api/test-db")
def test_db():
    try:
        conn = get_conn()
        cursor = conn.cursor()
        cursor.execute("SELECT COUNT(*) FROM RescueRequests")
        count = cursor.fetchone()[0]
        conn.close()
        return {"status": "success", "message": f"Kết nối DB thành công! Có {count} yêu cầu cứu hộ."}
    except Exception as e:
        return {"status": "error", "message": str(e)}

# API nhận yêu cầu cứu hộ
@app.post("/api/report")
def report(data: RescueReport):
    try:
        conn = get_conn()
        cursor = conn.cursor()
        cursor.execute("""
            INSERT INTO RescueRequests (Name, Phone, Message, Latitude, Longitude, Address, CreatedAt)
            VALUES (?, ?, ?, ?, ?, ?, GETDATE())
        """, (data.name, data.phone, data.message, data.lat, data.lng, data.address))
        conn.commit()
        conn.close()
        return {"status": "success", "message": "Đã gửi yêu cầu cứu hộ!"}
    except Exception as e:
        return {"status": "error", "message": str(e)}

# ✅ FIX: API lấy danh sách cứu hộ với error handling
@app.get("/api/locations")
def locations():
    try:
        conn = get_conn()
        cursor = conn.cursor()
        cursor.execute("""
            SELECT Id, Name, Phone, Message, Latitude, Longitude, 
                   ISNULL(Address, N'Không rõ') as Address, 
                   ISNULL(Status, 'Pending') as Status,
                   FORMAT(CreatedAt, 'dd/MM/yyyy HH:mm:ss') as CreatedAt
            FROM RescueRequests 
            WHERE Status != 'Rescued' OR Status IS NULL
            ORDER BY CreatedAt DESC
        """)
        
        columns = [column[0] for column in cursor.description]
        results = []
        
        for row in cursor.fetchall():
            row_dict = {}
            for i, column in enumerate(columns):
                value = row[i]
                # Convert float to proper format
                if isinstance(value, float):
                    row_dict[column] = value
                else:
                    row_dict[column] = value
            results.append(row_dict)
        
        conn.close()
        return {"status": "success", "data": results, "count": len(results)}
    except Exception as e:
        print(f"❌ Lỗi API locations: {e}")
        return {"status": "error", "message": str(e), "data": []}

# API đánh dấu đã cứu
@app.post("/api/mark-rescued/{id}")
def mark_rescued(id: int):
    try:
        conn = get_conn()
        cursor = conn.cursor()
        cursor.execute("""
            UPDATE RescueRequests 
            SET Status = 'Rescued', RescuedAt = GETDATE()
            WHERE Id = ?
        """, (id,))
        conn.commit()
        conn.close()
        return {"status": "success"}
    except Exception as e:
        return {"status": "error", "message": str(e)}

# API xóa yêu cầu
@app.delete("/api/delete/{id}")
def delete_request(id: int):
    try:
        conn = get_conn()
        cursor = conn.cursor()
        cursor.execute("DELETE FROM RescueRequests WHERE Id = ?", (id,))
        conn.commit()
        conn.close()
        return {"status": "success"}
    except Exception as e:
        return {"status": "error", "message": str(e)}