package org.apache.http.impl.cookie;

import org.apache.http.annotation.Contract;
import org.apache.http.annotation.ThreadingBehavior;
import org.apache.http.cookie.ClientCookie;
import org.apache.http.cookie.CommonCookieAttributeHandler;
import org.apache.http.cookie.Cookie;
import org.apache.http.cookie.CookieOrigin;
import org.apache.http.cookie.CookieRestrictionViolationException;
import org.apache.http.cookie.MalformedCookieException;
import org.apache.http.cookie.SetCookie;
import org.apache.http.util.Args;
import java.util.Locale;

@Contract(
   threading = ThreadingBehavior.IMMUTABLE
)
public class RFC2965DomainAttributeHandler implements CommonCookieAttributeHandler {
   public void parse(SetCookie cookie, String domain) throws MalformedCookieException {
      Args.notNull(cookie, "Cookie");
      if (domain == null) {
         throw new MalformedCookieException("Missing value for domain attribute");
      } else if (domain.trim().isEmpty()) {
         throw new MalformedCookieException("Blank value for domain attribute");
      } else {
         String s = domain.toLowerCase(Locale.ROOT);
         if (!domain.startsWith(".")) {
            s = '.' + s;
         }

         cookie.setDomain(s);
      }
   }

   public boolean domainMatch(String host, String domain) {
      boolean match = host.equals(domain) || domain.startsWith(".") && host.endsWith(domain);
      return match;
   }

   public void validate(Cookie cookie, CookieOrigin origin) throws MalformedCookieException {
      Args.notNull(cookie, "Cookie");
      Args.notNull(origin, "Cookie origin");
      String host = origin.getHost().toLowerCase(Locale.ROOT);
      if (cookie.getDomain() == null) {
         throw new CookieRestrictionViolationException("Invalid cookie state: domain not specified");
      } else {
         String cookieDomain = cookie.getDomain().toLowerCase(Locale.ROOT);
         if (cookie instanceof ClientCookie && ((ClientCookie)cookie).containsAttribute("domain")) {
            if (!cookieDomain.startsWith(".")) {
               throw new CookieRestrictionViolationException("Domain attribute \"" + cookie.getDomain() + "\" violates RFC 2109: domain must start with a dot");
            }

            int dotIndex = cookieDomain.indexOf(46, 1);
            if ((dotIndex < 0 || dotIndex == cookieDomain.length() - 1) && !cookieDomain.equals(".local")) {
               throw new CookieRestrictionViolationException("Domain attribute \"" + cookie.getDomain() + "\" violates RFC 2965: the value contains no embedded dots " + "and the value is not .local");
            }

            if (!this.domainMatch(host, cookieDomain)) {
               throw new CookieRestrictionViolationException("Domain attribute \"" + cookie.getDomain() + "\" violates RFC 2965: effective host name does not " + "domain-match domain attribute.");
            }

            String effectiveHostWithoutDomain = host.substring(0, host.length() - cookieDomain.length());
            if (effectiveHostWithoutDomain.indexOf(46) != -1) {
               throw new CookieRestrictionViolationException("Domain attribute \"" + cookie.getDomain() + "\" violates RFC 2965: " + "effective host minus domain may not contain any dots");
            }
         } else if (!cookie.getDomain().equals(host)) {
            throw new CookieRestrictionViolationException("Illegal domain attribute: \"" + cookie.getDomain() + "\"." + "Domain of origin: \"" + host + "\"");
         }

      }
   }

   public boolean match(Cookie cookie, CookieOrigin origin) {
      Args.notNull(cookie, "Cookie");
      Args.notNull(origin, "Cookie origin");
      String host = origin.getHost().toLowerCase(Locale.ROOT);
      String cookieDomain = cookie.getDomain();
      if (!this.domainMatch(host, cookieDomain)) {
         return false;
      } else {
         String effectiveHostWithoutDomain = host.substring(0, host.length() - cookieDomain.length());
         return effectiveHostWithoutDomain.indexOf(46) == -1;
      }
   }

   public String getAttributeName() {
      return "domain";
   }
}
