/**
 * Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.integration.tool.portfolio;

import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;

import com.opengamma.OpenGammaRuntimeException;
import com.opengamma.component.tool.AbstractTool;
import com.opengamma.integration.tool.IntegrationToolContext;
import com.opengamma.master.portfolio.PortfolioDocument;
import com.opengamma.master.portfolio.PortfolioSearchRequest;
import com.opengamma.master.portfolio.PortfolioSearchResult;
import com.opengamma.scripts.Scriptable;

/**
 * Tool to aggregate portfolios
 */
@Scriptable
public class PortfolioCopyMoveTool extends AbstractTool<IntegrationToolContext> {

  private static final String ORIGINAL_PORTFOLIO_NAME = "n";
  private static final String NEW_PORTFOLIO_NAME = "m";
  private static final String COPY_OPT = "d";

  //-------------------------------------------------------------------------
  /**
   * Main method to run the tool.
   * 
   * @param args  the standard tool arguments, not null
   */
  public static void main(String[] args) {  // CSIGNORE
    new PortfolioCopyMoveTool().invokeAndTerminate(args);
  }

  //-------------------------------------------------------------------------
  @Override
  protected void doRun() {
    PortfolioSearchRequest portfolioSearchRequest = new PortfolioSearchRequest();
    portfolioSearchRequest.setName(getCommandLine().getOptionValue(ORIGINAL_PORTFOLIO_NAME));
    PortfolioSearchResult portfolioSearchResult = getToolContext().getPortfolioMaster().search(portfolioSearchRequest);

    if (portfolioSearchResult.getDocuments().size() == 0) {
      throw new OpenGammaRuntimeException("No matching portfolio could be found, will not proceed.");
    } else if (portfolioSearchResult.getDocuments().size() > 1) {
      throw new OpenGammaRuntimeException("More than one portfolio matches, will not proceed.");
    } else {
      PortfolioDocument portfolioDocument = portfolioSearchResult.getFirstDocument();
      PortfolioSearchRequest portfolioSearchRequest1 = new PortfolioSearchRequest();
      portfolioSearchRequest1.setName(getCommandLine().getOptionValue(NEW_PORTFOLIO_NAME));
      PortfolioSearchResult portfolioSearchResult1 = getToolContext().getPortfolioMaster().search(portfolioSearchRequest1);
      if (portfolioSearchResult1.getDocuments().size() != 0) {
        throw new OpenGammaRuntimeException("A portfolio with the specified new name already exists, will not proceed.");
      }
      if (getCommandLine().hasOption(COPY_OPT)) {
        // Copy - check if new portfolio name already in use
        portfolioDocument.getPortfolio().setName(getCommandLine().getOptionValue(NEW_PORTFOLIO_NAME));
        getToolContext().getPortfolioMaster().add(portfolioDocument);
      } else {
        // Move
        portfolioDocument.getPortfolio().setName(getCommandLine().getOptionValue(NEW_PORTFOLIO_NAME));
        getToolContext().getPortfolioMaster().update(portfolioDocument);
      }
    }
  }

  protected Options createOptions(boolean contextProvided) {
    Options options = super.createOptions(contextProvided);

    Option origNameOption = new Option(
        ORIGINAL_PORTFOLIO_NAME, "origname", true, "The name of the OpenGamma portfolio to copy or rename");
    origNameOption.setRequired(true);
    options.addOption(origNameOption);

    Option newNameOption = new Option(
        NEW_PORTFOLIO_NAME, "newname", true, "The new name of the OpenGamma portfolio");
    newNameOption.setRequired(true);
    options.addOption(newNameOption);

    Option copyOption = new Option(
        COPY_OPT, "duplicate", false, "If specified, copy the portfolio instead of moving it (but share referenced positions and securities)");
    copyOption.setRequired(false);
    options.addOption(copyOption);

    return options;
  }
}