platform/external-system-impl/src/com/intellij/openapi/externalSystem/service/vcs/ExternalSystemVcsRegistrar.java

/*
 * Copyright 2000-2013 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.openapi.externalSystem.service.vcs;

import com.intellij.openapi.externalSystem.ExternalSystemManager;
import com.intellij.openapi.externalSystem.settings.AbstractExternalSystemSettings;
import com.intellij.openapi.externalSystem.settings.ExternalProjectSettings;
import com.intellij.openapi.externalSystem.settings.ExternalSystemSettingsListenerAdapter;
import com.intellij.openapi.externalSystem.util.ExternalSystemApiUtil;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vcs.ProjectLevelVcsManager;
import com.intellij.openapi.vcs.VcsDirectoryMapping;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.util.containers.ContainerUtilRt;
import com.intellij.vcsUtil.VcsUtil;
import org.jetbrains.annotations.NotNull;

import java.util.Collection;
import java.util.List;

/**
 * It's rather often when open-source project is backed by an external system and stored at public repo hosting
 * like github, bitbucket etc. Standard actions sequence looks as below then:
 * <pre>
 * <ol>
 *   <li>Clone target repo to local machine;</li>
 *   <li>Execute 'import from external system' wizard;</li>
 * </ol>
 * </pre>
 * The problem is that the ide detects unregistered vcs roots at the newly imported project and shows corresponding notification
 * with suggestion to configure vcs roots for the current project. We want to simplify that in a way to not showing the notification
 * but register them automatically. This class manages that.
 * 
 * @author Denis Zhdanov
 * @since 7/15/13 6:00 PM
 */
public class ExternalSystemVcsRegistrar {

  @SuppressWarnings("unchecked")
  public static void handle(@NotNull final Project project) {
    for (final ExternalSystemManager<?, ?, ?, ?, ?> manager : ExternalSystemApiUtil.getAllManagers()) {
      final AbstractExternalSystemSettings settings = manager.getSettingsProvider().fun(project);
      settings.subscribe(new ExternalSystemSettingsListenerAdapter() {
        @Override
        public void onProjectsLinked(@NotNull final Collection linked) {
          List<VcsDirectoryMapping> newMappings = ContainerUtilRt.newArrayList();
          final LocalFileSystem fileSystem = LocalFileSystem.getInstance();
          ProjectLevelVcsManager vcsManager = ProjectLevelVcsManager.getInstance(project);
          for (Object o : linked) {
            final ExternalProjectSettings settings = (ExternalProjectSettings)o;
            VirtualFile dir = fileSystem.refreshAndFindFileByPath(settings.getExternalProjectPath());
            if (dir == null) {
              continue;
            }
            if (!dir.isDirectory()) {
              dir = dir.getParent();
            }
            newMappings.addAll(VcsUtil.findRoots(dir, project));
          }

          // There is a possible case that no VCS mappings are configured for the current project. There is a single
          // mapping like <Project> - <No VCS> then. We want to replace it if only one mapping to the project root dir
          // has been detected then.
          List<VcsDirectoryMapping> oldMappings = vcsManager.getDirectoryMappings();
          if (oldMappings.size() == 1
              && newMappings.size() == 1
              && StringUtil.isEmpty(oldMappings.get(0).getVcs()))
          {
            VcsDirectoryMapping newMapping = newMappings.iterator().next();
            String detectedDirPath = newMapping.getDirectory();
            VirtualFile detectedDir = fileSystem.findFileByPath(detectedDirPath);
            if (detectedDir != null && detectedDir.equals(project.getBaseDir())) {
              newMappings.clear();
              newMappings.add(new VcsDirectoryMapping("", newMapping.getVcs()));
              vcsManager.setDirectoryMappings(newMappings);
              return;
            }
          }

          newMappings.addAll(oldMappings);
          vcsManager.setDirectoryMappings(newMappings);
        }
      });
    }
  }
}