/*
 * 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 <http://www.gnu.org/licenses/>.
 */
package com.juick.api;

import com.juick.xmpp.JID;
import com.juick.xmpp.Stream;
import com.juick.xmpp.StreamComponent;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.DriverManagerDataSource;

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 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;

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

    Connection sql;
    Connection sqlSearch;
    JdbcTemplate jdbc;
    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"));
            final String driverClassName = "com.mysql.jdbc.Driver";
            Class.forName(driverClassName);
            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=");
            DriverManagerDataSource dataSource = new DriverManagerDataSource();
            dataSource.setUrl(sql.getMetaData().getURL());
            dataSource.setDriverClassName(driverClassName);
            jdbc = new JdbcTemplate(dataSource);
            messages = new Messages(jdbc);
            users = new Users(jdbc);
            pm = new PM(jdbc);
            others = new Others(jdbc);

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

        } catch (IOException | ClassNotFoundException | SQLException e) {
            log(null, e);
        }
    }

    public void setupXmppComponent(final String password) {
        Thread thr = new Thread(() -> {
            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) {
                log("XMPP exception", e);
            }
        });
        thr.start();
    }

    @Override
    public void onStreamFail(String msg) {
        log("XMPP failed: " + 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");
        }

        int vuid = Utils.getHttpAuthUID(sql, request);
        if (vuid == 0) {
            vuid = Utils.getVisitorQueryStringUID(jdbc, 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 <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");
        }

        int vuid = Utils.getHttpAuthUID(sql, request);
        if (vuid == 0) {
            vuid = Utils.getVisitorQueryStringUID(jdbc, request);
        }
        if (vuid == 0) {
            response.sendError(401);
            return;
        }

        String uri = request.getRequestURI();
        switch (uri) {
            case "/post":
                break;
            case "/pm":
                pm.doPostPM(request, response, xmpp, vuid);
                break;
            default:
                response.sendError(405);
                break;
        }
    }

    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;
        }

        try (PrintWriter out = response.getWriter()) {
            if (callback != null) {
                out.print(callback + "(");
                out.print(json);
                out.print(")");
            } else {
                out.print(json);
            }
        }
    }
}