package io.mattw.youtube.commentsuite.db;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import static io.mattw.youtube.commentsuite.db.SQLLoader.INSERT_OR_MODERATED_COMMENTS;

public class ModeratedCommentsTable extends TableHelper<YouTubeComment> {

    private static final String INSERT_IGNORE = INSERT_OR_MODERATED_COMMENTS.toString().replace(":method", "IGNORE");
    private static final String INSERT_REPLACE = INSERT_OR_MODERATED_COMMENTS.toString().replace(":method", "REPLACE");

    private CommentDatabase database;

    public ModeratedCommentsTable(final Connection connection, final CommentDatabase database) {
        super(connection);
        this.database = database;
    }

    @Override
    public YouTubeComment to(ResultSet resultSet) throws SQLException {
        return database.comments().to(resultSet);
    }

    @Override
    public YouTubeComment get(String id) throws SQLException {
        try (PreparedStatement ps = preparedStatement(
                "SELECT * FROM comments_moderated WHERE comment_id = ?")) {
            ps.setString(1, id);

            try (ResultSet rs = ps.executeQuery()) {
                if (rs.next()) {
                    return to(rs);
                }
            }
        }

        return null;
    }

    @Override
    public List<YouTubeComment> getAll() throws SQLException {
        List<YouTubeComment> results = new ArrayList<>();

        try (PreparedStatement ps = preparedStatement(
                "SELECT * FROM comments_moderated")) {
            try (ResultSet rs = ps.executeQuery()) {
                while (rs.next()) {
                    results.add(to(rs));
                }
            }
        }

        return results;
    }

    @Override
    public boolean exists(String id) throws SQLException {
        try (PreparedStatement ps = preparedStatement(
                "SELECT comment_id FROM comments_moderated WHERE comment_id = ?")) {
            ps.setString(1, id);

            try (ResultSet rs = ps.executeQuery()) {
                return rs.next();
            }
        }
    }

    @Override
    public void insertAll(List<YouTubeComment> objects) throws SQLException {
        insert(objects, INSERT_IGNORE);
    }

    @Override
    public void deleteAll(List<YouTubeComment> objects) throws SQLException {
        try (PreparedStatement ps = preparedStatement(
                "DELETE FROM comments_moderated WHERE comment_id = ?")) {
            for (YouTubeComment comment : objects) {
                ps.setString(1, comment.getId());
                ps.addBatch();
            }

            ps.executeBatch();
        }
    }

    @Override
    public void updateAll(List<YouTubeComment> objects) throws SQLException {
        insert(objects, INSERT_REPLACE);
    }

    private void insert(List<YouTubeComment> objects, String insertSql) throws SQLException {
        try (PreparedStatement ps = preparedStatement(INSERT_OR_MODERATED_COMMENTS.toString())) {
            for (YouTubeComment ct : objects) {
                ps.setString(1, ct.getId());
                ps.setString(2, ct.getChannelId());
                ps.setString(3, ct.getVideoId());
                ps.setLong(4, ct.getPublished());
                ps.setString(5, ct.getCommentText());
                ps.setLong(6, ct.getLikes());
                ps.setLong(7, ct.getReplyCount());
                ps.setBoolean(8, ct.isReply());
                ps.setString(9, ct.getParentId());
                ps.setString(10, ct.getModerationStatus().name());
                ps.addBatch();
            }
            ps.executeBatch();
        }

        // Delete from comments if actually moderated usually made from in-app reply
        database.comments().deleteAll(objects);
    }

}

@@ -3055,47 +3055,17 @@
 ent(
-INSERT_OR_MODERATED_COMMENTS.toString()
+insertSql
 )) %7B

fix moderated comments table