// Client.java

import IDLInterchange.*;
import CosPropertyService.*;
import java.io.*;

public class Client2 {

  private static org.omg.CORBA.ORB orb;
  private static IDLInterchange.CConstrainCollection proxy;
  private static IDLInterchange.CCollection pResultTarget;
  private static boolean HaveItems = false;
  private static int total = 0;
  private static IDLInterchange.TItemsACState objects;
  private static ResultImpl ResultThread;
  private static String[] whatprop = new String[20];
  private static short how_many_prop;
  private static short MyMessageID = 0;
  private static short[] clientcookies = new short[100];
  private static short longclcookies = 0;
  
  public static void print_menu () {
    System.out.println("-----------------------------------------------");
    System.out.println("0- New Connection");
    System.out.println("1- Get Property Names");
    System.out.println("2- Search");
    System.out.println("3- Get Total Items");
    System.out.println("4- Get all docs");
    System.out.println("m,?- Print Menu");
    System.out.println("q- Quit");
    System.out.println("\n");
    return;
  }

  public static void testGetTotalItems() {
    System.out.println("Total Items in the result collection:" +
    pResultTarget.GetTotalItems());
    return;
  }

  public static void testRequestItemsProperties() {
    //Setting pMessageID
    CosPropertyService.Any pMessageID = new CosPropertyService.Any();
    ResultThread.current_MessageID = MyMessageID;
    pMessageID.a_short(MyMessageID++);

    //Setting pPropertyNames
    java.lang.String[] pPropertyNames = new java.lang.String[how_many_prop];
    for(int jj=0;jj<how_many_prop;jj++)
      pPropertyNames[jj] = whatprop[jj];

    //Setting pServerCookies
    CosPropertyService.Any[] pServerCookies =
                new CosPropertyService.Any[longclcookies];
    for(int i=0;i<longclcookies;i++) {
      pServerCookies[i] = new CosPropertyService.Any();
      pServerCookies[i].a_string(
             ResultThread.objects[clientcookies[i]].Get_prop("Cookie"));
    }

    //Setting pClientCookies
    CosPropertyService.Any[] pClientCookies =
                new CosPropertyService.Any[longclcookies];
    for(int i=0;i<longclcookies;i++) {
      pClientCookies[i] = new CosPropertyService.Any();
      pClientCookies[i].a_short(clientcookies[i]);
    }

    proxy.RequestItemsProperties(pMessageID,pPropertyNames,
                                 pServerCookies,pClientCookies,
                                 pResultTarget);

    return;
  }

  public static void testRequestItems (String stringcookie, int max) {
    //Setting pMessageID
    CosPropertyService.Any pMessageID = new CosPropertyService.Any();
    ResultThread.current_MessageID = MyMessageID;
    pMessageID.a_short(MyMessageID++);

    //Setting pPropertyNames
    java.lang.String[] pPropertyNames = new java.lang.String[how_many_prop];
    for(int jj=0;jj<how_many_prop;jj++)
      pPropertyNames[jj] = whatprop[jj];

    //Setting pMoreCookies
    CosPropertyService.Any pMoreCookie = new CosPropertyService.Any();
    pMoreCookie.a_string(stringcookie);

    //Setting pNumberofItems
    int pNumberofItems = max;

    //Calling the Proxy. Now we have to find it. Obviously, this is not
    // the right way to do it ;-)
    //Getting a Proxy (CConstrainCollection)
    proxy.RequestItems(pMessageID,pPropertyNames,pMoreCookie,
                       pNumberofItems,pResultTarget);
                       
    return;
  }

  public static void testRequestConstrain (String query, String maxhits) {
    //Testing RequestConstrain
    //Setting the parameters
    //Setting pMessageID
    CosPropertyService.Any pMessageID = new CosPropertyService.Any();
    ResultThread.current_MessageID = MyMessageID;
    pMessageID.a_short(MyMessageID++);

    //Setting pQuerySummary
    IDLInterchange.CQuery aQueryObject =
            IDLInterchange.CQueryHelper.bind(orb,"aQueryObject");

    //here we build the query for ACM web server
    IDLInterchange.TQueryDescription aQueryDescription =
                new IDLInterchange.TQueryDescription();
    CosPropertyService.Any any_query = new CosPropertyService.Any();
    any_query.a_string(query);
    IDLInterchange.RPNQuery0 rpnquery0 =
                        new IDLInterchange.RPNQuery0(0,any_query);
    aQueryDescription.QueryType0(rpnquery0);

    java.lang.String[] aItemPropertyNames =
                        new java.lang.String[how_many_prop];
    for(int jj=0;jj<how_many_prop;jj++)
      aItemPropertyNames[jj] = whatprop[jj];

    CosPropertyService.Property[] aMoreSummary =
                new CosPropertyService.Property[2];
    CosPropertyService.Any joquese = new CosPropertyService.Any();
    short joquese2 = 0;
    joquese.a_short(joquese2);
    CosPropertyService.Property property =
                new CosPropertyService.Property("",joquese);
    aMoreSummary[0] = property;
    aMoreSummary[1] = property;

    IDLInterchange.TQuerySummary pQuerySummary =
        new IDLInterchange.TQuerySummary(
                                         aQueryObject,aQueryDescription,
                                         aItemPropertyNames, aMoreSummary);

    //Setting pServicePrefSummary
    IDLInterchange.CServicePreferences aServicePreferences =
                new CServicePreferencesImpl();
    int aNumberOfItems = Integer.valueOf(maxhits).intValue();
    //Maximum numbers of docs go here!!!!!!!!!

    IDLInterchange.TServicePrefSummary pServicePrefSummary = 
        new IDLInterchange.TServicePrefSummary(aServicePreferences,
                                               aNumberOfItems,
                                               aMoreSummary);

    //Making call to RequestCosntrain

    proxy.RequestConstrain(pMessageID,pQuerySummary,pServicePrefSummary,
                           pResultTarget);
  }

  public static void testGetItemsPropertyNames (String stringcookie) {
    CosPropertyService.Any[] cookies = new CosPropertyService.Any[0];

    if ( stringcookie.equalsIgnoreCase("nothing")) {
      try {
       String[][] properties = proxy.GetItemsPropertyNames(cookies);

       System.out.println("GetItemsPropertyNames results");
       for(int i=0;i<properties[0].length;i++)
        System.out.println(properties[0][i]);
       ResultThread.objects[0] = new GeneralDoc(properties[0],
                                                properties[0]);
      }
      catch (Exception e) {
        System.out.println(e);
      }
    } else {
      try {
       String[][] properties = proxy.GetItemsPropertyNames(cookies);

       System.out.println("GetItemsPropertyNames results");
       for(int i=0;i<properties[0].length;i++)
        System.out.println(properties[0][i]);
      }
      catch (Exception e) {
        System.out.println(e);
      }
    }
    return;
    
  }

  public static void main(String[] args) {

    // Initialize the ORB.
    orb = org.omg.CORBA.ORB.init(args,null);

    //Creating the thread for the result collection
    if ( args.length != 0 ) {
      // We use Gatekeeper
      ResultThread = new ResultImpl(1,"ResultTarget");
    } else {
      ResultThread = new ResultImpl(0,"ResultTarget");
    }

    //Getting a Proxy (CConstrainCollection)
    proxy =
      IDLInterchange.CConstrainCollectionHelper.bind(orb,"Proxy");

    boolean QuitLoop = false;
    DataInputStream dis = new DataInputStream(System.in);
    String option = "";
    String maxhits = "";
    String keywords = "";
    String stringcookie = "";
    String ident = "";
    boolean done = false;
    longclcookies = 0;
    while (!QuitLoop) {
      System.out.println("\n");
      System.out.println("Option:");
      try {
      option = dis.readLine();
      if ( option.equalsIgnoreCase("m") || option.equalsIgnoreCase("?")) {
        print_menu();
        done = true;
      }
      if ( option.equalsIgnoreCase("0")) {
        if ( pResultTarget == null ) {
          System.out.println("Connecting...");
          GetResultTarget();
          testGetItemsPropertyNames("nothing");
        } else {
          System.out.println("Already connected");
        }
        done = true;
      }
      if ( option.equalsIgnoreCase("1")) {
        if ( pResultTarget != null ) {
          System.out.println("Get Property Names");
          testGetItemsPropertyNames("");
        } else {
          System.out.println("Not connected");
        }
        done = true;
      }
      if ( option.equalsIgnoreCase("2")) {
        if ( pResultTarget != null ) {
          ResultThread.pobjects = 1;
          System.out.println("Search");
          System.out.println("Enter keywords separated by spaces:");
          keywords = dis.readLine();
          System.out.println("Enter max number of hits [def: 5]");
          maxhits = dis.readLine();
          if ( maxhits.equalsIgnoreCase("")) {
            maxhits = "5";
          }
          WhatProperties(dis);
          HaveItems = false;
          total = 0;
          testRequestConstrain(keywords,maxhits);
          System.out.println("Query submitted");
          ResultThread.completed = false;
          ResultThread.total = 0;
        } else {
          System.out.println("Not connected");
        }
        done = true;
      }
      if ( option.equalsIgnoreCase("3")) {
        if ( pResultTarget != null ) {
          System.out.println("Get Total Items");
          testGetTotalItems();
        } else {
          System.out.println("Not connected");
        }
        done = true;
      }
      if ( option.equalsIgnoreCase("4")) {
        if ( pResultTarget != null ) {
          if (ResultThread.completed) {
            System.out.println("Transaction completed");
            PrintObjects();
          } else {
            System.out.println("Transaction not completed");
          }
        } else {
          System.out.println("Not connected");
        }
        done = true;
      }
      if ( option.equalsIgnoreCase("q")) {
        done = true;
        QuitLoop = true;
        ResultThread.stop();
      }
      if (!done) {
        print_menu();
      }
      }
      catch (Exception e) {
        System.err.println(e);
      }
    }
  }

  public static void GetResultTarget () {
        //Getting a Result (CCollection)
        pResultTarget =
              IDLInterchange.CCollectionHelper.bind(orb,"ResultTarget");
  }

  public static void PrintObjects () {
    try {
    DataInputStream dis = new DataInputStream(System.in);
    String wait;
    int aux = ResultThread.pobjects - 1;
    System.out.println("Got " + aux  + " objects");
    for(int i=1;i<ResultThread.pobjects;i++) {
      System.out.println(ResultThread.objects[i].toString());
      System.out.println("Click return");
      wait = dis.readLine();
    }
    }
    catch (Exception e) {
        System.out.println(e);
    }
  }

  public static void WhatProperties (DataInputStream dis)
                                throws IOException {
    System.out.println("What porperties?");
    boolean end = false;
    short i = 0;
    String option = "";
    how_many_prop = 0;
    while ( !end && (i<ResultThread.objects[0].length) ) {
      System.out.println( ResultThread.objects[0].names[i] +
                          " : [y] Yes, [n] No, [a] All, [q] quit");
      option = dis.readLine();
      if ( option.equalsIgnoreCase("y")) {
        whatprop[how_many_prop++] = ResultThread.objects[0].names[i];
      } else {
        if ( option.equalsIgnoreCase("n")) {
        } else {
          if ( option.equalsIgnoreCase("a")) {
            whatprop[how_many_prop++] = ResultThread.objects[0].names[i];
            end = true;
          } else {
            if ( option.equalsIgnoreCase("q")) {
              end = true;
            } else {
              // Nothing means Yes
              whatprop[how_many_prop++] = ResultThread.objects[0].names[i];
            }
          }
        }
      }
      i++;
    }
    if ( option.equalsIgnoreCase("a") ) {
      while (i<ResultThread.objects[0].length)  {
        whatprop[how_many_prop++] = ResultThread.objects[0].names[i];
        i++;
      }
    }
    System.out.println("Properties selected:");
    for(int j=0;j<how_many_prop;j++)
      System.out.println(whatprop[j]);
    return;
  }

  public static void select_docs(DataInputStream dis) {
    String wait = "";
    int aux = ResultThread.pobjects - 1;
    System.out.println("Got " + aux  + " objects");
    try {
      short i =1;
      boolean end = false;
      while ( !end && (i<ResultThread.pobjects) ) {
        System.out.println(ResultThread.objects[i].toString());
        System.out.println("[y] Yes, [n] No, [a] All, [q] Quit");
        wait = dis.readLine();
        if ( wait.equalsIgnoreCase("y")) {
          clientcookies[longclcookies++] = i;
        } else {
          if ( wait.equalsIgnoreCase("n")) {
          } else {
            if ( wait.equalsIgnoreCase("a")) {
              clientcookies[longclcookies++] = i;
              end = true;
            } else {
              if ( wait.equalsIgnoreCase("q")) {
                end = true;
              } else {
                // Nothing means Yes
                clientcookies[longclcookies++] = i;
              }
            }
          }
        }
        i++;
      }
      if ( wait.equalsIgnoreCase("a") ) {
        while (i<ResultThread.pobjects)  {
          clientcookies[longclcookies++] = i;;
          i++;
        }
      }
      System.out.println("Docs selected:");
      for(int j=0;j<longclcookies;j++)
        System.out.println(clientcookies[j]);
    }
    catch (Exception e) {
        System.out.println(e);
    }
    return;
  }
}