SetupInitialDataSessionBean.java

/** 
   Copyright 2013 Technische Universitat Wien (TUW), Distributed SystemsGroup E184.               
   
   This work was partially supported by the European Commission in terms of the CELAR FP7 project (FP7-ICT-2011-8 #317790).
 
   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.
*/

/**
 *  Author : Georgiana Copil - e.copil@dsg.tuwien.ac.at
 */
package at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.sessionbeans;

import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.dao.ResponsibilityDAO;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.dao.RoleDAO;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.dao.UserDAO;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.Role;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.User;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.interfaces.IResponsibility;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.interfaces.IRole;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.interfaces.IUser;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.entities.mapper.MapToCommunicationObjects;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.sessionbeans.interfaces.ISetupInitialDataBeanRemote;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.sessionbeans.interfaces.ISetupInitialDataSessionBean;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.utils.OMPLogger;
import at.ac.tuwien.dsg.rsybl.operationsmanagementplatform.utils.TestData;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import javax.activation.DataSource;
import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.ejb.ConcurrencyManagement;
import javax.ejb.ConcurrencyManagementType;
import javax.ejb.EJB;
import javax.ejb.Local;
import javax.ejb.LocalBean;
import javax.ejb.Remote;
import javax.ejb.Singleton;
import javax.ejb.Startup;
import javax.ejb.Stateless;
import javax.ejb.TransactionManagement;
import javax.ejb.TransactionManagementType;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.transaction.HeuristicMixedException;
import javax.transaction.HeuristicRollbackException;
import javax.transaction.NotSupportedException;
import javax.transaction.RollbackException;
import javax.transaction.SystemException;
import org.hibernate.exception.ConstraintViolationException;

/**
 *
 * @author Georgiana
 */
@Singleton
@Remote(ISetupInitialDataBeanRemote.class)
@Startup
@ConcurrencyManagement(ConcurrencyManagementType.CONTAINER)
@TransactionManagement(TransactionManagementType.BEAN)
public class SetupInitialDataSessionBean implements ISetupInitialDataSessionBean, ISetupInitialDataBeanRemote {

    @PersistenceContext
    protected EntityManager em;

    @Resource
    private javax.transaction.UserTransaction userTransaction;

    protected RoleDAO roleDAO;
    protected UserDAO userDAO;

    @PostConstruct
    public void init() {
        roleDAO = new RoleDAO();
        roleDAO.setEntityManager(em);
        userDAO = new UserDAO();
        userDAO.setEntityManager(em);
        populateWithITILRoles();
        initializeUsers();
    }

    @Override
    public void populateWithITILRoles() {
        try {
            userTransaction.begin();
            TestData testData = new TestData(em);
            testData.createInitialRoles();
            em.flush();
            userTransaction.commit();
        } catch (ConstraintViolationException | RollbackException | NotSupportedException | SystemException | HeuristicMixedException | HeuristicRollbackException e) {
            e.printStackTrace();
            try {
                userTransaction.rollback();
            } catch (Exception ex) {

            }
            OMPLogger.logger.info("failed populating with ITIL roles ~~~~ " + e.getMessage());
        }

    }

    @Override
    public void initializeUsers() {
        try {
            userTransaction.begin();
            User user = new User();
            user.setUsername("admin");
            user.setPassword("admin");
            user.setName("SysAdmin");
//            IRole role = (IRole) em.merge(roleDAO.findByRoleName("Systems Administrator"));

//            role.setRoleName("Systems Administrator");
//            user.addRole(role);
            em.persist(user);
            em.flush();
            User georgiana = new User();
            georgiana.setUsername("georgiana");
            georgiana.setPassword("georgiana");
            georgiana.setName("Georgiana Copil");
            georgiana.setEmail("cgeorgy1987@yahoo.com");
            IRole role = (IRole) em.merge(roleDAO.findByRoleName("Systems Administrator"));

            georgiana.addRole(role);
            role = (IRole) em.merge(roleDAO.findByRoleName("Configuration Librarian"));
            georgiana.addRole(role);
            georgiana.addRole(em.merge(roleDAO.findByRoleName("Incident Analyst")));
                        georgiana.addRole(em.merge(roleDAO.findByRoleName("Procurement Analysis")));


//            for (IRole irole : roleDAO.findAll()) {
//                georgiana.addRole(em.merge(irole));
//            }
            em.persist(georgiana);
            em.flush();
            User linh = new User();
            linh.setUsername("linh");
            linh.setPassword("linh");
            linh.setName("Hong-Linh Truong");
            linh.setEmail("X@x.com");
            linh.addRole(em.merge(roleDAO.findByRoleName("IT Financial Manager")));
            linh.addRole(em.merge(roleDAO.findByRoleName("Operations Manager")));
            linh.addRole(em.merge(roleDAO.findByRoleName("IT Quality Analyst")));
            em.persist(linh);
            em.flush();
               User schahram = new User();
            schahram.setUsername("schahram");
            schahram.setPassword("schahram");
            schahram.setName("Schahram Dustdar");
            schahram.setEmail("X@x.com");
            schahram.addRole(em.merge(roleDAO.findByRoleName("Service Manager")));
            em.persist(schahram);
            em.flush();
            userTransaction.commit();
        } catch (Exception e) {
            OMPLogger.logger.info("failed populating with users ~~~~ " + e.getMessage());
            try {
                userTransaction.rollback();
            } catch (Exception ex) {

            }
            OMPLogger.logger.info("failed populating with ITIL roles ~~~~ " + e.getMessage());
        }
    }

    public List<IRole> findAllRoles() {
        roleDAO = new RoleDAO();
        roleDAO.setEntityManager(em);
        return roleDAO.findAll();
    }

    public List<IResponsibility> findAllResponsibilities() {
        ResponsibilityDAO responsibilityDAO = new ResponsibilityDAO();
        responsibilityDAO.setEntityManager(em);
        return responsibilityDAO.findAll();
    }

    @Override
    public List<IRole> findAllRolesRemote() {
        roleDAO = new RoleDAO();
        roleDAO.setEntityManager(em);
        List<IRole> result = new ArrayList<IRole>();
        List<IRole> roles = roleDAO.findAll();
        for (IRole r : roles) {
            result.add(MapToCommunicationObjects.mapFromRole(r));
        }
        return result;
    }

    @Override
    public List<IResponsibility> findAllResponsibilitiesRemote() {
        ResponsibilityDAO responsibilityDAO = new ResponsibilityDAO();
        responsibilityDAO.setEntityManager(em);
        List<IResponsibility> result = new ArrayList<IResponsibility>();
        List<IResponsibility> roles = responsibilityDAO.findAll();
        for (IResponsibility r : roles) {
            result.add(MapToCommunicationObjects.mapFromResponsibility(r));
        }
        return result;
    }

    @Override
    public List<IResponsibility> findAllResponsibilitiesForRole(String name) {
        roleDAO = new RoleDAO();
        roleDAO.setEntityManager(em);
        List<IResponsibility> result = new ArrayList<IResponsibility>();
        Set<IResponsibility> roles = roleDAO.findByRoleName(name).getResponsabilities();
        for (IResponsibility r : roles) {
            result.add(MapToCommunicationObjects.mapFromResponsibility(r));
        }
        return result;
    }

    @Override
    public int findAuthorityForRole(String role) {
        roleDAO = new RoleDAO();
        roleDAO.setEntityManager(em);
        return roleDAO.findByRoleName(role).getAuthority();
    }
}