/**
* Copyright (C) 2013 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.integration.viewer.status;
import java.io.File;
import java.io.IOException;
import java.util.Collection;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.apache.commons.io.FileUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.opengamma.OpenGammaRuntimeException;
import com.opengamma.component.tool.AbstractTool;
import com.opengamma.financial.tool.ToolContext;
import com.opengamma.id.UniqueId;
import com.opengamma.integration.viewer.status.impl.BloombergReferencePortfolioMaker;
import com.opengamma.integration.viewer.status.impl.ViewStatusCalculationWorker;
import com.opengamma.master.portfolio.PortfolioSearchRequest;
import com.opengamma.master.portfolio.PortfolioSearchResult;
import com.opengamma.scripts.Scriptable;
/**
* The view status tool
*/
@Scriptable
public class ViewStatusTool extends AbstractTool<ToolContext> {
private static final Logger s_logger = LoggerFactory.getLogger(ViewStatusTool.class);
/**
* Main methog to run the tool.
*
* @param args the arguments, not null
*/
public static void main(String[] args) { //CSIGNORE
new ViewStatusTool().initAndRun(args, ToolContext.class);
System.exit(0);
}
@Override
protected void doRun() throws Exception {
ViewStatusOption option = ViewStatusOption.getViewStatusReporterOption(getCommandLine(), getToolContext());
String portfolioName = option.getPortfolioName();
UniqueId portfolioId = null;
if (portfolioName == null) {
portfolioId = createReferencePortfolio();
} else {
portfolioId = findPortfolioId(portfolioName);
}
if (portfolioId == null) {
throw new OpenGammaRuntimeException("Couldn't find portfolio " + portfolioName);
}
generateViewStatusReport(portfolioId, option);
}
private void generateViewStatusReport(final UniqueId portfolioId, final ViewStatusOption option) {
ViewStatusCalculationWorker calculationWorker = new ViewStatusCalculationWorker(getToolContext(), portfolioId, option);
ViewStatusResultAggregator resultAggregator = calculationWorker.run();
ViewStatusResultProducer resultProducer = new ViewStatusResultProducer();
String statusResult = resultProducer.statusResult(resultAggregator, option.getFormat(), option.getAggregateType());
try {
File outputFile = option.getOutputFile();
s_logger.debug("Writing status report into : {}", outputFile.getPath());
FileUtils.writeStringToFile(outputFile, statusResult);
} catch (IOException ex) {
throw new OpenGammaRuntimeException("Error writing view-status report to " + option.getOutputFile().toString(), ex);
}
}
private UniqueId findPortfolioId(final String portfolioName) {
final PortfolioSearchRequest searchRequest = new PortfolioSearchRequest();
searchRequest.setName(portfolioName);
final PortfolioSearchResult searchResult = getToolContext().getPortfolioMaster().search(searchRequest);
UniqueId portfolioId = null;
if (searchResult.getFirstPortfolio() != null) {
portfolioId = searchResult.getFirstPortfolio().getUniqueId().toLatest();
}
return portfolioId;
}
private UniqueId createReferencePortfolio() {
ToolContext toolContext = getToolContext();
BloombergReferencePortfolioMaker portfolioMaker = new BloombergReferencePortfolioMaker(toolContext.getPortfolioMaster(), toolContext.getPositionMaster(), toolContext.getSecurityMaster());
portfolioMaker.run();
return findPortfolioId(BloombergReferencePortfolioMaker.PORTFOLIO_NAME);
}
@SuppressWarnings("unchecked")
@Override
protected Options createOptions(boolean contextProvided) {
final Options toolOptions = super.createOptions(contextProvided);
Options viewStatusOptions = ViewStatusOption.createOptions();
for (Option option : (Collection<Option>) viewStatusOptions.getOptions()) {
s_logger.debug("adding {} to tool options", option);
toolOptions.addOption(option);
}
return toolOptions;
}
}