package eu.dnetlib.repo.manager.config;

import org.mitre.oauth2.model.ClientDetailsEntity.AuthMethod;
import org.mitre.oauth2.model.RegisteredClient;
import org.mitre.openid.connect.client.OIDCAuthenticationFilter;
import org.mitre.openid.connect.client.OIDCAuthenticationProvider;
import org.mitre.openid.connect.client.service.impl.*;
import org.mitre.openid.connect.config.ServerConfiguration;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.authentication.AuthenticationManager;
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;
import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
import org.springframework.security.web.authentication.LoginUrlAuthenticationEntryPoint;
import org.springframework.security.web.authentication.preauth.AbstractPreAuthenticatedProcessingFilter;

import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

@Configuration
@EnableWebSecurity
public class AaiSecurityConfiguration extends WebSecurityConfigurerAdapter {

    @Value("${webapp.dev.front}")
    private String logoutSuccessUrl;

    @Value("${oidc.issuer}")
    private String oidcIssuer;

    @Value("${oidc.id}")
    private String oidcId;

    @Value("${oidc.secret}")
    private String oidcSecret;

    @Value("${oidc.dev.home}")
    private String oidcDevHome;

    @Value("${webapp.dev.front}")
    private String webAppFrontEnd;

    private Map<String, String> userRoles = new HashMap<String, String>(){{
        put("urn:geant:openaire.eu:group:Super+Administrator#aai.openaire.eu", "ROLE_ADMIN");
        put("urn:geant:openaire.eu:group:Content+Provider+Dashboard+Administrator#aai.openaire.eu","ROLE_PROVIDE_ADMIN");
    }};

    @Bean
    @Override
    public AuthenticationManager authenticationManagerBean() throws Exception {
        return authenticationManager();
    }

    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
        auth.authenticationProvider(openIdConnectAuthenticationProvider());
    }

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http.csrf().disable()
                .anonymous().disable()
                .authorizeRequests()
                .anyRequest().authenticated()
                .and()
                    .httpBasic()
                    .authenticationEntryPoint(authenticationEntryPoint())
                .and()
                    .logout().logoutUrl("/openid_logout")
                    .invalidateHttpSession(true)
                    .deleteCookies("openAIRESession")
                    .logoutSuccessUrl(logoutSuccessUrl)
                .and()
                    .addFilterBefore(openIdConnectAuthenticationFilter(), AbstractPreAuthenticatedProcessingFilter.class)
        ;
    }

    @Bean
    public OIDCAuthenticationProvider openIdConnectAuthenticationProvider(){
        OIDCAuthenticationProvider oidcProvider = new OIDCAuthenticationProvider();
        oidcProvider.setAuthoritiesMapper(authoritiesMapper());
        return oidcProvider;
    }

    @Bean
    public OpenAireProviderAuthoritiesMapper authoritiesMapper(){
        OpenAireProviderAuthoritiesMapper authoritiesMapper = new OpenAireProviderAuthoritiesMapper(userRoles);
        return authoritiesMapper;
    }

    @Bean
    public StaticServerConfigurationService staticServerConfigurationService(){
        StaticServerConfigurationService staticServerConfigurationService = new StaticServerConfigurationService();
        Map<String, ServerConfiguration> servers = new HashMap<>();
        servers.put(oidcIssuer, serverConfiguration());
        staticServerConfigurationService.setServers(servers);
        return staticServerConfigurationService;
    }

    @Bean
    public StaticClientConfigurationService staticClientConfigurationService(){
        StaticClientConfigurationService staticClientConfigurationService = new StaticClientConfigurationService();
        Map<String, RegisteredClient> clients = new HashMap<>();
        clients.put(oidcIssuer,registeredClient());
        staticClientConfigurationService.setClients(clients);
        return staticClientConfigurationService;
    }

    @Bean
    public RegisteredClient registeredClient(){
        RegisteredClient registeredClient = new RegisteredClient();
        registeredClient.setClientId(oidcId);
        registeredClient.setClientSecret(oidcSecret);
        registeredClient.setScope(new HashSet<>(Collections.singletonList("openid")));
        registeredClient.setTokenEndpointAuthMethod(AuthMethod.SECRET_BASIC);
        registeredClient.setRedirectUris(new HashSet<>(Collections.singletonList(oidcDevHome)));
        return registeredClient;
    }

    @Bean
    public StaticAuthRequestOptionsService staticAuthRequestOptionsService(){
        return new StaticAuthRequestOptionsService();
    }

    @Bean
    public PlainAuthRequestUrlBuilder plainAuthRequestUrlBuilder(){
        return new PlainAuthRequestUrlBuilder();
    }

    @Bean
    public ServerConfiguration serverConfiguration(){
        ServerConfiguration serverConfiguration = new ServerConfiguration();
        serverConfiguration.setIssuer(oidcIssuer);
        serverConfiguration.setAuthorizationEndpointUri(oidcIssuer+"authorize");
        serverConfiguration.setTokenEndpointUri(oidcIssuer+"token");
        serverConfiguration.setUserInfoUri(oidcIssuer+"userinfo");
        serverConfiguration.setJwksUri(oidcIssuer+"jwk");
        serverConfiguration.setRevocationEndpointUri(oidcIssuer+"revoke");
        return serverConfiguration;
    }

    @Bean
    public LoginUrlAuthenticationEntryPoint authenticationEntryPoint(){
        return new LoginUrlAuthenticationEntryPoint("/openid_connect_login");
    }


    @Bean
    public OIDCAuthenticationFilter openIdConnectAuthenticationFilter() throws Exception {
        OIDCAuthenticationFilter oidc = new OIDCAuthenticationFilter();
        oidc.setAuthenticationManager(authenticationManagerBean());
        oidc.setIssuerService(staticSingleIssuerService());
        oidc.setServerConfigurationService(staticServerConfigurationService());
        oidc.setClientConfigurationService(staticClientConfigurationService());
        oidc.setAuthRequestOptionsService(staticAuthRequestOptionsService());
        oidc.setAuthRequestUrlBuilder(plainAuthRequestUrlBuilder());
        oidc.setAuthenticationSuccessHandler(frontEndRedirect());
        return oidc;
    }

    @Bean
    public StaticSingleIssuerService staticSingleIssuerService(){
        StaticSingleIssuerService staticSingleIssuerService = new StaticSingleIssuerService();
        staticSingleIssuerService.setIssuer(oidcIssuer);
        return staticSingleIssuerService;
    }

    @Bean(initMethod = "init")
    public FrontEndLinkURIAuthenticationSuccessHandler frontEndRedirect(){
        FrontEndLinkURIAuthenticationSuccessHandler frontEnd = new FrontEndLinkURIAuthenticationSuccessHandler();
        frontEnd.setFrontEndURI(webAppFrontEnd);
        return frontEnd;
    }

}