/*
 * 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.search;

import org.apache.commons.lang3.StringUtils;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.util.Assert;

import javax.sql.DataSource;
import java.util.Collections;
import java.util.List;

/**
 * Created by aalexeev on 11/18/16.
 */

/* Note
 * Don't use any spring's component annotation (such as @Repository, @Service, @Component, etc).
 * This class directly used by spring's search configuration class
 */
public class SphinxSearchServiceImpl implements SearchService {
    private static final int DEFAULT_MAX_RESULT = 25;

    private final NamedParameterJdbcTemplate namedParameterSearchJdbcTemplate;

    private int maxResult = DEFAULT_MAX_RESULT;


    public SphinxSearchServiceImpl(JdbcTemplate searchJdbcTemplate) {
        Assert.notNull(searchJdbcTemplate, "JdbcTemplate must be initialized");
        this.namedParameterSearchJdbcTemplate = new NamedParameterJdbcTemplate(searchJdbcTemplate);
    }

    public SphinxSearchServiceImpl(DataSource searchDataSource) {
        Assert.notNull(searchDataSource, "DataSource must be initialized");
        this.namedParameterSearchJdbcTemplate = new NamedParameterJdbcTemplate(searchDataSource);
    }

    @Override
    public List<Integer> searchInAllMessages(final String searchString, final int messageIdBefore) {
        if (StringUtils.isBlank(searchString))
            return Collections.emptyList();

        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource()
                .addValue("search", searchString)
                .addValue("before", messageIdBefore)
                .addValue("limit", maxResult);

        return namedParameterSearchJdbcTemplate.queryForList(
                "SELECT id AS message_id FROM messages WHERE MATCH(:search) " +
                        (messageIdBefore > 0 ?
                                " AND id < :before " : StringUtils.EMPTY) +
                        " ORDER BY id DESC LIMIT :limit",
                sqlParameterSource,
                Integer.class);
    }

    @Override
    public List<Integer> searchByStringAndUser(final String searchString, final int userId, int messageIdBefore) {
        if (StringUtils.isBlank(searchString))
            return Collections.emptyList();

        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource()
                .addValue("search", searchString)
                .addValue("userId", userId)
                .addValue("before", messageIdBefore)
                .addValue("limit", maxResult);

        return namedParameterSearchJdbcTemplate.queryForList(
                "SELECT id AS message_id FROM messages WHERE user_id = :userId AND MATCH(:search) " +
                        (messageIdBefore > 0 ?
                                " AND id < :before " : StringUtils.EMPTY) +
                        " ORDER BY id DESC LIMIT :limit",
                sqlParameterSource,
                Integer.class);
    }

    @Override
    public void setMaxResult(int maxResult) {
        if (maxResult <= 0)
            throw new IllegalArgumentException("maxResult value (" + maxResult + ") must be greater then 0");

        this.maxResult = maxResult;
    }
}