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

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 String modulename;

        /**
         * 
         * @param pycode     the python code to be zipped as pip-installable zip. It
         *                   is assumed that the code is just 1 file.
         * @param modulename The name of the module. Eg "a.b.c" if there is a file
         *                   a.b.c.py. The module name is equivalent to the java
         *                   package name. Note that you still need the java BASE
         *                   directory (typically src/main/java) to find the
         *                   original java file.
         */
        public PyModule(Block code, String modulename) {
                this.code = code;
                this.modulename = modulename;
        }

        public Block getCode() {
                return code;
        }

        public String getModuleName() {
                return modulename;
        }

        /**
         * 
         * @param javaBaseDir the java base dir from where packages are found,
         *                    typically src/main/java
         * @param modulename  the full.qualified.path.to.Class of the java module
         * @return PyModule containing the translated java file
         * @throws FileNotFoundException
         */
        public static PyModule fromJavaFile(Path javaBaseDir, String modulename)
                        throws FileNotFoundException {
                String modulepath = modulename.replace(".", "/");
                File javaFile = javaBaseDir.resolve(modulepath + ".java").toFile();
                if (!javaFile.exists())
                        throw new FileNotFoundException("There is no file " + modulename
                                        + ".java 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);

                Block translation = Translator.translate(res.getResult().get());
                return new PyModule(translation, modulename);
        }
}