source: java2python/src/main/java/tudelft/utilities/j2p/PyModule.java@ 390

package tudelft.utilities.j2p;

import java.io.File;
import java.io.FileNotFoundException;
import java.nio.file.Path;

import com.github.javaparser.JavaParser;
import com.github.javaparser.ParseResult;
import com.github.javaparser.ParserConfiguration;
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.symbolsolver.JavaSymbolSolver;
import com.github.javaparser.symbolsolver.resolution.typesolvers.CombinedTypeSolver;
import com.github.javaparser.symbolsolver.resolution.typesolvers.ReflectionTypeSolver;

import tudelft.utilities.j2p.formatting.Block;
import tudelft.utilities.j2p.t.Translator;

/**
 * A PyModule contains a python "file". This is the python equivalent to a java
 * class "file".
 */
public class PyModule {
        private final Block code;
        private final File source;

        /**
         * 
         * @param pycode     the python code to be zipped as pip-installable zip. It
         *                   is assumed that the code is just 1 file.
         * @param sourcefile the java source file of the program, relative to the
         *                   root of the source (typically src/main/java or so).
         *                   This relativeness allows us to derive the correct
         *                   filename, class path etc, without having to provide an
         *                   actual class object.
         */
        public PyModule(Block code, File sourcefile) {
                this.code = code;
                this.source = sourcefile;
        }

        public Block getCode() {
                return code;
        }

        /**
         * @return the java source file of the program, relative to the root of the
         *         source (typically src/main/java or so). This relativeness allows
         *         us to derive the correct filename, class path etc, without having
         *         to provide an actual class object.
         */
        public File getSourceFile() {
                return source;
        }

        /**
         * 
         * @param javaBaseDir the java base dir from where packages are found,
         *                    typically src/main/java
         * @param sourcefile  the java source file of the program, relative to the
         *                    root of the source (typically src/main/java or so).
         *                    This relativeness allows us to derive the correct
         *                    filename, class path etc, without having to provide an
         *                    actual class object.
         * @return PyModule containing the translated java file
         * @throws FileNotFoundException
         */
        public static PyModule fromJavaFile(Path javaBaseDir, File source)
                        throws FileNotFoundException {
                File javaFile = javaBaseDir.resolve(source.getPath()).toFile();
                if (!javaFile.exists())
                        throw new FileNotFoundException("There is no file " + source
                                        + " in the directory " + javaBaseDir);

                ParserConfiguration conf = new ParserConfiguration();
                CombinedTypeSolver typeSolver = new CombinedTypeSolver();
                typeSolver.add(new ReflectionTypeSolver(false));
                JavaSymbolSolver symbolSolver = new JavaSymbolSolver(typeSolver);
                conf.setSymbolResolver(symbolSolver);
                JavaParser parser = new JavaParser(conf);

                ParseResult<CompilationUnit> res = parser.parse(javaFile);
                try {
                        Block translation = Translator.translate(res.getResult().get());
                        return new PyModule(translation, source);
                } catch (Exception e) {
                        throw new IllegalArgumentException("Failed to translate " + source,
                                        e);
                }
        }

        @Override
        public String toString() {
                return "PyModule from " + source + ":\n" + code.toCode(
                                "from " + source.getPath().replace("/", ".") + " import .*");
        }

}