org.gatein.pc.samples.google.GoogleClippingPortlet Maven / Gradle / Ivy

/*
 * JBoss, a division of Red Hat
 * Copyright 2010, Red Hat Middleware, LLC, and individual
 * contributors as indicated by the @authors tag. See the
 * copyright.txt in the distribution for a full listing of
 * individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.gatein.pc.samples.google;

import javax.portlet.ActionRequest;
import javax.portlet.ActionResponse;
import javax.portlet.GenericPortlet;
import javax.portlet.PortletException;
import javax.portlet.PortletMode;
import javax.portlet.PortletPreferences;
import javax.portlet.RenderRequest;
import javax.portlet.RenderResponse;
import javax.xml.namespace.QName;
import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.net.URL;
import java.net.URLConnection;

/**
 * A simple portlet using Google's search services to extract interesting information (weather, map, ...) from first
 * result.
 *
 * @author Chris Laprun
 * @version $Revision: 9912 $
 */
public class GoogleClippingPortlet extends GenericPortlet
{
   protected static final String A = "beginningString portlet preference) is "
            + "contained in the result of issued query ('" + query
            + "', based on the value of the query portlet preference). You can adjust these values in "
            + "portlet.xml to tweak this portlet output...";
      }

      renderResponse.setContentType("text/html");
      PrintWriter printWriter = renderResponse.getWriter();
      printWriter.print(html);
   }

   private String getBegString(RenderRequest renderRequest)
   {
      return renderRequest.getPreferences().getValue(BEGINNING_STRING, DEFAULT_BEG);
   }

   private String process(String html, int begIndex) throws IOException
   {
      // extract table containing specific first result
      int tableIndex = html.indexOf(BEG_TABLE, begIndex);
      int endIndex = html.indexOf(END_TABLE, tableIndex);
      html = html.substring(tableIndex, endIndex + END_TABLE.length());
      html = postProcessHTML(html);

      return html;
   }

   protected String getZipCode(RenderRequest renderRequest)
   {
      String zip = renderRequest.getParameter(ZIP);
      if (zip == null)
      {
         zip = renderRequest.getPreferences().getValue(ZIP, DEFAULT_ZIP);
      }
      return zip;
   }

   @Override
   protected void doEdit(RenderRequest renderRequest, RenderResponse renderResponse) throws PortletException, IOException
   {
      renderResponse.setContentType("text/html");
      renderResponse.getWriter().print(
         "\n" +
            "   
\n" +
            "   \n" +
            "      Change location (zip code): \n" +
            "      \n" +
            "      \n" +
            "   \n" +
            "");
   }

   @Override
   public void processAction(ActionRequest actionRequest, ActionResponse actionResponse) throws PortletException, IOException
   {
      String zip = actionRequest.getParameter(ZIP);

      if (null != zip)
      {
         PortletPreferences prefs = actionRequest.getPreferences();
         prefs.setValue(ZIP, zip);
         prefs.store();
      }

      // set zip as render parameter
      actionResponse.setRenderParameter(ZIP, zip);

      // request view
      actionResponse.setPortletMode(PortletMode.VIEW);

      // send out zip event
      actionResponse.setEvent(QNAME, zip);
   }

   protected String postProcessHTML(String html)
   {
      // links should open in new windows
      html = html.replaceAll(A, A_TARGET_BLANK);

      // src attributes should be absolute
      html = html.replaceAll("src=/", "src=http://google.com/");
      html = html.replaceAll("src=\"/", "src=\"http://google.com/");

      // forms should open in new windows and have an absolute action URL
      html = html.replaceAll("action=\"/", "target='_blank' action=\"http://google.com/");

      return html;
   }

   protected String getQueryString(RenderRequest renderRequest, String zip)
   {
      return renderRequest.getPreferences().getValue(QUERY, DEFAULT_QUERY) + zip;
   }

   private byte[] getBytes(InputStream in, int bufferSize) throws IOException, IllegalArgumentException
   {
      ByteArrayOutputStream out = new ByteArrayOutputStream();

      byte[] buffer = new byte[bufferSize];
      while (true)
      {
         int i = in.read(buffer);
         if (i == 0)
         {
            continue;
         }
         if (i == -1)
         {
            break;
         }
         out.write(buffer, 0, i);
      }
      return out.toByteArray();
   }

}