package org.keycloak.models.sessions.mongo;

import org.keycloak.connections.mongo.api.context.MongoStoreInvocationContext;
import org.keycloak.models.ClientModel;
import org.keycloak.models.ClientSessionModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.RealmModel;
import org.keycloak.models.UserModel;
import org.keycloak.models.UserSessionModel;
import org.keycloak.models.sessions.mongo.entities.MongoClientSessionEntity;
import org.keycloak.models.sessions.mongo.entities.MongoUserSessionEntity;

import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
 */
public class ClientSessionAdapter extends AbstractMongoAdapter<MongoClientSessionEntity> implements ClientSessionModel {

    private KeycloakSession session;
    private MongoUserSessionProvider provider;
    private RealmModel realm;
    private MongoClientSessionEntity entity;

    public ClientSessionAdapter(KeycloakSession session, MongoUserSessionProvider provider, RealmModel realm, MongoClientSessionEntity entity, MongoStoreInvocationContext invContext) {
        super(invContext);
        this.session = session;
        this.provider = provider;
        this.realm = realm;
        this.entity = entity;
    }

    @Override
    public String getId() {
        return entity.getId();
    }

    @Override
    public RealmModel getRealm() {
        return session.realms().getRealm(entity.getRealmId());
    }

    @Override
    public ClientModel getClient() {
        return realm.getClientById(entity.getClientId());
    }

    @Override
    public UserSessionModel getUserSession() {
        if (entity.getSessionId() == null) return null;
        return provider.getUserSession(realm, entity.getSessionId());
    }

    @Override
    public void setUserSession(UserSessionModel userSession) {
        if (userSession == null) {
            if (entity.getSessionId() != null) {
                MongoUserSessionEntity userSessionEntity = provider.getUserSessionEntity(realm, entity.getSessionId());
                provider.getMongoStore().pullItemFromList(userSessionEntity, "clientSessions", entity.getSessionId(), invocationContext);
            }
            entity.setSessionId(null);
        } else {
            MongoUserSessionEntity userSessionEntity = provider.getUserSessionEntity(realm, userSession.getId());
            entity.setSessionId(userSessionEntity.getId());
            updateMongoEntity();

            provider.getMongoStore().pushItemToList(userSessionEntity, "clientSessions", entity.getId(), true, invocationContext);
        }
    }

    @Override
    public void setRedirectUri(String uri) {
        entity.setRedirectUri(uri);
        updateMongoEntity();
    }

    @Override
    public void setRoles(Set<String> roles) {
        if (roles == null) {
            entity.setRoles(null);
        } else {
            List<String> list = new LinkedList<String>();
            list.addAll(roles);
            entity.setRoles(list);
        }
        updateMongoEntity();
    }

    @Override
    public String getRedirectUri() {
        return entity.getRedirectUri();
    }

    @Override
    public int getTimestamp() {
        return entity.getTimestamp();
    }

    @Override
    public void setTimestamp(int timestamp) {
        entity.setTimestamp(timestamp);
        updateMongoEntity();
    }

    @Override
    public Action getAction() {
        return entity.getAction();
    }

    @Override
    public void setAction(Action action) {
        entity.setAction(action);
        updateMongoEntity();
    }

    @Override
    public Set<String> getRoles() {
        return entity.getRoles() != null ? new HashSet<String>(entity.getRoles()) : null;
    }

    @Override
    public Set<String> getProtocolMappers() {
        return entity.getProtocolMappers() != null ? new HashSet<String>(entity.getProtocolMappers()) : null;
    }

    @Override
    public void setProtocolMappers(Set<String> protocolMappers) {
        if (protocolMappers == null) {
            entity.setProtocolMappers(null);
        } else {
            List<String> list = new LinkedList<String>();
            list.addAll(protocolMappers);
            entity.setProtocolMappers(list);
        }
        updateMongoEntity();
    }

    @Override
    public String getNote(String name) {
        return entity.getNotes().get(name);
    }

    @Override
    public void setNote(String name, String value) {
        entity.getNotes().put(name, value);
        updateMongoEntity();
    }

    @Override
    public void removeNote(String name) {
        entity.getNotes().remove(name);
        updateMongoEntity();
    }

    @Override
    public void setUserSessionNote(String name, String value) {
        entity.getUserSessionNotes().put(name, value);
        updateMongoEntity();
    }

    @Override
    public Map<String, String> getUserSessionNotes() {
        Map<String, String> copy = new HashMap<>();
        copy.putAll(entity.getUserSessionNotes());
        return copy;
    }

    @Override
    public Map<String, UserSessionModel.AuthenticatorStatus> getAuthenticators() {
        return entity.getAuthenticatorStatus();
    }

    @Override
    public void setAuthenticatorStatus(String authenticator, UserSessionModel.AuthenticatorStatus status) {
        entity.getAuthenticatorStatus().put(authenticator, status);
        updateMongoEntity();

    }

    @Override
    public void setAuthenticatorStatus(Map<String, UserSessionModel.AuthenticatorStatus> status) {
        entity.setAuthenticatorStatus(status);
        updateMongoEntity();

    }

    @Override
    public UserModel getAuthenticatedUser() {
        return session.users().getUserById(entity.getAuthUserId(), realm);
    }

    @Override
    public void setAuthenticatedUser(UserModel user) {
        entity.setAuthUserId(user.getId());
        updateMongoEntity();

    }

    @Override
    public String getAuthMethod() {
        return entity.getAuthMethod();
    }

    @Override
    public void setAuthMethod(String method) {
        entity.setAuthMethod(method);
        updateMongoEntity();
    }

    @Override
    protected MongoClientSessionEntity getMongoEntity() {
        return entity;
    }
}
