0.8.1-rc1/curator/src/main/java/org/apache/oodt/cas/curation/service/PolicyResource.java - oodt - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You 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 org.apache.oodt.cas.curation.service;

 //OODT imports
 import org.apache.commons.lang.StringEscapeUtils;
 import org.apache.oodt.cas.filemgr.repository.XMLRepositoryManager;
 import org.apache.oodt.cas.filemgr.structs.Product;
 import org.apache.oodt.cas.filemgr.structs.ProductPage;
 import org.apache.oodt.cas.filemgr.structs.ProductType;
 import org.apache.oodt.cas.filemgr.structs.Query;
 import org.apache.oodt.cas.filemgr.structs.exceptions.RepositoryManagerException;

 //JDK imports
 import java.io.File;
 import java.io.FilenameFilter;
 import java.io.IOException;
 import java.net.MalformedURLException;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Vector;
 import java.util.logging.Level;
 import java.util.logging.Logger;

 //JAX-RS imports
 import javax.servlet.ServletContext;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import javax.ws.rs.DefaultValue;
 import javax.ws.rs.GET;
 import javax.ws.rs.Path;
 import javax.ws.rs.Produces;
 import javax.ws.rs.QueryParam;
 import javax.ws.rs.core.Context;
 import javax.ws.rs.core.UriInfo;

 //JSON imports
 import net.sf.json.JSONObject;

 @Path("policy")
 public class PolicyResource extends CurationService {

   @Context
   UriInfo uriInfo;

   private static final long serialVersionUID = -3757481221589264709L;

   private static final Logger LOG = Logger.getLogger(PolicyResource.class
       .getName());

   private static final FilenameFilter DIR_FILTER = new FilenameFilter() {

     public boolean accept(File dir, String name) {
       return new File(dir, name).isDirectory()
           && !new File(dir, name).getName().startsWith(".");
     }
   };

   public PolicyResource(@Context ServletContext context){

   }

   @GET
   @Path("browse")
   @Produces("text/plain")
   public String browseCatalog(
       @QueryParam("path") @DefaultValue("/") String path,
       @DefaultValue(FORMAT_HTML) @QueryParam("format") String format,
       @DefaultValue("1") @QueryParam("pageNum") Integer pageNum,
       @Context HttpServletRequest req, @Context HttpServletResponse res)
       throws IOException {

     // TODO: Send a not authorized response if not logged in. This should be a
     // utility method as a part of CurationService that every service interface
     // calls.

     String[] pathToks = tokenizeVirtualPath(path);
     String policy = null;
     String productType = null;

     if (pathToks == null) {
       LOG.log(Level.WARNING, "malformed path token string: "
           + Arrays.asList(pathToks));
       return "";
     }

     policy = pathToks.length > 0 ? pathToks[0]:null;
     productType = pathToks.length > 1 ? pathToks[1] : null;

     if (policy != null) {
       if (productType != null) {
         return getProductsForProductType(policy, productType, format, pageNum);
       } else {
         return getProductTypesForPolicy(policy, format);
       }
     } else {
       return getPolicies(format);
     }

   }

   private String getProductsForProductType(String policy,
       String productTypeName,
       String format, int pageNum) {

     ProductType productType;
     ProductPage page;
     try {
       productType = this.config.getFileManagerClient().getProductTypeByName(
           productTypeName);
       page = this.config.getFileManagerClient().pagedQuery(new Query(),
           productType, pageNum);
     } catch (Exception e) {
       e.printStackTrace();
       LOG.log(Level.WARNING, "Unable to obtain products for product type: ["
           + productTypeName + "]: Message: " + e.getMessage());
       return "";
     }

     if (format.equals(FORMAT_HTML)) {
       return encodeProductsAsHTML(page, policy, productTypeName);
     } else if (format.equals(FORMAT_JSON)) {
       return encodeProductsAsJSON(page, policy, productTypeName);
     } else {
       return UNKNOWN_OUT_FORMAT;
     }
   }

   private String encodeProductsAsHTML(ProductPage page, String policy,
       String productTypeName) {
     StringBuffer html = new StringBuffer();
     html.append("<ul class=\"fileTree\" >\r\n");

     for (Product product : page.getPageProducts()) {
       html.append(" <li class=\"file\">");
       html.append("<a href=\"#\" rel=\"/");
       html.append(policy);
       html.append("/");
       html.append(productTypeName);
       html.append("/");
       html.append(product.getProductId());
       html.append("\">");
       html.append(product.getProductName());
       html.append("</a>");
       html.append("</li>\r\n");
     }

     html.append("</ul>");
     return html.toString();
   }

   private String encodeProductsAsJSON(ProductPage page, String policy,
       String productTypeName) {
     return "NOT IMPLENTED YET";
   }

   private String getPolicies(String format) {
     String policyPath = this.cleanse(CurationService.config
         .getPolicyUploadPath());
     String[] policyDirs = new File(policyPath).list(DIR_FILTER);

     if (format.equals(FORMAT_HTML)) {
       return encodePoliciesAsHTML(policyDirs);
     } else if (format.equals(FORMAT_JSON)) {
       return encodePoliciesAsJSON(policyDirs);
     } else {
       return UNKNOWN_OUT_FORMAT;
     }

   }

   private String getProductTypesForPolicy(String policy, String format) {
     String[] typeNames = null;
     try {
       typeNames = this.getProductTypeNamesForPolicy(policy);
     } catch (Exception e) {
       e.printStackTrace();
       LOG.log(Level.WARNING,
           "Unable to obtain product type names for policy: [" + policy
               + "]: Message: " + e.getMessage());
       return "";
     }

     if (format.equals(FORMAT_HTML)) {
       return encodeProductTypesAsHTML(policy, typeNames);
     } else if (format.equals(FORMAT_JSON)) {
       return encodeProductTypesAsJSON(policy, typeNames);
     } else {
       return UNKNOWN_OUT_FORMAT;
     }

   }

   private String encodePoliciesAsHTML(String[] policyDirs) {
     StringBuffer out = new StringBuffer();
     out.append("<ul class=\"fileTree\" >");
     for (String policy : policyDirs) {
       out.append("<li class=\"directory collapsed\"><a href=\"#\" rel=\"/");
       out.append(StringEscapeUtils.escapeHtml(policy));
       out.append("/\">");
       out.append(StringEscapeUtils.escapeHtml(policy));
       out.append("</a></li>");
     }
     out.append("</ul>");
     return out.toString();
   }

   private String encodePoliciesAsJSON(String[] policyDirs) {
     Map<String, String> retMap = new HashMap<String, String>();
     for (String policyDir : policyDirs) {
       retMap.put("policy", policyDir);
     }
     JSONObject resObj = new JSONObject();
     resObj.put("policies", retMap);
     resObj.put("succeed", true);
     return resObj.toString();
   }

   private String encodeProductTypesAsHTML(String policy, String[] typeNames) {
     StringBuffer out = new StringBuffer();
     out.append("<ul class=\"fileTree\" >");
     for (String type : typeNames) {
       out
           .append("<li class=\"directory collapsed productType\"><a href=\"#\" rel=\"/");
       out.append(StringEscapeUtils.escapeHtml(policy));
       out.append("/");
       out.append(StringEscapeUtils.escapeHtml(type));
       out.append("/\">");
       out.append(StringEscapeUtils.escapeHtml(type));
       out.append("</a></li>");
     }

     out.append("</ul>");
     return out.toString();
   }

   private String encodeProductTypesAsJSON(String policy, String[] typeNames) {
     Map<String, Object> retMap = new HashMap<String, Object>();
     retMap.put("policy", policy);
     List<Map<String, String>> typeList = new Vector<Map<String, String>>();
     for (String typeName : typeNames) {
       Map<String, String> typeMap = new HashMap<String, String>();
       typeMap.put("name", typeName);
       typeList.add(typeMap);
     }
     retMap.put("productTypes", typeList);
     JSONObject resObj = new JSONObject();
     resObj.putAll(retMap);
     return resObj.toString();
   }

   private String[] getProductTypeNamesForPolicy(String policy)
       throws MalformedURLException, InstantiationException,
       RepositoryManagerException {
     String rootPolicyPath = this.cleanse(CurationService.config
         .getPolicyUploadPath());
     String policyPath = new File(rootPolicyPath + policy).toURL()
         .toExternalForm();
     String[] policies = { policyPath };
     XMLRepositoryManager repMgr = new XMLRepositoryManager(Arrays
         .asList(policies));
     List<ProductType> types = repMgr.getProductTypes();
     String[] typeNames = new String[types.size()];
     int i = 0;
     for (ProductType type : types) {
       typeNames[i] = type.getName();
       i++;
     }

     return typeNames;
   }

   private String cleanse(String origPath) {
     String retStr = origPath;
     if (!retStr.endsWith("/")) {
       retStr += "/";
     }
     return retStr;
   }

   private String[] tokenizeVirtualPath(String path) {
     String vPath = path;
     if (vPath.startsWith("/") && vPath.length() != 1) {
       vPath = vPath.substring(1);
     }
     String[] pathToks = vPath.split("/");
     LOG.log(Level.INFO, "origPath: ["+path+"]");
     LOG.log(Level.INFO, "pathToks: "+Arrays.asList(pathToks));
     return pathToks;
   }

 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You 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 org.apache.oodt.cas.curation.service;

	//OODT imports
	import org.apache.commons.lang.StringEscapeUtils;
	import org.apache.oodt.cas.filemgr.repository.XMLRepositoryManager;
	import org.apache.oodt.cas.filemgr.structs.Product;
	import org.apache.oodt.cas.filemgr.structs.ProductPage;
	import org.apache.oodt.cas.filemgr.structs.ProductType;
	import org.apache.oodt.cas.filemgr.structs.Query;
	import org.apache.oodt.cas.filemgr.structs.exceptions.RepositoryManagerException;

	//JDK imports
	import java.io.File;
	import java.io.FilenameFilter;
	import java.io.IOException;
	import java.net.MalformedURLException;
	import java.util.Arrays;
	import java.util.HashMap;
	import java.util.List;
	import java.util.Map;
	import java.util.Vector;
	import java.util.logging.Level;
	import java.util.logging.Logger;

	//JAX-RS imports
	import javax.servlet.ServletContext;
	import javax.servlet.http.HttpServletRequest;
	import javax.servlet.http.HttpServletResponse;
	import javax.ws.rs.DefaultValue;
	import javax.ws.rs.GET;
	import javax.ws.rs.Path;
	import javax.ws.rs.Produces;
	import javax.ws.rs.QueryParam;
	import javax.ws.rs.core.Context;
	import javax.ws.rs.core.UriInfo;

	//JSON imports
	import net.sf.json.JSONObject;

	@Path("policy")
	public class PolicyResource extends CurationService {

	@Context
	UriInfo uriInfo;

	private static final long serialVersionUID = -3757481221589264709L;

	private static final Logger LOG = Logger.getLogger(PolicyResource.class
	.getName());

	private static final FilenameFilter DIR_FILTER = new FilenameFilter() {

	public boolean accept(File dir, String name) {
	return new File(dir, name).isDirectory()
	&& !new File(dir, name).getName().startsWith(".");
	}
	};

	public PolicyResource(@Context ServletContext context){

	}

	@GET
	@Path("browse")
	@Produces("text/plain")
	public String browseCatalog(
	@QueryParam("path") @DefaultValue("/") String path,
	@DefaultValue(FORMAT_HTML) @QueryParam("format") String format,
	@DefaultValue("1") @QueryParam("pageNum") Integer pageNum,
	@Context HttpServletRequest req, @Context HttpServletResponse res)
	throws IOException {

	// TODO: Send a not authorized response if not logged in. This should be a
	// utility method as a part of CurationService that every service interface
	// calls.

	String[] pathToks = tokenizeVirtualPath(path);
	String policy = null;
	String productType = null;

	if (pathToks == null) {
	LOG.log(Level.WARNING, "malformed path token string: "
	+ Arrays.asList(pathToks));
	return "";
	}

	policy = pathToks.length > 0 ? pathToks[0]:null;
	productType = pathToks.length > 1 ? pathToks[1] : null;

	if (policy != null) {
	if (productType != null) {
	return getProductsForProductType(policy, productType, format, pageNum);
	} else {
	return getProductTypesForPolicy(policy, format);
	}
	} else {
	return getPolicies(format);
	}

	}

	private String getProductsForProductType(String policy,
	String productTypeName,
	String format, int pageNum) {

	ProductType productType;
	ProductPage page;
	try {
	productType = this.config.getFileManagerClient().getProductTypeByName(
	productTypeName);
	page = this.config.getFileManagerClient().pagedQuery(new Query(),
	productType, pageNum);
	} catch (Exception e) {
	e.printStackTrace();
	LOG.log(Level.WARNING, "Unable to obtain products for product type: ["
	+ productTypeName + "]: Message: " + e.getMessage());
	return "";
	}

	if (format.equals(FORMAT_HTML)) {
	return encodeProductsAsHTML(page, policy, productTypeName);
	} else if (format.equals(FORMAT_JSON)) {
	return encodeProductsAsJSON(page, policy, productTypeName);
	} else {
	return UNKNOWN_OUT_FORMAT;
	}
	}

	private String encodeProductsAsHTML(ProductPage page, String policy,
	String productTypeName) {
	StringBuffer html = new StringBuffer();
	html.append("<ul class=\"fileTree\" >\r\n");

	for (Product product : page.getPageProducts()) {
	html.append(" <li class=\"file\">");
	html.append("<a href=\"#\" rel=\"/");
	html.append(policy);
	html.append("/");
	html.append(productTypeName);
	html.append("/");
	html.append(product.getProductId());
	html.append("\">");
	html.append(product.getProductName());
	html.append("</a>");
	html.append("</li>\r\n");
	}

	html.append("</ul>");
	return html.toString();
	}

	private String encodeProductsAsJSON(ProductPage page, String policy,
	String productTypeName) {
	return "NOT IMPLENTED YET";
	}

	private String getPolicies(String format) {
	String policyPath = this.cleanse(CurationService.config
	.getPolicyUploadPath());
	String[] policyDirs = new File(policyPath).list(DIR_FILTER);

	if (format.equals(FORMAT_HTML)) {
	return encodePoliciesAsHTML(policyDirs);
	} else if (format.equals(FORMAT_JSON)) {
	return encodePoliciesAsJSON(policyDirs);
	} else {
	return UNKNOWN_OUT_FORMAT;
	}

	}

	private String getProductTypesForPolicy(String policy, String format) {
	String[] typeNames = null;
	try {
	typeNames = this.getProductTypeNamesForPolicy(policy);
	} catch (Exception e) {
	e.printStackTrace();
	LOG.log(Level.WARNING,
	"Unable to obtain product type names for policy: [" + policy
	+ "]: Message: " + e.getMessage());
	return "";
	}

	if (format.equals(FORMAT_HTML)) {
	return encodeProductTypesAsHTML(policy, typeNames);
	} else if (format.equals(FORMAT_JSON)) {
	return encodeProductTypesAsJSON(policy, typeNames);
	} else {
	return UNKNOWN_OUT_FORMAT;
	}

	}

	private String encodePoliciesAsHTML(String[] policyDirs) {
	StringBuffer out = new StringBuffer();
	out.append("<ul class=\"fileTree\" >");
	for (String policy : policyDirs) {
	out.append("<li class=\"directory collapsed\"><a href=\"#\" rel=\"/");
	out.append(StringEscapeUtils.escapeHtml(policy));
	out.append("/\">");
	out.append(StringEscapeUtils.escapeHtml(policy));
	out.append("</a></li>");
	}
	out.append("</ul>");
	return out.toString();
	}

	private String encodePoliciesAsJSON(String[] policyDirs) {
	Map<String, String> retMap = new HashMap<String, String>();
	for (String policyDir : policyDirs) {
	retMap.put("policy", policyDir);
	}
	JSONObject resObj = new JSONObject();
	resObj.put("policies", retMap);
	resObj.put("succeed", true);
	return resObj.toString();
	}

	private String encodeProductTypesAsHTML(String policy, String[] typeNames) {
	StringBuffer out = new StringBuffer();
	out.append("<ul class=\"fileTree\" >");
	for (String type : typeNames) {
	out
	.append("<li class=\"directory collapsed productType\"><a href=\"#\" rel=\"/");
	out.append(StringEscapeUtils.escapeHtml(policy));
	out.append("/");
	out.append(StringEscapeUtils.escapeHtml(type));
	out.append("/\">");
	out.append(StringEscapeUtils.escapeHtml(type));
	out.append("</a></li>");
	}

	out.append("</ul>");
	return out.toString();
	}

	private String encodeProductTypesAsJSON(String policy, String[] typeNames) {
	Map<String, Object> retMap = new HashMap<String, Object>();
	retMap.put("policy", policy);
	List<Map<String, String>> typeList = new Vector<Map<String, String>>();
	for (String typeName : typeNames) {
	Map<String, String> typeMap = new HashMap<String, String>();
	typeMap.put("name", typeName);
	typeList.add(typeMap);
	}
	retMap.put("productTypes", typeList);
	JSONObject resObj = new JSONObject();
	resObj.putAll(retMap);
	return resObj.toString();
	}

	private String[] getProductTypeNamesForPolicy(String policy)
	throws MalformedURLException, InstantiationException,
	RepositoryManagerException {
	String rootPolicyPath = this.cleanse(CurationService.config
	.getPolicyUploadPath());
	String policyPath = new File(rootPolicyPath + policy).toURL()
	.toExternalForm();
	String[] policies = { policyPath };
	XMLRepositoryManager repMgr = new XMLRepositoryManager(Arrays
	.asList(policies));
	List<ProductType> types = repMgr.getProductTypes();
	String[] typeNames = new String[types.size()];
	int i = 0;
	for (ProductType type : types) {
	typeNames[i] = type.getName();
	i++;
	}

	return typeNames;
	}

	private String cleanse(String origPath) {
	String retStr = origPath;
	if (!retStr.endsWith("/")) {
	retStr += "/";
	}
	return retStr;
	}

	private String[] tokenizeVirtualPath(String path) {
	String vPath = path;
	if (vPath.startsWith("/") && vPath.length() != 1) {
	vPath = vPath.substring(1);
	}
	String[] pathToks = vPath.split("/");
	LOG.log(Level.INFO, "origPath: ["+path+"]");
	LOG.log(Level.INFO, "pathToks: "+Arrays.asList(pathToks));
	return pathToks;
	}

	}