python/ide/src/com/jetbrains/python/newProject/actions/PyCharmNewProjectStep.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.jetbrains.python.newProject.actions;

import com.google.common.collect.Lists;
import com.intellij.openapi.actionSystem.DefaultActionGroup;
import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.project.DumbAware;
import com.intellij.platform.DirectoryProjectGenerator;
import com.intellij.util.NullableConsumer;
import com.jetbrains.python.newProject.PyFrameworkProjectGenerator;
import com.jetbrains.python.newProject.PythonBaseProjectGenerator;
import com.jetbrains.python.newProject.PythonProjectGenerator;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

public class PyCharmNewProjectStep extends DefaultActionGroup implements DumbAware {

  public PyCharmNewProjectStep(@NotNull final String name, @Nullable final Runnable runnable) {
    this(name, runnable, false);
  }

  public PyCharmNewProjectStep(@NotNull final String name, @Nullable final Runnable runnable, boolean isWelcomeScreen) {
    super(name, true);

    final NullableConsumer<AbstractProjectSettingsStep> callback = new GenerateProjectCallback(runnable);

    final ProjectSpecificAction action = new ProjectSpecificAction(callback, new PythonBaseProjectGenerator(), isWelcomeScreen);
    add(action);

    final DirectoryProjectGenerator[] generators = Extensions.getExtensions(DirectoryProjectGenerator.EP_NAME);
    if (generators.length == 0) {
      action.setPopup(false);
    }
    Arrays.sort(generators, new Comparator<DirectoryProjectGenerator>() {
      @Override
      public int compare(DirectoryProjectGenerator o1, DirectoryProjectGenerator o2) {
        if (o1 instanceof PyFrameworkProjectGenerator && !(o2 instanceof PyFrameworkProjectGenerator)) return -1;
        if (!(o1 instanceof PyFrameworkProjectGenerator) && o2 instanceof PyFrameworkProjectGenerator) return 1;
        return o1.getName().compareTo(o2.getName());
      }
    });

    List<DirectoryProjectGenerator> pluginSpecificGenerators = Lists.newArrayList();
    for (DirectoryProjectGenerator generator : generators) {
      if (generator instanceof PythonProjectGenerator)
        add(new ProjectSpecificAction(callback, generator, isWelcomeScreen));
      else
        pluginSpecificGenerators.add(generator);
    }

    if (!pluginSpecificGenerators.isEmpty()) {
      add(new PluginSpecificProjectsStep(callback, pluginSpecificGenerators, isWelcomeScreen));
    }
  }

  public PyCharmNewProjectStep() {
    this("Select Project Type", null, true);
  }
}