import json
from ..parser.models import ParsedCurl


def generate_rust(parsed: ParsedCurl) -> str:
    """Generate Rust reqwest code from parsed curl command."""
    lines = [
        "use reqwest;",
        "use std::collections::HashMap;",
        "use tokio;",
        "",
        "#[tokio::main]",
        "async fn main() -> Result<(), reqwest::Error> {",
        f'    let url = "{parsed.url}"',
        "",
    ]
    
    if parsed.headers or parsed.user_agent:
        lines.append("    let mut headers = HashMap::new();")
        for h in parsed.headers:
            lines.append(f'    headers.insert("{h.key}", "{h.value}");')
        if parsed.user_agent:
            lines.append(f'    headers.insert("User-Agent", "{parsed.user_agent}");')
        lines.append("")
    
    method_name = parsed.method.value.lower()
    
    if parsed.data:
        if parsed.data.data_type.value == "json" and parsed.data.json_data:
            json_str = json.dumps(parsed.data.json_data)
            lines.append(f"    let json_data: serde_json::Value = {json_str}.parse().unwrap();")
            lines.append("")
            lines.append("    let client = reqwest::Client::new();")
            if parsed.headers or parsed.user_agent:
                lines.append("    let response = client")
            else:
                lines.append("    let response = reqwest::Client::new()")
        else:
            lines.append(f"    let data = {repr(parsed.data.content)};")
            lines.append("")
            lines.append("    let client = reqwest::Client::new();")
            if parsed.headers or parsed.user_agent:
                lines.append("    let response = client")
            else:
                lines.append("    let response = reqwest::Client::new()")
    else:
        lines.append("    let client = reqwest::Client::new();")
        if parsed.headers or parsed.user_agent:
            lines.append("    let response = client")
        else:
            lines.append("    let response = reqwest::Client::new()")
    
    lines.append(f"        .{method_name}(url)")
    
    if parsed.headers or parsed.user_agent:
        lines.append("        .headers(headers)")
    
    if parsed.data and parsed.data.data_type.value == "json" and parsed.data.json_data:
        lines.append("        .json(&json_data)")
    elif parsed.data:
        lines.append(f"        .body(data)")
    
    if parsed.timeout:
        lines.append(f"        .timeout(std::time::Duration::from_secs({parsed.timeout}))")
    
    lines.append("        .send()")
    lines.append("        .await?;")
    lines.append("")
    lines.append("    if !response.status().is_success() {")
    f'        eprintln!("HTTP error: {{}}", response.status());'
    lines.append("        return Ok(());")
    lines.append("    }")
    lines.append("")
    lines.append("    let text = response.text().await?;")
    lines.append("    println!(\"{}\", text);")
    lines.append("")
    lines.append("    Ok(())")
    lines.append("}")
    
    return "\n".join(lines)