Java tutorial
/***************************************************************** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. ****************************************************************/ package org.apache.cayenne.modeler.util; import org.apache.cayenne.map.DbAttribute; import org.apache.cayenne.map.DbEntity; import org.apache.cayenne.map.DbRelationship; import org.apache.cayenne.map.Entity; import org.apache.cayenne.map.ObjAttribute; import org.apache.cayenne.map.ObjEntity; import org.apache.cayenne.modeler.editor.ObjAttributeTableModel; import org.apache.cayenne.util.CayenneMapEntry; import org.apache.commons.lang.StringUtils; import javax.swing.DefaultComboBoxModel; import javax.swing.JLabel; import javax.swing.JTable; import javax.swing.text.JTextComponent; import java.awt.Component; import java.util.ArrayList; import java.util.Collection; import java.util.Iterator; import java.util.List; import java.util.regex.Pattern; public class DbAttributePathComboBoxEditor extends PathChooserComboBoxCellEditor { private static final int DB_ATTRIBUTE_PATH_COLUMN = 3; private String savePath; private ObjAttributeTableModel model; @Override public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected, int row, int column) { this.model = (ObjAttributeTableModel) table.getModel(); this.row = row; treeModel = createTreeModelForComboBox(row); if (treeModel == null) { return new JLabel("You should select table for this ObjectEntity"); } initializeCombo(model, row, table); String dbAttributePath = ((JTextComponent) (comboBoxPathChooser).getEditor().getEditorComponent()) .getText(); previousEmbeddedLevel = StringUtils.countMatches(dbAttributePath, "."); return comboBoxPathChooser; } @Override public Object getCellEditorValue() { return model.getValueAt(row, DB_ATTRIBUTE_PATH_COLUMN); } @Override protected void initializeCombo(CayenneTableModel model, int row, final JTable table) { super.initializeCombo(model, row, table); comboBoxPathChooser.setSelectedItem( ((ObjAttributeTableModel) model).getAttribute(row).getValue().getDbAttributePath()); savePath = this.model.getAttribute(row).getValue().getDbAttributePath(); } @Override protected Object getCurrentNodeToInitializeCombo(CayenneTableModel model, int row) { return getCurrentNode(getPathToInitializeCombo(model, row)); } @Override protected String getPathToInitializeCombo(CayenneTableModel model, int row) { String pathString = ((ObjAttributeTableModel) model).getAttribute(row).getValue().getDbAttributePath(); if (pathString == null) { return ""; } String[] pathStrings = pathString.split(Pattern.quote(".")); String lastStringInPath = pathStrings[pathStrings.length - 1]; return pathString.replaceAll(lastStringInPath + '$', ""); } @Override protected void enterPressed(JTable table) { String dbAttributePath = ((JTextComponent) comboBoxPathChooser.getEditor().getEditorComponent()).getText(); Object currentNode = getCurrentNode(dbAttributePath); String[] pathStrings = dbAttributePath.split(Pattern.quote(".")); String lastStringInPath = pathStrings[pathStrings.length - 1]; if (ModelerUtil.getObjectName(currentNode).equals(lastStringInPath) && currentNode instanceof DbAttribute) { // in this case choose is made.. we save data if (table.getCellEditor() != null) { table.getCellEditor().stopCellEditing(); if (dbAttributePath.equals(savePath)) { return; } model.setUpdatedValueAt(dbAttributePath, row, DB_ATTRIBUTE_PATH_COLUMN); model.getAttribute(row).getValue().setDbAttributePath(dbAttributePath); } } else if (ModelerUtil.getObjectName(currentNode).equals(lastStringInPath) && currentNode instanceof DbRelationship) { // in this case we add dot to pathString (if it is missing) and show variants for currentNode if (dbAttributePath.charAt(dbAttributePath.length() - 1) != '.') { dbAttributePath = dbAttributePath + '.'; previousEmbeddedLevel = StringUtils.countMatches(dbAttributePath, "."); ((JTextComponent) (comboBoxPathChooser).getEditor().getEditorComponent()).setText(dbAttributePath); } List<String> currentNodeChildren = new ArrayList<>(); currentNodeChildren.addAll(getChildren(getCurrentNode(dbAttributePath), dbAttributePath)); comboBoxPathChooser.setModel(new DefaultComboBoxModel(currentNodeChildren.toArray())); comboBoxPathChooser.setSelectedItem(dbAttributePath); comboBoxPathChooser.showPopup(); comboBoxPathChooser.setPopupVisible(true); } } @Override protected EntityTreeModel createTreeModelForComboBox(int attributeIndexInTable) { ObjAttribute attribute = model.getAttribute(attributeIndexInTable).getValue(); Entity firstEntity = null; if (attribute.getDbAttribute() == null) { if (attribute.getParent() instanceof ObjEntity) { DbEntity dbEnt = ((ObjEntity) attribute.getParent()).getDbEntity(); if (dbEnt != null) { Collection<DbAttribute> attributes = dbEnt.getAttributes(); Collection<DbRelationship> rel = dbEnt.getRelationships(); if (!attributes.isEmpty()) { Iterator<DbAttribute> iterator = attributes.iterator(); firstEntity = iterator.next().getEntity(); } else if (!rel.isEmpty()) { Iterator<DbRelationship> iterator = rel.iterator(); firstEntity = iterator.next().getSourceEntity(); } } } } else { firstEntity = getFirstEntity(attribute); } if (firstEntity != null) { EntityTreeModel treeModel = new EntityTreeModel(firstEntity); treeModel.setFilter(new EntityTreeAttributeRelationshipFilter()); return treeModel; } return null; } private Entity getFirstEntity(ObjAttribute attribute) { Iterator<CayenneMapEntry> it = attribute.getDbPathIterator(); Entity firstEnt = attribute.getDbAttribute().getEntity(); boolean setEnt = false; while (it.hasNext()) { Object ob = it.next(); if (ob instanceof DbRelationship) { if (!setEnt) { firstEnt = ((DbRelationship) ob).getSourceEntity(); setEnt = true; } } else if (ob instanceof DbAttribute) { if (!setEnt) { firstEnt = ((DbAttribute) ob).getEntity(); } } } return firstEnt; } }