src/org/jetbrains/tfsIntegration/ui/checkoutwizard/ChooseLocalAndServerPathsStep.java

/*
 * Copyright 2000-2008 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 org.jetbrains.tfsIntegration.ui.checkoutwizard;

import com.intellij.ide.wizard.CommitStepException;
import com.intellij.openapi.application.WriteAction;
import com.intellij.openapi.util.Disposer;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.tfsIntegration.core.TFSBundle;

import javax.swing.*;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;

public class ChooseLocalAndServerPathsStep extends CheckoutWizardStep {

  public static final Object ID = new Object();

  private final LocalAndServerPathsForm myPathsForm = new LocalAndServerPathsForm();

  public ChooseLocalAndServerPathsStep(final CheckoutWizardModel model) {
    super("Choose Source and Destination Paths", model);
    Disposer.register(this, myPathsForm);

    myPathsForm.addListener(new ChangeListener() {
      @Override
      public void stateChanged(ChangeEvent e) {
        updateMessage();
        fireStateChanged();
      }
    });
  }

  @Override
  @NotNull
  public Object getStepId() {
    return ID;
  }

  @Override
  @Nullable
  public Object getNextStepId() {
    return SummaryStep.ID;
  }

  @Override
  @Nullable
  public Object getPreviousStepId() {
    return ChooseWorkspaceStep.ID;
  }

  @Override
  public boolean isComplete() {
    if (validateLocalPath(myPathsForm.getLocalPath()) != null) {
      return false;
    }
    if (validateServerPath(myPathsForm.getServerPath()) != null) {
      return false;
    }
    return true;
  }

  @Override
  public JComponent getComponent() {
    return myPathsForm.getContentPanel();
  }

  @Override
  public void _init() {
    myPathsForm.initialize(myModel.getServer(), myModel.getServerPath());
    updateMessage();
  }

  @Override
  public void commit(CommitType commitType) throws CommitStepException {
    if (validateLocalPath(myPathsForm.getLocalPath()) == null) {
      myModel.setDestinationFolder(myPathsForm.getLocalPath());
    }

    if (validateServerPath(myPathsForm.getServerPath()) == null) {
      myModel.setServerPath(myPathsForm.getServerPath());
    }
  }

  @Nullable
  private static String validateLocalPath(final String path) {
    if (StringUtil.isEmpty(path)) {
      return TFSBundle.message("destination.path.not.specified");
    }
    VirtualFile file = WriteAction.compute(() -> {
      // do VFS refresh to guarantee correct case if user enters with wrong case
      return LocalFileSystem.getInstance().refreshAndFindFileByPath(path);
    });
    if (file != null && file.exists() && !file.isDirectory()) {
      return TFSBundle.message("destination.path.is.not.a.file");
    }
    return null;
  }

  @Nullable
  private static String validateServerPath(String path) {
    if (StringUtil.isEmpty(path)) {
      return TFSBundle.message("source.path.is.empty");
    }
    return null;
  }

  private void updateMessage() {
    String errorMessage = validateServerPath(myPathsForm.getServerPath());
    if (errorMessage == null) {
      errorMessage = validateLocalPath(myPathsForm.getLocalPath());
    }
    if (errorMessage != null) {
      myPathsForm.setMessage(errorMessage, true);
    }
    else {
      myPathsForm.setMessage(TFSBundle.message("mapping.will.be.created", myModel.getNewWorkspaceName()), false);
    }
  }

  @Override
  public JComponent getPreferredFocusedComponent() {
    return myPathsForm.getPreferredFocusedComponent();
  }

  @Override
  public String getHelpId() {
    return "reference.checkoutTFS.choosepaths";
  }
}