MLE-27077 Added fix for invalid header for empty doc

Author: rjrudin

marklogic-client-api/src/main/java/com/marklogic/client/impl/OkHttpServices.java

@@ -34,7 +34,9 @@
 import jakarta.mail.BodyPart;
 import jakarta.mail.Header;
 import jakarta.mail.MessagingException;
+import jakarta.mail.internet.ContentDisposition;
 import jakarta.mail.internet.MimeMultipart;
+import jakarta.mail.internet.ParseException;
 import jakarta.mail.util.ByteArrayDataSource;
 import jakarta.xml.bind.DatatypeConverter;
 import okhttp3.*;
@@ -1808,16 +1810,50 @@ static private long getHeaderLength(String length) {
 
 	static private String getHeaderUri(BodyPart part) {
 		try {
-			if (part != null) {
-				return part.getFileName();
+			if (part == null) {
+				return null;
 			}
-			// if it's not found, just return null
+
+			try {
+				String filename = part.getFileName();
+				if (filename != null) {
+					return filename;
+				}
+			} catch (ParseException e) {
+				// Jakarta Mail's parser failed due to malformed Content-Disposition header.
+				// Check if MarkLogic sent a malformed "format=" parameter at the end, which violates RFC 2183.
+				String contentDisposition = getHeader(part, "Content-Disposition");
+				if (contentDisposition != null && contentDisposition.matches(".*;\\s*format\\s*=\\s*$")) {
+					// Remove the trailing "; format=" to fix the malformed header
+					String cleaned = contentDisposition.replaceFirst(";\\s*format\\s*=\\s*$", "").trim();
+					logger.debug("Removed trailing 'format=' from malformed Content-Disposition header: {} -> {}", contentDisposition, cleaned);
+					return extractFilenameFromContentDisposition(cleaned);
+				}
+				throw e;
+			}
+
 			return null;
 		} catch (MessagingException e) {
 			throw new MarkLogicIOException(e);
 		}
 	}
 
+	static private String extractFilenameFromContentDisposition(String contentDisposition) {
+		if (contentDisposition == null) {
+			return null;
+		}
+		try {
+			// Use Jakarta Mail's ContentDisposition parser to extract the filename parameter. This is the class
+			// that throws an error when "format=" exists in the value, but that has been removed already.
+			ContentDisposition cd = new ContentDisposition(contentDisposition);
+			return cd.getParameter("filename");
+		} catch (ParseException e) {
+			logger.warn("Failed to parse cleaned Content-Disposition header: {}; cause: {}",
+				contentDisposition, e.getMessage());
+			return null;
+		}
+	}
+
 	static private void updateVersion(DocumentDescriptor descriptor, Headers headers) {
 		updateVersion(descriptor, extractVersion(headers.get(HEADER_ETAG)));
 	}

marklogic-client-api/src/test/java/com/marklogic/client/test/document/ReadDocumentPageTest.java

@@ -4,25 +4,30 @@
 package com.marklogic.client.test.document;
 
 import com.marklogic.client.DatabaseClient;
-import com.marklogic.client.document.*;
+import com.marklogic.client.document.DocumentPage;
+import com.marklogic.client.document.DocumentRecord;
+import com.marklogic.client.document.DocumentWriteSet;
+import com.marklogic.client.document.JSONDocumentManager;
 import com.marklogic.client.io.BytesHandle;
 import com.marklogic.client.io.DocumentMetadataHandle;
 import com.marklogic.client.io.StringHandle;
 import com.marklogic.client.query.StructuredQueryBuilder;
 import com.marklogic.client.query.StructuredQueryDefinition;
+import com.marklogic.client.test.AbstractClientTest;
 import com.marklogic.client.test.Common;
-import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
-class ReadDocumentPageTest {
+class ReadDocumentPageTest extends AbstractClientTest {
 
+	/**
+	 * Verifies that the jakarta.mail library, instead of javax.mail, can probably read the URI.
+	 * See MLE-15748, which pertains to issues with javax.mail only allowing US-ASCII characters.
+	 */
 	@Test
 	void test() {
-		Common.deleteUrisWithPattern("/aaa-page/*");
-
 		final String uri = "/aaa-page/太田佳伸のＸＭＬファイル.xml";
 		DocumentRecord documentRecord;
 		try (DatabaseClient client = Common.newClient()) {
@@ -38,8 +43,24 @@ void test() {
 	}
 
 	@Test
-	@Disabled("Disabling for now because this seems to be a server bug.")
-	void testEmptyDocWithNoExtension() {
+	void emptyTextDocument() {
+		final String uri = "/sample/empty-file.txt";
+
+		try (DatabaseClient client = Common.newClient()) {
+			JSONDocumentManager documentManager = client.newJSONDocumentManager();
+			StructuredQueryDefinition query = new StructuredQueryBuilder().document(uri);
+			DocumentRecord documentRecord;
+			try (DocumentPage documentPage = documentManager.search(query, 1)) {
+				assertTrue(documentPage.hasNext(), "Expected a document in the page, but none was found.");
+				documentRecord = documentPage.next();
+			}
+			String actualUri = documentRecord.getUri();
+			assertEquals(uri, actualUri, "The URI of the empty document should match the one written.");
+		}
+	}
+
+	@Test
+	void emptyDocWithNoExtension() {
 		final String collection = "empty-binary-test";
 
 		try (DatabaseClient client = Common.newClient()) {