JavascriptServiceNameStrategy.java

package fr.adrienbrault.idea.symfony2plugin.action.generator.naming;

import com.google.gson.JsonObject;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.vfs.VfsUtil;
import com.jetbrains.php.lang.psi.elements.PhpClass;
import fr.adrienbrault.idea.symfony2plugin.Settings;
import fr.adrienbrault.idea.symfony2plugin.Symfony2ProjectComponent;
import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
import org.apache.commons.lang.StringUtils;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;

/**
 * @author Daniel Espendiller <daniel@espendiller.net>
 */
public class JavascriptServiceNameStrategy implements ServiceNameStrategyInterface {

    @Nullable
    @Override
    public String getServiceName(@NotNull ServiceNameStrategyParameter parameter) {

        String serviceJsNameStrategy = Settings.getInstance(parameter.getProject()).serviceJsNameStrategy;
        if(serviceJsNameStrategy == null || StringUtils.isBlank(serviceJsNameStrategy)) {
            return null;
        }

        try {
            Object eval = run(parameter.getProject(), parameter.getClassName(), serviceJsNameStrategy);
            if(!(eval instanceof String)) {
                return null;
            }
            return StringUtils.isNotBlank((String) eval) ? (String) eval : null;
        } catch (ScriptException e) {
            Symfony2ProjectComponent.getLogger().error(String.format("ScriptException: '%s' - Script: '%s'", e.getMessage(), serviceJsNameStrategy));
        }

        return null;
    }

    @Nullable
    public static Object run(@NotNull Project project, @NotNull String className, @NotNull String serviceJsNameStrategy) throws ScriptException {

        JsonObject jsonObject = new JsonObject();

        jsonObject.addProperty("className", className);
        jsonObject.addProperty("projectName", project.getName());
        jsonObject.addProperty("projectBasePath", project.getBasePath());
        jsonObject.addProperty("defaultNaming", new DefaultServiceNameStrategy().getServiceName(new ServiceNameStrategyParameter(project, className)));

        PhpClass aClass = PhpElementsUtil.getClass(project, className);
        if(aClass != null) {
            String relativePath = VfsUtil.getRelativePath(aClass.getContainingFile().getVirtualFile(), project.getBaseDir(), '/');
            if(relativePath != null) {
                jsonObject.addProperty("relativePath", relativePath);
            }
            jsonObject.addProperty("absolutePath", aClass.getContainingFile().getVirtualFile().toString());
        }

        ScriptEngine engine = new ScriptEngineManager().getEngineByName("JavaScript");
        if(engine == null) {
            return null;
        }

        return engine.eval("var __p = eval(" + jsonObject.toString() + "); result = function(args) { " + serviceJsNameStrategy + " }(__p)");
    }

}