/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package m.timeseries.delete;

import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;
import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import csip.annotations.*;
import javax.ws.rs.Path;
import csip.ModelDataService;
import csip.SessionLogger;
import java.util.ArrayList;
import java.util.List;
import org.bson.Document;
import org.codehaus.jettison.json.JSONArray;
import org.codehaus.jettison.json.JSONException;
import org.codehaus.jettison.json.JSONObject;

/**
 * Mongodb timeseries data deletion
 *
 * @author Dave Patterson
 */
@Name("Timeseries delete")
@Description("Delete a timeseries collection or document in mongo.")
@Author(org = "CSU")
@License(License.MIT)
@Path("m/delete/1.0")
public class V1_0 extends ModelDataService {

  MongoClient mongo;
  MongoDatabase db;
  JSONObject results = new JSONObject();

  protected void open(String mongoclienturi) {
    MongoClientURI u = new MongoClientURI(mongoclienturi);
    mongo = new MongoClient(u);
    if (u.getDatabase() == null) {
      throw new RuntimeException("Missing database in mongo uri: " + mongoclienturi);
    }
    if (!u.getDatabase().equals("csip_timeseries")) {
      throw new RuntimeException("Can only modify csip_timeseries collection");
    }
    db = mongo.getDatabase(u.getDatabase());
  }

  protected void close() {
    mongo.close();
  }

  @Override
  public void doProcess() throws Exception {
    String mongo_uri = getStringParam("mongo_uri");
    String mongo_collection = getStringParam("mongo_collection");
    JSONArray ids = getJSONArrayParam("ids", null);

    if (ids != null) {
      deleteIds(mongo_uri, mongo_collection, ids);
    } else {
      // If no ids specified, then delete the collection.
      deleteCollection(mongo_uri, mongo_collection);
    }
  }

  public void deleteCollection(String mongo_uri, String mongo_collection) throws JSONException {
    open(mongo_uri);
    MongoCollection<Document> collection = db.getCollection(mongo_collection);
    collection.drop();
    results.put("deleted_collection", mongo_collection);
    close();
  }

  public void deleteIds(String mongo_uri, String mongo_collection, JSONArray ids) throws JSONException {
    open(mongo_uri);
    MongoCollection<Document> collection = db.getCollection(mongo_collection);

    List<String> ids_list = new ArrayList();
    for (int i = 0; i < ids.length(); i++) {
      ids_list.add(ids.getString(i));
    }

    List<String> deleted_ids = new ArrayList();
    FindIterable<Document> c = collection.find();
    for (Document finddoc : c) {
      String find_id = (String) finddoc.get("_id");
      if (ids_list.contains(find_id)) {
        LOG.info("Deleting existing doc");
        collection.deleteOne(finddoc);
        deleted_ids.add(find_id);
      }
    }
    results.put("deleted_document_ids", new JSONArray(deleted_ids));
    close();
  }

  @Override
  protected void postProcess() throws Exception {
    putResult("result", results);
  }

  public static void main(String[] args) throws JSONException {
    V1_0 timeseries = new V1_0();
    JSONArray delete_ids = new JSONArray();
    delete_ids.put("test_ref");
    timeseries.deleteIds("mongodb://eds0.engr.colostate.edu:27017/csip_timeseries", "test_coll", delete_ids);
    timeseries.deleteCollection("mongodb://eds0.engr.colostate.edu:27017/csip_timeseries", "test_coll");
  }

  public static String[] toStringArray(List<String> data) {
    return (String[]) data.stream().map(String::toString).toArray();
  }
}