android-studio-plugin/src/com/google/gct/idea/appengine/wizard/NewCloudModuleDynamicWizardPath.java

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * 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.google.gct.idea.appengine.wizard;

import com.android.tools.idea.npw.FormFactorUtils;
import com.android.tools.idea.npw.ModuleTemplate;
import com.android.tools.idea.npw.NewModuleDynamicPath;
import com.android.tools.idea.wizard.WizardConstants;
import com.android.tools.idea.wizard.dynamic.DynamicWizardPath;
import com.android.tools.idea.wizard.dynamic.ScopedStateStore;
import com.google.common.collect.Lists;
import com.google.gct.idea.util.GctBundle;
import icons.GoogleCloudToolsIcons;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.*;
import java.io.File;
import java.util.List;

import static com.android.tools.idea.wizard.dynamic.ScopedStateStore.*;

/**
 * The global path used in the "create a new AppEngine module" wizard process.
 */
public class NewCloudModuleDynamicWizardPath extends DynamicWizardPath implements NewModuleDynamicPath {
  static final Key<String> KEY_MODULE_NAME = createKey(CloudModuleUtils.ATTR_MODULE_NAME, Scope.PATH, String.class);
  static final Key<String> KEY_PACKAGE_NAME = createKey(CloudModuleUtils.ATTR_PACKAGE_NAME, Scope.PATH, String.class);
  static final Key<String> KEY_CLIENT_MODULE_NAME = createKey("myClientModuleName", Scope.PATH, String.class);
  static final Key<File> KEY_SELECTED_TEMPLATE_FILE = createKey("myTemplateFile", Scope.WIZARD, File.class);

  @NonNls
  private static final String PATH_NAME = "New Cloud Module Path";

  List<ModuleTemplate> myModuleTemplates;

  @Override
  protected void init() {
    assert getProject() != null;
    initPrefilledValues();
    addStep(new NewCloudModuleDynamicWizardStep(getProject(), myWizard.getDisposable()));
  }

  private void initPrefilledValues() {
    final String examplePackageName;
    final String userName = System.getProperty("user.name");
    if (userName == null) {
      examplePackageName = GctBundle.message("appengine.wizard.prefilled_package_name");
    } else {
      examplePackageName = GctBundle.message("appengine.wizard.prefilled_user_package_name", userName);
    }
    myState.put(KEY_MODULE_NAME, GctBundle.message("appengine.wizard.prefilled_module_name"));
    myState.put(KEY_PACKAGE_NAME, examplePackageName);
  }

  @NotNull
  @Override
  public String getPathName() {
    return PATH_NAME;
  }

  @NotNull
  @Override
  public Iterable<ModuleTemplate> getModuleTemplates() {
    myModuleTemplates = Lists.newArrayList((ModuleTemplate)new NewCloudModuleTemplate());
    return myModuleTemplates;
  }

  @Override
  public boolean isPathVisible() {
    final ModuleTemplate moduleTemplate = myWizard.getState().get(WizardConstants.SELECTED_MODULE_TYPE_KEY);
    return moduleTemplate != null && myModuleTemplates != null && myModuleTemplates.contains(moduleTemplate);
  }

  @Override
  public boolean performFinishingActions() {
    final File templateFile = myState.get(KEY_SELECTED_TEMPLATE_FILE);
    final String newModuleName = myState.get(KEY_MODULE_NAME);
    final String packageName = myState.get(KEY_PACKAGE_NAME);
    final String clientModuleName = myState.get(KEY_CLIENT_MODULE_NAME);
    assert getProject() != null && templateFile != null && newModuleName != null && packageName != null;
    CloudModuleUtils.createModule(getProject(), templateFile, newModuleName, packageName, clientModuleName);
    return true;
  }

  /**
   * Each instance of this class represents an item that can be selected in the first page of the
   * New Module wizard.
   */
  private static final class NewCloudModuleTemplate implements ModuleTemplate {
    @NotNull
    @Override
    public String getName() {
      return GctBundle.message("appengine.wizard.gallery_title");
    }

    @Nullable
    @Override
    public String getDescription() {
      return GctBundle.message("appengine.wizard.gallery_description");
    }

    @Nullable
    @Override
    public Icon getIcon() {
      return GoogleCloudToolsIcons.CLOUD_MODULE;
    }

    @Override
    public void updateWizardState(@NotNull ScopedStateStore state) {
      // Do nothing.
    }

    @Nullable
    @Override
    public FormFactorUtils.FormFactor getFormFactor() {
      return null;
    }

    @Override
    public String toString() {
      return getName();
    }
  }
}