package org.apache.http.client.protocol;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.http.Header;
import org.apache.http.HeaderIterator;
import org.apache.http.HttpException;
import org.apache.http.HttpResponse;
import org.apache.http.HttpResponseInterceptor;
import org.apache.http.annotation.Contract;
import org.apache.http.annotation.ThreadingBehavior;
import org.apache.http.client.CookieStore;
import org.apache.http.cookie.Cookie;
import org.apache.http.cookie.CookieOrigin;
import org.apache.http.cookie.CookieSpec;
import org.apache.http.cookie.MalformedCookieException;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.Args;
import java.io.IOException;

@Contract(
   threading = ThreadingBehavior.IMMUTABLE
)
public class ResponseProcessCookies implements HttpResponseInterceptor {
   private final Log log = LogFactory.getLog(this.getClass());

   public void process(HttpResponse response, HttpContext context) throws HttpException, IOException {
      Args.notNull(response, "HTTP request");
      Args.notNull(context, "HTTP context");
      HttpClientContext clientContext = HttpClientContext.adapt(context);
      CookieSpec cookieSpec = clientContext.getCookieSpec();
      if (cookieSpec == null) {
         this.log.debug("Cookie spec not specified in HTTP context");
      } else {
         CookieStore cookieStore = clientContext.getCookieStore();
         if (cookieStore == null) {
            this.log.debug("Cookie store not specified in HTTP context");
         } else {
            CookieOrigin cookieOrigin = clientContext.getCookieOrigin();
            if (cookieOrigin == null) {
               this.log.debug("Cookie origin not specified in HTTP context");
            } else {
               HeaderIterator it = response.headerIterator("Set-Cookie");
               this.processCookies(it, cookieSpec, cookieOrigin, cookieStore);
               if (cookieSpec.getVersion() > 0) {
                  it = response.headerIterator("Set-Cookie2");
                  this.processCookies(it, cookieSpec, cookieOrigin, cookieStore);
               }

            }
         }
      }
   }

   private void processCookies(HeaderIterator iterator, CookieSpec cookieSpec, CookieOrigin cookieOrigin, CookieStore cookieStore) {
      while(iterator.hasNext()) {
         Header header = iterator.nextHeader();

         try {
            for(Cookie cookie : cookieSpec.parse(header, cookieOrigin)) {
               try {
                  cookieSpec.validate(cookie, cookieOrigin);
                  cookieStore.addCookie(cookie);
                  if (this.log.isDebugEnabled()) {
                     this.log.debug("Cookie accepted [" + formatCooke(cookie) + "]");
                  }
               } catch (MalformedCookieException ex) {
                  if (this.log.isWarnEnabled()) {
                     this.log.warn("Cookie rejected [" + formatCooke(cookie) + "] " + ex.getMessage());
                  }
               }
            }
         } catch (MalformedCookieException ex) {
            if (this.log.isWarnEnabled()) {
               this.log.warn("Invalid cookie header: \"" + header + "\". " + ex.getMessage());
            }
         }
      }

   }

   private static String formatCooke(Cookie cookie) {
      StringBuilder buf = new StringBuilder();
      buf.append(cookie.getName());
      buf.append("=\"");
      String v = cookie.getValue();
      if (v != null) {
         if (v.length() > 100) {
            v = v.substring(0, 100) + "...";
         }

         buf.append(v);
      }

      buf.append("\"");
      buf.append(", version:");
      buf.append(Integer.toString(cookie.getVersion()));
      buf.append(", domain:");
      buf.append(cookie.getDomain());
      buf.append(", path:");
      buf.append(cookie.getPath());
      buf.append(", expiry:");
      buf.append(cookie.getExpiryDate());
      return buf.toString();
   }
}
