void (@NotNull TransformationContext context) { if (!(context.getCodeClass() instanceof GroovyScriptClass scriptClass)) return; LightMethodBuilder mainMethod = new LightMethodBuilder(scriptClass.getManager(), GroovyLanguage.INSTANCE, "main") .setMethodReturnType(PsiTypes.voidType()) .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)); }