/*
 * Copyright 2016 The Netty Project
 *
 * The Netty Project 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 play.core.cookie.encoding;

import java.util.Iterator;

import static play.core.cookie.encoding.CookieUtil.*;

/**
 * A <a href="https://tools.ietf.org/html/rfc6265">RFC6265</a> compliant cookie encoder to be used
 * client side, so only name=value pairs are sent.
 *
 * <p>Note that multiple cookies are supposed to be sent at once in a single "Cookie" header.
 *
 * @see ClientCookieDecoder
 */
public final class ClientCookieEncoder extends CookieEncoder {

  /**
   * Strict encoder that validates that name and value chars are in the valid scope defined in
   * RFC6265
   */
  public static final ClientCookieEncoder STRICT = new ClientCookieEncoder(true);

  /** Lax instance that doesn't validate name and value */
  public static final ClientCookieEncoder LAX = new ClientCookieEncoder(false);

  private ClientCookieEncoder(boolean strict) {
    super(strict);
  }

  /**
   * Encodes the specified cookie into a Cookie header value.
   *
   * @param name the cookie name
   * @param value the cookie value
   * @return a Rfc6265 style Cookie header value
   */
  public String encode(String name, String value) {
    return encode(new DefaultCookie(name, value));
  }

  /**
   * Encodes the specified cookie into a Cookie header value.
   *
   * @param cookie specified the cookie
   * @return a Rfc6265 style Cookie header value
   */
  public String encode(Cookie cookie) {
    if (cookie == null) {
      throw new NullPointerException("cookie");
    }
    StringBuilder buf = new StringBuilder();
    encode(buf, cookie);
    return stripTrailingSeparator(buf);
  }

  /**
   * Encodes the specified cookies into a single Cookie header value.
   *
   * @param cookies some cookies
   * @return a Rfc6265 style Cookie header value, null if no cookies are passed.
   */
  public String encode(Cookie... cookies) {
    if (cookies == null) {
      throw new NullPointerException("cookies");
    }
    if (cookies.length == 0) {
      return null;
    }

    StringBuilder buf = new StringBuilder();
    for (Cookie c : cookies) {
      if (c == null) {
        break;
      }

      encode(buf, c);
    }
    return stripTrailingSeparatorOrNull(buf);
  }

  /**
   * Encodes the specified cookies into a single Cookie header value.
   *
   * @param cookies some cookies
   * @return a Rfc6265 style Cookie header value, null if no cookies are passed.
   */
  public String encode(Iterable<? extends Cookie> cookies) {
    if (cookies == null) {
      throw new NullPointerException("cookies");
    }
    Iterator<? extends Cookie> cookiesIt = cookies.iterator();
    if (!cookiesIt.hasNext()) {
      return null;
    }

    StringBuilder buf = new StringBuilder();
    while (cookiesIt.hasNext()) {
      Cookie c = cookiesIt.next();
      if (c == null) {
        break;
      }

      encode(buf, c);
    }
    return stripTrailingSeparatorOrNull(buf);
  }

  private void encode(StringBuilder buf, Cookie c) {
    final String name = c.name();
    final String value = c.value() != null ? c.value() : "";

    validateCookie(name, value);

    if (c.wrap()) {
      addQuoted(buf, name, value);
    } else {
      add(buf, name, value);
    }
  }
}
