package org.opendaylight.aaa.authn.mdsal.store;

import java.util.List;

import org.opendaylight.aaa.api.IDMStoreException;
import org.opendaylight.aaa.api.IDMStoreUtil;
import org.opendaylight.aaa.api.IIDMStore;
import org.opendaylight.aaa.api.model.Domain;
import org.opendaylight.aaa.api.model.Domains;
import org.opendaylight.aaa.api.model.Grant;
import org.opendaylight.aaa.api.model.Grants;
import org.opendaylight.aaa.api.model.Role;
import org.opendaylight.aaa.api.model.Roles;
import org.opendaylight.aaa.api.model.User;
import org.opendaylight.aaa.api.model.Users;

public class IDMStore implements IIDMStore{
    private final IDMMDSALStore mdsalStore;

    public IDMStore(IDMMDSALStore mdsalStore){
        this.mdsalStore = mdsalStore;
    }

    @Override
    public Domain writeDomain(Domain domain) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMDomain(mdsalStore.writeDomain(IDMObject2MDSAL.toMDSALDomain(domain)));
    }

    @Override
    public Domain readDomain(String domainid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMDomain(mdsalStore.readDomain(domainid));
    }

    @Override
    public Domain deleteDomain(String domainid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMDomain(mdsalStore.deleteDomain(domainid));
    }

    @Override
    public Domain updateDomain(Domain domain) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMDomain(mdsalStore.updateDomain(IDMObject2MDSAL.toMDSALDomain(domain)));
    }

    @Override
    public Domains getDomains() throws IDMStoreException {
        Domains domains = new Domains();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Domain> mdSalDomains = mdsalStore.getAllDomains();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Domain d:mdSalDomains){
            domains.getDomains().add(IDMObject2MDSAL.toIDMDomain(d));
        }
        return domains;
    }

    @Override
    public Role writeRole(Role role) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMRole(mdsalStore.writeRole(IDMObject2MDSAL.toMDSALRole(role)));
    }

    @Override
    public Role readRole(String roleid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMRole(mdsalStore.readRole(roleid));
    }

    @Override
    public Role deleteRole(String roleid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMRole(mdsalStore.deleteRole(roleid));
    }

    @Override
    public Role updateRole(Role role) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMRole(mdsalStore.writeRole(IDMObject2MDSAL.toMDSALRole(role)));
    }

    @Override
    public User writeUser(User user) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMUser(mdsalStore.writeUser(IDMObject2MDSAL.toMDSALUser(user)));
    }

    @Override
    public User readUser(String userid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMUser(mdsalStore.readUser(userid));
    }

    @Override
    public User deleteUser(String userid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMUser(mdsalStore.deleteUser(userid));
    }

    @Override
    public User updateUser(User user) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMUser(mdsalStore.writeUser(IDMObject2MDSAL.toMDSALUser(user)));
    }

    @Override
    public Grant writeGrant(Grant grant) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMGrant(mdsalStore.writeGrant(IDMObject2MDSAL.toMDSALGrant(grant)));
    }

    @Override
    public Grant readGrant(String grantid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMGrant(mdsalStore.readGrant(grantid));
    }

    @Override
    public Grant deleteGrant(String grantid) throws IDMStoreException {
        return IDMObject2MDSAL.toIDMGrant(mdsalStore.readGrant(grantid));
    }

    @Override
    public Roles getRoles() throws IDMStoreException {
        Roles roles = new Roles();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Role> mdSalRoles = mdsalStore.getAllRoles();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Role r:mdSalRoles){
            roles.getRoles().add(IDMObject2MDSAL.toIDMRole(r));
        }
        return roles;
    }

    @Override
    public Users getUsers() throws IDMStoreException {
        Users users = new Users();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.User> mdSalUsers = mdsalStore.getAllUsers();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.User u:mdSalUsers){
            users.getUsers().add(IDMObject2MDSAL.toIDMUser(u));
        }
        return users;
    }

    @Override
    public Users getUsers(String username, String domain) throws IDMStoreException {
        Users users = new Users();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.User> mdSalUsers = mdsalStore.getAllUsers();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.User u:mdSalUsers){
            if(u.getDomainid().equals(domain) && u.getName().equals(username)){
                users.getUsers().add(IDMObject2MDSAL.toIDMUser(u));
            }
        }
        return users;
    }

    @Override
    public Grants getGrants(String domainid, String userid) throws IDMStoreException {
        Grants grants = new Grants();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Grant> mdSalGrants = mdsalStore.getAllGrants();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Grant g:mdSalGrants){
            grants.getGrants().add(IDMObject2MDSAL.toIDMGrant(g));
        }
        return grants;
    }

    @Override
    public Grants getGrants(String userid) throws IDMStoreException {
        Grants grants = new Grants();
        List<org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Grant> mdSalGrants = mdsalStore.getAllGrants();
        for(org.opendaylight.yang.gen.v1.urn.aaa.yang.authn.claims.rev141029.authentication.Grant g:mdSalGrants){
            if(g.getUserid().equals(userid)){
                grants.getGrants().add(IDMObject2MDSAL.toIDMGrant(g));
            }
        }
        return grants;
    }

    @Override
    public Grant readGrant(String domainid, String userid, String roleid) throws IDMStoreException {
        return readGrant(IDMStoreUtil.createGrantid(userid, domainid, roleid));
    }
}