/*
 * 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.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 User me, final User to, final String body) {
        boolean ret = getJdbcTemplate().update(
                "INSERT INTO pm(user_id, user_id_to, txt) VALUES (?, ?, ?)",
                me.getUid(), to.getUid(), body) > 0;

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

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

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

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

    @Transactional(readOnly = true)
    @Override
    public List<User> getPMLastConversationsUsers(final User me, final int cnt) {
        return getJdbcTemplate().query(
                "SELECT pm_streams.user_id, users.nick, pm_streams.unread FROM pm_streams "
                        + "INNER JOIN users ON users.id = pm_streams.user_id "
                        + "WHERE pm_streams.user_id_to=? "
                        + "ORDER BY pm_streams.unread DESC, pm_streams.lastmessage DESC LIMIT ?",
                (rs, rowNum) -> {
                    com.juick.User u = new com.juick.User();
                    u.setUid(rs.getInt(1));
                    u.setName(rs.getString(2));
                    u.setUnreadCount(rs.getInt(3));
                    return u;
                },
                me.getUid(), cnt);
    }

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

        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 User me) {
        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;
                },
                me.getUid());
    }

    @Transactional(readOnly = true)
    @Override
    public List<com.juick.Message> getLastPMSent(final User me) {
        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;
                },
                me.getUid());
    }
}