import os, time, requests

API_BASE = "https://rvt-convert.paperengineer.workers.dev/api"

def convert_large_model(file_path: str, target_year: int, output_path: str):
    """Khuyên dùng cho tệp lớn (lên tới 1 GB) - Tải trực tiếp lên Cloudflare R2."""
    file_size = os.path.getsize(file_path)
    file_name = os.path.basename(file_path)

    # 1. Lấy Presigned URL tải trực tiếp lên đám mây
    res = requests.post(f"{API_BASE}/upload-url", json={
        "filename": file_name,
        "file_size": file_size,
        "content_type": "application/octet-stream"
    }).json()

    if res.get("r2_enabled"):
        # 2. Tải thẳng lên Cloudflare R2 (không tốn RAM máy chủ)
        with open(file_path, "rb") as f:
            requests.put(res["upload_url"], data=f, headers={"Content-Type": "application/octet-stream"}).raise_for_status()

        # 3. Kích hoạt tác vụ xử lý nền
        job = requests.post(f"{API_BASE}/convert-r2", json={
            "key": res["key"],
            "job_id": res["job_id"],
            "filename": file_name,
            "target_year": target_year
        }).json()
        job_id = job["job_id"]
    else:
        # Fallback chuyển đổi tiêu chuẩn
        with open(file_path, "rb") as f:
            job = requests.post(f"{API_BASE}/convert", files={"file": f}, data={"target_year": target_year}).json()
        job_id = job["job_id"]

    # 4. Kiểm tra tiến độ và tải về
    print(f"Task started: {job_id}")
    while True:
        status_res = requests.get(f"{API_BASE}/jobs/{job_id}").json()
        status = status_res["status"]
        progress = status_res.get("progress", 0)
        print(f"Progress: {progress}% ({status})")
        if status == "completed":
            break
        elif status == "failed":
            raise RuntimeError(status_res.get("error", "Conversion failed"))
        time.sleep(2)

    dl_res = requests.get(f"{API_BASE}/jobs/{job_id}/download")
    with open(output_path, "wb") as f_out:
        f_out.write(dl_res.content)
    print(f"Saved to: {output_path}")

# Khởi chạy chuyển đổi
convert_large_model("Building2026.rvt", 2023, "Building2023.rvt")

using System.IO;
using System.Net.Http;
using System.Threading.Tasks;

public class PaperConvertClient
{
    private static readonly HttpClient client = new HttpClient();
    private const string BaseUrl = "https://rvt-convert.paperengineer.workers.dev/api";

    public async Task ConvertAndSaveAsync(string inputPath, int targetYear, string outputPath)
    {
        using var form = new MultipartFormDataContent();
        using var fileContent = new ByteArrayContent(await File.ReadAllBytesAsync(inputPath));
        form.Add(fileContent, "file", Path.GetFileName(inputPath));
        form.Add(new StringContent(targetYear.ToString()), "target_year");

        var response = await client.PostAsync($"{BaseUrl}/convert", form);
        response.EnsureSuccessStatusCode();
        // Poll status and download the converted model...
    }
}