1 files changed, 149 insertions, 0 deletions
diff --git a/src/plugins/db.core/src/com/motorolamobility/studio/android/db/core/ui/TableNode.java b/src/plugins/db.core/src/com/motorolamobility/studio/android/db/core/ui/TableNode.java
new file mode 100644
index 0000000..5219268
--- /dev/null
+++ b/src/plugins/db.core/src/com/motorolamobility/studio/android/db/core/ui/TableNode.java
@@ -0,0 +1,149 @@
+/*
+ * Copyright (C) 2012 The Android Open Source Project
+ *
+ * Licensed 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 com.motorolamobility.studio.android.db.core.ui;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.eclipse.core.runtime.IStatus;
+import org.eclipse.core.runtime.Status;
+import org.eclipse.datatools.modelbase.sql.tables.Column;
+import org.eclipse.datatools.modelbase.sql.tables.Table;
+import org.eclipse.datatools.sqltools.data.internal.ui.editor.TableDataEditorInput;
+import org.eclipse.datatools.sqltools.data.internal.ui.extract.ExtractDataWizard;
+import org.eclipse.datatools.sqltools.data.internal.ui.load.LoadDataWizard;
+import org.eclipse.emf.common.util.EList;
+import org.eclipse.jface.resource.ImageDescriptor;
+import org.eclipse.jface.wizard.WizardDialog;
+import org.eclipse.swt.widgets.Display;
+import org.eclipse.ui.IWorkbenchPage;
+import org.eclipse.ui.PartInitException;
+import org.eclipse.ui.PlatformUI;
+
+import com.motorolamobility.studio.android.db.core.DbCoreActivator;
+import com.motorolamobility.studio.android.db.core.i18n.DbCoreNLS;
+import com.motorolamobility.studio.android.db.core.model.DbModel;
+
+public class TableNode extends AbstractTreeNode implements ITableNode
+{
+
+    private final Table table;
+
+    private final DbModel model;
+
+    /**
+     * @param table
+     * @param parent
+     */
+    public TableNode(Table table, DbModel dbModel, IDbNode parent)
+    {
+        super(table.getName(), table.getName(), parent);
+        this.table = table;
+        this.model = dbModel;
+    }
+
+    public IStatus browseTableContents()
+    {
+        IStatus browseTableContentsStatus =
+                new Status(IStatus.OK, DbCoreActivator.PLUGIN_ID,
+                        DbCoreNLS.TableNode_BrowsingTableContentsSuccessStatus);
+        final IWorkbenchPage workbenchPage =
+                PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();
+
+        try
+        {
+            workbenchPage.openEditor(new TableDataEditorInput(table),
+                    "org.eclipse.datatools.sqltools.data.internal.ui.editor.tableDataEditor"); //$NON-NLS-1$
+
+        }
+        catch (PartInitException e)
+        {
+            //Display error message!
+            browseTableContentsStatus =
+                    new Status(IStatus.ERROR, DbCoreActivator.PLUGIN_ID,
+                            DbCoreNLS.TableNode_BrowsingTableContentsErrorStatus, e);
+        }
+        return browseTableContentsStatus;
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.AbstractTreeNode#refresh()
+     */
+    @SuppressWarnings("unchecked")
+    @Override
+    public void refresh()
+    {
+        clear();
+
+        EList<Column> columns = table.getColumns();
+        List<ITreeNode> columnNodes = new ArrayList<ITreeNode>(columns.size());
+        for (Column column : columns)
+        {
+            ColumnNode columnNode = new ColumnNode(column, model, this);
+            columnNodes.add(columnNode);
+        }
+        putChildren(columnNodes);
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.AbstractTreeNode#isLeaf()
+     */
+    @Override
+    public boolean isLeaf()
+    {
+        return false;
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.AbstractTreeNode#getIcon()
+     */
+    @Override
+    public ImageDescriptor getIcon()
+    {
+        return getSpecificIcon("org.eclipse.datatools.connectivity.sqm.core.ui", //$NON-NLS-1$
+                "icons/table.gif"); //$NON-NLS-1$
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.ITableNode#sampleDbContents()
+     */
+    public void sampleDbContents()
+    {
+        model.sampleContents(table);
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.ITableNode#extractData()
+     */
+    public void extractData()
+    {
+        ExtractDataWizard wiz = new ExtractDataWizard(table);
+        WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wiz);
+        dialog.create();
+        dialog.open();
+    }
+
+    /* (non-Javadoc)
+     * @see com.motorolamobility.studio.android.db.core.ui.ITableNode#loadData()
+     */
+    public void loadData()
+    {
+        LoadDataWizard wiz = new LoadDataWizard(table);
+        WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wiz);
+        dialog.create();
+        dialog.open();
+    }
+}