Java GroovyScriptClass.getResolveScope Examples

Example #1

File: BaseScriptTransformationSupport.java Project: jameshnsears/intellij-community

  @Override
  public void applyTransformation(@NotNull TransformationContext context) {
    if (!(context.getCodeClass() instanceof GroovyScriptClass)) return;
    GroovyScriptClass scriptClass = (GroovyScriptClass) context.getCodeClass();

    LightMethodBuilder mainMethod =
        new LightMethodBuilder(scriptClass.getManager(), GroovyLanguage.INSTANCE, "main")
            .setMethodReturnType(PsiType.VOID)
            .addParameter(
                "args",
                new PsiArrayType(
                    PsiType.getJavaLangString(
                        scriptClass.getManager(), scriptClass.getResolveScope())))
            .addModifiers(PsiModifier.PUBLIC, PsiModifier.STATIC);

    LightMethodBuilder runMethod =
        new LightMethodBuilder(scriptClass.getManager(), GroovyLanguage.INSTANCE, "run")
            .setMethodReturnType(TypesUtil.getJavaLangObject(scriptClass))
            .addModifier(PsiModifier.PUBLIC);

    context.addMethod(runMethod, true);
    context.addMethod(mainMethod, true);

    context.setSuperType(getBaseClassType(scriptClass));
  }