source: java2python/src/test/java/tudelft/utilities/j2p/PyTest.java@ 353

package tudelft.utilities.j2p;

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.List;

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;

public class PyTest {
        private static final String TESTFILE = "test.py";
        private static final File TESTDIR = new File("test/");

        /**
         * 
         * @param pycode       the python code to run
         * @param fullclaspath a "package/class/path/filename.py" name containing
         *                     the full filename where the file should be placed.
         *                     This must be a RELATIVE name, we will prefix it with
         *                     {@link #TESTDIR} as needed
         * @return the text lines that the code printed to stdout.
         * @throws Exception
         */
        public List<String> runPython(String pycode, String fullclaspath)
                        throws Exception {
                File f = TESTDIR.toPath().resolve(fullclaspath).toFile();
                f.getParentFile().mkdirs();// create dir, massaging java...
                FileOutputStream fos = new FileOutputStream(f);
                BufferedOutputStream bos = new BufferedOutputStream(fos);
                // convert string to byte array
                byte[] bytes = pycode.getBytes();
                // write byte array to file
                bos.write(bytes);
                bos.close();
                fos.close();

                Process process = Runtime.getRuntime().exec("python " + fullclaspath,
                                new String[0], TESTDIR);
                List<String> errors = readAll(process.getErrorStream());
                if (!errors.isEmpty())
                        throw new IllegalStateException("Python failed. " + errors);

                return readAll(process.getInputStream());
        }

        private List<String> readAll(InputStream in) throws IOException {
                BufferedReader Buffered_Reader = new BufferedReader(
                                new InputStreamReader(in));
                List<String> lines = new LinkedList<>();

                String line;
                while ((line = Buffered_Reader.readLine()) != null) {
                        lines.add(line);
                }
                return lines;
        }

        /**
         * @param javaFile a java file name to be translated
         * @return a string containing python code translated from the java.
         * @throws FileNotFoundException
         */
        public String translate(String javaFile) throws FileNotFoundException {
                ParserConfiguration conf = new ParserConfiguration();
                CombinedTypeSolver typeSolver = new CombinedTypeSolver();
                typeSolver.add(new ReflectionTypeSolver(false));
                // FIXME add sources of jars source code etc
                // typeSolver.add(new JavaParserTypeSolver(new File(SOURCE_PATH)));
                // typeSolver.add(new JarTypeSolver(jarSourceFile));
                JavaSymbolSolver symbolSolver = new JavaSymbolSolver(typeSolver);
                conf.setSymbolResolver(symbolSolver);
                JavaParser parser = new JavaParser(conf);

                ParseResult<CompilationUnit> res = parser.parse(new File(javaFile));

                Block translation = new Translator().translate(res.getResult().get());
                return translation.toString();
        }

}