/*
 * Copyright 2017 The authors
 * 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.struts2.dom.struts.action;

import com.intellij.codeInspection.LocalQuickFixAndIntentionActionOnPsiElement;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Iconable;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.pom.Navigatable;
import com.intellij.psi.*;
import com.intellij.psi.codeStyle.CodeStyleManager;
import com.intellij.psi.codeStyle.JavaCodeStyleManager;
import com.intellij.util.OpenSourceUtil;
import icons.Struts2Icons;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.*;

/**
 * Create action-method.
 *
 * @author Yann Cébron
 */
public class CreateActionMethodQuickFix extends LocalQuickFixAndIntentionActionOnPsiElement implements Iconable {

  private final String methodName;

  public CreateActionMethodQuickFix(final PsiClass actionClass,
                                    final String methodName) {
    super(actionClass);
    this.methodName = methodName;
  }

  @Override
  protected boolean isAvailable() {
    return StringUtil.isNotEmpty(methodName) &&
           super.isAvailable();
  }

  @NotNull
  public String getText() {
    return "Create action-method '" + methodName + "'";
  }

  @NotNull
  public String getFamilyName() {
    return "Create action-method";
  }

  @Override
  public Icon getIcon(final int flags) {
    return Struts2Icons.Action;
  }

  @Override
  public void invoke(@NotNull final Project project,
                     @NotNull final PsiFile psiFile,
                     @Nullable("is null when called from inspection") final Editor editor,
                     @NotNull final PsiElement startPsiElement, @NotNull final PsiElement endPsiElement) {
    final PsiClass actionClass = (PsiClass) startPsiElement;

    final PsiElementFactory elementFactory = JavaPsiFacade.getInstance(project).getElementFactory();
    PsiMethod actionMethod = elementFactory.createMethodFromText("public java.lang.String " + methodName + "() throws java.lang.Exception { return \"success\"; }",
                                                                 actionClass);

    final JavaCodeStyleManager javaCodeStyleManager = JavaCodeStyleManager.getInstance(project);
    actionMethod = (PsiMethod) javaCodeStyleManager.shortenClassReferences(actionMethod);
    final CodeStyleManager codestylemanager = CodeStyleManager.getInstance(project);
    actionMethod = (PsiMethod) codestylemanager.reformat(actionMethod);

    final PsiMethod element = (PsiMethod) actionClass.add(actionMethod);

    //noinspection ConstantConditions
    OpenSourceUtil.navigate((Navigatable) element.getBody().getNavigationElement());
  }

}