/*
 * Juick
 * Copyright (C) 2008-2011, Ugnich Anton
 *
 * 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.server;

import com.juick.Tag;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;

/**
 *
 * @author Ugnich Anton
 */
public class TagQueries {

    private static final Logger LOGGER = Logger.getLogger(TagQueries.class.getName());

    public static com.juick.Tag getTag(JdbcTemplate sql, int tid) {
        return sql.queryForObject("SELECT synonym_id,name FROM tags WHERE tag_id=?",
                new Object[]{tid}, (rs, num) -> {
                    Tag ret = new Tag();
                    ret.TID = tid;
                    ret.SynonymID = rs.getInt(1);
                    ret.Name = rs.getString(2);
                    return ret;
                });
    }

    public static com.juick.Tag getTag(JdbcTemplate sql, String tag, boolean autoCreate) {
        Tag ret = null;
        try {
            ret = sql.queryForObject("SELECT tag_id,synonym_id,name FROM tags WHERE name=?",
                    new Object[]{tag}, (rs, rowNum) -> {
                        Tag ret1 = new Tag();
                        ret1.TID = rs.getInt(1);
                        ret1.SynonymID = rs.getInt(2);
                        ret1.Name = rs.getString(3);
                        return ret1;
                    });
        } catch (EmptyResultDataAccessException e) {
            // tag not found
        }
        if (ret == null && autoCreate) {
            ret = new com.juick.Tag();
            ret.TID = createTag(sql, tag);
            ret.Name = tag;
        }

        return ret;
    }

    public static List<com.juick.Tag> getTags(JdbcTemplate sql, String[] tags, boolean autoCreate) {
        List<Tag> ret = new ArrayList<>();

        for (String tag : tags) {
            if (!tag.isEmpty()) {
                Tag t = getTag(sql, tag, autoCreate);
                if (t != null) {
                    ret.add(t);
                }
            }
        }

        return ret;
    }

    public static boolean getTagNoIndex(Connection sql, int tag_id) {
        boolean ret = false;

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT noindex FROM tags WHERE tag_id=?");
            stmt.setInt(1, tag_id);
            rs = stmt.executeQuery();
            if (rs.first()) {
                ret = rs.getInt(1) == 1;
            }
        } catch (SQLException e) {
            LOGGER.log(Level.SEVERE, "sql exception", e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        return ret;
    }

    public static int createTag(JdbcTemplate sql, String name) {
        KeyHolder holder = new GeneratedKeyHolder();
        sql.update(con -> {
            PreparedStatement stmt = con.prepareStatement("INSERT INTO tags(name) VALUES (?)",
                    Statement.RETURN_GENERATED_KEYS);
                stmt.setString(1, name);                
                return stmt;
        }, holder);

        return holder.getKey().intValue();        
    }

    public static List<com.juick.Tag> getUserTagsAll(Connection sql, int uid) {
        List<com.juick.Tag> tags = new ArrayList<>();

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT tags.name,COUNT(messages.message_id) FROM (messages INNER JOIN messages_tags ON (messages.user_id=? AND messages.message_id=messages_tags.message_id)) INNER JOIN tags ON messages_tags.tag_id=tags.tag_id GROUP BY tags.tag_id ORDER BY tags.name ASC");
            stmt.setInt(1, uid);
            rs = stmt.executeQuery();
            rs.beforeFirst();
            while (rs.next()) {
                com.juick.Tag t = new com.juick.Tag();
                t.Name = rs.getString(1);
                t.UsageCnt = rs.getInt(2);
                tags.add(t);
            }
        } catch (SQLException e) {
            LOGGER.log(Level.SEVERE, "sql exception", e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }
        return tags;
    }

    public static List<String> getUserBLTags(Connection sql, int uid) {
        List<String> tags = new ArrayList<>();

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT tags.name FROM tags INNER JOIN bl_tags ON (bl_tags.user_id=? AND bl_tags.tag_id=tags.tag_id) ORDER BY tags.name");
            stmt.setInt(1, uid);
            rs = stmt.executeQuery();
            rs.beforeFirst();
            while (rs.next()) {
                tags.add(rs.getString(1));
            }
        } catch (SQLException e) {
            LOGGER.log(Level.SEVERE, "sql exception", e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }
        return tags;
    }
}