src/main/java/org/hedgecode/chess/uci/ConsoleAcceptor.java

   1 /*
   2  * Copyright (c) 2018. Developed by Hedgecode.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.hedgecode.chess.uci;
  18
  19 import java.io.InputStream;
  20 import java.io.OutputStream;
  21 import java.io.PrintWriter;
  22 import java.util.List;
  23 import java.util.Scanner;
  24
  25 import org.hedgecode.chess.uci.command.OptionParams;
  26 import org.hedgecode.chess.uci.command.OptionType;
  27
  28 /**
  29  *
  30  *
  31  * @author Dmitry Samoshin aka gotty
  32  */
  33 public class ConsoleAcceptor implements Acceptor {
  34
  35     private class ConsoleReceiver implements Receiver<String> {
  36
  37         private Scanner scanner;
  38         private boolean isActive;
  39
  40         ConsoleReceiver(InputStream input) {
  41             scanner = new Scanner(input);
  42             isActive = true;
  43         }
  44
  45         @Override
  46         public boolean isActive() {
  47             return isActive;
  48         }
  49
  50         @Override
  51         public boolean hasCommand() {
  52             return isActive && scanner.hasNextLine();
  53         }
  54
  55         @Override
  56         public String receiveCommand() {
  57             return scanner.nextLine();
  58         }
  59
  60         @Override
  61         public void stop() {
  62             isActive = false;
  63         }
  64
  65         @Override
  66         public void close() {
  67             scanner.close();
  68         }
  69     }
  70
  71     private ConsoleReceiver receiver;
  72     private PrintWriter writer;
  73
  74     ConsoleAcceptor(InputStream input, OutputStream output) {
  75         receiver = new ConsoleReceiver(input);
  76         writer = new PrintWriter(output, true);
  77     }
  78
  79     public Receiver<String> receiver() {
  80         return receiver;
  81     }
  82
  83     @Override
  84     public void start() {
  85     }
  86
  87     @Override
  88     public void id(String params) {
  89         console(UCIConstants.ID, params);
  90     }
  91
  92     @Override
  93     public void uciOK(String params) {
  94         console(UCIConstants.UCI_OK, params);
  95     }
  96
  97     @Override
  98     public void readyOK(String params) {
  99         console(UCIConstants.READY_OK, params);
 100     }
 101
 102     @Override
 103     public void setOption(String name) {
 104         console(
 105                 UCIConstants.OPTION,
 106                 String.format(
 107                         "%s %s %s %s",
 108                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.BUTTON.type()
 109                 )
 110         );
 111     }
 112
 113     @Override
 114     public void setOption(String name, boolean def) {
 115         console(
 116                 UCIConstants.OPTION,
 117                 String.format(
 118                         "%s %s %s %s %s %b",
 119                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.CHECK.type(),
 120                         OptionParams.OPTION_DEFAULT, def
 121                 )
 122         );
 123     }
 124
 125     @Override
 126     public void setOption(String name, int def, int min, int max) {
 127         console(
 128                 UCIConstants.OPTION,
 129                 String.format(
 130                         "%s %s %s %s %s %d %s %d %s %d",
 131                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.SPIN.type(),
 132                         OptionParams.OPTION_DEFAULT, def, OptionParams.OPTION_MIN, min, OptionParams.OPTION_MAX, max
 133                 )
 134         );
 135     }
 136
 137     @Override
 138     public void setOption(String name, String def) {
 139         console(
 140                 UCIConstants.OPTION,
 141                 String.format(
 142                         "%s %s %s %s %s %s",
 143                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.STRING.type(),
 144                         OptionParams.OPTION_DEFAULT, def == null ? "" : def
 145                 )
 146         );
 147     }
 148
 149     @Override
 150     public void setOption(String name, String def, List<String> vars) {
 151         StringBuilder varOptions = new StringBuilder();
 152         for (String var : vars)
 153             varOptions.append(
 154                     String.format("%s %s ", OptionParams.OPTION_VAR, var == null ? "" : var)
 155             );
 156         console(
 157                 UCIConstants.OPTION,
 158                 String.format(
 159                         "%s %s %s %s %s %s %s",
 160                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.COMBO.type(),
 161                         OptionParams.OPTION_DEFAULT, def == null ? "" : def, varOptions.toString().trim()
 162                 )
 163         );
 164     }
 165
 166     @Override
 167     public void terminate(String params) {
 168         receiver.stop();
 169     }
 170
 171     private void console(String name, String params) {
 172         writer.println(
 173                 String.format(
 174                         "%s %s",
 175                         name,
 176                         params != null ? params : ""
 177                 ).trim()
 178         );
 179     }
 180
 181 }