protected boolean doSetIcon(
DefaultMutableTreeNode node, @Nullable String path, Component component) {
if (StringUtil.isNotEmpty(path) && !new File(path).isFile()) {
Messages.showErrorDialog(
component,
IdeBundle.message("error.file.not.found.message", path),
IdeBundle.message("title.choose.action.icon"));
return false;
}
String actionId = getActionId(node);
if (actionId == null) return false;
final AnAction action = ActionManager.getInstance().getAction(actionId);
if (action != null && action.getTemplatePresentation() != null) {
if (StringUtil.isNotEmpty(path)) {
Image image = null;
try {
image =
ImageLoader.loadFromStream(
VfsUtil.convertToURL(VfsUtil.pathToUrl(path.replace(File.separatorChar, '/')))
.openStream());
} catch (IOException e) {
LOG.debug(e);
}
Icon icon = new File(path).exists() ? IconLoader.getIcon(image) : null;
if (icon != null) {
if (icon.getIconWidth() > EmptyIcon.ICON_18.getIconWidth()
|| icon.getIconHeight() > EmptyIcon.ICON_18.getIconHeight()) {
Messages.showErrorDialog(
component,
IdeBundle.message("custom.icon.validation.message"),
IdeBundle.message("title.choose.action.icon"));
return false;
}
node.setUserObject(Pair.create(actionId, icon));
mySelectedSchema.addIconCustomization(actionId, path);
}
} else {
node.setUserObject(Pair.create(actionId, null));
mySelectedSchema.removeIconCustomization(actionId);
final DefaultMutableTreeNode nodeOnToolbar = findNodeOnToolbar(actionId);
if (nodeOnToolbar != null) {
editToolbarIcon(actionId, nodeOnToolbar);
node.setUserObject(nodeOnToolbar.getUserObject());
}
}
return true;
}
return false;
}
/** This gets called whenever one of the ConfigElements we are modeling has changed its name. */
public void nameChanged(ConfigElementEvent evt) {
ConfigElement src = (ConfigElement) evt.getSource();
DefaultMutableTreeNode root = (DefaultMutableTreeNode) getRoot();
DefaultMutableTreeNode name_node = (DefaultMutableTreeNode) root.getChildAt(0);
name_node.setUserObject(src.getName());
fireTreeNodesChanged(
this, new Object[] {getPathToRoot(root)}, new int[] {0}, new Object[] {name_node});
}
private void editToolbarIcon(String actionId, DefaultMutableTreeNode node) {
final AnAction anAction = ActionManager.getInstance().getAction(actionId);
if (isToolbarAction(node)
&& anAction.getTemplatePresentation() != null
&& anAction.getTemplatePresentation().getIcon() == null) {
final int exitCode =
Messages.showOkCancelDialog(
IdeBundle.message("error.adding.action.without.icon.to.toolbar"),
IdeBundle.message("title.unable.to.add.action.without.icon.to.toolbar"),
Messages.getInformationIcon());
if (exitCode == Messages.OK) {
mySelectedSchema.addIconCustomization(actionId, null);
anAction.getTemplatePresentation().setIcon(AllIcons.Toolbar.Unknown);
anAction.setDefaultIcon(false);
node.setUserObject(Pair.create(actionId, AllIcons.Toolbar.Unknown));
myActionsTree.repaint();
setCustomizationSchemaForCurrentProjects();
}
}
}
/** Sets the value for the given node at the given column to the given value. */
public void setValueAt(Object value, Object node, int col) {
DefaultMutableTreeNode tree_node = (DefaultMutableTreeNode) node;
Object node_value = tree_node.getUserObject();
switch (col) {
// Name (not supported)
case 0:
break;
// Value
case 1:
DefaultMutableTreeNode parent_node = (DefaultMutableTreeNode) tree_node.getParent();
// First child of root is always the element name
if (parent_node == getRoot() && parent_node.getIndex((TreeNode) node) == 0) {
ConfigElement elt = (ConfigElement) parent_node.getUserObject();
elt.setName((String) value);
tree_node.setUserObject(value);
} else if (node_value instanceof PropertyDefinition) {
// Hey, we're editing a property definition. If it's type is not a
// configuration element, we're probably editing a summary list of
// the valuesof the children.
if (((PropertyDefinition) node_value).getType() != ConfigElement.class) {
ConfigElement elt = (ConfigElement) parent_node.getUserObject();
PropertyDefinition prop_def = (PropertyDefinition) node_value;
StringTokenizer tokenizer = new StringTokenizer((String) value, ", ");
int idx = 0;
while (tokenizer.hasMoreTokens()) {
String token = tokenizer.nextToken();
// Make sure we don't overrun the property values
if ((idx >= prop_def.getPropertyValueDefinitionCount()) && (!prop_def.isVariable())) {
break;
}
// Convert the value to the appropriate type
Object new_value = null;
Class type = prop_def.getType();
if (type == Boolean.class) {
new_value = new Boolean(token);
} else if (type == Integer.class) {
new_value = new Integer(token);
} else if (type == Float.class) {
new_value = new Float(token);
} else if (type == String.class) {
new_value = new String(token);
} else if (type == ConfigElementPointer.class) {
new_value = new ConfigElementPointer(token);
}
setProperty(new_value, elt, prop_def.getToken(), idx);
// Get the node for the current property value and update it
if (idx < tree_node.getChildCount()) {
DefaultMutableTreeNode child_node =
(DefaultMutableTreeNode) tree_node.getChildAt(idx);
child_node.setUserObject(new_value);
} else {
// Insert the new property
DefaultMutableTreeNode new_node = new DefaultMutableTreeNode(new_value);
tree_node.add(new_node);
fireTreeNodesInserted(
this,
new Object[] {getPathToRoot(tree_node)},
new int[] {tree_node.getIndex(new_node)},
new Object[] {new_node});
}
++idx;
}
}
} else {
// Parent is a ConfigElement ... must be a single-valued property
if (parent_node.getUserObject() instanceof ConfigElement) {
ConfigElement elt = (ConfigElement) parent_node.getUserObject();
int desc_idx = parent_node.getIndex(tree_node);
// If the parent is the root, take into account the extra name
// and type nodes
if (parent_node == getRoot()) {
desc_idx -= 2;
}
PropertyDefinition prop_def =
(PropertyDefinition) elt.getDefinition().getPropertyDefinitions().get(desc_idx);
setProperty(value, elt, prop_def.getToken(), 0);
tree_node.setUserObject(value);
} else {
// Parent must be a PropertyDefinition
PropertyDefinition prop_def = (PropertyDefinition) parent_node.getUserObject();
int value_idx = parent_node.getIndex(tree_node);
DefaultMutableTreeNode elt_node = (DefaultMutableTreeNode) parent_node.getParent();
ConfigElement elt = (ConfigElement) elt_node.getUserObject();
setProperty(value, elt, prop_def.getToken(), value_idx);
tree_node.setUserObject(value);
}
}
fireTreeNodesChanged(
this,
new Object[] {getPathToRoot(parent_node)},
new int[] {parent_node.getIndex(tree_node)},
new Object[] {tree_node});
break;
default:
throw new IllegalArgumentException("Invalid column: " + col);
}
}
/* Clear all existing nodes from the tree model and rebuild from scratch.
*/
protected void refreshTree() {
DefaultMutableTreeNode propertiesNode;
DefaultMutableTreeNode leaf;
// First clear the existing tree by simply enumerating
// over the root node's children and removing them one by one.
while (treeModel.getChildCount(rootNode) > 0) {
DefaultMutableTreeNode child = (DefaultMutableTreeNode) treeModel.getChild(rootNode, 0);
treeModel.removeNodeFromParent(child);
child.removeAllChildren();
child.removeFromParent();
}
treeModel.nodeStructureChanged(rootNode);
treeModel.reload();
tScrollPane.repaint();
// Now rebuild the tree below its root
try {
// Start by naming the root node from its URL:
rootNode.setUserObject(dMeta.getURL());
// get metadata about user tables by building a vector of table names
String usertables[] = {"TABLE", "GLOBAL TEMPORARY", "VIEW"};
ResultSet result = dMeta.getTables(null, null, null, usertables);
Vector tables = new Vector();
// sqlbob@users Added remarks.
Vector remarks = new Vector();
while (result.next()) {
tables.addElement(result.getString(3));
remarks.addElement(result.getString(5));
}
result.close();
// For each table, build a tree node with interesting info
for (int i = 0; i < tables.size(); i++) {
String name = (String) tables.elementAt(i);
DefaultMutableTreeNode tableNode = makeNode(name, rootNode);
ResultSet col = dMeta.getColumns(null, null, name, null);
// sqlbob@users Added remarks.
String remark = (String) remarks.elementAt(i);
if ((remark != null) && !remark.trim().equals("")) {
makeNode(remark, tableNode);
}
// With a child for each column containing pertinent attributes
while (col.next()) {
String c = col.getString(4);
DefaultMutableTreeNode columnNode = makeNode(c, tableNode);
String type = col.getString(6);
makeNode("Type: " + type, columnNode);
boolean nullable = col.getInt(11) != DatabaseMetaData.columnNoNulls;
makeNode("Nullable: " + nullable, columnNode);
}
col.close();
DefaultMutableTreeNode indexesNode = makeNode("Indices", tableNode);
ResultSet ind = dMeta.getIndexInfo(null, null, name, false, false);
String oldiname = null;
// A child node to contain each index - and its attributes
while (ind.next()) {
DefaultMutableTreeNode indexNode = null;
boolean nonunique = ind.getBoolean(4);
String iname = ind.getString(6);
if ((oldiname == null || !oldiname.equals(iname))) {
indexNode = makeNode(iname, indexesNode);
makeNode("Unique: " + !nonunique, indexNode);
oldiname = iname;
}
// And the ordered column list for index components
makeNode(ind.getString(9), indexNode);
}
ind.close();
}
// Finally - a little additional metadata on this connection
propertiesNode = makeNode("Properties", rootNode);
makeNode("User: "******"ReadOnly: " + cConn.isReadOnly(), propertiesNode);
makeNode("AutoCommit: " + cConn.getAutoCommit(), propertiesNode);
makeNode("Driver: " + dMeta.getDriverName(), propertiesNode);
makeNode("Product: " + dMeta.getDatabaseProductName(), propertiesNode);
makeNode("Version: " + dMeta.getDatabaseProductVersion(), propertiesNode);
} catch (SQLException se) {
propertiesNode = makeNode("Error getting metadata:", rootNode);
makeNode(se.getMessage(), propertiesNode);
makeNode(se.getSQLState(), propertiesNode);
}
treeModel.nodeStructureChanged(rootNode);
treeModel.reload();
tScrollPane.repaint();
}
