/*
 * 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 java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;

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

    public static com.juick.Tag getTag(Connection sql, int tid) {
        com.juick.Tag ret = null;

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT synonym_id,name FROM tags WHERE tag_id=?");
            stmt.setInt(1, tid);
            rs = stmt.executeQuery();
            if (rs.first()) {
                ret = new com.juick.Tag();
                ret.TID = tid;
                ret.SynonymID = rs.getInt(1);
                ret.Name = rs.getString(2);
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        return ret;
    }

    public static com.juick.Tag getTag(Connection sql, String tag, boolean autoCreate) {
        com.juick.Tag ret = null;

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("SELECT tag_id,synonym_id,name FROM tags WHERE name=?");
            stmt.setString(1, tag);
            rs = stmt.executeQuery();
            if (rs.first()) {
                ret = new com.juick.Tag();
                ret.TID = rs.getInt(1);
                ret.SynonymID = rs.getInt(2);
                ret.Name = rs.getString(3);
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        if (ret == null && autoCreate) {
            ret = new com.juick.Tag();
            ret.TID = createTag(sql, tag);
            ret.Name = tag;
        }

        return ret;
    }

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

        for (int i = 0; i < tags.length; i++) {
            if (!tags[i].isEmpty()) {
                com.juick.Tag t = getTag(sql, tags[i], 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) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        return ret;
    }

    public static int createTag(Connection sql, String name) {
        int ret = 0;

        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            stmt = sql.prepareStatement("INSERT INTO tags(name) VALUES (?)", Statement.RETURN_GENERATED_KEYS);
            stmt.setString(1, name);
            stmt.executeUpdate();
            rs = stmt.getGeneratedKeys();
            if (rs.first()) {
                ret = rs.getInt(1);
            }
        } catch (SQLException e) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }

        return ret;
    }

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

        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) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }
        return tags;
    }

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

        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) {
            System.err.println(e);
        } finally {
            Utils.finishSQL(rs, stmt);
        }
        return tags;
    }
}