package wrappers;

import java.io.BufferedReader;
import java.io.Console;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Set;

public class Client2 {
	static boolean DEBUG = true;
	static final Console c = System.console();
	static Mongo2Interface jdi = null;

	public static void main(String[] args) {
		int rc, m, i = 0;
		boolean brc;
		Set<String> columns;
		String[] parms = new String[4];

		jdi = new Mongo2Interface(DEBUG);

		parms[0] = getLine("URI: ", false);
		parms[1] = getLine("DB: ", false);
		parms[2] = null;
		parms[3] = null;

		if (parms[0] == null)
			parms[0] = "mongodb://localhost:27017";

		if (parms[1] == null)
			parms[1] = "test";

		rc = jdi.MongoConnect(parms);

		if (rc == 0) {
			String name, pipeline, query, fields;
			System.out.println("Successfully connected to " + parms[1]);

			while ((name = getLine("Collection: ", false)) != null) {
				if (jdi.GetCollection(name))
					System.out.println("GetCollection failed");
				else
					System.out.println("Collection size: " + jdi.GetCollSize());

				pipeline = getLine("Pipeline: ", false);

				if (pipeline == null) {
					query = getLine("Filter: ", false);
					fields = getLine("Proj: ", false);
					brc = jdi.FindColl(query, fields);
				} else
					brc = jdi.AggregateColl(pipeline);

				System.out.println("Returned brc = " + brc);

				if (!brc) {
					for (i = 0; i < 10; i++) {
						m = jdi.ReadNext();

						if (m > 0) {
							columns = jdi.GetColumns();

							for (String col : columns)
								System.out.println(col + "=" + jdi.GetField(col));

							if (pipeline == null) {
								if (name.equalsIgnoreCase("gtst"))
									System.out.println("gtst=" + jdi.GetField("*"));

								if (name.equalsIgnoreCase("inventory")) {
									System.out.println("warehouse=" + jdi.GetField("instock.0.warehouse"));
									System.out.println("quantity=" + jdi.GetField("instock.1.qty"));
								} // endif inventory

								if (name.equalsIgnoreCase("restaurants")) {
									System.out.println("score=" + jdi.GetField("grades.0.score"));
									System.out.println("date=" + jdi.GetField("grades.0.date"));
								} // endif restaurants
						
							} // endif pipeline

						} else if (m < 0) {
							System.out.println("ReadNext: " + jdi.GetErrmsg());
							break;
						} else
							break;

					} // endfor i

				} // endif brc

			} // endwhile name

			rc = jdi.MongoDisconnect();
			System.out.println("Disconnect returned " + rc);
		} else
			System.out.println(jdi.GetErrmsg() + " rc=" + rc);

	} // end of main

	// ==================================================================
	private static String getLine(String p, boolean b) {
		String response;

		if (c != null) {
			// Standard console mode
			if (b) {
				response = new String(c.readPassword(p));
			} else
				response = c.readLine(p);

		} else {
			// For instance when testing from Eclipse
			BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

			System.out.print(p);

			try {
				// Cannot suppress echo for password entry
				response = in.readLine();
			} catch (IOException e) {
				response = "";
			} // end of try/catch

		} // endif c

		return (response.isEmpty()) ? null : response;
	} // end of getLine

} // end of class Client