package main;
import com.mongodb.MongoClient;
import com.mongodb.MongoException;
import com.mongodb.WriteConcern;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.DBCursor;
import com.mongodb.ServerAddress;


public class App {
  MongoClient mongoClient;
  
  public App() throws Exception{
    System.out.println("Setting up client");
    mongoClient = new MongoClient("127.0.0.1", 27017);
    System.out.println("Client set up");
   //statement.execute("DROP TABLE IF EXISTS test;");
   //statement.execute("CREATE TABLE test (city varchar(20), state char(2));");
  }
  
  public void insertData(String city, String state) {
    System.out.println("Getting database");
    DB db = mongoClient.getDB("test");
    System.out.println("Got database");
    DBCollection coll = db.getCollection("testCollection");
    System.out.println("Got collection");
    BasicDBObject doc = new BasicDBObject("city", city).append("state", state);
    System.out.println("Got object");
    coll.insert(doc);
  }

  public String[] getData(String city, String state) {
    String[] toReturn = null;
    DBCursor cursor = null;
      
    try {
      DB db = mongoClient.getDB("test");
      DBCollection coll = db.getCollection("testCollection");
      BasicDBObject doc = new BasicDBObject("city", "Seattle").append("state", "WA");
      coll.insert(doc);
      if (city != null && state != null) {
        BasicDBObject query = new BasicDBObject("city", city).append("state", state);
        cursor = coll.find(query);
      } else if (city != null) {
        BasicDBObject query = new BasicDBObject("city", city);
        cursor = coll.find(query);
      } else if (state != null) {
        BasicDBObject query = new BasicDBObject("state", state);
        cursor = coll.find(query);
      } else {
        cursor = coll.find();
      }
      
      int number = cursor.size();
      toReturn = new String[number];
      int i = 0;
      while (cursor.hasNext()) {
        DBObject item = cursor.next();
        toReturn[i] = item.get("city") + ", " + item.get("state");
        i++;
      }
    } finally {
      try {
        cursor.close();
      } catch (Exception ex) {
        cursor = null;
      }
      return toReturn;
    }
  }
}