JstdGenerateNewTestAction.java

package com.google.jstestdriver.idea.assertFramework.jstd.codeInsight;

import com.intellij.javascript.testFramework.codeInsight.AbstractJsGenerateAction;
import com.intellij.javascript.testFramework.codeInsight.GenerateActionContext;
import com.intellij.javascript.testFramework.codeInsight.JsGeneratorUtils;
import com.google.jstestdriver.idea.assertFramework.jstd.JstdTestCaseStructure;
import com.google.jstestdriver.idea.assertFramework.jstd.JstdTestFileStructure;
import com.google.jstestdriver.idea.assertFramework.jstd.JstdTestFileStructureBuilder;
import com.intellij.javascript.testFramework.util.JsPsiUtils;
import com.intellij.lang.javascript.psi.JSArgumentList;
import com.intellij.lang.javascript.psi.JSCallExpression;
import com.intellij.lang.javascript.psi.JSExpression;
import com.intellij.lang.javascript.psi.JSObjectLiteralExpression;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class JstdGenerateNewTestAction extends AbstractJsGenerateAction {

  @NotNull
  @Override
  public String getHumanReadableDescription() {
    return "JsTestDriver Test";
  }

  @Override
  public boolean isEnabled(@NotNull GenerateActionContext context) {
    JstdTestFileStructureBuilder builder = JstdTestFileStructureBuilder.getInstance();
    JstdTestFileStructure fileStructure = builder.fetchCachedTestFileStructure(context.getJsFile());
    if (fileStructure.getTestCaseCount() == 0) {
      return false;
    }
    Runnable testGenerator = buildGenerator(fileStructure, context);
    return testGenerator != null;
  }

  @Override
  public void actionPerformed(@NotNull GenerateActionContext context) {
    JstdTestFileStructureBuilder builder = JstdTestFileStructureBuilder.getInstance();
    JstdTestFileStructure fileStructure = builder.fetchCachedTestFileStructure(context.getJsFile());
    Runnable testGenerator = buildGenerator(fileStructure, context);
    if (testGenerator != null) {
      testGenerator.run();
    }
  }

  @Nullable
  private static Runnable buildGenerator(@NotNull JstdTestFileStructure fileStructure, @NotNull GenerateActionContext context) {
    int caretOffset = context.getDocumentCaretOffset();
    JstdTestCaseStructure jstdTestCaseStructure = fileStructure.findEnclosingTestCaseByOffset(caretOffset);
    if (jstdTestCaseStructure != null) {
      JSObjectLiteralExpression testsObjectLiteral = jstdTestCaseStructure.getTestsObjectsLiteral();
      if (testsObjectLiteral != null) {
        return new TestGeneratorOnObjectLiteral(testsObjectLiteral, context);
      } else {
        if (jstdTestCaseStructure.getTestCount() == 0) {
          JSCallExpression callExpression = jstdTestCaseStructure.getEnclosingCallExpression();
          JSArgumentList argumentList = callExpression.getArgumentList();
          JSExpression[] arguments = JsPsiUtils.getArguments(argumentList);
          if (arguments.length == 1 && arguments[0] != null) {
            return new TestGeneratorOnNewlyCreatedObjectLiteral(argumentList, context);
          }
        }
      }
    } else {
      for (JstdTestCaseStructure testCaseStructure : fileStructure.getTestCaseStructures()) {
        JSObjectLiteralExpression testsObjectLiteral = testCaseStructure.getTestsObjectsLiteral();
        if (testsObjectLiteral != null && JsPsiUtils.containsOffsetStrictly(testsObjectLiteral.getTextRange(), caretOffset)) {
          return new TestGeneratorOnObjectLiteral(testsObjectLiteral, context);
        }
      }
    }
    return null;
  }

  private static class TestGeneratorOnObjectLiteral implements Runnable {

    private final JSObjectLiteralExpression myTestsObjectLiteral;
    private final GenerateActionContext myContext;

    public TestGeneratorOnObjectLiteral(@NotNull JSObjectLiteralExpression testsObjectLiteral, GenerateActionContext context) {
      myTestsObjectLiteral = testsObjectLiteral;
      myContext = context;
    }

    public void run() {
      JsGeneratorUtils.generateProperty(myTestsObjectLiteral, myContext, "\"test ${name}\": function() {|}");
    }
  }

  private static class TestGeneratorOnNewlyCreatedObjectLiteral implements Runnable {

    private final JSArgumentList myArgumentList;
    private final GenerateActionContext myContext;

    public TestGeneratorOnNewlyCreatedObjectLiteral(@NotNull JSArgumentList argumentList,
                                                    @NotNull GenerateActionContext context) {
      myArgumentList = argumentList;
      myContext = context;
    }

    @Override
    public void run() {
      JsGeneratorUtils.generateObjectLiteralWithPropertyAsArgument(myContext, "{\n\"test ${name}\": function() {|}\n}", myArgumentList, 1);
    }
  }
}