Пример #1
0
  @Override
  public boolean close() {
    //  로그인이 안되었을 경우 로그인 창이 남아 있도록...(https://github.com/hangum/TadpoleForDBTools/issues/31)
    if (!SessionManager.isLogin()) return false;

    return super.close();
  }
  /**
   * mongodb update
   *
   * @param txtLimitCount
   * @param txtMacCount
   * @param txtFindPage
   * @param txtResultPage
   * @throws Exception
   */
  public static void updateMongoDBUserInfoData(
      String txtLimitCount, String txtMacCount, String txtFindPage, String txtResultPage)
      throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    // 	MONGO_DEFAULT_LIMIT
    userInfoData.setName(PreferenceDefine.MONGO_DEFAULT_LIMIT);
    userInfoData.setValue0(txtLimitCount);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    // MONGO_DEFAULT_MAX_COUNT
    userInfoData.setName(PreferenceDefine.MONGO_DEFAULT_MAX_COUNT);
    userInfoData.setValue0(txtMacCount);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    // MONGO_DEFAULT_FIND
    userInfoData.setName(PreferenceDefine.MONGO_DEFAULT_FIND);
    userInfoData.setValue0(txtFindPage);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    // MONGO_DEFAULT_RESULT
    userInfoData.setName(PreferenceDefine.MONGO_DEFAULT_RESULT);
    userInfoData.setValue0(txtResultPage);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
  /**
   * general 정보의 export delimit을 저장합니다.
   *
   * @param sessionTimeOut
   * @throws Exception
   */
  public static void updateGeneralExportDelimitData(String delimit) throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    // 	select 제한  갯수
    userInfoData.setName(PreferenceDefine.EXPORT_DILIMITER);
    userInfoData.setValue0(delimit);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
  /**
   * general 정보의 session time을 저장합니다.
   *
   * @param sessionTimeOut
   * @throws Exception
   */
  public static void updateGeneralUserInfoData(String sessionTimeOut) throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    // 	select 제한  갯수
    userInfoData.setName(PreferenceDefine.SESSION_DFEAULT_PREFERENCE);
    userInfoData.setValue0(sessionTimeOut);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
  /**
   * default home page
   *
   * @param use
   * @throws Exception
   */
  public static void updateDefaultHomePageUse(String use) throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    // 	select 제한  갯수
    userInfoData.setName(PreferenceDefine.DEFAULT_HOME_PAGE_USE);
    userInfoData.setValue0(use);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
  /**
   * @param userDB
   * @param title
   * @param desc
   * @param cronExp
   * @param listSchedule
   */
  public static ScheduleMainDAO addSchedule(
      final UserDBDAO userDB,
      String title,
      String desc,
      String cronExp,
      List<ScheduleDAO> listSchedule)
      throws Exception {
    int userSeq = SessionManager.getUserSeq();

    ScheduleMainDAO dao = new ScheduleMainDAO();
    dao.setSeq(userSeq);
    dao.setUser_seq(SessionManager.getUserSeq());
    dao.setDb_seq(userDB.getSeq());
    dao.setTitle(title);
    dao.setDescription(desc);
    dao.setCron_exp(cronExp);

    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    dao = (ScheduleMainDAO) sqlClient.insert("scheduleMainInsert", dao);

    // save schedule
    for (ScheduleDAO scheduleDAO : listSchedule) {
      scheduleDAO.setSchedule_main_seq(dao.getSeq());
      ScheduleDAO retScheduleDAO = (ScheduleDAO) sqlClient.insert("scheduleInsert", scheduleDAO);

      // sql
      String[] sqls = SQLUtil.makeResourceDataArays(scheduleDAO.getSql());
      ScheduleDetailDAO detailDao = null;
      for (String sql : sqls) {
        detailDao = new ScheduleDetailDAO();
        detailDao.setSchedule_seq(retScheduleDAO.getSeq());
        detailDao.setDatas(sql);

        sqlClient.insert("scheduleDetailInsert", detailDao);
      }
    }

    return dao;
  }
  /**
   * update SQLFormatter
   *
   * @param userdb
   * @param tabSize
   * @param resultSelect
   * @param sqlFormatIn
   */
  public static void updateSQLFormatterInfoData(
      String tabSize, String sqlFormatDecode, String sqlFormatIn) throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    userInfoData.setName(PreferenceDefine.DEFAULT_TAB_SIZE_PREFERENCE);
    userInfoData.setValue0(tabSize);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    userInfoData.setName(PreferenceDefine.SQL_FORMATTER_DECODE_PREFERENCE);
    userInfoData.setValue0(sqlFormatDecode);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    userInfoData.setName(PreferenceDefine.SQL_FORMATTER_IN_PREFERENCE);
    userInfoData.setValue0(sqlFormatIn);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
  /**
   * 신규 사용자의 기본 유저 데이터 정보를 저장합니다.
   *
   * @param userdb
   * @param limitSelect
   * @param resultSelect
   * @param oraclePlan
   */
  public static void updateRDBUserInfoData(
      String limitSelect, String resultSelect, String oraclePlan) throws Exception {
    SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
    UserInfoDataDAO userInfoData = new UserInfoDataDAO();
    userInfoData.setUser_seq(SessionManager.getSeq());

    // 	select 제한  갯수
    userInfoData.setName(PreferenceDefine.SELECT_LIMIT_COUNT);
    userInfoData.setValue0(limitSelect);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    // 검색 결과 페이지 당 보여주는 갯수
    userInfoData.setName(PreferenceDefine.SELECT_RESULT_PAGE_PREFERENCE);
    userInfoData.setValue0(resultSelect);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$

    // ORACLE PLAN TABLE
    userInfoData.setName(PreferenceDefine.ORACLE_PLAN_TABLE);
    userInfoData.setValue0(oraclePlan);
    sqlClient.update("userInfoDataUpdate", userInfoData); // $NON-NLS-1$
  }
 /**
  * 사용자 정보 데이터.
  *
  * @param key
  * @return
  * @throws Exception
  */
 public static Object getUserInfoData(String key) throws Exception {
   SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
   return sqlClient.queryForObject("getUserInfoData", SessionManager.getSeq()); // $NON-NLS-1$
 }
  @Override
  public boolean performOk() {
    String txtResultType = comboRDBResultType.getText();
    String txtSelectLimit = textSelectLimit.getText();
    String txtResultPage = textResultPage.getText();
    String txtQueryTimtout = textQueryTimeout.getText();
    String txtOraclePlan = textOraclePlan.getText();
    String txtRDBNumberColumnIsComman = comboRDBNumberComma.getText();
    String txtFontInfo = lblUserFont.getText();
    String txtCommitCount = textCommitCount.getText();
    String txtShownInTheColumn = textShowInTheColumn.getText();

    if (!NumberUtils.isNumber(txtSelectLimit)) {
      MessageDialog.openError(
          getShell(),
          "Confirm",
          Messages.get().DefaultPreferencePage_0
              + Messages.get().RDBPreferencePage_0); // $NON-NLS-1$
      textSelectLimit.setFocus();
      return false;
    }

    if (!NumberUtils.isNumber(txtResultPage)) {
      MessageDialog.openError(
          getShell(),
          "Confirm",
          Messages.get().DefaultPreferencePage_other_labelText_1
              + Messages.get().RDBPreferencePage_0); // $NON-NLS-1$
      textResultPage.setFocus();
      return false;
    }

    if (!NumberUtils.isNumber(txtQueryTimtout)) {
      MessageDialog.openError(
          getShell(), "Confirm", "Query timeout is " + Messages.get().RDBPreferencePage_0);
      textQueryTimeout.setFocus();
      return false;
    }

    if (!NumberUtils.isNumber(txtCommitCount)) {
      MessageDialog.openError(
          getShell(), "Confirm", "Commit count is " + Messages.get().RDBPreferencePage_0);
      textCommitCount.setFocus();
      return false;
    }

    if ("".equals(txtOraclePlan)) { // $NON-NLS-1$
      MessageDialog.openError(
          getShell(), "Confirm", Messages.get().RDBPreferencePage_3); // $NON-NLS-1$
      return false;
    }

    if (!NumberUtils.isNumber(txtShownInTheColumn)) {
      MessageDialog.openError(getShell(), "Confirm", Messages.get().RDBPreferencePage_0);
      textShowInTheColumn.setFocus();
      return false;
    }

    // 테이블에 저장
    try {
      TadpoleSystem_UserInfoData.updateRDBUserInfoData(
          txtSelectLimit,
          txtResultPage,
          txtQueryTimtout,
          txtOraclePlan,
          txtRDBNumberColumnIsComman,
          txtFontInfo,
          txtCommitCount,
          txtShownInTheColumn,
          txtResultType);

      // session 데이터를 수정한다.
      SessionManager.setUserInfo(PreferenceDefine.RDB_RESULT_TYPE, txtResultType);
      SessionManager.setUserInfo(PreferenceDefine.SELECT_LIMIT_COUNT, txtSelectLimit);
      SessionManager.setUserInfo(PreferenceDefine.SELECT_RESULT_PAGE_PREFERENCE, txtResultPage);
      SessionManager.setUserInfo(PreferenceDefine.SELECT_QUERY_TIMEOUT, txtQueryTimtout);

      SessionManager.setUserInfo(PreferenceDefine.ORACLE_PLAN_TABLE, txtOraclePlan);
      SessionManager.setUserInfo(
          PreferenceDefine.RDB_RESULT_NUMBER_IS_COMMA, txtRDBNumberColumnIsComman);
      SessionManager.setUserInfo(PreferenceDefine.RDB_RESULT_FONT, txtFontInfo);
      SessionManager.setUserInfo(PreferenceDefine.RDB_COMMIT_COUNT, txtCommitCount);
      SessionManager.setUserInfo(
          PreferenceDefine.RDB_CHARACTER_SHOW_IN_THE_COLUMN, txtShownInTheColumn);

    } catch (Exception e) {
      logger.error("RDBPreference saveing", e);

      MessageDialog.openError(
          getShell(),
          "Confirm",
          Messages.get().RDBPreferencePage_5 + e.getMessage()); // $NON-NLS-1$
      return false;
    }

    return super.performOk();
  }
 /**
  * find user schedule
  *
  * @return
  * @throws Exception
  */
 public static List<ScheduleMainDAO> findUserScheduleMain() throws Exception {
   SqlMapClient sqlClient = TadpoleSQLManager.getInstance(TadpoleSystemInitializer.getUserDB());
   return sqlClient.queryForList("getUserScheduleMain", SessionManager.getUserSeq());
 }
Пример #12
0
  @Override
  protected void okPressed() {
    String strEmail = StringUtils.trimToEmpty(textEMail.getText());
    String strPass = StringUtils.trimToEmpty(textPasswd.getText());

    if (!validation(strEmail, strPass)) return;

    try {
      UserDAO userDao = TadpoleSystem_UserQuery.login(strEmail, strPass);

      // firsttime email confirm
      if (PublicTadpoleDefine.YES_NO.NO.name().equals(userDao.getIs_email_certification())) {
        InputDialog inputDialog =
            new InputDialog(
                getShell(),
                Messages.get().LoginDialog_10,
                Messages.get().LoginDialog_17,
                "",
                null); //$NON-NLS-3$ //$NON-NLS-1$
        if (inputDialog.open() == Window.OK) {
          if (!userDao.getEmail_key().equals(inputDialog.getValue())) {
            throw new Exception(Messages.get().LoginDialog_19);
          } else {
            TadpoleSystem_UserQuery.updateEmailConfirm(strEmail);
          }
        } else {
          throw new Exception(Messages.get().LoginDialog_20);
        }
      }

      if (PublicTadpoleDefine.YES_NO.NO.name().equals(userDao.getApproval_yn())) {
        MessageDialog.openError(
            getParentShell(), Messages.get().LoginDialog_7, Messages.get().LoginDialog_27);

        return;
      }

      // Check the allow ip
      String strAllowIP = userDao.getAllow_ip();
      boolean isAllow = IPFilterUtil.ifFilterString(strAllowIP, RequestInfoUtils.getRequestIP());
      if (logger.isDebugEnabled())
        logger.debug(
            Messages.get().LoginDialog_21
                + userDao.getEmail()
                + Messages.get().LoginDialog_22
                + strAllowIP
                + Messages.get().LoginDialog_23
                + RequestInfoUtils.getRequestIP());
      if (!isAllow) {
        logger.error(
            Messages.get().LoginDialog_21
                + userDao.getEmail()
                + Messages.get().LoginDialog_22
                + strAllowIP
                + Messages.get().LoginDialog_26
                + RequestInfoUtils.getRequestIP());
        MessageDialog.openError(
            getParentShell(), Messages.get().LoginDialog_7, Messages.get().LoginDialog_28);
        return;
      }

      if (PublicTadpoleDefine.YES_NO.YES.name().equals(userDao.getUse_otp())) {
        OTPLoginDialog otpDialog = new OTPLoginDialog(getShell());
        otpDialog.open();

        if (!GoogleAuthManager.getInstance()
            .isValidate(userDao.getOtp_secret(), otpDialog.getIntOTPCode())) {
          throw new Exception(Messages.get().LoginDialog_2);
        }
      }

      // 로그인 유지.
      registLoginID(userDao.getEmail());

      SessionManager.addSession(userDao);

      // save login_history
      TadpoleSystem_UserQuery.saveLoginHistory(userDao.getSeq());
    } catch (Exception e) {
      logger.error(
          String.format(
              "Login exception. request email is %s, reason %s",
              strEmail, e.getMessage())); // $NON-NLS-1$
      MessageDialog.openError(getParentShell(), Messages.get().LoginDialog_29, e.getMessage());

      textPasswd.setFocus();
      return;
    }

    super.okPressed();
  }
  @Override
  public boolean connection() {
    if (!isValidate()) return false;

    String dbUrl = "";
    if (comboConnType.getText().equals("SID")) {
      dbUrl =
          String.format(
              DBDefine.ORACLE_DEFAULT.getDB_URL_INFO(),
              textHost.getText(),
              textPort.getText(),
              textDatabase.getText());
    } else if (comboConnType.getText().equals("Service Name")) {
      dbUrl =
          String.format(
              "jdbc:oracle:thin:@%s:%s/%s",
              textHost.getText(), textPort.getText(), textDatabase.getText());
    }

    userDB = new UserDBDAO();
    userDB.setTypes(DBDefine.ORACLE_DEFAULT.getDBToString());
    userDB.setUrl(dbUrl);
    userDB.setDb(textDatabase.getText());
    userDB.setGroup_name(comboGroup.getText().trim());
    userDB.setDisplay_name(textDisplayName.getText());
    userDB.setOperation_type(
        DBOperationType.getNameToType(comboOperationType.getText()).toString());
    userDB.setHost(textHost.getText());
    userDB.setPasswd(textPassword.getText());
    userDB.setPort(textPort.getText());
    //		userDB.setLocale(comboLocale.getText().trim());
    userDB.setUsers(textUser.getText());

    // 기존 데이터 업데이트
    if (oldUserDB != null) {
      if (!MessageDialog.openConfirm(null, "Confirm", Messages.SQLiteLoginComposite_13))
        return false; //$NON-NLS-1$

      if (!checkDatabase(userDB)) return false;

      try {
        TadpoleSystem_UserDBQuery.updateUserDB(userDB, oldUserDB, SessionManager.getSeq());
      } catch (Exception e) {
        logger.error(Messages.SQLiteLoginComposite_8, e);
        Status errStatus =
            new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
        ExceptionDetailsErrorDialog.openError(
            getShell(), "Error", Messages.SQLiteLoginComposite_5, errStatus); // $NON-NLS-1$

        return false;
      }

      // 신규 데이터 저장.
    } else {
      // db가 정상적인지 채크해본다
      try {
        SqlMapClient sqlClient = TadpoleSQLManager.getInstance(userDB);
        List showTables = sqlClient.queryForList("tableList", textDatabase.getText());

      } catch (Exception e) {
        logger.error(Messages.OracleLoginComposite_7, e);
        Status errStatus =
            new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
        ExceptionDetailsErrorDialog.openError(
            getShell(), "Error", Messages.OracleLoginComposite_10, errStatus); // $NON-NLS-1$

        return false;
      }

      try {
        TadpoleSystem_UserDBQuery.newUserDB(userDB, SessionManager.getSeq());
      } catch (Exception e) {
        logger.error("Oracle db info save", e);
        Status errStatus =
            new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
        ExceptionDetailsErrorDialog.openError(
            getShell(), "Error", Messages.OracleLoginComposite_11, errStatus); // $NON-NLS-1$
      }
    }

    return true;
  }
/**
 * Tadpole DB Hub ERD editor
 *
 * @author hangum
 */
public class TadpoleRDBEditor extends GraphicalEditor { // WithFlyoutPalette {
  public static final String ID = "com.hangum.tadpole.rdb.erd.core.editor"; // $NON-NLS-1$
  /** Logger for this class */
  private static final Logger logger = Logger.getLogger(TadpoleRDBEditor.class);

  /** 사용자 seq */
  private final int user_seq = SessionManager.getUserSeq();

  /** first init data */
  private DB db;

  private UserDBDAO userDB;
  private UserDBResourceDAO userDBErd;
  /** 처음로드될때부터 모든 테이블 로드 인지 */
  private boolean isAllTable = false;

  /** short key handler */
  private KeyHandler keyHandler;

  /** dnd */
  TableTransferFactory tableTransFactory = new TableTransferFactory();

  public TadpoleRDBEditor() {
    setEditDomain(new DefaultEditDomain(this));
  }

  @Override
  protected void initializeGraphicalViewer() {
    //		super.initializeGraphicalViewer();

    Job job =
        new Job("ERD Initialize") {
          @Override
          public IStatus run(IProgressMonitor monitor) {
            monitor.beginTask("Painting table object", IProgressMonitor.UNKNOWN);

            try {
              RdbFactory factory = RdbFactory.eINSTANCE;

              if (db == null) {

                // 모든 table 정보를 가져온다.
                if (isAllTable) {
                  db = TadpoleModelUtils.INSTANCE.getDBAllTable(monitor, userDB);
                  // 부분 테이블 정보를 처리한다.
                } else {
                  db = factory.createDB();
                }

                db.setDbType(userDB.getDbms_type() + " (" + userDB.getDisplay_name() + ")");
              }

              // 하위 호환을 위한 코드 .
              if (db.getStyle() == null) {
                Style style = RdbFactory.eINSTANCE.createStyle();
                style.setDb(db);
                db.setStyle(style);
              }

            } catch (Exception e) {
              logger.error("ERD Initialize excepiton", e);

              return new Status(Status.WARNING, Activator.PLUGIN_ID, e.getMessage());
            } finally {
              monitor.done();
            }

            /////////////////////////////////////////////////////////////////////////////////////////
            return Status.OK_STATUS;
          }
        };

    // job의 event를 처리해 줍니다.
    final TadpoleRDBEditor rdbEditor = this;
    job.addJobChangeListener(
        new JobChangeAdapter() {

          public void done(IJobChangeEvent event) {
            final IJobChangeEvent jobEvent = event;
            getSite()
                .getShell()
                .getDisplay()
                .asyncExec(
                    new Runnable() {
                      public void run() {
                        if (!jobEvent.getResult().isOK()) {

                          // 아래의 try문은 이슈 169의 오류를 검증하기 위한 코드입니다.
                          // https://github.com/hangum/TadpoleForDBTools/issues/169
                          //  근본적인 에러는 해결안됨.  그러나 프로그램에서 에러나고 죽는 그런 경우는 해결.
                          //
                          try {
                            Exception e = new Exception(jobEvent.getResult().getException());
                            Status errStatus =
                                new Status(
                                    IStatus.ERROR,
                                    Activator.PLUGIN_ID,
                                    e.getMessage(),
                                    e); //$NON-NLS-1$
                            ExceptionDetailsErrorDialog.openError(
                                PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(),
                                "Error",
                                Messages.get().TadpoleModelUtils_2,
                                errStatus); //$NON-NLS-1$
                          } catch (Exception e) {
                            logger.error(
                                "https://github.com/hangum/TadpoleForDBTools/issues/169 검증오류...",
                                e);
                          }

                          // 오류가 발생했을때는 기본 정보로
                          RdbFactory factory = RdbFactory.eINSTANCE;
                          db = factory.createDB();
                          db.setDbType(userDB.getDbms_type());
                          db.setId(userDB.getUsers());
                          db.setUrl(userDB.getHost());
                        }
                        getGraphicalViewer().setContents(db);

                        // dnd
                        getGraphicalViewer()
                            .addDropTargetListener(
                                new TableTransferDropTargetListener(
                                    rdbEditor, getGraphicalViewer(), userDB, db));
                      }
                    }); // end display.asyncExec
          } // end done
        }); // end job

    job.setName(userDB.getDisplay_name());
    job.setUser(true);
    job.schedule();
  }

  @Override
  public void dispose() {
    super.dispose();
  }

  @Override
  protected void configureGraphicalViewer() {
    super.configureGraphicalViewer();

    GraphicalViewer viewer = getGraphicalViewer();
    viewer.setEditPartFactory(new TadpoleEditPartFactory());

    // zoom menu
    zoomContribution(viewer);

    // layout action
    createDiagramAction(viewer);

    // context menu
    ContextMenuProvider provider = new TadpoleERDContextMenuProvider(viewer, getActionRegistry());
    viewer.setContextMenu(provider);

    // key handler
    configureKeyHandler();

    // grid and geometry
    configureGeometry();
    configureGrid();
  }

  /** configure key handler */
  private void configureKeyHandler() {
    GraphicalViewer viewer = getGraphicalViewer();

    keyHandler = new KeyHandler();
    //		keyHandler.put(KeyStroke.getPressed('a', 0x61, SWT.COMMAND),
    //	getActionRegistry().getAction(ActionFactory.SELECT_ALL.getId()));
    keyHandler.put(
        KeyStroke.getPressed('s', 0x61, SWT.CTRL),
        getActionRegistry().getAction(ActionFactory.SAVE.getId()));

    keyHandler.put(
        KeyStroke.getPressed('z', 0x7a, SWT.CTRL),
        getActionRegistry().getAction(ActionFactory.UNDO.getId()));
    keyHandler.put(
        KeyStroke.getPressed('z', 0x7a, SWT.CTRL | SWT.SHIFT),
        getActionRegistry().getAction(ActionFactory.REDO.getId()));
    keyHandler.put(
        KeyStroke.getPressed('a', 0x61, SWT.CTRL),
        getActionRegistry().getAction(ActionFactory.SELECT_ALL.getId()));

    keyHandler.put(
        KeyStroke.getPressed(SWT.DEL, 127, 0),
        getActionRegistry().getAction(ActionFactory.DELETE.getId()));
    keyHandler.put(
        KeyStroke.getPressed('+', SWT.KEYPAD_ADD, 0),
        getActionRegistry().getAction(GEFActionConstants.ZOOM_IN));
    keyHandler.put(
        KeyStroke.getPressed('-', SWT.KEYPAD_SUBTRACT, 0),
        getActionRegistry().getAction(GEFActionConstants.ZOOM_IN));

    viewer.setProperty(
        MouseWheelHandler.KeyGenerator.getKey(SWT.NONE), MouseWheelZoomHandler.SINGLETON);
    viewer.setKeyHandler(keyHandler);
  }

  /** configure grid */
  private void configureGrid() {
    GraphicalViewer viewer = getGraphicalViewer();
    viewer.setProperty(SnapToGrid.PROPERTY_GRID_ENABLED, true);
    viewer.setProperty(SnapToGrid.PROPERTY_GRID_VISIBLE, true);

    IAction action = new ToggleGridAction(viewer);
    getActionRegistry().registerAction(action);
  }

  /** configure geometry */
  private void configureGeometry() {
    GraphicalViewer viewer = getGraphicalViewer();
    viewer.setProperty(SnapToGeometry.PROPERTY_SNAP_ENABLED, true);

    IAction action = new ToggleSnapToGeometryAction(viewer);
    getActionRegistry().registerAction(action);
  }

  private void zoomContribution(GraphicalViewer viewer) {
    double[] zoomLevels;
    List<String> zoomContributions;

    ScalableRootEditPart rootEditPart = new ScalableRootEditPart();
    viewer.setRootEditPart(rootEditPart);

    ZoomManager manager = rootEditPart.getZoomManager();
    getActionRegistry().registerAction(new ZoomInAction(manager));
    getActionRegistry().registerAction(new ZoomOutAction(manager));

    zoomLevels = new double[] {0.25, 0.5, 0.75, 1.0, 1.5, 2.0, 2.5, 3.0, 5.0, 10.0, 20.0};
    manager.setZoomLevels(zoomLevels);

    zoomContributions = new ArrayList<String>();
    //		zoomContributions.add(ZoomManager.FIT_ALL);
    //		zoomContributions.add(ZoomManager.FIT_HEIGHT);
    //		zoomContributions.add(ZoomManager.FIT_WIDTH);
    manager.setZoomLevelContributions(zoomContributions);
  }

  private void createDiagramAction(GraphicalViewer viewer) {
    ActionRegistry registry = getActionRegistry();
    AutoLayoutAction autoLayoutAction = new AutoLayoutAction(this, getGraphicalViewer());
    registry.registerAction(autoLayoutAction);
    getSelectionActions().add(autoLayoutAction.getId());

    IAction action = new TableSelectionAction(this, getGraphicalViewer());
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    ERDViewStyleAction erdStyledAction = new ERDViewStyleAction(this, getGraphicalViewer());
    registry.registerAction(erdStyledAction);
    getSelectionActions().add(ERDViewStyleAction.ID);

    ERDRefreshAction refreshAction = new ERDRefreshAction(this, getGraphicalViewer());
    registry.registerAction(refreshAction);
    getSelectionActions().add(refreshAction.getId());
  }

  @Override
  protected void createActions() {
    super.createActions();
    ActionRegistry registry = getActionRegistry();

    IAction action = new MatchWidthAction(this);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new MatchHeightAction(this);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.LEFT);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.RIGHT);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.TOP);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.BOTTOM);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.CENTER);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());

    action = new AlignmentAction((IWorkbenchPart) this, PositionConstants.MIDDLE);
    registry.registerAction(action);
    getSelectionActions().add(action.getId());
  }

  @Override
  public void init(IEditorSite site, IEditorInput input) throws PartInitException {
    super.init(site, input);
    loadDBRsource(input);
  }

  private void configureKeyboardShortcu() {
    //		getGraphicalViewer().getKeyHandler();
    //		GraphicalViewerKeyHandler keyHandler = new GraphicalViewerKeyHandler(getGraphicalViewer());
    //		keyHandler.put(KeyStroke.getPressed(SWT.F2, 0),
    // getActionRegistry().getAction(GEFActionConstants.DIRECT_EDIT));
    //		keyHandler.put(KeyStroke.getPressed(SWT.F3, 0),
    // getActionRegistry().getAction(ResizeToContentsAction.RESIZE_TO_CONTENTS_ID));
    //		getGraphicalViewer().setKeyHandler(keyHandler);
    //
    //		getGraphicalViewer().getKeyHandler();
    //		GraphicalViewerKeyHandler keyHandler = new GraphicalViewerKeyHandler(getGraphicalViewer());
    //		keyHandler.put(KeyStroke.getPressed(SWT.F3, 0),
    // getActionRegistry().getAction(GEFActionConstants.SAVE));
    ////		keyHandler.put(KeyStroke.getPressed(SWT.F3, 0),
    // getActionRegistry().getAction(ResizeToContentsAction.RESIZE_TO_CONTENTS_ID));
    //		getGraphicalViewer().setKeyHandler(keyHandler);
  }

  /**
   * resource load
   *
   * @param input
   */
  private void loadDBRsource(IEditorInput input) {
    TadpoleRDBEditorInput erdInput = (TadpoleRDBEditorInput) input;
    userDB = erdInput.getUserDBDAO();
    isAllTable = erdInput.isAllTable();

    // 신규로드 인지 기존 파일 로드 인지 검사합니다.
    if (null != erdInput.getUserDBERD()) {
      userDBErd = erdInput.getUserDBERD();

      // load resouce
      try {
        String xmlString = TadpoleSystem_UserDBResource.getResourceData(userDBErd);

        // 처음 로드 할때 ResourceSet에 instance가 등록 되어 있어야 합니다.
        /** <code>TadpolePackage.eNS_URI</code> */
        ResourceSet resourceSet = new ResourceSetImpl();
        if (resourceSet.getPackageRegistry().get("http://com.hangum.tadpole.rdb.model.ERDInfo")
            == null) {
          resourceSet
              .getPackageRegistry()
              .put("http://com.hangum.tadpole.rdb.model.ERDInfo", RdbPackage.eINSTANCE.getClass());
        }

        //
        XMLResourceImpl resource = new XMLResourceImpl();
        resource.setEncoding("UTF-8");
        resource.load(new InputSource(new StringReader(xmlString)), null);
        db = (DB) resource.getContents().get(0);

      } catch (Exception e) {
        logger.error("Load ERD Resource", e); // $NON-NLS-1$

        Status errStatus =
            new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
        ExceptionDetailsErrorDialog.openError(
            getSite().getShell(),
            "Error",
            Messages.get().TadpoleEditor_0,
            errStatus); //$NON-NLS-1$
      }

      setPartName(isAllTable ? "All " + userDBErd.getName() : userDBErd.getName());
      setTitleToolTip(userDB.getDisplay_name());
    } else {
      setPartName(isAllTable ? "All " + userDB.getDisplay_name() : userDB.getDisplay_name());
      setTitleToolTip(userDB.getDisplay_name());
    }

    // google analytic
    AnalyticCaller.track(TadpoleRDBEditor.ID, userDB.getDbms_type());
  }

  //	@Override
  //	protected PaletteRoot getPaletteRoot() {
  //		return null;
  //	}
  //
  //	@Override
  //	public void doSaveAs() {
  //		super.doSaveAs();
  //	}

  //	/**
  //	 * export images
  //	 *
  //	 * RAP currently, GC drawing is only supported on Canvas, but not on image.
  //	 */
  //	public void exportImage() {
  //		try {
  //           IFigure figure =
  // ((AbstractGraphicalEditPart)getGraphicalViewer().getRootEditPart()).getFigure();
  //           File file = new File ("/Users/hangum/Downloads/gefout.png");
  //           if (file.exists ()) {
  //               if (!MessageDialog.openQuestion(null, "prompted", "The file already exists. Want
  // to re-cover it?")) {
  //                   return;
  //               }
  //           } else {
  //        	   file.createNewFile ();
  //           }
  //
  //           if (figure instanceof Viewport) {
  //	           ((Viewport)figure).setViewLocation(0, 0);
  //           }
  //
  //           Dimension size = figure.getPreferredSize ();
  //           Image image = new Image (Display.getDefault (), size.width, size.height);
  ////           Drawable drawable = (Drawable)image;
  //
  //           GC gc = new GC(getSite().getShell().getDisplay());
  //           gc.drawImage(image, size.width, size.height);
  //           SWTGraphics graphics = new SWTGraphics(gc);
  //           figure.paint (graphics);
  //
  //           ImageLoader loader = new ImageLoader ();
  //           loader.data = new ImageData []{image.getImageData ()};
  //           loader.save("/Users/hangum/Downloads/gefout.png",
  // SWT.IMAGE_PNG);//FileFormat.FORMATS);
  //
  //           graphics.dispose();
  //           gc.dispose();
  //           image.dispose();
  //
  //        } catch (Exception e) {
  //        	e.printStackTrace();
  //	     } finally {
  //        }
  //	}
  @Override
  public void doSave(IProgressMonitor monitor) {
    //		exportImage();

    // 신규 저장이면
    if (userDBErd == null) {

      // file 이름 dialog
      ResourceSaveDialog rsDialog =
          new ResourceSaveDialog(null, null, userDB, PublicTadpoleDefine.RESOURCE_TYPE.ERD, "");
      if (rsDialog.open() == Window.OK) {

        try {
          // erd 정보 디비저장
          userDBErd =
              TadpoleSystem_UserDBResource.saveResource(
                  userDB, rsDialog.getRetResourceDao(), createResourceToString());
          userDBErd.setParent(userDB);

          // command stack 초기화
          getCommandStack().markSaveLocation();

          // title 수정
          setPartName(userDBErd.getName());

          // managerView tree refresh
          // 뒤에 시간을붙인것은 한번 저장한 db_seq는 업데이지 않는 오류를 방지하기위해...
          //
          PlatformUI.getPreferenceStore()
              .setValue(
                  PublicTadpoleDefine.SAVE_FILE,
                  ""
                      + userDBErd.getDb_seq()
                      + ":"
                      + System.currentTimeMillis()); // $NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$

        } catch (Exception e) {
          logger.error(Messages.get().TadpoleEditor_9, e);

          Status errStatus =
              new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
          ExceptionDetailsErrorDialog.openError(
              getSite().getShell(),
              "Error",
              Messages.get().TadpoleEditor_3,
              errStatus); //$NON-NLS-1$
        }
      }

      // 기존 리소스를 가지고 있었으면
    } else {

      try {
        TadpoleSystem_UserDBResource.updateResource(userDBErd, createResourceToString());
        getCommandStack().markSaveLocation();
      } catch (Exception e) {
        logger.error(Messages.get().TadpoleEditor_12, e);

        Status errStatus =
            new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
        ExceptionDetailsErrorDialog.openError(
            getSite().getShell(),
            "Error",
            Messages.get().TadpoleEditor_1,
            errStatus); //$NON-NLS-1$
      }
    }
  }

  /**
   * model to string
   *
   * @return
   * @throws Exception
   */
  private String createResourceToString() throws Exception {
    XMLResourceImpl resource = new XMLResourceImpl();
    resource.setEncoding("UTF-8");
    resource.getContents().add(db);

    XMLProcessor processor = new XMLProcessor();
    return processor.saveToString(resource, null);
  }

  @Override
  public void commandStackChanged(EventObject event) {
    firePropertyChange(PROP_DIRTY);
    super.commandStackChanged(event);
  }

  @Override
  public Object getAdapter(Class type) {
    if (type == ZoomManager.class) {
      return ((ScalableRootEditPart) getGraphicalViewer().getRootEditPart()).getZoomManager();
    }

    //		if(type == IContentOutlinePage.class) {
    //			return new OutlinePage();
    //		}

    return super.getAdapter(type);
  }

  public UserDBResourceDAO getUserDBErd() {
    return userDBErd;
  }

  /**
   * outline page
   *
   * @author hangum
   */
  protected class OutlinePage extends ContentOutlinePage {
    private SashForm sash;
    //		private ScrollableThumbnail thumbnail;
    //		private DisposeListener disposeListener;

    public OutlinePage() {
      super(new TreeViewer());
    }

    @Override
    public void createControl(Composite parent) {
      sash = new SashForm(parent, SWT.VERTICAL);

      getViewer().createControl(sash);
      getViewer().setEditDomain(getEditDomain());
      getViewer().setEditPartFactory(new TadpoleTreeEditPartFactory());
      getViewer().setContents(db);
      getSelectionSynchronizer().addViewer(getViewer());

      //			Canvas canvas = new Canvas(sash, SWT.BORDER);
      //			LightweightSystem lws = new LightweightSystem(canvas);
      //			RootEditPart rep = getGraphicalViewer().getRootEditPart();
      //			if (rep instanceof ScalableFreeformRootEditPart) {
      //				ScalableFreeformRootEditPart root = (ScalableFreeformRootEditPart) rep;
      //				thumbnail = new ScrollableThumbnail((Viewport) root.getFigure());
      //				thumbnail.setBorder(new MarginBorder(3));
      //				thumbnail.setSource(root.getLayer(LayerConstants.PRINTABLE_LAYERS));
      //				lws.setContents(thumbnail);
      //				disposeListener = new DisposeListener() {
      //					public void widgetDisposed(DisposeEvent e) {
      //						if (thumbnail != null) {
      //							thumbnail.deactivate();
      //							thumbnail = null;
      //						}
      //					}
      //				};
      //				getGraphicalViewer().getControl().addDisposeListener(disposeListener);
      //			}

    }

    @Override
    public void init(IPageSite pageSite) {
      super.init(pageSite);

      IActionBars bars = getSite().getActionBars();
      bars.setGlobalActionHandler(
          ActionFactory.UNDO.getId(), getActionRegistry().getAction(ActionFactory.UNDO.getId()));
      bars.setGlobalActionHandler(
          ActionFactory.REDO.getId(), getActionRegistry().getAction(ActionFactory.REDO.getId()));
      bars.setGlobalActionHandler(
          ActionFactory.DELETE.getId(),
          getActionRegistry().getAction(ActionFactory.DELETE.getId()));
      bars.updateActionBars();

      getViewer().setKeyHandler(keyHandler);
    }

    @Override
    public Control getControl() {
      return sash;
    }

    @Override
    public void dispose() {
      getSelectionSynchronizer().removeViewer(getViewer());
      //			if(getGraphicalViewer().getControl() != null &&
      // !getGraphicalViewer().getControl().isDisposed())
      //				getGraphicalViewer().getControl().removeDisposeListener(disposeListener);

      super.dispose();
    }
  }

  public DB getDb() {
    return db;
  }

  /** @return the userDB */
  public UserDBDAO getUserDB() {
    return userDB;
  }
}