package org.apidesign.exceptions.trycatchredo;

import java.awt.EventQueue;

import java.awt.event.ActionEvent;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.net.URL;

import javax.swing.Action;

import javax.swing.JOptionPane;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

import static org.junit.Assert.*;

/**

 *

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

 */

public class IOManagerTest {

    public IOManagerTest() {

    }

    @Before

    public void setUp() {

        MemoryURL.initialize();

    }

    @After

    public void tearDown() {

    }

    @Test

    public void simpleWrite() throws Exception {

        URL u = new URL("memory://simpleWrite.txt");

        MemoryURL.registerURL(u.toExternalForm(), "", null);

        final Action a = IOManager.createSaveAction(u, "Hello World!");

        EventQueue.invokeAndWait(new Runnable() {

            public void run() {

                a.actionPerformed(new ActionEvent(this, 0, ""));

            }

        });

        String out = MemoryURL.getOutputForURL(u.toExternalForm());

        assertEquals("Hello World!", out);

    }

    @Test

    public void writeWithAQuestion() throws Exception {

        URL u = new URL("memory://queryEncoding.txt");

        MemoryURL.registerURL(u.toExternalForm(), "", new QueryStream());

        final Action a = IOManager.createSaveAction(u, "Ask a Question");

        EventQueue.invokeAndWait(new Runnable() {

            public void run() {

                a.actionPerformed(new ActionEvent(this, 0, ""));

            }

        });

        String out = MemoryURL.getOutputForURL(u.toExternalForm());

        assertEquals("Text is reversed", "noitseuQ a ksA", out);

    }

    private static class QueryStream extends OutputStream {

        Boolean reverse;

        ByteArrayOutputStream arr = new ByteArrayOutputStream();

        @Override

        public synchronized void write(byte[] b, int off, int len) throws IOException {

            if (reverse == null) {

                throw new QueryException();

            }

            arr.write(b, off, len);

        }

        @Override

        public synchronized void write(int b) throws IOException {

            if (reverse == null) {

                throw new QueryException();

            }

            arr.write(b);

        }

        @Override

        public String toString() {

            if (reverse == null) {

                return "Reverse question was not answered yet!";

            }

            if (reverse) {

                StringBuilder sb = new StringBuilder();

                sb.append(arr.toString());

                sb.reverse();

                return sb.toString();

            }

            return arr.toString();

        }

        private class QueryException extends UserQuestionException {

            @Override

            public JOptionPane getQuestionPane() {

                JOptionPane p = new JOptionPane("Store in reverse way?");

                p.setOptionType(JOptionPane.YES_NO_CANCEL_OPTION);

                return p;

            }

            @Override

            public void confirm(Object option) {

                if (option.equals(JOptionPane.YES_OPTION)) {

                    reverse = Boolean.TRUE;

                    return;

                }

                if (option.equals(JOptionPane.NO_OPTION)) {

                    reverse = Boolean.FALSE;

                    return;

                }

            }

        } // end of QueryException

    } // end of QueryStream

}