auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/InputIterator.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.helpers;
  23
  24 import java.io.BufferedReader;
  25 import java.io.IOException;
  26 import java.io.PrintStream;
  27 import java.util.Iterator;
  28 import java.util.NoSuchElementException;
  29
  30 public class InputIterator implements Iterable<String> {
  31         private BufferedReader in;
  32         private final PrintStream out;
  33         private final String prompt, instructions;
  34
  35         public InputIterator(BufferedReader in, PrintStream out, String prompt, String instructions) {
  36                 this.in = in;
  37                 this.out = out;
  38                 this.prompt = prompt;
  39                 this.instructions = instructions;
  40         }
  41
  42         @Override
  43         public Iterator<String> iterator() {
  44                 out.println(instructions);
  45                 return new Iterator<String>() {
  46                         String input;
  47                         @Override
  48                         public boolean hasNext() {
  49                                 out.append(prompt);
  50                                 try {
  51                                         input = in.readLine();
  52                                 } catch (IOException e) {
  53                                         input = null;
  54                                         return false;
  55                                 }
  56                                 return input.length()>0;
  57                         }
  58
  59                         @Override
  60                         public String next() {
  61                                 if(!hasNext()) {
  62                                         throw new NoSuchElementException();
  63                                 }
  64                                 return input;
  65                         }
  66
  67                         @Override
  68                         public void remove() {
  69                         }
  70                 };
  71         }
  72 }
  73