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.portlet.usersadmin.util;
016    
017    import com.liferay.portal.kernel.dao.orm.ActionableDynamicQuery;
018    import com.liferay.portal.kernel.exception.PortalException;
019    import com.liferay.portal.kernel.log.Log;
020    import com.liferay.portal.kernel.log.LogFactoryUtil;
021    import com.liferay.portal.kernel.search.BaseIndexer;
022    import com.liferay.portal.kernel.search.BooleanQuery;
023    import com.liferay.portal.kernel.search.Document;
024    import com.liferay.portal.kernel.search.Field;
025    import com.liferay.portal.kernel.search.SearchContext;
026    import com.liferay.portal.kernel.search.SearchEngineUtil;
027    import com.liferay.portal.kernel.search.Summary;
028    import com.liferay.portal.kernel.search.filter.BooleanFilter;
029    import com.liferay.portal.kernel.spring.osgi.OSGiBeanProperties;
030    import com.liferay.portal.kernel.util.GetterUtil;
031    import com.liferay.portal.kernel.util.Validator;
032    import com.liferay.portal.kernel.workflow.WorkflowConstants;
033    import com.liferay.portal.model.Contact;
034    import com.liferay.portal.model.User;
035    import com.liferay.portal.service.ContactLocalServiceUtil;
036    import com.liferay.portal.service.UserLocalServiceUtil;
037    
038    import java.util.LinkedHashMap;
039    import java.util.Locale;
040    
041    import javax.portlet.PortletRequest;
042    import javax.portlet.PortletResponse;
043    
044    /**
045     * @author Raymond Aug??
046     * @author Zsigmond Rab
047     * @author Hugo Huijser
048     */
049    @OSGiBeanProperties
050    public class ContactIndexer extends BaseIndexer {
051    
052            public static final String CLASS_NAME = Contact.class.getName();
053    
054            public ContactIndexer() {
055                    setStagingAware(false);
056            }
057    
058            @Override
059            public String getClassName() {
060                    return CLASS_NAME;
061            }
062    
063            @Override
064            public void postProcessSearchQuery(
065                            BooleanQuery searchQuery, BooleanFilter fullQueryBooleanFilter,
066                            SearchContext searchContext)
067                    throws Exception {
068    
069                    addSearchTerm(searchQuery, searchContext, "city", false);
070                    addSearchTerm(searchQuery, searchContext, "country", false);
071                    addSearchTerm(searchQuery, searchContext, "emailAddress", false);
072                    addSearchTerm(searchQuery, searchContext, "firstName", false);
073                    addSearchTerm(searchQuery, searchContext, "fullName", false);
074                    addSearchTerm(searchQuery, searchContext, "lastName", false);
075                    addSearchTerm(searchQuery, searchContext, "middleName", false);
076                    addSearchTerm(searchQuery, searchContext, "region", false);
077                    addSearchTerm(searchQuery, searchContext, "screenName", false);
078                    addSearchTerm(searchQuery, searchContext, "street", false);
079                    addSearchTerm(searchQuery, searchContext, "zip", false);
080    
081                    LinkedHashMap<String, Object> params =
082                            (LinkedHashMap<String, Object>)searchContext.getAttribute("params");
083    
084                    if (params != null) {
085                            String expandoAttributes = (String)params.get("expandoAttributes");
086    
087                            if (Validator.isNotNull(expandoAttributes)) {
088                                    addSearchExpando(searchQuery, searchContext, expandoAttributes);
089                            }
090                    }
091            }
092    
093            @Override
094            protected void doDelete(Object obj) throws Exception {
095                    Contact contact = (Contact)obj;
096    
097                    deleteDocument(contact.getCompanyId(), contact.getContactId());
098            }
099    
100            @Override
101            protected Document doGetDocument(Object obj) throws Exception {
102                    Contact contact = (Contact)obj;
103    
104                    if (contact.isUser()) {
105                            User user = UserLocalServiceUtil.fetchUserByContactId(
106                                    contact.getContactId());
107    
108                            if ((user == null) || user.isDefaultUser() ||
109                                    (user.getStatus() != WorkflowConstants.STATUS_APPROVED)) {
110    
111                                    return null;
112                            }
113                    }
114    
115                    Document document = getBaseModelDocument(CLASS_NAME, contact);
116    
117                    document.addKeyword(Field.COMPANY_ID, contact.getCompanyId());
118                    document.addDate(Field.MODIFIED_DATE, contact.getModifiedDate());
119                    document.addKeyword(Field.USER_ID, contact.getUserId());
120                    document.addKeyword(Field.USER_NAME, contact.getFullName());
121    
122                    document.addText("emailAddress", contact.getEmailAddress());
123                    document.addText("firstName", contact.getFirstName());
124                    document.addText("fullName", contact.getFullName());
125                    document.addText("jobTitle", contact.getJobTitle());
126                    document.addText("lastName", contact.getLastName());
127                    document.addText("middleName", contact.getMiddleName());
128    
129                    return document;
130            }
131    
132            @Override
133            protected String doGetSortField(String orderByCol) {
134                    if (orderByCol.equals("email-address")) {
135                            return "emailAddress";
136                    }
137                    else if (orderByCol.equals("first-name")) {
138                            return "firstName";
139                    }
140                    else if (orderByCol.equals("job-title")) {
141                            return "jobTitle";
142                    }
143                    else if (orderByCol.equals("last-name")) {
144                            return "lastName";
145                    }
146                    else {
147                            return orderByCol;
148                    }
149            }
150    
151            @Override
152            protected Summary doGetSummary(
153                    Document document, Locale locale, String snippet,
154                    PortletRequest portletRequest, PortletResponse portletResponse) {
155    
156                    return null;
157            }
158    
159            @Override
160            protected void doReindex(Object obj) throws Exception {
161                    Contact contact = (Contact)obj;
162    
163                    Document document = getDocument(contact);
164    
165                    if (document != null) {
166                            SearchEngineUtil.updateDocument(
167                                    getSearchEngineId(), contact.getCompanyId(), document,
168                                    isCommitImmediately());
169                    }
170            }
171    
172            @Override
173            protected void doReindex(String className, long classPK) throws Exception {
174                    Contact contact = ContactLocalServiceUtil.getContact(classPK);
175    
176                    doReindex(contact);
177            }
178    
179            @Override
180            protected void doReindex(String[] ids) throws Exception {
181                    long companyId = GetterUtil.getLong(ids[0]);
182    
183                    reindexContacts(companyId);
184            }
185    
186            protected void reindexContacts(long companyId) throws PortalException {
187                    final ActionableDynamicQuery actionableDynamicQuery =
188                            ContactLocalServiceUtil.getActionableDynamicQuery();
189    
190                    actionableDynamicQuery.setCompanyId(companyId);
191                    actionableDynamicQuery.setPerformActionMethod(
192                            new ActionableDynamicQuery.PerformActionMethod() {
193    
194                                    @Override
195                                    public void performAction(Object object) {
196                                            Contact contact = (Contact)object;
197    
198                                            try {
199                                                    Document document = getDocument(contact);
200    
201                                                    if (document != null) {
202                                                            actionableDynamicQuery.addDocument(document);
203                                                    }
204                                            }
205                                            catch (PortalException pe) {
206                                                    if (_log.isWarnEnabled()) {
207                                                            _log.warn(
208                                                                    "Unable to index contact " +
209                                                                            contact.getContactId(),
210                                                                    pe);
211                                                    }
212                                            }
213                                    }
214    
215                            });
216                    actionableDynamicQuery.setSearchEngineId(getSearchEngineId());
217    
218                    actionableDynamicQuery.performActions();
219            }
220    
221            private static final Log _log = LogFactoryUtil.getLog(ContactIndexer.class);
222    
223    }