archive/net.sourceforge.phpeclipse.quantum.sql/src/com/quantum/view/bookmark/EntityNode.java

   1 package com.quantum.view.bookmark;
   2
   3 import java.sql.SQLException;
   4
   5 import com.quantum.model.Column;
   6 import com.quantum.model.Entity;
   7 import com.quantum.model.EntityHolder;
   8 import com.quantum.model.NotConnectedException;
   9 import com.quantum.model.SchemaHolder;
  10 import com.quantum.model.Table;
  11 import com.quantum.model.View;
  12
  13 /**
  14  * @author BC
  15  */
  16 public class EntityNode extends TreeNode implements EntityHolder {
  17
  18     private Entity entity;
  19     private boolean longFormName;
  20
  21     public EntityNode(TreeNode parent, Entity entity) {
  22         this(parent, entity, false);
  23     }
  24     public EntityNode(TreeNode parent, Entity entity, boolean longFormName) {
  25         super(parent);
  26         this.entity = entity;
  27         this.longFormName = longFormName;
  28
  29         if (parent instanceof SchemaHolder) {
  30             SchemaHolder schemaHolder = (SchemaHolder) parent;
  31             if (!schemaHolder.getSchema().getDisplayName().equals(entity.getSchema())) {
  32                 this.longFormName = true;
  33             }
  34         }
  35     }
  36
  37     public Object[] getChildren() throws NotConnectedException, SQLException {
  38         initializeChildren();
  39         if (this.children.size() > 0) {
  40             return (ColumnNode[]) this.children.toArray(new ColumnNode[this.children.size()]);
  41         } else {
  42             return BookmarkListNode.EMPTY_ARRAY;
  43         }
  44     }
  45
  46     protected synchronized void initializeChildren() throws NotConnectedException, SQLException {
  47         this.children.clear();
  48         Column[] columns = this.entity.getColumns();
  49         for (int i = 0, length = (columns == null) ? 0 : columns.length;
  50             i < length;
  51             i++) {
  52             this.children.add(new ColumnNode(this, columns[i]));
  53         }
  54         // TODO: fire property change event
  55     }
  56
  57     public boolean hasChildren() {
  58         if (!isSequence()) {
  59             return true;
  60         } else {
  61             return false;
  62         }
  63     }
  64
  65     public Entity getEntity() {
  66         return this.entity;
  67     }
  68
  69     public String getName() {
  70         return this.entity.getName();
  71     }
  72
  73     public String getLabelName() {
  74         return (this.longFormName) ? this.entity.getQualifiedName() : this.entity.getName();
  75     }
  76
  77     public boolean isTable() {
  78         return Entity.TABLE_TYPE.equals(this.entity.getType());
  79     }
  80
  81     public boolean isView() {
  82         return Entity.VIEW_TYPE.equals(this.entity.getType());
  83     }
  84
  85     public boolean isSequence() {
  86         return Entity.SEQUENCE_TYPE.equals(this.entity.getType());
  87     }
  88
  89     protected String getImageName() {
  90         if (isSequence()) {
  91             return "sequence.gif";
  92         } else if (isView()) {
  93             return "view.gif";
  94         } else {
  95             return (this.entity.exists() == null || this.entity.exists().booleanValue())
  96                 ? "bigtable.gif" : "missingtable.gif";
  97         }
  98     }
  99
 100     public String getLabelDecorations(LabelDecorationInstructions instructions) {
 101         String decoration = null;
 102         if (instructions.isSizeVisible()) {
 103             Integer size = getSize();
 104             if (size != null) {
 105                 decoration = ((decoration == null) ? "" : decoration)
 106                     + "[" + size + "]";
 107             }
 108         }
 109         return decoration;
 110     }
 111
 112     private Integer getSize() {
 113         if (isTable()) {
 114             return ((Table) this.entity).getSize();
 115         } else if (isView()) {
 116             return ((View) this.entity).getSize();
 117         } else {
 118             return null;
 119         }
 120     }
 121
 122     public int compareTo(Object o) {
 123         if (o instanceof EntityNode) {
 124             EntityNode that = (EntityNode) o;
 125             return this.entity.getQualifiedName().compareTo(
 126                 that.entity.getQualifiedName());
 127         } else {
 128             return super.compareTo(o);
 129         }
 130     }
 131 }