org.benjp.services.mongodb.TokenServiceImpl.java Source code

Introduction

Here is the source code for org.benjp.services.mongodb.TokenServiceImpl.java

Source

/*
 * Copyright (C) 2012 eXo Platform SAS.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.benjp.services.mongodb;

import com.mongodb.*;
import org.benjp.listener.ConnectionManager;
import org.benjp.model.UserBean;
import org.benjp.utils.MessageDigester;
import org.benjp.utils.PropertyManager;

import javax.enterprise.context.ApplicationScoped;
import javax.inject.Named;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@Named("tokenService")
@ApplicationScoped
public class TokenServiceImpl implements org.benjp.services.TokenService {
    private int validity_ = -1;

    private DB db() {
        return ConnectionManager.getInstance().getDB();
    }

    public String getToken(String user) {
        String passphrase = PropertyManager.getProperty(PropertyManager.PROPERTY_PASSPHRASE);
        String in = user + passphrase;
        String token = MessageDigester.getHash(in);
        //System.out.println("getToken :: user="+user+" ; token="+token);
        return token;
    }

    public boolean hasUserWithToken(String user, String token) {
        DBCollection coll = db().getCollection(M_USERS_COLLECTION);
        BasicDBObject query = new BasicDBObject();
        query.put("user", user);
        query.put("token", token);
        DBCursor cursor = coll.find(query);
        return (cursor.hasNext());
    }

    public void addUser(String user, String token) {
        if (!hasUserWithToken(user, token)) {
            //System.out.println("TOKEN SERVICE :: ADDING :: " + user + " : " + token);
            //      removeUser(user);
            DBCollection coll = db().getCollection(M_USERS_COLLECTION);

            BasicDBObject query = new BasicDBObject();
            query.put("user", user);
            DBCursor cursor = coll.find(query);
            if (cursor.hasNext()) {
                DBObject doc = cursor.next();
                doc.put("token", token);
                doc.put("validity", System.currentTimeMillis());
                doc.put("isDemoUser", user.startsWith(ANONIM_USER));
                coll.save(doc, WriteConcern.SAFE);
            } else {
                BasicDBObject doc = new BasicDBObject();
                doc.put("_id", user);
                doc.put("user", user);
                doc.put("token", token);
                doc.put("validity", System.currentTimeMillis());
                doc.put("isDemoUser", user.startsWith(ANONIM_USER));
                coll.insert(doc);
            }
        }
    }

    /*
      private void removeUser(String user)
      {
        DBCollection coll = db().getCollection(M_USERS_COLLECTION);
        BasicDBObject query = new BasicDBObject();
        query.put("user", user);
        DBCursor cursor = coll.find(query);
        while (cursor.hasNext())
        {
          DBObject doc = cursor.next();
          coll.remove(doc);
        }
      }
    */

    public void updateValidity(String user, String token) {
        DBCollection coll = db().getCollection(M_USERS_COLLECTION);
        BasicDBObject query = new BasicDBObject();
        query.put("user", user);
        query.put("token", token);
        DBCursor cursor = coll.find(query);
        while (cursor.hasNext()) {
            DBObject doc = cursor.next();
            doc.put("validity", System.currentTimeMillis());
            coll.save(doc, WriteConcern.SAFE);
        }
    }

    public HashMap<String, UserBean> getActiveUsersFilterBy(String user, boolean withUsers, boolean withPublic,
            boolean isAdmin) {
        return getActiveUsersFilterBy(user, withUsers, withPublic, isAdmin, 0);
    }

    public HashMap<String, UserBean> getActiveUsersFilterBy(String user, boolean withUsers, boolean withPublic,
            boolean isAdmin, int limit) {
        HashMap<String, UserBean> users = new HashMap<String, UserBean>();
        DBCollection coll = db().getCollection(M_USERS_COLLECTION);
        BasicDBObject query = new BasicDBObject();
        query.put("validity", new BasicDBObject("$gt", System.currentTimeMillis() - getValidity())); //check token not updated since 10sec + status interval (15 sec)
        if (isAdmin) {
            if (withPublic && !withUsers) {
                query.put("isDemoUser", true);
            } else if (!withPublic && withUsers) {
                query.put("isDemoUser", false);
            }
        } else {
            query.put("isDemoUser", user.startsWith(ANONIM_USER));
        }
        if (limit < 0)
            limit = 0;
        DBCursor cursor = coll.find(query).limit(limit);
        while (cursor.hasNext()) {
            DBObject doc = cursor.next();
            String target = doc.get("user").toString();
            if (!user.equals(target)) {
                UserBean userBean = new UserBean();
                userBean.setName(target);
                if (doc.get("fullname") != null)
                    userBean.setFullname(doc.get("fullname").toString());
                if (doc.get("status") != null)
                    userBean.setStatus(doc.get("status").toString());
                users.put(target, userBean);
            }
        }

        return users;
    }

    public boolean isUserOnline(String user) {
        DBCollection coll = db().getCollection(M_USERS_COLLECTION);
        BasicDBObject query = new BasicDBObject();
        query.put("user", user);
        query.put("validity", new BasicDBObject("$gt", System.currentTimeMillis() - getValidity())); //check token not updated since 10sec + status interval (15 sec)
        DBCursor cursor = coll.find(query);
        return cursor.hasNext();
    }

    public boolean isDemoUser(String user) {
        return user.startsWith(ANONIM_USER);
    }

    private int getValidity() {
        if (validity_ == -1) {
            validity_ = 25000;
            try {
                validity_ = new Integer(PropertyManager.getProperty(PropertyManager.PROPERTY_TOKEN_VALIDITY));
            } catch (Exception e) {
                //do nothing if exception happens, keep 15000 value (=> statusInterval should set)
            }

        }
        return validity_;
    }

}