com.hangum.tadpole.rdb.core.util.GrantCheckerUtils.java Source code

Introduction

Here is the source code for com.hangum.tadpole.rdb.core.util.GrantCheckerUtils.java

Source

/*******************************************************************************
 * Copyright (c) 2015 hangum.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     hangum - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.rdb.core.util;

import org.eclipse.jface.dialogs.MessageDialog;

import com.hangum.tadpole.ace.editor.core.define.EditorDefine;
import com.hangum.tadpole.commons.libs.core.define.PublicTadpoleDefine;
import com.hangum.tadpole.engine.permission.PermissionChecker;
import com.hangum.tadpole.engine.query.dao.system.UserDBDAO;
import com.hangum.tadpole.engine.security.TadpoleSecurityManager;
import com.hangum.tadpole.engine.sql.util.SQLUtil;
import com.hangum.tadpole.rdb.core.Messages;
import com.hangum.tadpole.rdb.core.editors.main.utils.RequestQuery;

/**
 * grant checker
 * 
 * @author hangum
 *
 */
public class GrantCheckerUtils {

    /**
     * execute query
     * 
     * @param userDB
     * @return
     * @throws Exception
     */
    public static boolean ifExecuteQuery(UserDBDAO userDB) throws Exception {
        // security check.
        if (!TadpoleSecurityManager.getInstance().isLock(userDB)) {
            throw new Exception(Messages.get().ResultMainComposite_1);
        }

        if (PublicTadpoleDefine.YES_NO.YES.name().equals(userDB.getQuestion_dml())
                || PermissionChecker.isProductBackup(userDB)) {
            MessageDialog dialog = new MessageDialog(null, Messages.get().Confirm, null,
                    Messages.get().GrantCheckerUtils_0, MessageDialog.ERROR,
                    new String[] { Messages.get().YES, Messages.get().NO }, 1);
            if (dialog.open() == 1)
                return false;
        }

        return true;
    }

    /**
     *    ?? .
     * 
     * - DB lock ???
     * - dml ?  ???
     * - ?,  ? select  ?
     * 
     * @param userDB
     * @param reqQuery
     * @throws Exception
     */
    public static boolean ifExecuteQuery(UserDBDAO userDB, RequestQuery reqQuery) throws Exception {
        // security check.
        if (!TadpoleSecurityManager.getInstance().isLock(userDB)) {
            throw new Exception(Messages.get().ResultMainComposite_1);
        }

        // ? ? .
        if (PublicTadpoleDefine.YES_NO.YES.name().equals(userDB.getQuestion_dml())
                || PermissionChecker.isProductBackup(userDB)) {
            boolean isDDLQuestion = !reqQuery.isStatement();
            if (reqQuery.getExecuteType() == EditorDefine.EXECUTE_TYPE.ALL) {
                for (String strSQL : reqQuery.getOriginalSql().split(PublicTadpoleDefine.SQL_DELIMITER)) {
                    if (!SQLUtil.isStatement(strSQL)) {
                        isDDLQuestion = true;
                        break;
                    }
                }
            }

            if (isDDLQuestion) {
                MessageDialog dialog = new MessageDialog(null, Messages.get().Execute, null,
                        Messages.get().GrantCheckerUtils_0, MessageDialog.QUESTION,
                        new String[] { Messages.get().YES, Messages.get().NO }, 1);
                if (dialog.open() != MessageDialog.OK)
                    return false;
            }
        }

        return true;
    }
}