/*

 * Quoridor server and related libraries

 * Copyright (C) 2009-2010 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, either version 3 of the License.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. Look for COPYING file in the top folder.

 * If not, see http://www.gnu.org/licenses/.

 */

package cz.xelfi.quoridor.freemarkerdor;

import org.glassfish.grizzly.http.server.HttpServer;

import java.util.Collection;

import java.util.Arrays;

import com.sun.jersey.api.client.Client;

import com.sun.jersey.api.client.ClientResponse;

import com.sun.jersey.api.client.WebResource;

import com.sun.jersey.core.header.MediaTypes;

import com.sun.jersey.core.util.MultivaluedMapImpl;

import cz.xelfi.quoridor.webidor.resources.Quoridor;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.net.URI;

import java.util.HashSet;

import java.util.Locale;

import java.util.concurrent.Callable;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import javax.ws.rs.core.Cookie;

import javax.ws.rs.core.MediaType;

import javax.ws.rs.core.MultivaluedMap;

import org.junit.AfterClass;

import org.junit.Before;

import org.junit.BeforeClass;

import org.junit.Test;

import static org.junit.Assert.*;

/**

 *

 * @author Jaroslav Tulach <jtulach@netbeans.org>

 */

public class UITest extends Object {

    private static File dir;

    private static HttpServer stopAPI;

    private static HttpServer stopStatistics;

    private static Callable<Void> stop;

    private WebResource webResource;

    private WebResource apiResource;

    private WebResource statResource;

    public UITest() throws Exception {

    }

    @BeforeClass

    public static void localeEnglish() throws Exception {

        Locale.setDefault(Locale.ENGLISH);

        dir = File.createTempFile("quoridor", ".dir");

        dir.delete();

        dir.mkdirs();

        File games = new File(dir, "games");

        copyResource("wrong.encoding", new File(games, "fdcc275c-55ad-4b7e-a776-bde3876b2f2c"));

        System.setProperty("quoridor.dir", dir.getPath());

        stopAPI = Quoridor.start(9990);

        stopStatistics = Quoridor.start(9992);

        stop = UI.startServers(9991, "http://localhost:9990", "http://localhost:9992", null);

        File passwd = new File(dir, "passwd");

        FileOutputStream os = new FileOutputStream(passwd);

        os.write("test=pes\nJarda=darda\n".getBytes("UTF-8"));

        os.close();

    }

    @Before

    public void setUp() throws Exception {

        Client client = new Client();

        webResource = client.resource(new URI("http://localhost:9991/"));

        apiResource = client.resource(new URI("http://localhost:9990/"));

        statResource = client.resource(new URI("http://localhost:9992/"));

    }

    @AfterClass

    public static void cleanUpAll() throws Exception {

        deleteRec(dir);

        if (stop != null) {

            stop.call();

        }

        if (stopAPI != null) {

            stopAPI.stop();

        }

    }

    static void deleteRec(File dir) throws IOException {

        if (dir == null) {

            return;

        }

        File[] arr = dir.listFiles();

        if (arr != null) {

            for (File f : arr) {

                deleteRec(f);

            }

        }

        dir.delete();

    }

    @Test public void testApplicationWadl() {

        String serviceWadl = webResource.path("application.wadl").

                accept(MediaTypes.WADL).get(String.class);

        assertTrue(serviceWadl.length() > 0);

    }

    @Test public void testAPIWadl() {

        String serviceWadl = apiResource.path("application.wadl").

                accept(MediaTypes.WADL).get(String.class);

        assertTrue(serviceWadl.length() > 0);

    }

    @Test public void testStatWadl() {

        String serviceWadl = statResource.path("application.wadl").

                accept(MediaTypes.WADL).get(String.class);

        assertTrue(serviceWadl.length() > 0);

    }

    @Test public void testGetIndexPage() throws Exception {

        String logJarda = apiResource.path("login").

            queryParam("name", "Jarda").

            queryParam("password", "darda").

            accept(MediaType.TEXT_PLAIN).

            put(String.class);

        assertNotNull("Logged in OK", logJarda);

        String res = webResource.accept("text/html").get(String.class);

        if (res.indexOf("Quoridor") == -1) {

            fail("Wrong index.html:\n" + res);

        }

        if (res.indexOf("Login") == -1) {

            fail("Wrong index.html:\n" + res);

        }

        if (res.indexOf("action=\"/login\"") == -1) {

            fail("Wrong index.html:\n" + res);

        }

        if (res.toLowerCase().indexOf("error") != -1) {

            fail("There was an error:\n" + res);

        }

        res = webResource.cookie(Cookie.valueOf("login=" + logJarda)).accept("text/html").get(String.class);

        if (res.indexOf("action=\"/games/create\"") == -1) {

            fail(res);

        }

    }

    @Test public void testLogin() throws Exception {

        MultivaluedMap formData = new MultivaluedMapImpl();

        formData.add("name", "test");

        formData.add("password", "pes");

        ClientResponse response = webResource.path("login").

            accept("text/html").type("application/x-www-form-urlencoded").

            post(ClientResponse.class, formData);

        final String res = response.getEntity(String.class);

        assertEquals("OK", ClientResponse.Status.OK, response.getClientResponseStatus());

        if (res.indexOf("You are logged in as test") == -1) {

            fail("res: " + res);

        }

    }

    @Test public void testCreateGameWrongUsers() throws Exception {

        String logTest = apiResource.path("login").

            queryParam("name", "test").

            queryParam("password", "pes").

            accept(MediaType.TEXT_PLAIN).

            put(String.class);

        ClientResponse res = webResource.path("games/create").

            queryParam("white", "unknown1").

            queryParam("black", "unknown2").

            cookie(Cookie.valueOf("login=" + logTest)).

            accept("text/html").

            get(ClientResponse.class);

        final String txt = res.getEntity(String.class);

        assertEquals("Rejected, unknown user\n" + txt, ClientResponse.Status.NOT_FOUND, res.getClientResponseStatus());

        if (txt.indexOf("You (test) must be") == -1) {

            fail(txt);

        }

    }

    @Test public void testCreateGameOK() throws Exception {

        Collection<String> origGamesNames= Arrays.asList(new File(dir, "games").list());

        int origGames = origGamesNames.size();

        String logTest = apiResource.path("login").

            queryParam("name", "test").

            queryParam("password", "pes").

            accept(MediaType.TEXT_PLAIN).

            put(String.class);

        String logJarda = apiResource.path("login").

            queryParam("name", "Jarda").

            queryParam("password", "darda").

            accept(MediaType.TEXT_PLAIN).

            put(String.class);

        ClientResponse res = webResource.path("games/create").

            queryParam("white", "Jarda").

            queryParam("black", "test").

            cookie(Cookie.valueOf("login=" + logTest)).

            accept("text/html").

            get(ClientResponse.class);

        final String txt = res.getEntity(String.class);

        assertEquals("OK\n" + txt, ClientResponse.Status.OK, res.getClientResponseStatus());

        String myGame;

        {

            String[] games = new File(dir, "games").list();

            assertEquals("One new game", origGames + 1, games.length);

            HashSet<String> newNames = new HashSet<String>();

            newNames.addAll(Arrays.asList(games));

            newNames.removeAll(origGamesNames);

            assertEquals("Just one name left: " + newNames, 1, newNames.size());

            myGame = newNames.iterator().next();

        }

        if (txt.indexOf(myGame) == -1) {

            fail(myGame + " expected inside of:\n" + txt);

        }

        ClientResponse page = webResource.path("games/" + myGame).

            cookie(Cookie.valueOf("login=" + logJarda)).

            get(ClientResponse.class);

        String ptxt = page.getEntity(String.class);

        assertEquals("OK:\n" + ptxt, ClientResponse.Status.OK, page.getClientResponseStatus());

        Pattern p = Pattern.compile(".*<img[^>]*src=\"([^\"]*)\"");

        Matcher m = p.matcher(ptxt);

        assertTrue("image found\n" + ptxt, m.find());

        InputStream img1 = webResource.path(m.group(1)).get(InputStream.class);

        assertNotNull("image found", img1);

        ClientResponse move = apiResource.path("games/" + myGame).

            queryParam("loginID", logJarda).

            queryParam("player", "Jarda").queryParam("move", "N").put(ClientResponse.class);

        assertEquals("Move OK:\n" + move.getEntity(String.class), ClientResponse.Status.OK, move.getClientResponseStatus());

        InputStream img2 = webResource.path(m.group(1)).get(InputStream.class);

        assertNotNull("image found", img2);

        ClientResponse page2 = webResource.path("games/" + myGame).

            cookie(Cookie.valueOf("login=" + logJarda)).

            get(ClientResponse.class);

        String ptxt2 = page2.getEntity(String.class);

        assertEquals("OK:\n" + ptxt2, ClientResponse.Status.OK, page2.getClientResponseStatus());

        Matcher m2 = p.matcher(ptxt2);

        assertTrue("image found\n" + ptxt2, m2.find());

        InputStream img3 = webResource.path(m2.group(1)).get(InputStream.class);

        assertNotNull("image found", img3);

        int diff = 0;

        int cnt = 0;

        for (;;) {

            cnt++;

            int b1 = img1.read();

            int b2 = img2.read();

            int b3 = img3.read();

            assertEquals(b1, b2);

            if (b3 != b1) {

                diff++;

            }

            if (b1 == -1 || b3 == -1) break;

        }

        if (diff == 0) {

//            fail("There shall be difference in the streams. Read bytes " + cnt);

        }

    }

    @Test public void testGetWrongGame() throws Exception {

        String logTest = apiResource.path("login").

            queryParam("name", "test").

            queryParam("password", "pes").

            accept(MediaType.TEXT_PLAIN).

            put(String.class);

        String xml = apiResource.path("games/fdcc275c-55ad-4b7e-a776-bde3876b2f2c").

            queryParam("name", "test").

            queryParam("password", "pes").

            accept(MediaType.TEXT_XML).

            get(String.class);

        if (xml.indexOf((char)0x1b) >= 0) {

            fail(xml.replace((char)0x1b, 'X'));

        }

        ClientResponse res = webResource.path("games/fdcc275c-55ad-4b7e-a776-bde3876b2f2c").

            queryParam("white", "Jarda").

            queryParam("black", "test").

            cookie(Cookie.valueOf("login=" + logTest)).

            accept("text/html").

            get(ClientResponse.class);

        final String txt = res.getEntity(String.class);

        assertEquals("OK\n" + txt, ClientResponse.Status.OK, res.getClientResponseStatus());

    }

    private static void copyResource(String res, File file) throws IOException {

        InputStream is = UITest.class.getResourceAsStream(res);

        file.getParentFile().mkdirs();

        FileOutputStream os = new FileOutputStream(file);

        byte[] arr = new byte[4092];

        for (;;) {

            int len = is.read(arr);

            if (len == -1) {

                break;

            }

            os.write(arr, 0, len);

        }

        is.close();

        os.close();

    }

}