plugins/ui-designer/src/com/intellij/uiDesigner/snapShooter/SnapShotTreeModel.java

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * 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.intellij.uiDesigner.snapShooter;

import com.intellij.openapi.ui.Messages;

import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;
import java.io.IOException;

/**
 * @author yole
 */
public class SnapShotTreeModel implements TreeModel {
  private final SnapShotClient myClient;
  private final SnapShotRemoteComponent myRoot;

  public SnapShotTreeModel(final SnapShotClient client) {
    myClient = client;
    myRoot = new SnapShotRemoteComponent(0, Object.class.getName(), "", "Root");
  }

  public Object getRoot() {
    return myRoot;
  }

  public Object getChild(Object parent, int index) {
    SnapShotRemoteComponent component = (SnapShotRemoteComponent) parent;
    return checkGetChildren(component) [index];
  }

  private SnapShotRemoteComponent[] checkGetChildren(final SnapShotRemoteComponent component) {
    if (component.getChildren() == null) {
      try {
        component.setChildren(myClient.listChildren(component.getId()));
      }
      catch (IOException e) {
        reportDisconnection(myClient);
        return new SnapShotRemoteComponent[0];
      }
    }
    return component.getChildren();
  }

  private static void reportDisconnection(final SnapShotClient client) {
    Messages.showMessageDialog("Disconnected from remote application", "Create Form Snapshot",
                               Messages.getErrorIcon());
    client.setDisconnected();
  }

  public int getChildCount(Object parent) {
    SnapShotRemoteComponent component = (SnapShotRemoteComponent) parent;
    return checkGetChildren(component).length;
  }

  public boolean isLeaf(Object node) {
    SnapShotRemoteComponent component = (SnapShotRemoteComponent) node;
    return checkGetChildren(component).length == 0;
  }

  public void valueForPathChanged(TreePath path, Object newValue) {
  }

  public int getIndexOfChild(Object parent, Object child) {
    SnapShotRemoteComponent component = (SnapShotRemoteComponent) parent;
    final SnapShotRemoteComponent[] snapShotRemoteComponents = checkGetChildren(component);
    for(int i=0; i<snapShotRemoteComponents.length; i++) {
      if (snapShotRemoteComponents [i] == child) {
        return i;
      }
    }
    return -1;
  }

  public void addTreeModelListener(TreeModelListener l) {
  }

  public void removeTreeModelListener(TreeModelListener l) {
  }
}