V4__replace_old_scopes.java

/*
 * Copyright (C) 2013 tarent AG
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package db.migration;

import org.flywaydb.core.api.migration.MigrationChecksumProvider;
import org.flywaydb.core.api.migration.jdbc.JdbcMigration;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

public class V4__replace_old_scopes implements JdbcMigration, MigrationChecksumProvider {

    private final Map<String, Long> internalIds = new HashMap<>();
    private Connection connection;

    @Override
    public Integer getChecksum() {
        return 1546262391;
    }

    @Override
    public void migrate(Connection connection) throws Exception {
        this.connection = connection;
        removeScopesFromClient("auth-server", "GET", "POST", "PUT", "PATCH", "DELETE");
        addScopesToClient("auth-server", "ADMIN");
    }

    private void removeScopesFromClient(String clientId, String... scopes) throws SQLException {
        long internalId = toInternalId(clientId);
        if (internalId < 0) {
            return;
        }
        for (String scope : scopes) {
            try (PreparedStatement statement = connection
                    .prepareStatement("delete from osiam_client_scopes where id = ? and scope = ?")) {
                statement.setLong(1, internalId);
                statement.setString(2, scope);
                statement.execute();
            }
        }
    }

    private void addScopesToClient(String clientId, String... scopes) throws SQLException {
        long internalId = toInternalId(clientId);
        if (internalId < 0) {
            return;
        }
        for (String scope : scopes) {
            try (PreparedStatement statement = connection
                    .prepareStatement("insert into osiam_client_scopes (id, scope) values (?, ?)")) {
                statement.setLong(1, internalId);
                statement.setString(2, scope);
                statement.execute();
            }
        }
    }

    private long toInternalId(String clientId) {
        if (!internalIds.containsKey(clientId)) {
            try (PreparedStatement statement = connection
                    .prepareStatement("select internal_id from osiam_client where id = ?")) {
                statement.setString(1, clientId);
                ResultSet resultSet = statement.executeQuery();
                if (resultSet.next()) {
                    internalIds.put(clientId, resultSet.getLong(1));
                } else {
                    internalIds.put(clientId, -1L);
                }
            } catch (SQLException e) {
                internalIds.put(clientId, -1L);
            }
        }
        return internalIds.get(clientId);
    }
}