path: root/src/java/com/juick/http/www/Blogs.java

/*
 * Juick
 * Copyright (C) 2008-2011, Ugnich Anton
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.juick.http.www;

import com.juick.server.MessagesQueries;
import com.juick.server.TagQueries;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Locale;
import java.util.ResourceBundle;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 *
 * @author Ugnich Anton
 */
public class Blogs {

    private int relatedTagsStartMID = 0;

    public Blogs(Connection sql) {
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT MIN(message_id) FROM messages WHERE ts>DATE_ADD(NOW(),INTERVAL -6 MONTH)");
            rs = stmt.executeQuery();
            if (rs.first()) {
                relatedTagsStartMID = rs.getInt(1);
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }
    }

    protected void doGet(Connection sql, Connection sqlSearch, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String paramTagStrQuery = request.getParameter("tag");
        if (paramTagStrQuery != null) {
            String url = "/tag/" + URLEncoder.encode(paramTagStrQuery, "UTF-8");
            String paramBeforeStr = request.getParameter("before");
            if (paramBeforeStr != null) {
                url += "?before=" + paramBeforeStr;
            }
            Utils.sendPermanentRedirect(response, url);
        }

        com.juick.User visitor = Utils.getVisitorUser(sql, request);
        Locale locale = request.getLocale();
        ResourceBundle rb = ResourceBundle.getBundle("Blogs", locale);

        String title;
        ArrayList<Integer> mids;

        String paramShow = request.getParameter("show");

        String requesturi = request.getRequestURI();

        int paramTag = 0;
        String paramTagStr = null;
        if (requesturi.startsWith("/tag/")) {
            paramTagStr = URLDecoder.decode(request.getRequestURI().substring(5), "UTF-8");
            paramTag = TagQueries.getTagID(sql, paramTagStr, false);
            if (paramTag == 0) {
                response.sendError(404);
                return;
            }
        } else if (requesturi.startsWith("/places/")) {
            //TODO перенести сюда блок, который страницей ниже
        }

        int paramBefore = 0;
        String paramBeforeStr = request.getParameter("before");
        if (paramBeforeStr != null) {
            try {
                paramBefore = Integer.parseInt(paramBeforeStr);
            } catch (NumberFormatException e) {
            }
        }

        String paramSearch = request.getParameter("search");
        if (paramSearch != null && paramSearch.length() > 64) {
            paramSearch = null;
        }

        com.juick.Place place = null;
        int visitor_uid = visitor != null ? visitor.UID : 0;

        if (paramShow == null) {
            if (paramTag > 0) {
                title = "*" + Utils.encodeHTML(paramTagStr);
                mids = MessagesQueries.getTag(sql, paramTag, visitor_uid, paramBefore);
            } else if (paramSearch != null) {
                title = rb.getString("Search") + ": " + Utils.encodeHTML(paramSearch);
                mids = MessagesQueries.getSearch(sql, sqlSearch, Utils.encodeSphinx(paramSearch), paramBefore);
            } else if (requesturi.matches("^\\/places\\/\\d+$")) {
                int place_id = 0;
                try {
                    place_id = Integer.parseInt(requesturi.substring(8));
                } catch (NumberFormatException e) {
                }
                place = com.juick.server.PlacesQueries.getPlace(sql, place_id);
                if (place != null) {
                    title = place.name;
                    mids = MessagesQueries.getPlace(sql, place_id, visitor_uid, paramBefore);
                } else {
                    response.sendError(404);
                    return;
                }
            } else {
                title = rb.getString("Last messages");
                mids = MessagesQueries.getAll(sql, visitor_uid, paramBefore);
            }
        } else if (paramShow.equals("my")) {
            if (visitor != null) {
                title = rb.getString("My feed");
                mids = MessagesQueries.getMyFeed(sql, visitor.UID, paramBefore);
            } else {
                response.sendError(404);
                return;
            }
        } else if (paramShow.equals("private")) {
            if (visitor != null) {
                title = rb.getString("Private");
                mids = MessagesQueries.getPrivate(sql, visitor.UID, paramBefore);
            } else {
                response.sendError(404);
                return;
            }
        } else if (paramShow.equals("discuss")) {
            if (visitor != null) {
                title = rb.getString("Discussions");
                mids = MessagesQueries.getDiscussions(sql, visitor.UID, paramBefore);
            } else {
                response.sendError(404);
                return;
            }
        } else if (paramShow.equals("recommended")) {
            if (visitor != null) {
                title = rb.getString("Recommended");
                mids = MessagesQueries.getRecommended(sql, visitor.UID, paramBefore);
            } else {
                response.sendError(404);
                return;
            }
        } else if (paramShow.equals("top")) {
            title = rb.getString("Popular");
            mids = MessagesQueries.getPopular(sql, paramBefore);
        } else if (paramShow.equals("photos")) {
            title = rb.getString("With photos");
            mids = MessagesQueries.getPhotos(sql, visitor_uid, paramBefore);
        } else {
            response.sendError(404);
            return;
        }

        response.setContentType("text/html; charset=UTF-8");
        PrintWriter out = response.getWriter();
        try {
            PageTemplates.pageHead(out, title, null);
            PageTemplates.pageNavigation(out, locale, visitor, paramSearch);

            out.println("<div id=\"topwrapper\">");
            out.println("<div id=\"wrapper\">");
            out.println("<div id=\"content\">");

            if (place != null) {
                out.println("<div id=\"geomap\"><img src=\"http://maps.google.com/maps/api/staticmap?center=" + place.lat + "," + place.lon + "&amp;markers=color:orange|" + place.lat + "," + place.lon + "&amp;zoom=14&amp;size=600x300&amp;maptype=hybrid&amp;sensor=false\" width=\"600\" height=\"300\" alt=\"Map\"/></div>");
            }

            if (mids.size() > 0) {
                out.println("<ul>");
                PageTemplates.printMessages(out, sql, null, mids, visitor, locale);
                out.println("</ul>");
            }

            if (mids.size() == 20) {
                String nextpage = "?before=" + mids.get(mids.size() - 1);
                if (paramShow != null) {
                    nextpage += "&amp;show=" + paramShow;
                }
                if (paramSearch != null) {
                    nextpage += "&amp;search=" + URLEncoder.encode(paramSearch, "UTF-8");
                }
                out.println("<p class=\"page\"><a href=\"" + nextpage + "\">Older →</a></p>");
            }

            out.println("</div>");
            out.println("</div>"); // wrapper


            out.println("<div id=\"column\" class=\"abs\">");
            if (visitor != null) {
                out.println("  <ul>");
                out.println("    <li><a href=\"/?show=my\">" + rb.getString("My feed") + "</a></li>");
                out.println("    <li><a href=\"/?show=private\">" + rb.getString("Private") + "</a></li>");
                out.println("    <li><a href=\"/?show=discuss\">" + rb.getString("Discussions") + "</a></li>");
                out.println("    <li><a href=\"/?show=recommended\">" + rb.getString("Recommended") + "</a></li>");
                out.println("  </ul>");
                out.println("  <hr/>");
            }
            out.println("  <ul>");
            out.println("    <li><a href=\"/\">" + rb.getString("All messages") + "</a></li>");
            out.println("    <li><a href=\"/?show=top\">" + rb.getString("Popular") + "</a></li>");
            out.println("    <li><a href=\"/?show=photos\">" + rb.getString("With photos") + "</a></li>");
            out.println("  </ul>");
            out.println("  <hr/>");
            out.println("  <p>" + getTags(sql, 36) + "</p>");
            out.println("</div>");

            out.println("<div id=\"rcol\" class=\"abs\">");
            PageTemplates.pageYandexAd(out, visitor == null ? 2 : 3); // разный ID для залогиненых и нет
            if (paramTag > 0 && relatedTagsStartMID > 0) {
                String related = getRelatedTags(sql, paramTag, 10);
                if (!related.isEmpty()) {
                    out.println("<h3>Related tags:</h3>");
                    out.println("<ul id=\"reltags\">" + related + "</ul>");
                }
            }
            out.println("</div>");

            out.println("</div>"); // topwrapper

            PageTemplates.pageDialogTemplate(out, locale);
            PageTemplates.pageFooter(request, out, locale, visitor, true);
        } finally {
            out.close();
        }
    }

    private String getTags(Connection sql, int cnt) {
        String ret = "";
        com.juick.Tag tags[] = new com.juick.Tag[cnt];

        int maxUsageCnt = 0;
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT tags.name AS name,COUNT(DISTINCT messages.user_id) AS cnt FROM (messages INNER JOIN messages_tags ON (messages.ts>TIMESTAMPADD(DAY,-3,NOW()) AND messages.message_id=messages_tags.message_id)) INNER JOIN tags ON messages_tags.tag_id=tags.tag_id WHERE tags.tag_id NOT IN (SELECT tag_id FROM tags_ignore) GROUP BY tags.tag_id ORDER BY cnt DESC LIMIT ?");
            stmt.setInt(1, cnt);
            rs = stmt.executeQuery();
            rs.beforeFirst();
            cnt = 0;
            while (rs.next()) {
                tags[cnt] = new com.juick.Tag();
                tags[cnt].Name = rs.getString(1);
                tags[cnt].UsageCnt = rs.getInt(2);
                if (tags[cnt].UsageCnt > maxUsageCnt) {
                    maxUsageCnt = tags[cnt].UsageCnt;
                }
                cnt++;
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        Arrays.sort(tags, 0, cnt);

        for (int i = 0; i < cnt; i++) {
            String tag = Utils.encodeHTML(tags[i].Name);
            try {
                tag = "<a href=\"/tag/" + URLEncoder.encode(tags[i].Name, "UTF-8") + "\">" + tag + "</a>";
            } catch (UnsupportedEncodingException e) {
            }

            if (tags[i].UsageCnt > maxUsageCnt / 3 * 2) {
                ret += "<big>" + tag + "</big> ";
            } else if (tags[i].UsageCnt > maxUsageCnt / 3) {
                ret += "<small>" + tag + "</small> ";
            } else {
                ret += tag + " ";
            }
        }

        return ret;
    }

    private String getRelatedTags(Connection sql, int tag_id, int cnt) {
        String ret = "";

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT tags.name,COUNT(message_id) AS cnt FROM messages_tags INNER JOIN tags ON messages_tags.tag_id=tags.tag_id WHERE messages_tags.message_id>" + relatedTagsStartMID + " AND messages_tags.message_id IN (SELECT message_id FROM messages_tags WHERE tag_id=?) AND messages_tags.tag_id NOT IN (SELECT tag_id FROM tags_ignore) GROUP BY messages_tags.tag_id HAVING cnt>1 ORDER BY cnt DESC LIMIT 1,?");
            stmt.setInt(1, tag_id);
            stmt.setInt(2, cnt);
            rs = stmt.executeQuery();
            rs.beforeFirst();
            while (rs.next()) {
                try {
                    ret += "<li><a href=\"/tag/" + URLEncoder.encode(rs.getString(1), "UTF-8") + "\">" + Utils.encodeHTML(rs.getString(1)) + "</a></li>";
                } catch (UnsupportedEncodingException e) {
                }
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        return ret;
    }
}