ApplicationNode.java

/*
* 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.devices.model;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.osgi.util.NLS;
import org.eclipse.ui.plugin.AbstractUIPlugin;

import com.motorola.studio.android.adt.DDMSUtils;
import com.motorolamobility.studio.android.db.core.CanRefreshStatus;
import com.motorolamobility.studio.android.db.core.exception.MotodevDbException;
import com.motorolamobility.studio.android.db.core.model.TableModel;
import com.motorolamobility.studio.android.db.core.ui.AbstractTreeNode;
import com.motorolamobility.studio.android.db.core.ui.IDbNode;
import com.motorolamobility.studio.android.db.core.ui.ITreeNode;
import com.motorolamobility.studio.android.db.core.ui.action.IDbCreatorNode;
import com.motorolamobility.studio.android.db.devices.DbDevicesPlugin;
import com.motorolamobility.studio.android.db.devices.i18n.DbDevicesNLS;
import com.motorolamobility.studio.android.db.devices.utils.DeviceDbUtils;

/**
 * This class represents a tree node for an application installed on a device.
 */
public class ApplicationNode extends AbstractTreeNode implements IDbCreatorNode
{

    private String appName;

    private String serialNumber;

    @SuppressWarnings("unused")
    private ApplicationNode()
    {
        //Forcing user to use a proper constructor
    }

    /**
     * Creates a new Application node for the app 
     * @param appName the application name
     * @param parent this node parent
     */
    public ApplicationNode(String appName, DeviceNode parent)
    {
        super(parent);
        serialNumber = parent.getSerialNumber();
        this.appName = appName;
        setId(appName);
        setName(appName);
        ImageDescriptor icon =
                AbstractUIPlugin.imageDescriptorFromPlugin("com.android.ide.eclipse.adt", //$NON-NLS-1$
                        "icons/android.png"); //$NON-NLS-1$
        setIcon(icon);
    }

    /* (non-Javadoc)
     * @see com.motorolamobility.studio.android.db.core.ui.AbstractTreeNode#canRefresh()
     */
    @Override
    public IStatus canRefresh()
    {
        IStatus status = null;
        boolean hasConnectedDb = false;

        //Search for connected dbs
        for (ITreeNode treeNode : getChildren())
        {
            if (treeNode instanceof DeviceDbNode)
            {
                DeviceDbNode dbNode = (DeviceDbNode) treeNode;
                hasConnectedDb = dbNode.isConnected();
                break;
            }
        }

        if (hasConnectedDb)
        {
            status =
                    new CanRefreshStatus(CanRefreshStatus.ASK_USER, DbDevicesPlugin.PLUGIN_ID,
                            NLS.bind(
                            DbDevicesNLS.ApplicationNode_ConnectedDbs_Refresh_Message, getName()));
        }

        return status != null ? status : Status.OK_STATUS;
    }

    /* (non-Javadoc)
     * @see com.motorolamobility.studio.andkroid.db.core.ui.AbstractTreeNode#refresh()
     */
    @Override
    public void refresh()
    {
        clear();
        try
        {
            List<String> applicationDatabases =
                    DDMSUtils.getApplicationDatabases(serialNumber, appName);

            List<ITreeNode> dbNodes = new ArrayList<ITreeNode>(applicationDatabases.size());

            for (String dbLocation : applicationDatabases)
            {
                IPath dbPath = DeviceDbUtils.getRemoteDbPath(appName, dbLocation);
                DeviceDbNode deviceDbNode = new DeviceDbNode(dbPath, serialNumber, this);
                dbNodes.add(deviceDbNode);
            }

            putChildren(dbNodes);
            setNodeStatus(Status.OK_STATUS);
        }
        catch (IOException e)
        {
            setNodeStatus(new Status(IStatus.ERROR, DbDevicesPlugin.PLUGIN_ID, e.getMessage()));
        }
    }

    /* (non-Javadoc)
     * @see com.motorolamobility.studio.android.db.core.ui.action.IDbCreatorNode#createDb(java.lang.String)
     */
    public IStatus createDb(String dbName)
    {
        return createDb(dbName, null);
    }

    /* (non-Javadoc)
     * @see com.motorolamobility.studio.android.db.core.ui.action.IDbCreatorNode#createDb(java.lang.String, java.util.List)
     */
    public IStatus createDb(String dbName, List<TableModel> tables)
    {
        IStatus status = null;

        IPath remoteDbPath = DeviceDbUtils.getRemoteDbPath(appName, dbName);
        try
        {
            DeviceDbNode dbNode = new DeviceDbNode(remoteDbPath, serialNumber, this, true);
            if (tables != null)
            {
                dbNode.createTables(tables);
            }
            putChild(dbNode);
        }
        catch (MotodevDbException e)
        {
            status =
                    new Status(IStatus.ERROR, DbDevicesPlugin.PLUGIN_ID, NLS.bind(
                            DbDevicesNLS.ApplicationNode_Could_Not_Create_Database,
                            remoteDbPath.toString(), serialNumber));
        }

        return status != null ? status : Status.OK_STATUS;
    }

    /* (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.action.IDbCreatorNode#deleteDb(com.motorolamobility.studio.android.db.core.ui.IDbNode)
     */
    public IStatus deleteDb(IDbNode dbNode)
    {
        IStatus status = dbNode.deleteDb();

        if (status.isOK())
        {
            removeChild(dbNode);
        }

        return status;
    }

}