Arakne / Araknemu

 * This file is part of Araknemu.

 *

 * Araknemu is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Lesser General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * Araknemu 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 Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with Araknemu.  If not, see <https://www.gnu.org/licenses/>.

 *

 * Copyright (c) 2017-2019 Vincent Quatrevieux

 */

package fr.quatrevieux.araknemu.data.world.repository.implementation.sql;

import fr.quatrevieux.araknemu.core.dbal.executor.QueryExecutor;

import fr.quatrevieux.araknemu.core.dbal.repository.Record;

import fr.quatrevieux.araknemu.core.dbal.repository.RepositoryException;

import fr.quatrevieux.araknemu.core.dbal.repository.RepositoryUtils;

import fr.quatrevieux.araknemu.data.world.entity.environment.npc.Question;

import fr.quatrevieux.araknemu.data.world.entity.environment.npc.ResponseAction;

import fr.quatrevieux.araknemu.data.world.repository.environment.npc.ResponseActionRepository;

import org.apache.commons.lang3.StringUtils;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.Collections;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

/**

 * Response action repository implementation for SQL database

 *

 * @see ResponseAction

 */

final class SqlResponseActionRepository implements ResponseActionRepository {

    private final QueryExecutor executor;

    private final RepositoryUtils<ResponseAction> utils;

    public SqlResponseActionRepository(QueryExecutor executor) {

        this.executor = executor;

        utils = new RepositoryUtils<>(this.executor, new Loader());

    }

    @Override

    public void initialize() throws RepositoryException {

        try {

            executor.query(

                "CREATE TABLE NPC_RESPONSE_ACTION(" +

                    "RESPONSE_ID INTEGER," +

                    "ACTION VARCHAR(12)," +

                    "ARGUMENTS VARCHAR(255)," +

                    "PRIMARY KEY (RESPONSE_ID, ACTION)" +

                ")"

            );

        } catch (SQLException e) {

            throw new RepositoryException(e);

        }

    }

    @Override

    public void destroy() throws RepositoryException {

        try {

            executor.query("DROP TABLE NPC_RESPONSE_ACTION");

        } catch (SQLException e) {

            throw new RepositoryException(e);

        }

    }

    @Override

    public ResponseAction get(ResponseAction entity) throws RepositoryException {

        return utils.findOne(

            "SELECT * FROM NPC_RESPONSE_ACTION WHERE RESPONSE_ID = ? AND ACTION = ?",

            stmt -> {

                stmt.setInt(1, entity.responseId());

                stmt.setString(2, entity.action());

            }

        );

    }

    @Override

    public boolean has(ResponseAction entity) throws RepositoryException {

        return utils.aggregate(

            "SELECT COUNT(*) FROM NPC_RESPONSE_ACTION WHERE RESPONSE_ID = ? AND ACTION = ?",

            stmt -> {

                stmt.setInt(1, entity.responseId());

                stmt.setString(2, entity.action());

            }

        ) > 0;

    }

    @Override

    public Map<Integer, List<ResponseAction>> all() throws RepositoryException {

        return

            utils

                .findAll("SELECT * FROM NPC_RESPONSE_ACTION")

                .stream()

                .collect(Collectors.groupingBy(ResponseAction::responseId))

        ;

    }

    @Override

    @SuppressWarnings("array.access.unsafe.high.constant") // checker do not handle switch case

    public Map<Integer, List<ResponseAction>> byQuestion(Question question) {

        switch (question.responseIds().length) {

            case 0:

                return Collections.emptyMap();

            case 1:

                final List<ResponseAction> ras = utils.findAll(

                    "SELECT * FROM NPC_RESPONSE_ACTION WHERE RESPONSE_ID = ?",

                    rs -> rs.setInt(1, question.responseIds()[0])

                );

                return ras.isEmpty() ? Collections.emptyMap() : Collections.singletonMap(question.responseIds()[0], ras);

            default:

                return utils.findAll(

                        "SELECT * FROM NPC_RESPONSE_ACTION WHERE RESPONSE_ID IN(" + StringUtils.repeat("?, ", question.responseIds().length - 1) + "?)",

                        rs -> {

                            for (int i = 0; i < question.responseIds().length; ++i) {

                                rs.setInt(i + 1, question.responseIds()[i]);

                            }

                        }

                    )

                    .stream()

                    .collect(Collectors.groupingBy(ResponseAction::responseId))

                ;

        }

    }

        }

        @Override

        public ResponseAction fillKeys(ResponseAction entity, ResultSet keys) {

            throw new RepositoryException("Read-only entity");

        }

    }

}