java/idea-ui/src/com/intellij/ide/util/newProjectWizard/StepSequence.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /*
  18  * User: anna
  19  * Date: 08-Jul-2007
  20  */
  21 package com.intellij.ide.util.newProjectWizard;
  22
  23 import com.intellij.ide.util.projectWizard.ModuleWizardStep;
  24 import com.intellij.openapi.module.ModuleType;
  25 import org.jetbrains.annotations.NonNls;
  26 import org.jetbrains.annotations.Nullable;
  27
  28 import java.util.*;
  29
  30 public class StepSequence {
  31   private final List<ModuleWizardStep> myCommonSteps = new ArrayList<ModuleWizardStep>();
  32   private final Map<String, StepSequence> mySpecificSteps = new HashMap<String, StepSequence>();
  33   @NonNls private String myType;
  34   private StepSequence myParentSequence;
  35
  36   public StepSequence() {
  37     this(null);
  38   }
  39
  40   public StepSequence(final StepSequence stepSequence) {
  41     myParentSequence = stepSequence;
  42   }
  43
  44   public void addCommonStep(ModuleWizardStep step){
  45     myCommonSteps.add(step);
  46   }
  47
  48   public void addSpecificSteps(String type, StepSequence sequence){
  49     mySpecificSteps.put(type, sequence);
  50   }
  51
  52   public List<ModuleWizardStep> getCommonSteps() {
  53     return myCommonSteps;
  54   }
  55
  56   public StepSequence getSpecificSteps(String type) {
  57     return mySpecificSteps.get(type);
  58   }
  59
  60   public Set<String> getTypes() {
  61     return mySpecificSteps.keySet();
  62   }
  63
  64   @Nullable
  65   public ModuleWizardStep getNextStep(ModuleWizardStep step) {
  66     final StepSequence stepSequence = mySpecificSteps.get(myType);
  67     if (myCommonSteps.contains(step)) {
  68       final int idx = myCommonSteps.indexOf(step);
  69       if (idx < myCommonSteps.size() - 1) {
  70         return myCommonSteps.get(idx + 1);
  71       }
  72       if (stepSequence != null && stepSequence.getCommonSteps().size() > 0) {
  73         return stepSequence.getCommonSteps().get(0);
  74       }
  75     }
  76     if (stepSequence != null) {
  77       return stepSequence.getNextStep(step);
  78     }
  79     return null;
  80   }
  81
  82   @Nullable
  83   public ModuleWizardStep getPreviousStep(ModuleWizardStep step) {
  84     if (myCommonSteps.contains(step)) {
  85       final int idx = myCommonSteps.indexOf(step);
  86       if (idx > 0) {
  87         return myCommonSteps.get(idx - 1);
  88       }
  89       if (myParentSequence != null) {
  90         final List<ModuleWizardStep> commonSteps = myParentSequence.getCommonSteps();
  91         if (!commonSteps.isEmpty()) {
  92           return commonSteps.get(commonSteps.size() - 1);
  93         }
  94       }
  95     }
  96     final StepSequence stepSequence = mySpecificSteps.get(myType);
  97     return stepSequence != null ? stepSequence.getPreviousStep(step) : null;
  98   }
  99
 100   public void setType(@NonNls final String type) {
 101     if (type == null) {
 102       myType = ModuleType.EMPTY.getId();
 103     } else {
 104       myType = type;
 105     }
 106   }
 107
 108   public String getSelectedType() {
 109     return myType;
 110   }
 111 }