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)
}

pub fn initiate_multipart_upload_xml(bucket: &str, key: &str, upload_id: &str) -> (HeaderMap, String) {
    let mut headers = HeaderMap::new();
    headers.insert("Content-Type", "application/xml".parse().unwrap());

    let xml = format!(
        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>
<InitiateMultipartUploadResult xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">
  <Bucket>{}</Bucket>
  <Key>{}</Key>
  <UploadId>{}</UploadId>
</InitiateMultipartUploadResult>",
        bucket, key, upload_id
    );

    (headers, xml)
}

pub fn complete_multipart_upload_xml(bucket: &str, key: &str, etag: &str) -> (HeaderMap, String) {
    let mut headers = HeaderMap::new();
    headers.insert("Content-Type", "application/xml".parse().unwrap());

    let xml = format!(
        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>
<CompleteMultipartUploadResult xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">
  <Location>http://localhost:11438/s3/{}/{}</Location>
  <Bucket>{}</Bucket>
  <Key>{}</Key>
  <ETag>{}</ETag>
</CompleteMultipartUploadResult>",
        bucket, key, bucket, key, etag
    );

    (headers, xml)
}