/*
 * Copyright 2021 Haulmont.
 *
 * Licensed 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
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 io.jmix.security.configurer;

import io.jmix.core.session.SessionProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.configurers.AbstractHttpConfigurer;
import org.springframework.security.core.session.SessionRegistry;
import org.springframework.security.web.authentication.session.SessionAuthenticationStrategy;

public class SessionManagementConfigurer extends AbstractHttpConfigurer<SessionManagementConfigurer, HttpSecurity> {
    @Override
    public void setBuilder(HttpSecurity http) {
        super.setBuilder(http);
        initSessionManagement(http);
    }

    private void initSessionManagement(HttpSecurity http) {
        try {
            ApplicationContext applicationContext = http.getSharedObject(ApplicationContext.class);

            SessionAuthenticationStrategy sessionAuthenticationStrategy = applicationContext.getBean(SessionAuthenticationStrategy.class);
            SessionRegistry sessionRegistry = applicationContext.getBean(SessionRegistry.class);
            SessionProperties sessionProperties = applicationContext.getBean(SessionProperties.class);

            http.sessionManagement().sessionAuthenticationStrategy(sessionAuthenticationStrategy)
                    .maximumSessions(sessionProperties.getMaximumSessionsPerUser())
                    .sessionRegistry(sessionRegistry);
        } catch (Exception e) {
            throw new RuntimeException("Error while init security", e);
        }
    }
}
