AdminUtils.java

/**
 * Copyright (C) 2015 Orion Health (Orchestral Development Ltd)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package xbdd.webapp.resource.feature;

import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Pattern;

import javax.inject.Inject;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;

import xbdd.webapp.factory.MongoDBAccessor;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.DuplicateKeyException;

@Path("/rest/admin")
public class AdminUtils {

	private final MongoDBAccessor client;

	@Inject
	public AdminUtils(final MongoDBAccessor client,
			@Context final HttpServletRequest req) {
		this.client = client;
		if (!req.isUserInRole("admin")) {
			throw new WebApplicationException();
		}
	}
	
	@DELETE
	@Path("/delete/{product}")
	@Produces("application/json")
	public Response softDeleteEntireProduct(@PathParam("product") final String product,
			@Context final HttpServletRequest req,
			@Context final HttpServletResponse response) throws IOException {
		
		final DB db = this.client.getDB("bdd");
		final DBCollection collection = db.getCollection("summary");
		final DBCollection targetCollection = db.getCollection("deletedSummary");
		
		final BasicDBObject query = new BasicDBObject("coordinates.product",product);
		
		final DBCursor cursor = collection.find(query);
		DBObject doc;
		
		while(cursor.hasNext()) {
			doc = cursor.next();
			//kill the old id
			doc.removeField("_id");
			try {
				targetCollection.insert(doc);
			} catch (Throwable e) {
				return Response.status(500).build();
			}
		}
		
		collection.remove(query);
		
		return Response.ok().build();
	}
	
	@DELETE
	@Path("/delete/{product}/{version}")
	@Produces("application/json")
	public Response softDeleteSingleVersion(@PathParam("product") final String product,
			@PathParam("version") final String version,
			@Context final HttpServletRequest req,
			@Context final HttpServletResponse response) throws IOException {
		
		final DB db = this.client.getDB("bdd");
		final DBCollection collection = db.getCollection("summary");
		final DBCollection targetCollection = db.getCollection("deletedSummary");
		
		final Pattern productReg = java.util.regex.Pattern.compile("^"+product+"/"+version+"$");
		final BasicDBObject query = new BasicDBObject("_id", productReg);
		
		final DBCursor cursor = collection.find(query);
		DBObject doc;
		
		while(cursor.hasNext()) {
			doc = cursor.next();
			//kill the old id
			doc.removeField("_id");
			try {
				targetCollection.insert(doc);
			} catch (Throwable e) {
				return Response.status(500).build();
			}
		}
		
		collection.remove(query);
		
		return Response.ok().build();
	}
	
	@DELETE
	@Path("/delete/{product}/{version}/{build}")
	@Produces("application/json")
	public Response softDeleteSingleBuild(@PathParam("product") final String product,
			@PathParam("build") final String build,
			@PathParam("version") final String version,
			@Context final HttpServletRequest req,
			@Context final HttpServletResponse response) throws IOException {
		
		final DB db = this.client.getDB("bdd");
		final DBCollection collection = db.getCollection("summary");
		final DBCollection targetCollection = db.getCollection("deletedSummary");
		
		final Pattern productReg = java.util.regex.Pattern.compile("^"+product+"/"+version+"$");
		final BasicDBObject query = new BasicDBObject("_id", productReg);
		
		final DBCursor cursor = collection.find(query);
		
		while (cursor.hasNext()) {
			DBObject doc = cursor.next();
			DBObject backupDoc = doc;
			//Make sure the backup document only has the deleted build number
			try {
				final String[] singleBuild = {build};
				backupDoc.put("builds", singleBuild);
				targetCollection.insert(backupDoc);
			} catch (DuplicateKeyException e) {
				//The backup document already exists, possibly already deleted a build
				//Lets add the deleted build to the existing document
				targetCollection.update(new BasicDBObject("_id",backupDoc.get("_id")), new BasicDBObject("$push", new BasicDBObject("builds",build)));
			} catch(Exception e) {
				return Response.status(500).build();
			}
			//Remove the build number from the current document and push it back into the collection
			try {
				collection.update(new BasicDBObject("_id",doc.get("_id")), new BasicDBObject("$pull",new BasicDBObject("builds", build)));
			} catch(Exception e) {
				System.out.println(e);
				return Response.status(500).build();
			}
		}
		return Response.ok().build();
	}

	
	//Lets register the helper class that replaces the instances of the old product name with the new one
	private DBObject renameDoc(String product, String newname, DBObject doc) {
		doc.put("_id", ((String) doc.get("_id")).replaceAll(product+"/", newname+"/"));
		if (doc.containsField("coordinates")) {
			DBObject coordinates = (DBObject) doc.get("coordinates");
			coordinates.put("product",newname);
			doc.put("coordinates", coordinates);
		}
		return doc;
	}
	
	public static class Product {
	    public String name;
	}
	
	@PUT @Consumes(MediaType.APPLICATION_JSON)
	@Path("/{product}")
	@Produces("application/json")
	public Response renameProduct(@PathParam("product") final String product,
			final Product renameObject) {
		
		final DB db = this.client.getDB("bdd");
		final List<DBCollection> collections = Arrays.asList(db.getCollection("summary"), db.getCollection("features"), db.getCollection("reportStats"), db.getCollection("testingTips"));
		
		final Pattern productReg = Pattern.compile("^"+product+"/");
		final BasicDBObject query = new BasicDBObject("_id", productReg);
		
		//Before anything lets check the new name isn't already in use
		
		final BasicDBObject existsQuery = new BasicDBObject("_id", Pattern.compile("^"+renameObject.name+"/"));
		final int summaryCount = db.getCollection("summary").find(existsQuery).count();
		final int delCount = db.getCollection("deletedSummary").find(existsQuery).count();
		
		if (delCount + summaryCount != 0) {
			throw new WebApplicationException();
		}
		
		//We need to rename the product everywhere
		//First up are all the collection with the product in the _id attribute
		for (DBCollection collectioni : collections) {
			DBCursor cursor = collectioni.find(query);
			while(cursor.hasNext()) {
				DBObject doc = cursor.next();
				String id = (String) doc.get("_id");
				doc = renameDoc(product, renameObject.name, doc);
				collectioni.insert(doc);
				collectioni.remove(new BasicDBObject("_id", id));
			}
		}
		
		//Then we deal with the environments collection where only the coordinates.product is set
		final DBCollection[] noIDCollections = {db.getCollection("environments"),db.getCollection("deletedSummary")};
		final BasicDBObject enviroQuery = new BasicDBObject("coordinates.product", product);
		
		for (int i=0;i<noIDCollections.length;i++) {
			final DBCursor enviroCursor = noIDCollections[i].find(enviroQuery);
			
			while (enviroCursor.hasNext()) {
				DBObject doc = enviroCursor.next();
				DBObject coordinates = (DBObject) doc.get("coordinates");
				coordinates.put("product",renameObject.name);
				DBObject updateDoc = new BasicDBObject("$set", new BasicDBObject("coordinates", coordinates));
				noIDCollections[i].update(new BasicDBObject("_id", doc.get("_id")), updateDoc);
			}
		}
		
		//Then we correct the name in any users favourites object
		final DBCollection userCollection = db.getCollection("users");
		final BasicDBObject favouriteQuery = new BasicDBObject("favourites."+product, new BasicDBObject("$exists",true));
		final DBCursor users = userCollection.find(favouriteQuery);
		
		while(users.hasNext()) {
			DBObject doc = users.next();
			DBObject favs = (DBObject) doc.get("favourites");
			favs.put(renameObject.name, favs.get(product));
			BasicDBObject updateDoc = new BasicDBObject("$set",new BasicDBObject("favourites",favs));
			updateDoc.put("$unset", new BasicDBObject(product, ""));
			userCollection.update(new BasicDBObject("_id", doc.get("_id")), updateDoc);
		}
		
		return Response.ok().build();
	}
}