/*
 * Copyright (c) 2005-2016 Vincent Vandenschrick. All rights reserved.
 *
 *  This file is part of the Jspresso framework.
 *
 *  Jspresso is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  Jspresso is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with Jspresso.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.jspresso.framework.util.http;

import java.util.HashMap;
import java.util.Map;

import javax.servlet.http.Cookie;

import org.jspresso.framework.util.preferences.IPreferencesStore;

/**
 * An implementation of preference store relying on browser cookies.
 *
 * @author Vincent Vandenschrick
 */
public class CookiePreferencesStore implements IPreferencesStore {

  private String storePath;
  private Map<String, String> sessionStore;

  /**
   * Constructs a new {@code CookiePreferencesStore} instance.
   */
  public CookiePreferencesStore() {
    this.storePath = GLOBAL_STORE;
    this.sessionStore = new HashMap<>();
  }

  /**
   * Sets the path of this store.
   *
   * @param storePath
   *          the preferences store path.
   */
  @Override
  public void setStorePath(String... storePath) {
    if (storePath != null && storePath.length > 0) {
      StringBuilder buff = new StringBuilder();
      for (String aStorePath : storePath) {
        buff.append(aStorePath).append('.');
      }
      this.storePath = buff.toString();
    } else {
      this.storePath = GLOBAL_STORE;
    }
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public String getPreference(String key) {
    String completeKey = storePath + key;
    if (sessionStore.containsKey(completeKey)) {
      return sessionStore.get(completeKey);
    } else {
      if (HttpRequestHolder.getServletRequest() != null) {
        Cookie[] cookies = HttpRequestHolder.getServletRequest().getCookies();
        if (cookies != null) {
          for (Cookie cooky : cookies) {
            if (completeKey.equals(cooky.getName())) {
              return cooky.getValue();
            }
          }
        }
      }
    }
    return null;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public void putPreference(String key, String value) {
    String completeKey = storePath + key;
    sessionStore.put(completeKey, value);
    if (HttpRequestHolder.getServletResponse() != null) {
      Cookie cookie = new Cookie(completeKey, value);
      cookie.setMaxAge(Integer.MAX_VALUE);
      HttpRequestHolder.getServletResponse().addCookie(cookie);
    }
  }

  /**
   * Deletes the cookie storing the preference.
   * <p>
   * {@inheritDoc}
   */
  @Override
  public void removePreference(String key) {
    String completeKey = storePath + key;
    sessionStore.remove(completeKey);
    if (HttpRequestHolder.getServletResponse() != null) {
      Cookie cookie = new Cookie(completeKey, "");
      cookie.setMaxAge(0);
      HttpRequestHolder.getServletResponse().addCookie(cookie);
    }
  }

}