org.finos.legend.engine.shared.javaCompiler.SingleFileCompiler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of legend-engine-shared-javaCompiler Show documentation

There is a newer version: 4.67.8

Show newest version

// Copyright 2020 Goldman Sachs
//
// 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 org.finos.legend.engine.shared.javaCompiler;

import org.codehaus.commons.compiler.CompileException;
import org.codehaus.janino.ClassLoaderIClassLoader;
import org.codehaus.janino.IClassLoader;
import org.codehaus.janino.Parser;
import org.codehaus.janino.Scanner;
import org.codehaus.janino.UnitCompiler;
import org.codehaus.janino.util.ClassFile;
import org.eclipse.collections.api.factory.Maps;
import org.eclipse.collections.api.map.MutableMap;

import java.io.IOException;
import java.io.Reader;
import java.util.Map;

public class SingleFileCompiler
{
    public static Map compileFile(StringJavaSource source) throws IOException, CompileException
    {
        return compileFile(source, null);
    }

    public static Map compileFile(StringJavaSource source, ClassLoader parentClassLoader) throws IOException, CompileException
    {
        IClassLoader classLoader = new ClassLoaderIClassLoader((parentClassLoader == null) ? Thread.currentThread().getContextClassLoader() : parentClassLoader);

        ClassFile[] classFiles;
        try (Reader reader = source.openReader(true))
        {
            Parser parser = new Parser(new Scanner(source.getName(), reader));
            UnitCompiler unitCompiler = new UnitCompiler(parser.parseAbstractCompilationUnit(), classLoader);
            classFiles = unitCompiler.compileUnit(false, false, false);
        }

        MutableMap classes = Maps.mutable.ofInitialCapacity(classFiles.length);
        for (ClassFile cf : classFiles)
        {
            classes.put(cf.getThisClassName(), cf.toByteArray());
        }
        return classes;
    }
}