package org.exist.xquery.functions.request;

import org.exist.xmldb.UserManagementService;
import org.apache.commons.io.output.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import org.apache.commons.httpclient.HttpException;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.methods.ByteArrayRequestEntity;
import org.apache.commons.httpclient.methods.PostMethod;
import org.exist.http.RESTTest;
import org.exist.xmldb.EXistResource;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.fail;
import org.xmldb.api.DatabaseManager;
import org.xmldb.api.base.Collection;
import org.xmldb.api.base.XMLDBException;
import org.xmldb.api.modules.BinaryResource;
import static org.apache.commons.codec.binary.Base64.encodeBase64String;

/**
 * @author Adam Retter <adam.retter@googlemail.com>
 */
public class GetDataTest extends RESTTest {

    private final static String CONTAINER_ELEMENT_NAME = "data";
    private final static String XQUERY = wrapInElement("{request:get-data()}");
    private final static String XQUERY_FILENAME = "test-get-data.xql";

    private static Collection root;

    private static String wrapInElement(String value) {
        return value == null || value.length() == 0 ? "<" + CONTAINER_ELEMENT_NAME + "/>" : "<" + CONTAINER_ELEMENT_NAME + ">" + value + "</" + CONTAINER_ELEMENT_NAME + ">";
    }

    @BeforeClass
    public static void beforeClass() throws XMLDBException {
        root = DatabaseManager.getCollection("xmldb:exist://localhost:" + existWebServer.getPort() + "/xmlrpc/db", "admin", "");
        BinaryResource res = (BinaryResource)root.createResource(XQUERY_FILENAME, "BinaryResource");
        ((EXistResource) res).setMimeType("application/xquery");
        res.setContent(XQUERY);
        root.storeResource(res);
        UserManagementService ums = (UserManagementService)root.getService("UserManagementService", "1.0");
        ums.chmod(res, 0777);
    }

    @AfterClass
    public static void afterClass() throws XMLDBException {
        BinaryResource res = (BinaryResource)root.getResource(XQUERY_FILENAME);
        root.removeResource(res);
    }

    @Test
    public void retrieveEmpty() {
        PostMethod post = new PostMethod(getCollectionRootUri() + "/" + XQUERY_FILENAME);

        post.setRequestHeader("Content-Type", "application/octet-stream");

        testRequest(post, wrapInElement("").getBytes());
    }
    
    @Test
    public void retrieveBinary() {
        PostMethod post = new PostMethod(getCollectionRootUri() + "/" + XQUERY_FILENAME);

        final String testData = "12345";

        post.setRequestHeader("Content-Type", "application/octet-stream");
        post.setRequestEntity(new ByteArrayRequestEntity(testData.getBytes()));

        testRequest(post, wrapInElement(encodeBase64String(testData.getBytes()).trim()).getBytes());
    }

    @Test
    public void retrieveXml() {
        PostMethod post = new PostMethod(getCollectionRootUri() + "/" + XQUERY_FILENAME);

        final String testData = "<a><b><c>hello</c></b></a>";

        post.setRequestHeader("Content-Type", "text/xml");
        post.setRequestEntity(new ByteArrayRequestEntity(testData.getBytes()));

        testRequest(post, wrapInElement("\n\t" + testData + "\n").getBytes(), true);
    }

    @Test
    public void retrieveMalformedXmlFallbackToString() {
        PostMethod post = new PostMethod(getCollectionRootUri() + "/" + XQUERY_FILENAME);

        final String testData = "<a><b></a>";

        post.setRequestHeader("Content-Type", "text/xml");
        post.setRequestEntity(new ByteArrayRequestEntity(testData.getBytes()));

        testRequest(post, wrapInElement(testData.replace("<", "<").replace(">", ">")).getBytes());
    }

    @Test
    public void retrieveString() {
        PostMethod post = new PostMethod(getCollectionRootUri() + "/" + XQUERY_FILENAME);

        final String testData = "12345";

        post.setRequestEntity(new ByteArrayRequestEntity(testData.getBytes()));

        testRequest(post, wrapInElement(testData).getBytes());
    }
    
    private void testRequest(HttpMethod method, byte expectedResponse[]) {
        testRequest(method, expectedResponse, false);
    }
    
    private void testRequest(HttpMethod method, byte expectedResponse[], boolean stripWhitespaceAndFormatting) {
        try {
            int httpResult = client.executeMethod(method);

            assertEquals(HttpStatus.SC_OK, httpResult);
            
            byte buf[] = new byte[1024];
            int read = -1;
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            InputStream is = method.getResponseBodyAsStream();
            while((read = is.read(buf)) > -1) {
                baos.write(buf, 0, read);
            }
            
            byte actualResponse[] = baos.toByteArray();
            
            if(stripWhitespaceAndFormatting) {
                expectedResponse = new String(expectedResponse).replace("\n", "").replace("\t", "").replace(" ", "").getBytes();
                actualResponse = new String(actualResponse).replace("\n", "").replace("\t","").replace(" ", "").getBytes();
            }
            
            assertArrayEquals(expectedResponse, actualResponse);
            
        } catch(HttpException he) {
            fail(he.getMessage());
        } catch(IOException ioe) {
            fail(ioe.getMessage());
        } finally {
            method.releaseConnection();
        }
    }
}