path: root/src/main/java/com/juick/http/www/Main.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.UserQueries;
import com.juick.xmpp.JID;
import com.juick.xmpp.Stream;
import com.juick.xmpp.StreamComponent;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.Socket;
import java.net.URLEncoder;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;
import javax.servlet.ServletException;
import javax.servlet.annotation.MultipartConfig;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import ru.sape.Sape;

/**
 *
 * @author Ugnich Anton
 */
@WebServlet(name = "Main", urlPatterns = {"/"})
@MultipartConfig(fileSizeThreshold = 1024 * 1024, maxRequestSize = 1024 * 1024 * 10)
public class Main extends HttpServlet implements Stream.StreamListener {

    Connection sql;
    Connection sqlSearch;
    String sqlSearchConnStr = "jdbc:mysql://127.0.0.1:9306?autoReconnect=true&useUnicode=yes&characterEncoding=utf8&maxAllowedPacket=512000";
    Stream xmpp;
    Home home = new Home();
    Discover discover = new Discover();
    PM pm = new PM();
    Login login = new Login();
    Help help = new Help();
    User pagesUser = new User();
    UserThread pagesUserThread = new UserThread();
    NewMessage pagesNewMessage = new NewMessage();
    FacebookLogin loginFacebook = new FacebookLogin();
    VKontakteLogin loginVK = new VKontakteLogin();
    SignUp signup = new SignUp();
    Settings settings = new Settings();
    RSS rss = new RSS();

    @Override
    public void init() throws ServletException {

        super.init();
        try {
            Properties conf = new Properties();
            conf.load(new FileInputStream("/etc/juick/www.conf"));

            Class.forName("com.mysql.jdbc.Driver");
            sql = DriverManager.getConnection("jdbc:mysql://localhost/juick?autoReconnect=true&user=" + conf.getProperty("mysql_username", "") + "&password=" + conf.getProperty("mysql_password", ""));
            sqlSearch = null; // init this on search, timeout is too low

            setupXmppComponent(conf.getProperty("xmpp_password"));

            PageTemplates.sape = new Sape(conf.getProperty("sape_user"), "juick.com", 2000, 3600);
        } catch (Exception e) {
            log(null, e);
        }
    }
    public void closeSqlSearch() {
        if (sqlSearch != null) {
            try {
                sqlSearch.close();
                sqlSearch = null;
            } catch (SQLException e) {
                log("An error on closing sql search connection", e);
            }
        }
    };
    public Connection getSqlSearch() {
        if (sqlSearch == null) {
            try {
                sqlSearch = DriverManager.getConnection(sqlSearchConnStr, "", "");
            }
            catch (Exception e) {
                log("Couldn't open sqlSearch connection",e);
            }
        }
        return sqlSearch;
    }
    public void setupXmppComponent(final String password) {
        Thread thr = new Thread(new Runnable() {

            @Override
            public void run() {
                try {
                    Socket socket = new Socket("localhost", 5347);
                    xmpp = new StreamComponent(new JID("", "www.juick.com", ""), socket.getInputStream(), socket.getOutputStream(), password);
                    xmpp.addListener(Main.this);
                    xmpp.startParsing();
                } catch (IOException e) {
                    System.err.println(e);
                }
            }
        });
        thr.start();
    }

    @Override
    public void onStreamFail(String msg) {
        System.err.println("XMPP STREAM FAIL: " + msg);
    }

    @Override
    public void onStreamReady() {
        System.err.println("XMPP STREAM READY");
    }

    @Override
    public void destroy() {
        super.destroy();
        if (sql != null) {
            try {
                sql.close();
                sql = null;
            } catch (SQLException e) {
                log(null, e);
            }
        }
        if (sqlSearch != null) {
            try {
                sqlSearch.close();
                sqlSearch = null;
            } catch (SQLException e) {
                log(null, e);
            }
        }
    }

    /**
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        if (request.getCharacterEncoding() == null) {
            request.setCharacterEncoding("UTF-8");
        }
        String uri = request.getRequestURI();

        if (uri.equals("/")) {
            String tag = request.getParameter("tag");
            if (tag != null) {
                Utils.sendPermanentRedirect(response, "/tag/" + URLEncoder.encode(tag, "UTF-8"));
            } else {
                com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
                home.doGet(sql, getSqlSearch(), request, response, visitor);
                closeSqlSearch();
            }
        } else if (uri.equals("/post")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor != null) {
                pagesNewMessage.doGetNewMessage(sql, request, response, visitor);
            } else {
                Utils.sendTemporaryRedirect(response, "/login");
            }
        } else if (uri.equals("/login")) {
            if (request.getQueryString() == null) {
                login.doGetLoginForm(sql, request, response);
            } else {
                login.doGetLogin(sql, request, response);
            }
        } else if (uri.startsWith("/pm/")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor == null) {
                Utils.sendTemporaryRedirect(response, "/login");
            } else {
                if (uri.equals("/pm/inbox")) {
                    pm.doGetInbox(sql, request, response, visitor);
                } else if (uri.equals("/pm/sent")) {
                    pm.doGetSent(sql, request, response, visitor);
                } else {
                    Errors.doGet404(sql, request, response);
                }
            }
        } else if (uri.startsWith("/rss/")) {
            String uname = uri.substring(5);
            int uid = UserQueries.getUIDbyName(sql, uname);
            if (uid > 0) {
                rss.doGet(sql, request, response, uid, uname);
            } else {
                response.sendError(404);
            }
        } else if (uri.equals("/logout")) {
            login.doGetLogout(sql, request, response);
        } else if (uri.equals("/settings")) {
            settings.doGet(sql, request, response);
        } else if (uri.equals("/_fblogin")) {
            loginFacebook.doGet(sql, request, response);
        } else if (uri.equals("/_vklogin")) {
            loginVK.doGet(sql, request, response);
        } else if (uri.equals("/signup")) {
            signup.doGet(sql, request, response);
        } else if (uri.equals("/help") || uri.equals("/help/")) {
            help.doRedirectToHelpIndex(sql, request, response);
        } else if (uri.startsWith("/help/")) {
            help.doGetHelp(sql, request, response);
        } else if (uri.startsWith("/tag/")) {
            discover.doGet(sql, getSqlSearch(), request, response);
            closeSqlSearch();
        } else if (uri.matches("^/\\d+$")) {
            String strID = request.getRequestURI().substring(1);
            int mid = 0;
            try {
                mid = Integer.parseInt(strID);
            } catch (NumberFormatException e) {
            }
            if (mid > 0) {
                com.juick.User author = com.juick.server.MessagesQueries.getMessageAuthor(sql, mid);
                if (author != null) {
                    Utils.sendPermanentRedirect(response, "/" + author.UName + "/" + mid);
                    return;
                }
            }
            Errors.doGet404(sql, request, response);
        } else if (uri.matches("^/[^/]+$")) {
            com.juick.User user = com.juick.server.UserQueries.getUserByName(sql, request.getRequestURI().substring(1));
            if (user != null) {
                Utils.sendPermanentRedirect(response, "/" + user.UName + "/");
            } else {
                Errors.doGet404(sql, request, response);
            }
        } else if (uri.matches("^/.+/.*")) {
            String uriparts[] = uri.split("/");
            com.juick.User user = com.juick.server.UserQueries.getUserByName(sql, uriparts[1]);
            if (user != null && user.UName.equals(uriparts[1]) && user.Banned == false) {
                if (uriparts.length == 2) { // http://juick.com/username/
                    pagesUser.doGetBlog(sql, getSqlSearch(), request, response, user);
                    closeSqlSearch();
                } else if (uriparts[2].equals("tags")) {
                    pagesUser.doGetTags(sql, request, response, user);
                } else if (uriparts[2].equals("friends")) {
                    pagesUser.doGetFriends(sql, request, response, user);
                } else if (uriparts[2].equals("readers")) {
                    pagesUser.doGetReaders(sql, request, response, user);
                } else {
                    int mid = 0;
                    try {
                        mid = Integer.parseInt(uriparts[2]);
                    } catch (NumberFormatException e) {
                    }
                    if (mid > 0) {
                        com.juick.User author = com.juick.server.MessagesQueries.getMessageAuthor(sql, mid);
                        if (author != null) {
                            if (!author.UName.equals(user.UName)) {
                                Utils.sendPermanentRedirect(response, "/" + author.UName + "/" + mid);
                            } else {
                                pagesUserThread.doGetThread(sql, request, response, mid);
                            }
                        } else {
                            Errors.doGet404(sql, request, response);
                        }
                    } else {
                        Errors.doGet404(sql, request, response);
                    }
                }
            } else if (user != null && user.Banned == false) {
                Utils.sendPermanentRedirect(response, "/" + user.UName + "/" + (uriparts.length > 2 ? uriparts[2] : ""));
            } else {
                Errors.doGet404(sql, request, response);
            }
        } else {
            Errors.doGet404(sql, request, response);
        }
    }

    /** 
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        if (request.getCharacterEncoding() == null) {
            request.setCharacterEncoding("UTF-8");
        }

        String uri = request.getRequestURI();
        if (uri.equals("/post")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor != null) {
                pagesNewMessage.doPostMessage(sql, request, response, xmpp, visitor);
            } else {
                response.sendError(403);
            }
        } else if (uri.equals("/comment")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor != null) {
                pagesNewMessage.doPostComment(sql, request, response, xmpp, visitor);
            } else {
                response.sendError(403);
            }
        } else if (uri.equals("/like")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor != null) {
                pagesNewMessage.doPostRecomm(sql, request, response, xmpp, visitor);
            } else {
                response.sendError(403);
            }
        } else if (uri.equals("/pm/send")) {
            com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
            if (visitor != null) {
                pm.doPostPM(sql, request, response, xmpp, visitor);
            } else {
                response.sendError(403);
            }
        } else if (uri.equals("/login")) {
            login.doPostLogin(sql, request, response);
        } else if (uri.equals("/signup")) {
            signup.doPost(sql, request, response);
        } else if (uri.equals("/settings")) {
            settings.doPost(sql, request, response);
        } else {
            response.sendError(405);
        }
    }
}