/*
 * Copyright (C) 2008-2022, 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.model.Chat;
import com.juick.model.User;
import com.juick.model.Message;
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 ChatServiceImpl extends BaseJdbcService implements ChatService {

    @Transactional
    @Override
    public boolean createMessage(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(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) -> {
                    Chat u = new 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<Message> getChat(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 OFFSET 0 ROWS FETCH NEXT 20 ROWS ONLY",
                sqlParameterSource,
                (rs, rowNum) -> {
                    Message msg = new 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<Message> getInbox(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 OFFSET 0 ROWS FETCH NEXT 20 ROWS ONLY",
                (rs, num) -> {
                    Message msg = new 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<Message> getOutbox(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 OFFSET 0 ROWS FETCH NEXT 20 ROWS ONLY",
                (rs, num) -> {
                    Message msg = new 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);
    }
}