chesshog-uci/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 ConsoleTransmitter implements Transmitter<String> {
  36
  37         private Scanner scanner;
  38         private boolean isActive;
  39
  40         ConsoleTransmitter(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 transmitCommand() {
  57             return scanner.nextLine();
  58         }
  59
  60         @Override
  61         public void addCommand(String command) {
  62             // not supported for console acceptor
  63         }
  64
  65         @Override
  66         public void stop() {
  67             isActive = false;
  68         }
  69
  70         @Override
  71         public void close() {
  72             scanner.close();
  73         }
  74     }
  75
  76     private ConsoleTransmitter transmitter;
  77     private PrintWriter writer;
  78
  79     ConsoleAcceptor(InputStream input, OutputStream output) {
  80         transmitter = new ConsoleTransmitter(input);
  81         writer = new PrintWriter(output, true);
  82     }
  83
  84     public Transmitter<String> transmitter() {
  85         return transmitter;
  86     }
  87
  88     @Override
  89     public void start() {
  90     }
  91
  92     @Override
  93     public void id(String params) {
  94         console(UCIConstants.ID, params);
  95     }
  96
  97     @Override
  98     public void uciOK(String params) {
  99         console(UCIConstants.UCI_OK, params);
 100     }
 101
 102     @Override
 103     public void readyOK(String params) {
 104         console(UCIConstants.READY_OK, params);
 105     }
 106
 107     @Override
 108     public void setOption(String name) {
 109         console(
 110                 UCIConstants.OPTION,
 111                 String.format(
 112                         "%s %s %s %s",
 113                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.BUTTON.type()
 114                 )
 115         );
 116     }
 117
 118     @Override
 119     public void setOption(String name, boolean def) {
 120         console(
 121                 UCIConstants.OPTION,
 122                 String.format(
 123                         "%s %s %s %s %s %b",
 124                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.CHECK.type(),
 125                         OptionParams.OPTION_DEFAULT, def
 126                 )
 127         );
 128     }
 129
 130     @Override
 131     public void setOption(String name, int def, int min, int max) {
 132         console(
 133                 UCIConstants.OPTION,
 134                 String.format(
 135                         "%s %s %s %s %s %d %s %d %s %d",
 136                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.SPIN.type(),
 137                         OptionParams.OPTION_DEFAULT, def, OptionParams.OPTION_MIN, min, OptionParams.OPTION_MAX, max
 138                 )
 139         );
 140     }
 141
 142     @Override
 143     public void setOption(String name, String def) {
 144         console(
 145                 UCIConstants.OPTION,
 146                 String.format(
 147                         "%s %s %s %s %s %s",
 148                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.STRING.type(),
 149                         OptionParams.OPTION_DEFAULT, def == null ? "" : def
 150                 )
 151         );
 152     }
 153
 154     @Override
 155     public void setOption(String name, String def, List<String> vars) {
 156         StringBuilder varOptions = new StringBuilder();
 157         for (String var : vars)
 158             varOptions.append(
 159                     String.format("%s %s ", OptionParams.OPTION_VAR, var == null ? "" : var)
 160             );
 161         console(
 162                 UCIConstants.OPTION,
 163                 String.format(
 164                         "%s %s %s %s %s %s %s",
 165                         OptionParams.OPTION_NAME, name, OptionParams.OPTION_TYPE, OptionType.COMBO.type(),
 166                         OptionParams.OPTION_DEFAULT, def == null ? "" : def, varOptions.toString().trim()
 167                 )
 168         );
 169     }
 170
 171     @Override
 172     public void terminate(String params) {
 173         transmitter.stop();
 174     }
 175
 176     private void console(String name, String params) {
 177         writer.println(
 178                 String.format(
 179                         "%s %s",
 180                         name,
 181                         params != null ? params : ""
 182                 ).trim()
 183         );
 184     }
 185
 186 }