Cookies.java

package org.smartly.packages.http.impl.util.vtool;


import org.json.JSONObject;
import org.smartly.commons.util.JsonWrapper;
import org.smartly.commons.util.LocaleUtils;
import org.smartly.packages.velocity.impl.vtools.IVLCTool;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.text.DecimalFormatSymbols;
import java.util.HashMap;
import java.util.Map;

public class Cookies implements IVLCTool {

    public static final String NAME = "cookies";

    public static final String HEADER_SET_COOKIE = "Set-Cookie";
    public static final String HEADER_COOKIE = "Cookie";

    private final HttpServletRequest _request;
    private final HttpServletResponse _response;
    private final Map<String, String> __cookies;
    private JSONObject _authCookie;


    public Cookies(final HttpServletRequest httprequest, final HttpServletResponse httpresponse) {
        _request = httprequest;
        _response = httpresponse;
        __cookies = new HashMap<String, String>();
    }

    @Override
    public String getName() {
        return NAME;
    }


    public Cookie[] getCookiesArray() {
        return _request.getCookies();
    }

    public Cookie getCookie(final String name) {
        final Cookie[] cookies = this.getCookiesArray();
        for (final Cookie cookie : cookies) {
            if (cookie.getName().equalsIgnoreCase(name)) {
                return cookie;
            }
        }
        return null;
    }

    public String getCookieValue(final String name) {
        final Cookie cookie = this.getCookie(name);
        return null != cookie ? cookie.getValue() : "";
    }

    public Map<String, String> getCookies() {
        if (__cookies.size() == 0) {
            this.initCookies();
        }
        return __cookies;
    }

    public void addCookie(final Cookie cookie) {
        _response.addCookie(cookie);
        __cookies.put(cookie.getName(), cookie.getValue());
    }

    public void addCookie(final String name, final String value) {
        final Cookie cookie = new Cookie(name, value);
        cookie.setPath("/");
        this.addCookie(cookie);
    }

    public boolean hasCookie(final String name) {
        if (__cookies.size() == 0) {
            this.initCookies();
        }
        return __cookies.containsKey(name);
    }

    public boolean hasAuth() {
        return null != _authCookie;
    }

    public JSONObject getAuth() {
        if (null == _authCookie) {
            _authCookie = this.getAuthCookie();
        }
        return _authCookie;
    }

    public String getLang() {
        return this.getAuthLang();
    }

    public void setLang(final String lang) {
        this.setAuthLang(lang);
    }

    public String getAuthLang() {
        return this.getAuth().optString("lang");
    }

    public void setAuthLang(final String lang) {
        this.getAuth().putOpt("lang", lang);
        this.addCookie("lang", lang);
    }

    // ------------------------------------------------------------------------
    //                      p r i v a t e
    // ------------------------------------------------------------------------

    private void initCookies() {
        final Cookie[] cookies = this.getCookiesArray();
        for (final Cookie cookie : cookies) {
            __cookies.put(cookie.getName(), cookie.getValue());
        }
    }

    /**
     * exports.userid = authcookie.userid || exports.userid;
     * exports.accountid = authcookie.accountid || exports.accountid || exports.userid;
     * exports.username = authcookie.username || exports.username;
     * exports.lang = authcookie.lang || exports.lang;
     * exports.currency = authcookie.curr || exports.currency;
     * exports.country = authcookie.country || exports.country;
     * //decimals =  authCookie.userid||decimals;
     * exports.decimalsep = authcookie.sep || exports.decimalsep;
     * exports.discount = authcookie.discount || exports.discount;
     *
     * @return
     */
    private JSONObject getAuthCookie() {
        final JSONObject result = new JSONObject();

        // userid
        if (!this.hasCookie("userid")) {
            this.addCookie("userid", "0");
        }
        JsonWrapper.put(result, "userid", this.getCookieValue("userid"));

        // userid
        if (!this.hasCookie("accountid")) {
            this.addCookie("accountid", "0");
        }
        JsonWrapper.put(result, "accountid", this.getCookieValue("accountid"));

        // accountid
        if (!this.hasCookie("accountid")) {
            this.addCookie("accountid", "0");
        }
        JsonWrapper.put(result, "accountid", this.getCookieValue("accountid"));

        // lang
        String lang = getLangCode();
        if (!this.hasCookie("lang")) {
            this.addCookie("lang", lang);
        } else {
            lang = this.getCookieValue("lang");
        }
        JsonWrapper.put(result, "lang", lang);

        // country
        String country = getCountry(lang);
        if (!this.hasCookie("country")) {
            this.addCookie("country", country);
        } else {
            country = this.getCookieValue("country");
        }
        JsonWrapper.put(result, "country", country);

        // currency
        final DecimalFormatSymbols dfs = getDfs(lang, country);
        //if (!this.hasCookie("currency")) {
        this.addCookie("currency", dfs.getCurrency().getCurrencyCode());
        //}
        JsonWrapper.put(result, "currency", dfs.getCurrency().getCurrencyCode());

        // decimalsep
        //if (!this.hasCookie("decimalsep")) {
        this.addCookie("decimalsep", "" + dfs.getDecimalSeparator());
        //}
        JsonWrapper.put(result, "decimalsep", "" + dfs.getDecimalSeparator());

        // discount
        if (!this.hasCookie("discount")) {
            this.addCookie("discount", "0");
        }
        JsonWrapper.put(result, "discount", this.getCookieValue("discount"));

        // exports.userenabled = null != authcookie.userenabled ? authcookie.userenabled : false;
        if (!this.hasCookie("userenabled")) {
            this.addCookie("userenabled", "false");
        }
        JsonWrapper.put(result, "userenabled", this.getCookieValue("userenabled"));

        return result;
    }

    private String getLangCode() {
        return Req.getLang(_request);
    }

    private static String getCountry(final String lang) {
        return LocaleUtils.getCountry(LocaleUtils.getLocaleFromString(lang)).getCountry();
    }

    private static DecimalFormatSymbols getDfs(final String lang, final String country) {
        return LocaleUtils.getDecimalFormatSymbols(LocaleUtils.getLocale(lang, country));
    }


}