/**
 * Licensed to Apereo under one or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information regarding copyright ownership. Apereo
 * licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not use
 * this file except in compliance with the License. You may obtain a copy of the License at the
 * following location:
 *
 * <p>http://www.apache.org/licenses/LICENSE-2.0
 *
 * <p>Unless required by applicable law or agreed to in writing, software distributed under the
 * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apereo.portal.portlet.container.cache;

import javax.portlet.CacheControl;
import org.apache.commons.lang.StringUtils;
import org.apereo.portal.portlet.rendering.PortletResourceOutputHandler;

/**
 * CacheControl where calls to {@link #setExpirationTime(int)}, {@link #setPublicScope(boolean)},
 * and {@link #setETag(String)} result setting headers Last-Modified, CacheControl and ETag
 *
 */
public class HeaderSettingCacheControl implements CacheControl {
    private final CacheControl cacheControl;
    private final PortletResourceOutputHandler portletResourceOutputHandler;
    private final long lastModified = System.currentTimeMillis();

    public HeaderSettingCacheControl(
            CacheControl cacheControl, PortletResourceOutputHandler portletResourceOutputHandler) {
        this.cacheControl = cacheControl;
        this.portletResourceOutputHandler = portletResourceOutputHandler;
    }

    private void setCacheHeaders(int time, boolean publicScope) {
        PortletCachingHeaderUtils.setCachingHeaders(
                time, publicScope, lastModified, this.portletResourceOutputHandler);
    }

    @Override
    public int getExpirationTime() {
        return this.cacheControl.getExpirationTime();
    }

    @Override
    public void setExpirationTime(int time) {
        this.cacheControl.setExpirationTime(time);
        this.setCacheHeaders(time, this.isPublicScope());
    }

    @Override
    public boolean isPublicScope() {
        return this.cacheControl.isPublicScope();
    }

    @Override
    public void setPublicScope(boolean publicScope) {
        this.cacheControl.setPublicScope(publicScope);
        this.setCacheHeaders(this.getExpirationTime(), publicScope);
    }

    @Override
    public String getETag() {
        return this.cacheControl.getETag();
    }

    @Override
    public void setETag(String token) {
        token = StringUtils.trimToNull(token);
        this.cacheControl.setETag(token);
        PortletCachingHeaderUtils.setETag(token, portletResourceOutputHandler);
    }

    @Override
    public boolean useCachedContent() {
        return this.cacheControl.useCachedContent();
    }

    @Override
    public void setUseCachedContent(boolean useCachedContent) {
        this.cacheControl.setUseCachedContent(useCachedContent);
    }
}