GoogleSearchTag.java

/*************************************************************************
 * Copyright 2014 Infield Design
 *
 * 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 Gaetan Marmasse <gaetan@infielddesign.com>
 *
 *************************************************************************/

package com.infield.googlesearch.taglib;

import javax.servlet.jsp.tagext.SimpleTagSupport;

import java.io.IOException;
import java.util.LinkedList;

import javax.servlet.jsp.JspException;

import com.day.cq.wcm.api.Page;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ResourceResolver;

import com.adobe.granite.xss.XSSAPI;
import com.infield.googlesearch.GoogleSearchService;
import com.infield.googlesearch.model.ResultItem;
import com.infield.googlesearch.model.ResultList;
import org.apache.sling.api.resource.ValueMap;

public class GoogleSearchTag extends SimpleTagSupport {

    private String q;
    private String currentTab;
    private long numberOfResults;
    private long numberOfPages;
    private SlingHttpServletRequest request;
    private Page currentPage;

    @Override
    public void doTag() throws JspException, IOException {

        ResourceResolver resourceResolver = request.getResourceResolver();
        XSSAPI xssAPI = resourceResolver.adaptTo(XSSAPI.class);
        String q = xssAPI.encodeForHTML(this.q);
        String appName= "", apikey = "", contextNumber = "";

        if (currentPage != null) {
            Page rootPage = currentPage.getAbsoluteParent(1);

            ValueMap properties = rootPage.getProperties();
            String[] cloudServices = properties.get("cq:cloudserviceconfigs", new String[0]);
            for (String cloudService : cloudServices) {
                if (cloudService.contains("/etc/cloudservices/googlecustomsearch")) {
                    Resource searchConfigResource = resourceResolver.getResource(cloudService);
                    Page searchConfigPage = (searchConfigResource != null) ? searchConfigResource.adaptTo(Page.class) : null;

                    if (searchConfigPage!= null) {
                        ValueMap searchConfigProperties = searchConfigPage.getContentResource().adaptTo(ValueMap.class);

                        appName = searchConfigProperties.get("appname", "");
                        apikey = searchConfigProperties.get("apikey", "");
                        contextNumber = searchConfigProperties.get("cx", "");
                    }
                }
            }
        }

        if (!"".equals(appName) && !"".equals(apikey) && !"".equals(contextNumber) &&
                this.q.length() > 0 && this.currentTab.length() > 0 && this.numberOfResults > 0 && this.numberOfPages > 0) {

            ResultList resultList = new GoogleSearchService(apikey, appName, contextNumber)
                    .getResults(q, currentTab, numberOfResults, numberOfPages);

            LinkedList<ResultItem> resultItems = resultList.getResultItems();

            getJspContext().setAttribute("resultList", resultList);
            getJspContext().setAttribute("resultItems", resultItems);
        }
    }

    /**
     * @return the q, the query
     */
    public String getQ() {
        return q;
    }

    /**
     * @param q the query to set
     */
    public void setQ(String q) {
        this.q = q;
    }

    /**
     * @return the currentTab
     */
    public String getCurrentTab() {
        return currentTab;
    }

    /**
     * @param currentTab the currentTab to set
     */
    public void setCurrentTab(String currentTab) {
        this.currentTab = currentTab;
    }

    /**
     * @return the numberOfResults
     */
    public long getNumberOfResults() {
        return numberOfResults;
    }

    /**
     * @param numberOfResults the numberOfResults to set
     */
    public void setNumberOfResults(long numberOfResults) {
        this.numberOfResults = numberOfResults;
    }

    /**
     * @return the numberOfPages
     */
    public long getNumberOfPages() {
        return numberOfPages;
    }

    /**
     * @param numberOfPages the numberOfPages to set
     */
    public void setNumberOfPages(long numberOfPages) {
        this.numberOfPages = numberOfPages;
    }

    /**
     * @return the request
     */
    public SlingHttpServletRequest getRequest() {
        return request;
    }

    /**
     * @param request set request
     */
    public void setRequest(SlingHttpServletRequest request) {
        this.request = request;
    }

    /**
     * @return the currentPage
     */
    public Page getCurrentPage() {
        return currentPage;
    }

    /**
     * @param currentPage set currentPage object
     */
    public void setCurrentPage(Page currentPage) {
        this.currentPage = currentPage;
    }
}