SearchResultsBeanPopulator.java

/*
 * Copyright (c) 2006, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * 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 org.wso2.carbon.registry.search.services.utils;

import org.wso2.carbon.registry.core.Collection;
import org.wso2.carbon.registry.core.RegistryConstants;
import org.wso2.carbon.registry.core.Resource;
import org.wso2.carbon.registry.core.TaggedResourcePath;
import org.wso2.carbon.registry.core.exceptions.RegistryException;
import org.wso2.carbon.registry.core.secure.AuthorizationFailedException;
import org.wso2.carbon.registry.core.session.UserRegistry;
import org.wso2.carbon.registry.search.beans.SearchResultsBean;
import org.wso2.carbon.registry.common.ResourceData;
import org.wso2.carbon.registry.common.TagCount;
import org.wso2.carbon.registry.common.utils.CommonUtil;

import java.lang.String;
import java.util.Arrays;
import java.util.Comparator;
import java.util.Map;
import java.util.Calendar;

public class SearchResultsBeanPopulator {
    public static SearchResultsBean populate(UserRegistry userRegistry, String searchType, String criteria) {    	
        SearchResultsBean searchResultsBean = new SearchResultsBean();

        try {
            ResourceData[] resourceDataList;
            if (searchType.equalsIgnoreCase("Tag")) {
                resourceDataList = searchByTags(criteria, userRegistry);
            } else {
                resourceDataList = searchByContent(criteria, userRegistry);
            }
            Arrays.sort(resourceDataList, new Comparator<ResourceData>() {
                public int compare(ResourceData o1, ResourceData o2) {
                    return o1.getResourcePath().toLowerCase().compareTo(
                            o2.getResourcePath().toLowerCase());
                }
            });
            searchResultsBean.setResourceDataList(resourceDataList);

        } catch (RegistryException e) {

            String msg = "Failed to generate search results. " + e.getMessage();
            searchResultsBean.setErrorMessage(msg);
            searchResultsBean.setResourceDataList(new ResourceData[0]);
        }

        return searchResultsBean;
    }

    //metadata search - search by tag
    private static ResourceData [] searchByTags(String tag, UserRegistry registry)
            throws RegistryException {

        ResourceData[] resourceDataList = new ResourceData[0];

        if (tag != null && tag.length() != 0) {
            TaggedResourcePath[] taggedPaths = registry.getResourcePathsWithTag(tag);
            resourceDataList = new ResourceData [taggedPaths.length];
            for (int i=0; i<taggedPaths.length; i++) {
                String resultPath = taggedPaths[i].getResourcePath();

                ResourceData resourceData = new ResourceData();
                resourceData.setResourcePath(resultPath);

                if (resultPath != null) {
                    if (resultPath.equals(RegistryConstants.ROOT_PATH)) {
                        resourceData.setName(RegistryConstants.ROOT_PATH);
                    } else {
                        String[] parts = resultPath.split(RegistryConstants.PATH_SEPARATOR);
                        resourceData.setName(parts[parts.length - 1]);
                    }
                }

                try {
                    Resource child = registry.get(resultPath);
                    String actualPath = child.getProperty("registry.actualpath");
                    if (actualPath != null && registry.resourceExists(actualPath)) {
                        child = registry.get(actualPath);
                    }
                    resourceData.setResourceType(child instanceof Collection ?
                            "collection" : "resource");
                    resourceData.setAuthorUserName(child.getAuthorUserName());
                    resourceData.setDescription(child.getDescription());
                    resourceData.setAverageRating(registry.getAverageRating(child.getPath()));
                    Calendar createdDateTime = Calendar.getInstance();
                    createdDateTime.setTime(child.getCreatedTime());
                    resourceData.setCreatedOn(createdDateTime);

                    Map counts = taggedPaths[i].getTagCounts();
                    Object [] keySet = counts.keySet().toArray();
                    TagCount [] tagCounts = new TagCount [counts.size()];
                    for(int j=0; j<counts.size(); j++) {
                        TagCount tagCount = new TagCount();
                        tagCount.setKey((String)keySet[j]);
                        tagCount.setValue(new Long((String)counts.get(keySet[j])));
                        tagCounts[j] = tagCount;
                    }
                    resourceData.setTagCounts(tagCounts);
                    
                    CommonUtil.populateAverageStars(resourceData);

                    child.discard();

                    resourceDataList[i] = resourceData;

                } catch (AuthorizationFailedException e) {
                    // do not show unauthorized resources in search results
                }
            }
        }
        return resourceDataList;
    }

    //content search

         private static ResourceData [] searchByContent (String content, UserRegistry registry)
            throws RegistryException {

        ResourceData [] resourceDataList = new ResourceData[0];

        if(content != null && content.length() != 0){
            String [] paths;
            try {
                paths = (String [])registry.searchContent(content).getContent();
            } catch (Exception e) {
                return new ResourceData[0];
            }
            resourceDataList = new ResourceData [paths.length];
            for(int i=0; i<paths.length; i++){
                ResourceData resourceData = new ResourceData();
                resourceData.setResourcePath(paths[i]);

                if (paths[i] != null) {
                    if (paths[i].equals(RegistryConstants.ROOT_PATH)) {
                        resourceData.setName(RegistryConstants.ROOT_PATH);
                    } else {
                        String[] parts = paths[i].split(RegistryConstants.PATH_SEPARATOR);
                        resourceData.setName(parts[parts.length - 1]);
                    }
                }

                try {
                    Resource child = registry.get(paths[i]);
                    String actualPath = child.getProperty("registry.actualpath");
                    if (actualPath != null && registry.resourceExists(actualPath)) {
                        child = registry.get(actualPath);
                    }
                    resourceData.setResourceType(child instanceof Collection ?
                            "collection" : "resource");
                    resourceData.setAuthorUserName(child.getAuthorUserName());
                    resourceData.setDescription(child.getDescription());
                    resourceData.setAverageRating(registry.getAverageRating(child.getPath()));
                    Calendar createdDateTime = Calendar.getInstance();
                    createdDateTime.setTime(child.getCreatedTime());
                    resourceData.setCreatedOn(createdDateTime);
                    CommonUtil.populateAverageStars(resourceData);

                    child.discard();

                    resourceDataList[i] = resourceData;

                } catch (AuthorizationFailedException e) {
                    // do not show unauthorized resources in search results
                }
            }
        }

        return resourceDataList;
    }
}