/*
* 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.filemgr.tools;
//OODT imports
import org.apache.oodt.cas.filemgr.structs.Product;
import org.apache.oodt.cas.filemgr.structs.exceptions.CatalogException;
import org.apache.oodt.cas.filemgr.structs.exceptions.ConnectionException;
import org.apache.oodt.cas.filemgr.system.XmlRpcFileManagerClient;
import org.apache.oodt.cas.metadata.Metadata;
import org.apache.oodt.commons.xml.XMLUtils;
//JDK imports
import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* @author mattmann
* @version $Revision$
*
* <p>
* A simple tool to write out a .product {@link Product} file containing product
* information and {@link Reference}s.
* </p>.
*/
public final class ProductDumper {
/* our log stream */
private static final Logger LOG = Logger.getLogger(ProductDumper.class
.getName());
/* our file manager client */
private XmlRpcFileManagerClient fmClient = null;
private final static String FILENAME = "Filename";
private final static String PRODUCT_NAME = "CAS.ProductName";
public ProductDumper(String fmUrlStr) throws InstantiationException {
try {
this.fmClient = new XmlRpcFileManagerClient(new URL(fmUrlStr));
} catch (MalformedURLException e) {
LOG.log(Level.SEVERE, "malformed file manager url: [" + fmUrlStr
+ "]", e);
throw new InstantiationException(e.getMessage());
} catch (ConnectionException e) {
LOG.log(Level.SEVERE, "unable to connect to file manager: ["
+ fmUrlStr + "]", e);
throw new InstantiationException(e.getMessage());
}
}
private Product getProduct(String productId) {
Product product = null;
try {
product = this.fmClient.getProductById(productId);
} catch (Exception e) {
throw new RuntimeException("Unable to retrieve product:["
+ productId + "] by id");
}
try {
product.setProductReferences(this.fmClient
.getProductReferences(product));
} catch (CatalogException e) {
LOG.log(Level.WARNING,
"Unable to obtain product references! Message: "
+ e.getMessage());
}
return product;
}
private void writeProductFileToDir(Product p, String fullProdFilePath) {
try {
XMLUtils.writeXmlFile(p.toXML(), fullProdFilePath);
} catch (Exception e) {
LOG.log(Level.WARNING, "product xml file not generated: reason: "
+ e.getMessage(), e);
}
}
private Metadata getMetadata(Product product) {
Metadata met = null;
try {
met = this.fmClient.getMetadata(product);
} catch (Exception e) {
throw new RuntimeException("Unable to get metadata for product: ["
+ product.getProductName() + "]");
}
return met;
}
/**
* Dumps a {@link Product} in XML format identified by the given
* <code>productId</code>. The {@link Product}XML is written to the
* local directory <code>.</code>, where this program was invoked from.
*
* @param productId
* The string identifier of the product to dump {@link Product}
* info from.
*/
public void dumpProduct(String productId) {
dumpProduct(productId, new File(".").getAbsolutePath());
}
/**
* Dumps a {@link Product} in XML format identified by the given
* <code>productId</code>. The {@link Product}XML is written to the
* specified <code>outDirPath</code>.
*
* @param productId
* The string identifier of the product to dump {@link Product}
* info from.
* @param outDirPath
* The path on the local filesystem to write the {@link Product}
* xml file to.
*/
public void dumpProduct(String productId, String outDirPath) {
Product product = getProduct(productId);
String fullProdFilePath = outDirPath;
fullProdFilePath = (fullProdFilePath.endsWith("/")) ? fullProdFilePath
: fullProdFilePath + "/";
Metadata met = getMetadata(product);
String filename = met.getMetadata(FILENAME) != null ? met
.getMetadata(FILENAME) : met.getMetadata(PRODUCT_NAME);
fullProdFilePath += filename + ".product";
writeProductFileToDir(product, fullProdFilePath);
}
/**
* @param args
*/
public static void main(String[] args) throws Exception {
String fileManagerUrlStr = null, productId = null, outDirPath = null;
String usage = "ProductDumper --url <filemgr url> --productId <id> [--out <dir path>]\n";
for (int i = 0; i < args.length; i++) {
if (args[i].equals("--url")) {
fileManagerUrlStr = args[++i];
} else if (args[i].equals("--productId")) {
productId = args[++i];
} else if (args[i].equals("--out")) {
outDirPath = args[++i];
}
}
if (fileManagerUrlStr == null || productId == null) {
System.err.println(usage);
System.exit(1);
}
ProductDumper dumper = new ProductDumper(fileManagerUrlStr);
if (outDirPath != null) {
dumper.dumpProduct(productId, outDirPath);
} else
dumper.dumpProduct(productId);
}
}