ccc.cli.NewDBQueries.java Source code

Introduction

Here is the source code for ccc.cli.NewDBQueries.java

Source

/*-----------------------------------------------------------------------------
 * Copyright (c) 2009 Civic Computing Ltd.
 * All rights reserved.
 *
 * This file is part of Content Control.
 *
 * Content Control is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Content Control 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Content Control.  If not, see http://www.gnu.org/licenses/.
 *
 * Revision      $Rev$
 * Modified by   $Author$
 * Modified on   $Date$
 *
 * Changes: See subversion log.
 *-----------------------------------------------------------------------------
 */
package ccc.cli;

import static ccc.api.types.DBC.*;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Date;
import java.util.UUID;

import org.apache.commons.dbutils.DbUtils;

import ccc.commons.Encryption;

/**
 * Queries for data migration.
 *
 * @author Civic Computing Ltd
 */
class NewDBQueries {

    private final Connection _connection;

    /**
     * Constructor.
     *
     * @param conn Connection
     */
    public NewDBQueries(final Connection conn) {
        require().notNull(conn);
        _connection = conn;
    }

    /**
     * Creates migration user.
     *
     * @param username The username for the new user.
     * @param email The email for the new user.
     * @param password The password for the new user.
     *
     * @return UUID of the created user.
     */
    public UUID insertMigrationUser(final String username, final String email, final String password) {
        final UUID uid = UUID.randomUUID();
        final byte[] hash = Encryption.hash(password, uid.toString());

        PreparedStatement ps = null;

        try {
            // insert user
            ps = _connection.prepareStatement(
                    "INSERT INTO users (id, email, username, vn, hash, name) " + "VALUES (?,?,?,?,?,?)");
            ps.setString(1, uid.toString());
            ps.setString(2, email);
            ps.setString(3, username);
            ps.setInt(4, 0);
            ps.setBytes(5, hash);
            ps.setString(6, username);
            ps.executeUpdate();

            final PreparedStatement gs = _connection.prepareStatement("select id from groups where name in "
                    + "('ADMINISTRATOR', 'SITE_BUILDER', 'CONTENT_CREATOR')");
            final ResultSet groups = gs.executeQuery();

            // insert permission
            while (groups.next()) {
                ps = _connection.prepareStatement("INSERT INTO user_roles (user_id, group_id) " + "VALUES (?, ?)");
                ps.setString(1, uid.toString());
                ps.setString(2, groups.getString(1));
                ps.executeUpdate();
            }

            groups.close();
            gs.close();

            _connection.commit();
        } catch (final SQLException e) {
            throw new RuntimeException(e);
        } finally {
            DbUtils.closeQuietly(ps);
        }
        return uid;
    }

    /**
     * Change migration user password.
     *
     * @param muid UUID of the migration user;
     */
    public void changeMigrationUserPw(final UUID muid) {

        final UUID pwId = UUID.randomUUID();
        final byte[] hash = Encryption.hash("" + new Date().getTime(), pwId.toString());

        PreparedStatement ps = null;

        try {
            // update password
            ps = _connection.prepareStatement("UPDATE users SET hash=?, vn=1 WHERE user_id = ?");
            ps.setBytes(1, hash);
            ps.setString(2, muid.toString());
            ps.executeUpdate();
            _connection.commit();

        } catch (final SQLException e) {
            throw new RuntimeException(e);
        } finally {
            DbUtils.closeQuietly(ps);
        }
    }

    /**
     * Update file store and lucene paths in the setting table.
     *
     * @param path The new path.
     */
    public void updateSettingPaths(final String path) {
        PreparedStatement ps = null;

        try {
            // update file store path
            ps = _connection.prepareStatement("UPDATE settings SET value=? WHERE name = 'FILE_STORE_PATH'");
            ps.setString(1, path + "filestore");
            ps.executeUpdate();
            ps = _connection.prepareStatement("UPDATE settings SET value=? WHERE name = 'LUCENE_INDEX_PATH'");
            ps.setString(1, path + "lucene");
            ps.executeUpdate();
            _connection.commit();

        } catch (final SQLException e) {
            throw new RuntimeException(e);
        } finally {
            DbUtils.closeQuietly(ps);
        }
    }
}