001    /**
002     * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.lar.messaging;
016    
017    import com.liferay.portal.kernel.exception.PortalException;
018    import com.liferay.portal.kernel.exception.SystemException;
019    import com.liferay.portal.kernel.messaging.BaseMessageStatusMessageListener;
020    import com.liferay.portal.kernel.util.ArrayUtil;
021    import com.liferay.portal.model.CompanyConstants;
022    import com.liferay.portal.model.User;
023    import com.liferay.portal.security.auth.CompanyThreadLocal;
024    import com.liferay.portal.security.auth.PrincipalThreadLocal;
025    import com.liferay.portal.security.permission.PermissionChecker;
026    import com.liferay.portal.security.permission.PermissionCheckerFactoryUtil;
027    import com.liferay.portal.security.permission.PermissionThreadLocal;
028    import com.liferay.portal.service.ServiceContext;
029    import com.liferay.portal.service.ServiceContextThreadLocal;
030    import com.liferay.portal.service.UserLocalServiceUtil;
031    import com.liferay.portal.util.PortalUtil;
032    
033    import java.io.Serializable;
034    
035    import java.util.HashMap;
036    import java.util.Map;
037    
038    /**
039     * @author Levente Hud??k
040     */
041    public abstract class BasePublisherMessageListener
042            extends BaseMessageStatusMessageListener {
043    
044            protected void initThreadLocals(
045                            long userId, Map<String, String[]> parameterMap)
046                    throws PortalException {
047    
048                    User user = UserLocalServiceUtil.getUserById(userId);
049    
050                    CompanyThreadLocal.setCompanyId(user.getCompanyId());
051    
052                    PrincipalThreadLocal.setName(userId);
053    
054                    PermissionChecker permissionChecker = null;
055    
056                    try {
057                            permissionChecker = PermissionCheckerFactoryUtil.create(user);
058                    }
059                    catch (Exception e) {
060                            throw new SystemException(e);
061                    }
062    
063                    PermissionThreadLocal.setPermissionChecker(permissionChecker);
064    
065                    ServiceContext serviceContext = new ServiceContext();
066    
067                    serviceContext.setCompanyId(user.getCompanyId());
068                    serviceContext.setPathMain(PortalUtil.getPathMain());
069                    serviceContext.setSignedIn(!user.isDefaultUser());
070                    serviceContext.setUserId(user.getUserId());
071    
072                    Map<String, Serializable> attributes =
073                            new HashMap<String, Serializable>();
074    
075                    for (Map.Entry<String, String[]> entry : parameterMap.entrySet()) {
076                            String param = entry.getKey();
077                            String[] values = entry.getValue();
078    
079                            if (ArrayUtil.isNotEmpty(values)) {
080                                    if (values.length == 1) {
081                                            attributes.put(param, values[0]);
082                                    }
083                                    else {
084                                            attributes.put(param, values);
085                                    }
086                            }
087                    }
088    
089                    serviceContext.setAttributes(attributes);
090    
091                    ServiceContextThreadLocal.pushServiceContext(serviceContext);
092            }
093    
094            protected void resetThreadLocals() {
095                    CompanyThreadLocal.setCompanyId(CompanyConstants.SYSTEM);
096                    PermissionThreadLocal.setPermissionChecker(null);
097                    PrincipalThreadLocal.setName(null);
098                    ServiceContextThreadLocal.popServiceContext();
099            }
100    
101    }