com.commercetools.sunrise.sessions.country.DefaultCountryInSession Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of common_2.11 Show documentation

common

There is a newer version: 1.0.0-M10

package com.commercetools.sunrise.sessions.country;

import com.commercetools.sunrise.sessions.DataFromResourceStoringOperations;
import com.commercetools.sunrise.sessions.SessionStrategy;
import com.neovisionaries.i18n.CountryCode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import play.Configuration;

import javax.annotation.Nullable;
import javax.inject.Inject;
import javax.inject.Singleton;
import java.util.Optional;

/**
 * Keeps the current country in session.
 */
@Singleton
public class DefaultCountryInSession extends DataFromResourceStoringOperations implements CountryInSession {

    private static final Logger LOGGER = LoggerFactory.getLogger(CountryInSession.class);
    private static final String DEFAULT_COUNTRY_SESSION_KEY = "sunrise-country";

    private final String countrySessionKey;
    private final SessionStrategy session;

    @Inject
    public DefaultCountryInSession(final SessionStrategy session, final Configuration configuration) {
        this.countrySessionKey = configuration.getString("session.country", DEFAULT_COUNTRY_SESSION_KEY);
        this.session = session;
    }

    @Override
    protected final Logger getLogger() {
        return LOGGER;
    }

    protected final String getCountrySessionKey() {
        return countrySessionKey;
    }

    protected final SessionStrategy getSession() {
        return session;
    }

    @Override
    public Optional findCountry() {
        return session.findValueByKey(countrySessionKey).map(CountryCode::getByCode);
    }

    @Override
    public void store(@Nullable final CountryCode countryCode) {
        super.store(countryCode);
    }

    @Override
    public void remove() {
        super.remove();
    }

    @Override
    protected void storeAssociatedData(final CountryCode countryCode) {
        session.overwriteValueByKey(countrySessionKey, countryCode.getAlpha2());
    }

    @Override
    protected void removeAssociatedData() {
        session.removeValueByKey(countrySessionKey);
    }
}