use axum::http::HeaderMap;
use serde_json::Value;

pub fn list_buckets_xml(buckets: &[String]) -> (HeaderMap, String) {
    let mut headers = HeaderMap::new();
    headers.insert("Content-Type", "application/xml".parse().unwrap());
    
    let bucket_entries = buckets
        .iter()
        .map(|b| format!(
            "<Bucket><Name>{}</Name><CreationDate>2026-05-27T00:00:00Z</CreationDate></Bucket>",
            b
        ))
        .collect::<Vec<_>>()
        .join("\n    ");
    
    let xml = format!(
        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>
<ListAllMyBucketsResult xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">
  <Owner>
    <ID>owner-id</ID>
    <DisplayName>MarkBase</DisplayName>
  </Owner>
  <Buckets>
    {}
  </Buckets>
</ListAllMyBucketsResult>",
        bucket_entries
    );
    
    (headers, xml)
}

pub fn list_objects_xml(bucket_name: &str, objects: &[Value]) -> (HeaderMap, String) {
    let mut headers = HeaderMap::new();
    headers.insert("Content-Type", "application/xml".parse().unwrap());
    
    let object_entries = objects
        .iter()
        .map(|obj| {
            let key = obj.get("Key").and_then(|k| k.as_str()).unwrap_or("");
            let last_modified = obj.get("LastModified").and_then(|l| l.as_str()).unwrap_or("");
            let etag = obj.get("ETag").and_then(|e| e.as_str()).unwrap_or("");
            let size = obj.get("Size").and_then(|s| s.as_i64()).unwrap_or(0);
            
            format!(
                "<Contents>
      <Key>{}</Key>
      <LastModified>{}</LastModified>
      <ETag>{}</ETag>
      <Size>{}</Size>
    </Contents>",
                key, last_modified, etag, size
            )
        })
        .collect::<Vec<_>>()
        .join("\n    ");
    
    let xml = format!(
        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>
<ListBucketResult xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">
  <Name>{}</Name>
  <Prefix></Prefix>
  <Marker></Marker>
  <MaxKeys>1000</MaxKeys>
  <IsTruncated>false</IsTruncated>
  {}
</ListBucketResult>",
        bucket_name, object_entries
    );
    
    (headers, xml)
}