| /* |
| * 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.policymgr; |
| |
| //OODT imports |
| import org.apache.oodt.cas.filemgr.repository.XMLRepositoryManager; |
| import org.apache.oodt.cas.filemgr.structs.ProductType; |
| import org.apache.oodt.cas.metadata.SerializableMetadata; |
| |
| //JDK imports |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.FileNotFoundException; |
| import java.io.FilenameFilter; |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Vector; |
| import java.util.logging.Level; |
| import java.util.logging.Logger; |
| |
| //APACHE imports |
| import net.sf.json.JSONArray; |
| import net.sf.json.JSONObject; |
| |
| import org.apache.commons.lang.StringUtils; |
| |
| /** |
| * |
| * |
| * A management interface for dealing with CAS filemgr policy, and for allowing |
| * for browsing and traversal of the staging area used in ingestion. |
| * |
| * @author mattmann |
| * @version $Revision$ |
| * |
| */ |
| public class CurationPolicyManager { |
| |
| private static final Logger LOG = Logger |
| .getLogger(CurationPolicyManager.class.getName()); |
| |
| private String stagingAreaPath; |
| |
| private String policyDirectoryPath; |
| |
| public CurationPolicyManager() { |
| this(null, null); |
| } |
| |
| public CurationPolicyManager(String policyDirectoryPath, |
| String stagingAreaPath) { |
| this.policyDirectoryPath = policyDirectoryPath; |
| this.stagingAreaPath = stagingAreaPath; |
| } |
| |
| public Map<String, ProductType> getProductTypes(String dsCollection) { |
| Map<String, ProductType> typeMap = new HashMap<String, ProductType>(); |
| XMLRepositoryManager xmlRepo = getRepo(dsCollection); |
| List<ProductType> types = safeGetProductTypes(xmlRepo); |
| |
| for (ProductType type : types) { |
| typeMap.put(type.getName(), type); |
| } |
| |
| return typeMap; |
| } |
| |
| public List<ProductType> typesToList(Map<String, ProductType> types) { |
| List<ProductType> typeList = new Vector<ProductType>(); |
| for (String typeStr : types.keySet()) { |
| typeList.add(types.get(typeStr)); |
| } |
| |
| return typeList; |
| } |
| |
| public String getDirectoryAreaAsJSON(String base, String directory, |
| boolean showFiles) { |
| String startingPath = (base + "/" + directory); |
| startingPath = StringUtils.replace(startingPath, "source", "/"); |
| String f[] = getFilesInDirectory(startingPath, showFiles); |
| |
| List<Map<String, Object>> items = new ArrayList<Map<String, Object>>(); |
| for (int i = 0; i < f.length; i++) { |
| Map<String, Object> entry = new HashMap<String, Object>(); |
| String children[] = getFilesInDirectory(startingPath + "/" + f[i], |
| showFiles); |
| entry.put("text", f[i]); |
| entry.put("id", directory + "/" + f[i]); |
| entry.put("expanded", false); |
| entry.put("hasChildren", children != null && (children.length > 0)); |
| entry.put("isFile", new File(startingPath + "/" + f[i]).isFile()); |
| items.add(entry); |
| } |
| |
| return JSONArray.fromObject(items).toString(); |
| } |
| |
| public String getExistingPoliciesAsHTML() { |
| String f[] = getFilesInDirectory(this.policyDirectoryPath, false); |
| |
| List<Map<String, Object>> items = new ArrayList<Map<String, Object>>(); |
| for (int i = 0; i < f.length; i++) { |
| Map<String, Object> entry = new HashMap<String, Object>(); |
| entry |
| .put("text", "<a href=\\\"" + f[i] |
| + "\\\" onclick=\\\"return treeSelection(this);\\\">" + f[i] |
| + "</a>"); |
| entry.put("id", f[i].toString()); |
| entry.put("expanded", false); |
| entry.put("hasChildren", false); |
| } |
| |
| return JSONArray.fromObject(items).toString(); |
| } |
| |
| public String getDatasetsByPolicyAsJSON(String dsCollection) { |
| XMLRepositoryManager xmlRepo = getRepo(dsCollection); |
| List<ProductType> types = safeGetProductTypes(xmlRepo); |
| |
| Iterator<ProductType> it = types.iterator(); |
| |
| List<Map<String, Object>> items = new ArrayList<Map<String, Object>>(); |
| while (it.hasNext()) { |
| ProductType t = it.next(); |
| Map<String, Object> entry = new HashMap<String, Object>(); |
| entry.put("text", "<a href=\\\"" + t.getName() |
| + "\\\" onclick=\\\"return treeSelection(this);\\\">" + t.getName() |
| + "</a>"); |
| entry.put("id", t.getName()); |
| entry.put("expanded", false); |
| entry.put("hasChildren", false); |
| } |
| |
| return JSONArray.fromObject(items).toString(); |
| } |
| |
| public String getMetFileAsJSON(String base, String file) { |
| Map<String, Object> responseMap = new HashMap<String, Object>(); |
| if (file != null) { |
| responseMap.put("file", file); |
| } |
| |
| try { |
| SerializableMetadata metadata = new SerializableMetadata("UTF-8", false); |
| metadata |
| .loadMetadataFromXmlStream(new FileInputStream(base + "/" + file)); |
| responseMap.put("metadata", metadata.getHashtable()); |
| } catch (InstantiationException e) { |
| // no op |
| } catch (FileNotFoundException e) { |
| // no op |
| } catch (IOException e) { |
| // no op |
| } |
| |
| JSONObject metJSON = JSONObject.fromObject(responseMap); |
| return metJSON.toString(); |
| } |
| |
| public void writeMetFileFromJSON(String base, String json) { |
| |
| } |
| |
| /** |
| * @return the stagingAreaPath |
| */ |
| public String getStagingAreaPath() { |
| return stagingAreaPath; |
| } |
| |
| /** |
| * @param stagingAreaPath |
| * the stagingAreaPath to set |
| */ |
| public void setStagingAreaPath(String stagingAreaPath) { |
| this.stagingAreaPath = stagingAreaPath; |
| } |
| |
| /** |
| * @return the policyDirectoryPath |
| */ |
| public String getPolicyDirectoryPath() { |
| return policyDirectoryPath; |
| } |
| |
| /** |
| * @param policyDirectoryPath |
| * the policyDirectoryPath to set |
| */ |
| public void setPolicyDirectoryPath(String policyDirectoryPath) { |
| this.policyDirectoryPath = policyDirectoryPath; |
| } |
| |
| private List<ProductType> safeGetProductTypes(XMLRepositoryManager xmlRepo) { |
| try { |
| return xmlRepo.getProductTypes(); |
| } catch (Exception ignore) { |
| return null; |
| } |
| |
| } |
| |
| private XMLRepositoryManager getRepo(String dsCollection) { |
| XMLRepositoryManager xmlRepo = null; |
| String url = "file://" + this.policyDirectoryPath + "/" + dsCollection; |
| |
| try { |
| xmlRepo = new XMLRepositoryManager(Collections.singletonList(url)); |
| } catch (Exception e) { |
| LOG.log(Level.WARNING, "Error reading product types from: [" + url |
| + "]: Message: " + e.getMessage()); |
| } |
| |
| return xmlRepo; |
| } |
| |
| private String[] getFilesInDirectory(String directory, final boolean showFiles) { |
| File dir = new File(directory); |
| FilenameFilter filter = new FilenameFilter() { |
| |
| public boolean accept(File dir, String name) { |
| |
| return !name.startsWith(".") |
| && (showFiles || !new File(dir, name).isFile()); |
| } |
| }; |
| return dir.list(filter); |
| } |
| } |