/*
 * 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.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) {
        return getJdbcTemplate().update(
                "INSERT INTO pm(user_id, user_id_to, txt) VALUES (?, ?, ?)",
                uidFrom, uid_to, body) > 0;
    }

    @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 List<Chat> getLastChats(final User user) {
        return getJdbcTemplate().query(
                "SELECT l.user_id, users.nick, l.last, pm.txt FROM pm "
                        + "INNER JOIN users ON users.id = pm.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.user_id_to=? "
                        + "ORDER BY 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).trim());
                    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);

        return 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).trim());
                    msg.setCreated(rs.getTimestamp(3).toInstant());
                    return msg;
                });
    }

    @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).trim());
                    msg.setCreated(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).trim());
                    msg.setCreated(rs.getTimestamp(4).toInstant());
                    return msg;
                },
                uid);
    }
}