UserDAO.java

/*
 * Copyright 2012 Nodeable Inc
 *
 *    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 com.streamreduce.core.dao;

import com.google.code.morphia.Datastore;
import com.google.code.morphia.query.Query;
import com.streamreduce.core.model.APIAuthenticationToken;
import com.streamreduce.core.model.Account;
import com.streamreduce.core.model.User;

import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.regex.Pattern;

import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Repository;
import org.springframework.util.Assert;

@Repository("userDAO")
public class UserDAO extends SobaObjectDAO<User> {

    @Autowired
    protected UserDAO(@Qualifier(value = "businessDBDatastore") Datastore datastore) {
        super(datastore);
    }

    public User findUser(String username) {
        Assert.hasText(username);
        return ds.createQuery(entityClazz)
                .field("username").equal(Pattern.compile("^\\Q" + username + "\\E$", Pattern.CASE_INSENSITIVE))
                .get();
    }

    public User findUser(String signupKey, String userId) {
        Assert.hasText(signupKey);
        Assert.hasText(userId);
        return ds.createQuery(entityClazz)
                .field("secretKey").equal(signupKey)
                .field("_id").equal(new ObjectId(userId))
                .get();
    }

    public User findInvitedUser(String inviteKey, String accountId) {
        Assert.hasText(inviteKey);
        Assert.hasText(accountId);

        User user = ds.createQuery(entityClazz)
                .field("secretKey").equal(inviteKey)
                .get();

        if (user != null) {
            if ((user.getAccount().getId().toString().equals(accountId))) {
                return user;
            }
        }
        return null;
    }

    public List<User> allEnabledUsersForAccount(Account account) {
        return ds.createQuery(entityClazz)
                .field("account").equal(account)
                .field("userStatus").notEqual(User.UserStatus.DISABLED)
                .asList();

    }

    public User findUserForAlias(Account account, String alias) {
        return ds.createQuery(entityClazz)
                .field("alias").equal(Pattern.compile("^\\Q" + alias + "\\E$", Pattern.CASE_INSENSITIVE))
                .field("account").equal(account)
                .get();
    }

    public User findUserForUsername(Account account, String username) {
        return ds.createQuery(entityClazz)
                .field("username").equal(Pattern.compile("^\\Q" + username + "\\E$", Pattern.CASE_INSENSITIVE))
                .field("account").equal(account)
                .get();
    }

    public User findUserInAccount(Account account, String name) {
        Query<User> query = ds.createQuery(entityClazz);
        query.field("account").equal(account);
        query.or(
                query.criteria("username").equal(name),
                query.criteria("alias").equal(Pattern.compile("^\\Q" + name + "\\E$", Pattern.CASE_INSENSITIVE))
        );
        return query.get();
    }

    public APIAuthenticationToken findAuthToken(String authenticationToken) {
        User user = findByAuthToken(authenticationToken);
        if (user != null) {
            return user.getAuthenticationToken();
        }
        return null;
    }

    public User findByAuthToken(String authenticationToken) {
        Assert.hasText(authenticationToken);
        APIAuthenticationToken authToken = new APIAuthenticationToken(authenticationToken);

        return ds.createQuery(entityClazz)
                .field("authenticationToken").equal(authToken)
                .get();
    }


    public Set<User> getActiveLoggedInUsers(Account account, Long maxInactivity) {
        Query<User> q = createQuery();
        q.field("account").equal(account);
        q.field("lastActivity").greaterThanOrEq(System.currentTimeMillis() - maxInactivity);
        return new HashSet<>(q.asList());
    }

}