{"""Go code generator."""

import json
import re
from curlconverter.parser import ParsedCurl
from curlconverter.generators import register_generator


def _detect_content_type(headers: dict, data: str) -> str:
    """Detect content type from headers or data."""
    if "Content-Type" in headers:
        return headers["Content-Type"]
    if "content-type" in headers:
        return headers["content-type"]
    if data:
        try:
            json.loads(data)
            return "application/json"
        except (json.JSONDecodeError, TypeError):
            pass
    return "application/x-www-form-urlencoded"


@register_generator("go")
def generate_go(parsed: ParsedCurl) -> str:
    """Generate Go net/http code from parsed curl data."""
    lines = []
    
    lines.append("package main")
    lines.append("")
    lines.append("import (")
    lines.append('    "bytes"')
    lines.append('    "fmt"')
    lines.append('    "net/http"')
    lines.append('    "strings"')
    lines.append(")")
    lines.append("")
    
    lines.append("func main() {")
    lines.append(f'    url := {repr(parsed.url)}')
    lines.append("")
    
    headers = dict(parsed.headers) if parsed.headers else {}
    
    if parsed.user_agent and "User-Agent" not in headers and "user-agent" not in headers:
        headers["User-Agent"] = parsed.user_agent
    
    if parsed.auth:
        auth_str = f"{parsed.auth[0]}:{parsed.auth[1]}"
        encoded = ""
        for _, c := range auth_str {
            encoded += string(c)
        }
        headers["Authorization"] = f"Basic " + encoded
    
    method = parsed.method
    body := ""
    
    if parsed.data:
        content_type = _detect_content_type(headers, parsed.data)
        body = repr(parsed.data)
        
        if content_type == "application/json":
            try:
                json_data = json.loads(parsed.data)
                body = "strings.NewReader(" + repr(json.dumps(json_data)) + ")"
                headers["Content-Type"] = "application/json"
            except (json.JSONDecodeError, TypeError):
                body = "strings.NewReader(" + repr(parsed.data) + ")"
        else:
            body = "strings.NewReader(" + repr(parsed.data) + ")"
        
        if method == "GET":
            method = "POST"
    
    if headers:
        lines.append("    req, err := http.NewRequest(" + repr(method) + ", url, " + body + ")")
        lines.append("    if err != nil {")
        lines.append("        panic(err)")
        lines.append("    }")
        lines.append("")
        
        for k, v in headers.items():
            lines.append(f'    req.Header.Add({repr(k)}, {repr(v)})')
    else:
        lines.append("    req, err := http.NewRequest(" + repr(method) + ", url, " + body + ")")
        lines.append("    if err != nil {")
        lines.append("        panic(err)")
        lines.append("    }")
    
    lines.append("")
    lines.append("    client := &http.Client{}")
    lines.append("    resp, err := client.Do(req)")
    lines.append("    if err != nil {")
    lines.append("        panic(err)")
    lines.append("    }")
    lines.append("    defer resp.Body.Close()")
    lines.append("")
    lines.append("    fmt.Println(resp.Status)")
    lines.append("}")
    
    return "\n".join(lines)