package org.forgerock.openam.sts.tokengeneration.service;

import static org.forgerock.http.routing.RoutingMode.STARTS_WITH;

import static org.forgerock.json.resource.http.CrestHttp.newHttpHandler;

import static org.forgerock.openam.audit.AuditConstants.Component.STS;

import javax.inject.Inject;

import javax.inject.Named;

import java.util.Collections;

import java.util.Set;

import com.google.inject.Key;

import com.google.inject.TypeLiteral;

import org.forgerock.http.Handler;

import org.forgerock.http.handler.Handlers;

import org.forgerock.json.resource.CollectionResourceProvider;

import org.forgerock.openam.http.HttpRoute;

import org.forgerock.openam.http.HttpRouteProvider;

import org.forgerock.openam.rest.RestRouter;

import org.forgerock.openam.rest.authz.STSTokenGenerationServiceAuthzModule;

import org.forgerock.openam.sts.tokengeneration.CTSTokenPersistence;

import org.forgerock.openam.sts.tokengeneration.config.TokenGenerationServiceInjectorHolder;

import org.forgerock.openam.sts.tokengeneration.oidc.OpenIdConnectTokenGeneration;

import org.forgerock.openam.sts.tokengeneration.saml2.SAML2TokenGeneration;

import org.forgerock.openam.sts.tokengeneration.state.RestSTSInstanceState;

import org.forgerock.openam.sts.tokengeneration.state.STSInstanceStateProvider;

import org.forgerock.openam.sts.tokengeneration.state.SoapSTSInstanceState;

import org.forgerock.util.Function;

import org.forgerock.util.promise.NeverThrowsException;

import org.slf4j.Logger;

public class TokenGenerationServiceHttpRouteProvider implements HttpRouteProvider {

private RestRouter rootRouter;

private org.forgerock.http.Filter authenticationFilter;

@Inject

public void setRouters(RestRouter router) {

this.rootRouter = router;

}

@Inject

public void setAuthenticationFilter(@Named("AuthenticationFilter") org.forgerock.http.Filter authenticationFilter) {

this.authenticationFilter = authenticationFilter;

}

@Override

public Set<HttpRoute> get() {

return Collections.singleton(HttpRoute.newHttpRoute(STARTS_WITH, "sts-tokengen", new Function<Void, Handler, NeverThrowsException>() {

@Override

public Handler apply(Void value) throws NeverThrowsException {

CollectionResourceProvider tokenGenerationService =

new TokenGenerationService(

TokenGenerationServiceInjectorHolder.getInstance(Key.get(SAML2TokenGeneration.class)),

TokenGenerationServiceInjectorHolder.getInstance(Key.get(OpenIdConnectTokenGeneration.class)),

TokenGenerationServiceInjectorHolder.getInstance(Key.get(new TypeLiteral<STSInstanceStateProvider<RestSTSInstanceState>>(){})),

TokenGenerationServiceInjectorHolder.getInstance(Key.get(new TypeLiteral<STSInstanceStateProvider<SoapSTSInstanceState>>(){})),

TokenGenerationServiceInjectorHolder.getInstance(Key.get(CTSTokenPersistence.class)),

TokenGenerationServiceInjectorHolder.getInstance(Key.get(Logger.class)));

rootRouter.route("")

.auditAs(STS)

.authorizeWith(STSTokenGenerationServiceAuthzModule.class)

.toCollection(tokenGenerationService);

return Handlers.chainOf(newHttpHandler(rootRouter.getRouter()), authenticationFilter);

}

}));

}

}