package org.apidesign.exceptions.trycatchredo.usage;

 import org.apidesign.exceptions.trycatchredo.api.UserQuestionException;

 import java.io.ByteArrayOutputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import javax.swing.JOptionPane;

 // BEGIN: trycatchredo.stream

 public final class QueryStream extends OutputStream {

     private ByteArrayOutputStream arr = new ByteArrayOutputStream();

     /** this field can be manipulated by the QueryException */

     Boolean reverse;

     @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: trycatchredo.stream