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

import java.net.URI;
import java.util.Collection;
import java.util.List;

import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriBuilder;
import javax.ws.rs.core.UriInfo;

import org.threeten.bp.LocalDate;

import com.opengamma.core.holiday.Holiday;
import com.opengamma.core.holiday.HolidaySource;
import com.opengamma.core.holiday.HolidayType;
import com.opengamma.id.ExternalIdBundle;
import com.opengamma.id.ObjectId;
import com.opengamma.id.UniqueId;
import com.opengamma.id.VersionCorrection;
import com.opengamma.util.ArgumentChecker;
import com.opengamma.util.fudgemsg.FudgeBooleanWrapper;
import com.opengamma.util.money.Currency;
import com.opengamma.util.rest.AbstractDataResource;

/**
 * RESTful resource for holidays.
 * <p>
 * The holidays resource receives and processes RESTful calls to the holiday source.
 */
@Path("holidaySource")
public class DataHolidaySourceResource extends AbstractDataResource {

  /**
   * The holiday source.
   */
  private final HolidaySource _holSource;

  /**
   * Creates the resource, exposing the underlying source over REST.
   * 
   * @param holidaySource  the underlying holiday source, not null
   */
  public DataHolidaySourceResource(final HolidaySource holidaySource) {
    ArgumentChecker.notNull(holidaySource, "holidaySource");
    _holSource = holidaySource;
  }

  //-------------------------------------------------------------------------
  /**
   * Gets the holiday source.
   * 
   * @return the holiday source, not null
   */
  public HolidaySource getHolidaySource() {
    return _holSource;
  }

  //-------------------------------------------------------------------------
  @GET
  public Response getHateaos(@Context UriInfo uriInfo) {
    return hateoasResponse(uriInfo);
  }

  @GET
  @Path("holidays/{holidayId}")
  public Response get(
      @PathParam("holidayId") String idStr,
      @QueryParam("version") String version,
      @QueryParam("versionAsOf") String versionAsOf,
      @QueryParam("correctedTo") String correctedTo) {
    
    final ObjectId objectId = ObjectId.parse(idStr);
    if (version != null) {
      final Holiday result = getHolidaySource().get(objectId.atVersion(version));
      return responseOkObject(result);
    } else {
      final VersionCorrection vc = VersionCorrection.parse(versionAsOf, correctedTo);
      Holiday result = getHolidaySource().get(objectId, vc);
      return responseOkObject(result);
    }
  }

  //-------------------------------------------------------------------------
  /**
   * Builds a URI.
   * 
   * @param baseUri  the base URI, not null
   * @param uniqueId  the unique identifier, may be null
   * @return the URI, not null
   */
  public static URI uriGet(URI baseUri, UniqueId uniqueId) {
    UriBuilder bld = UriBuilder.fromUri(baseUri).path("holidays/{holidayId}");
    if (uniqueId.getVersion() != null) {
      bld.queryParam("version", uniqueId.getVersion());
    }
    return bld.build(uniqueId.getObjectId());
  }

  /**
   * Builds a URI.
   *
   * @param baseUri  the base URI, not null
   * @param objectId  the object identifier, may be null
   * @param vc  the version-correction, null means latest
   * @return the URI, not null
   */
  public static URI uriGet(URI baseUri, ObjectId objectId, VersionCorrection vc) {
    UriBuilder bld = UriBuilder.fromUri(baseUri).path("holidays/{holidayId}");
    if (vc != null) {
      bld.queryParam("versionAsOf", vc.getVersionAsOfString());
      bld.queryParam("correctedTo", vc.getCorrectedToString());
    }
    return bld.build(objectId);
  }

  /**
   * Builds a URI.
   *
   * @param baseUri  the base URI, not null
   * @param currency  the currency, not null
   * @return the URI, not null
   */
  public static URI uriGet(URI baseUri, Currency currency) {
    return UriBuilder.fromUri(baseUri).path("holidaySearches/retrieve")
        .queryParam("currency", ArgumentChecker.notNull(currency, "currency").getCode())
        .build();
  }

  /**
   * Builds a URI.
   *
   * @param baseUri  the base URI, not null
   * @param holidayType  the holiday type, not null
   * @param regionOrExchangeIds  the ids, not null
   * @return the URI, not null
   */
  public static URI uriGet(URI baseUri, HolidayType holidayType, ExternalIdBundle regionOrExchangeIds) {
    return UriBuilder.fromUri(baseUri).path("holidaySearches/retrieve")
        .queryParam("holidayType", ArgumentChecker.notNull(holidayType, "holidayType").name())
        .queryParam("id", ArgumentChecker.notNull(regionOrExchangeIds, "regionOrExchangeIds").toStringList().toArray())
        .build();
  }

  @GET
  @Path("holidaySearches/retrieve")
  public Response get(
      @QueryParam("holidayType") HolidayType holidayType,
      @QueryParam("currency") String currencyCode,
      @QueryParam("id") List<String> externalIdStrs) {

    Collection<Holiday> result = holidayType == HolidayType.CURRENCY ?
        getHolidaySource().get(Currency.of(currencyCode)) :
        getHolidaySource().get(holidayType, ExternalIdBundle.parse(externalIdStrs));
    return responseOkObject(result);
  }

  // deprecated
  //-------------------------------------------------------------------------
  @GET
  @Path("holidaySearches/check")
  public Response check(
      @QueryParam("date") String localDateStr,
      @QueryParam("holidayType") HolidayType holidayType,
      @QueryParam("currency") String currencyCode,
      @QueryParam("id") List<String> externalIdStrs) {

    LocalDate date = LocalDate.parse(localDateStr);
    if (holidayType == HolidayType.CURRENCY) {
      boolean result = getHolidaySource().isHoliday(date, Currency.of(currencyCode));
      return responseOkObject(FudgeBooleanWrapper.of(result));
    } else {
      final ExternalIdBundle bundle = ExternalIdBundle.parse(externalIdStrs);
      boolean result = getHolidaySource().isHoliday(date, holidayType, bundle);
      return responseOkObject(FudgeBooleanWrapper.of(result));
    }
  }

  /**
   * Builds a URI.
   *
   * @param baseUri  the base URI, not null
   * @param date  the date, not null
   * @param holidayType  the holiday type, not null
   * @param currency  the currency, may be null
   * @param regionOrExchangeIds  the ids, may be null
   * @return the URI, not null
   */
  public static URI uriSearchCheck(URI baseUri, LocalDate date, HolidayType holidayType, Currency currency, ExternalIdBundle regionOrExchangeIds) {
    UriBuilder bld = UriBuilder.fromUri(baseUri).path("holidaySearches/check")
        .queryParam("date", date.toString())
        .queryParam("holidayType", holidayType.name());
    if (currency != null) {
      bld.queryParam("currency", currency.getCode());
    }
    if (regionOrExchangeIds != null) {
      bld.queryParam("id", regionOrExchangeIds.toStringList().toArray());
    }
    return bld.build();
  }
}