// Hw8AndrewFinger.java
// <NAME, ANDREW ID, SECTION>

import java.io.*;
import java.net.*;
import java.util.*;

class Hw8AndrewFinger {
  public static String[] andrewFinger(String key) {
    ArrayList<String> lines = new ArrayList<String>();
    try {
      String line, host = "andrew.cmu.edu";
      Socket socket = new Socket(host, 79);
      BufferedReader in = new BufferedReader(
                              new InputStreamReader(socket.getInputStream()));
      PrintStream out = new PrintStream(socket.getOutputStream());
      out.println(key);
      while ((line = in.readLine()) != null)
        lines.add(line);
      socket.close();
    }
    catch (Exception e ) { System.out.println("Error: " + e); }
    return (String[])lines.toArray(new String[]{});
  }

  public static void testAndrewFinger() {
    Scanner scanner = new Scanner(System.in);
    while (true) {
      System.out.print("Andrew search key [q to quit] --> " );
      String key = scanner.next();
      if (key.equals("q")) return;
      System.out.println("-------------------------------");
      System.out.println("finger output:");
      String[] fingerOutput = andrewFinger(key);
      for (int i=0; i<fingerOutput.length; i++)
        System.out.println(fingerOutput[i]);
      System.out.println("-------------------------------");
      System.out.println();
    }
  }


  // This method takes a search key and -- using the andrewFinger
  // method already written for you, and searching the result for
  // lines labeled as "login name" -- returns a sorted non-null
  // (though perhaps empty) array of Strings containing all the
  // andrew id's that the finger program returns for the given key.
  public static String[] getAndrewIds(String key) {
    return null;
  }

  public static void testGetAndrewIds() {
    // this is a partial test.  You may wish to add more tests.
    assert(Arrays.equals(getAndrewIds("stehlik"),
                         new String[]{"kstehlik", "mjs"}));
    assert(Arrays.equals(getAndrewIds("@"),
                         new String[]{ }));
  }

  public static void main(String[] args) {
    // testGetAndrewIds();
    testAndrewFinger();
  }
}