/* * Juick * Copyright (C) 2008-2013, 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 . */ package com.juick.api; 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.io.PrintWriter; import java.net.Socket; 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; /** * * @author Ugnich Anton */ @WebServlet(name = "Main", urlPatterns = {"/"}) @MultipartConfig public class Main extends HttpServlet implements Stream.StreamListener { Connection sql; Connection sqlSearch; Stream xmpp; Messages messages; Users users; PM pm; Others others; @Override public void init() throws ServletException { super.init(); try { Properties conf = new Properties(); conf.load(new FileInputStream("/etc/juick/api.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 = DriverManager.getConnection("jdbc:mysql://127.0.0.1:9306/juick?autoReconnect=true&characterEncoding=utf8&maxAllowedPacket=512000&relaxAutoCommit=true&user=root&password="); messages = new Messages(sql); users = new Users(sql); pm = new PM(sql); others = new Others(sql); setupXmppComponent(conf.getProperty("xmpp_password")); } catch (Exception e) { log(null, e); } } 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("", "api.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 GET 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"); } int vuid = Utils.getHttpAuthUID(sql, request); if (vuid == 0) { vuid = Utils.getVisitorQueryStringUID(sql, request); } String uri = request.getRequestURI(); if (uri.equals("/home")) { if (vuid > 0) { messages.doGetHome(request, response, vuid); } else { response.sendError(401); } } else if (uri.equals("/users")) { users.doGetUsers(request, response, vuid); } else if (uri.equals("/users/read")) { users.doGetUserRead(request, response, vuid); } else if (uri.equals("/users/readers")) { users.doGetUserReaders(request, response, vuid); } else if (uri.equals("/pm")) { if (vuid > 0) { pm.doGetPM(request, response, vuid); } else { response.sendError(401); } } else if (uri.equals("/groups_pms")) { if (vuid > 0) { others.doGetGroupsPMs(request, response, vuid); } else { response.sendError(401); } } else if (uri.equals("/messages/recommended")) { if (vuid > 0) { messages.doGetRecommended(request, response, vuid); } else { response.sendError(401); } } else if (uri.equals("/messages/set_popular") && vuid == 3694) { messages.doSetPopular(request, response, xmpp); } else if (uri.equals("/messages/set_privacy") && vuid > 0) { messages.doSetPrivacy(request, response, xmpp, vuid); } else { response.sendError(404); } } /** * Handles the HTTP POST 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"); } int vuid = Utils.getHttpAuthUID(sql, request); if (vuid == 0) { vuid = Utils.getVisitorQueryStringUID(sql, request); } if (vuid == 0) { response.sendError(401); return; } String uri = request.getRequestURI(); if (uri.equals("/post")) { } else if (uri.equals("/pm")) { pm.doPostPM(request, response, xmpp, vuid); } else { response.sendError(405); } } public static void replyJSON(HttpServletRequest request, HttpServletResponse response, String json) throws IOException { response.setContentType("application/json; charset=UTF-8"); response.setHeader("Access-Control-Allow-Origin", "*"); String callback = request.getParameter("callback"); if (callback != null && (callback.length() > 64 || !callback.matches("[a-zA-Z0-9\\-\\_]+"))) { callback = null; } PrintWriter out = response.getWriter(); try { if (callback != null) { out.print(callback + "("); out.print(json); out.print(")"); } else { out.print(json); } } finally { out.close(); } } }