/*
 * Copyright (C) 2008-2017, Juick
 *
 * 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.service;

import com.juick.Chat;
import com.juick.User;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.SqlParameterSource;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;

/**
 * Created by aalexeev on 11/13/16.
 */
@Repository
public class PMQueriesServiceImpl extends BaseJdbcService implements PMQueriesService {

    @Transactional
    @Override
    public boolean createPM(final int uidFrom, final int uid_to, final String body) {
        boolean ret = getJdbcTemplate().update(
                "INSERT INTO pm(user_id, user_id_to, txt) VALUES (?, ?, ?)",
                uidFrom, uid_to, body) > 0;

        if (ret) {
            try {
                getJdbcTemplate().update(
                        "INSERT INTO pm_streams(user_id, user_id_to, lastmessage, unread) VALUES (?, ?, NOW(), 1)",
                        uidFrom, uid_to);
            } catch (DuplicateKeyException e) {
                // ignore
            }
        }
        return ret;
    }

    @Transactional
    @Override
    public boolean addPMinRoster(final int uid, final String jid) {
        return getJdbcTemplate().update(
                "INSERT INTO pm_inroster(user_id, jid) VALUES (?, ?) ON DUPLICATE KEY UPDATE user_id=user_id", uid, jid) > 0;
    }

    @Transactional
    @Override
    public boolean removePMinRoster(final int uid, final String jid) {
        return getJdbcTemplate().update(
                "DELETE FROM pm_inroster WHERE user_id = ? AND jid = ?", uid, jid) > 0;
    }

    @Transactional
    @Override
    public boolean havePMinRoster(final int uid, final String jid) {
        List<Integer> res = getJdbcTemplate().queryForList(
                "SELECT 1 FROM pm_inroster WHERE user_id = ? AND jid = ?",
                Integer.class,
                uid, jid);
        return res.size() > 0;
    }

    @Transactional(readOnly = true)
    @Override
    public String getLastView(final int uidFrom, final int uidTo) {
        List<String> list = getJdbcTemplate().queryForList(
                "SELECT lastview FROM pm_streams WHERE user_id = ? AND user_id_to = ?",
                String.class,
                uidFrom, uidTo);

        return list.isEmpty() ?
                null : list.get(0);
    }

    @Transactional(readOnly = true)
    @Override
    public List<Chat> getLastChats(final User user) {
        return getJdbcTemplate().query(
                "SELECT pm_streams.user_id, users.nick, l.last, pm.txt, pm_streams.unread FROM pm_streams "
                        + "INNER JOIN users ON users.id = pm_streams.user_id "
                        + "INNER JOIN pm ON pm.user_id = pm_streams.user_id "
                        + "INNER JOIN (SELECT user_id, MAX(ts) AS last FROM pm "
                        + "WHERE user_id_to=? GROUP BY user_id) l ON l.last = pm.ts "
                        + "WHERE pm_streams.user_id_to=? "
                        + "ORDER BY pm_streams.unread DESC, l.last DESC",
                (rs, rowNum) -> {
                    com.juick.Chat u = new com.juick.Chat();
                    u.setUid(rs.getInt(1));
                    u.setName(rs.getString(2));
                    u.setLastMessageTimestamp(rs.getTimestamp(3).toInstant());
                    u.setLastMessageText(rs.getString(4));
                    u.setUnreadCount(rs.getInt(5));
                    return u;
                },
                user.getUid(), user.getUid());
    }

    @Transactional
    @Override
    public List<com.juick.Message> getPMMessages(final int uid, final int uidTo) {
        SqlParameterSource sqlParameterSource = new MapSqlParameterSource()
                .addValue("uid", uid)
                .addValue("uidTo", uidTo);

        List<com.juick.Message> msgs = getNamedParameterJdbcTemplate().query(
                "SELECT pm.user_id, pm.txt, pm.ts, users.nick FROM pm INNER JOIN users ON users.id=pm.user_id WHERE (user_id = :uid AND user_id_to = :uidTo) "
                        + "OR (user_id_to = :uid AND user_id = :uidTo) ORDER BY ts DESC LIMIT 20",
                sqlParameterSource,
                (rs, rowNum) -> {
                    com.juick.Message msg = new com.juick.Message();
                    int uuid = rs.getInt(1);
                    User user = new User();
                    user.setUid(uuid);
                    user.setName(rs.getString(4));
                    msg.setUser(user);
                    msg.setText(rs.getString(2));
                    msg.setTimestamp(rs.getTimestamp(3).toInstant());
                    return msg;
                });

        getNamedParameterJdbcTemplate().update(
                "UPDATE pm_streams SET lastview = NOW(), unread = 0 WHERE user_id_to = :uid AND user_id = :uidTo",
                sqlParameterSource);

        return msgs;
    }

    @Transactional(readOnly = true)
    @Override
    public List<com.juick.Message> getLastPMInbox(final int uid) {
        return getJdbcTemplate().query(
                "SELECT pm.user_id, users.nick, pm.txt, pm.ts " +
                        "FROM pm INNER JOIN users ON pm.user_id=users.id WHERE pm.user_id_to=? ORDER BY pm.ts DESC LIMIT 20",
                (rs, num) -> {
                    com.juick.Message msg = new com.juick.Message();
                    msg.setUser(new User());
                    msg.getUser().setUid(rs.getInt(1));
                    msg.getUser().setName(rs.getString(2));
                    msg.setText(rs.getString(3));
                    msg.setTimestamp(rs.getTimestamp(4).toInstant());
                    return msg;
                },
                uid);
    }

    @Transactional(readOnly = true)
    @Override
    public List<com.juick.Message> getLastPMSent(final int uid) {
        return getJdbcTemplate().query(
                "SELECT pm.user_id_to, users.nick, pm.txt, " +
                        "pm.ts FROM pm INNER JOIN users ON pm.user_id_to=users.id " +
                        "WHERE pm.user_id=? ORDER BY pm.ts DESC LIMIT 20",
                (rs, num) -> {
                    com.juick.Message msg = new com.juick.Message();
                    msg.setUser(new User());
                    msg.getUser().setUid(rs.getInt(1));
                    msg.getUser().setName(rs.getString(2));
                    msg.setText(rs.getString(3));
                    msg.setTimestamp(rs.getTimestamp(4).toInstant());
                    return msg;
                },
                uid);
    }
}