1 files changed, 103 insertions, 0 deletions
diff --git a/src/main/java/org/glavo/javah/Main.java b/src/main/java/org/glavo/javah/Main.java
new file mode 100644
index 0000000..3c2be04
--- /dev/null
+++ b/src/main/java/org/glavo/javah/Main.java
@@ -0,0 +1,103 @@
+package org.glavo.javah;
+
+import picocli.CommandLine;
+
+import java.io.File;
+import java.io.InputStream;
+import java.io.OutputStreamWriter;
+import java.io.PrintWriter;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.Arrays;
+import java.util.List;
+import java.util.jar.Manifest;
+import java.util.stream.Stream;
+
+import static picocli.CommandLine.*;
+
+@Command(name = "gjavah", version = "gjavah ${gjavah.version}", sortOptions = false)
+public class Main {
+    @Option(names = {"-p", "--module-path"}, description = "Path from which to search  modules")
+    private String modulePath;
+    @Option(names = {"-cp", "-classpath", "--classpath", "--class-path"}, description = "Path from which to search classes")
+    private String classpath;
+
+    @Option(names = {"-version", "--version"}, description = "Print version information")
+    private boolean showVersion;
+
+    @Option(names = {"-h", "--help", "-?"}, usageHelp = true, description = "Print this message")
+    private boolean showHelp;
+
+    @Option(names = {"-d"}, description = "Output directory")
+    private Path outputDir = Paths.get("").toAbsolutePath().normalize();
+
+    @Parameters(paramLabel = "classes")
+    private List<String> classes;
+
+    public static void main(String[] args) throws Exception {
+        Main m = new Main();
+        CommandLine cm = new CommandLine(m);
+        if (args == null || args.length == 0) {
+            cm.usage(System.err);
+            System.exit(-1);
+        }
+
+        cm.parseArgs(args);
+        if (m.showHelp) {
+            cm.usage(System.out);
+            return;
+        }
+        if (m.showVersion) {
+            try (InputStream in = Main.class.getResourceAsStream("/META-INF/MANIFEST.MF")) {
+                System.setProperty("gjavah.version", new Manifest(in).getMainAttributes().getValue("GJavah-Version"));
+                cm.printVersionHelp(System.out);
+            }
+            return;
+        }
+        if (m.classes == null || m.classes.isEmpty()) {
+            cm.usage(System.err);
+            System.exit(-1);
+        }
+
+        JavahTask task = new JavahTask();
+        if (m.modulePath != null) {
+            Arrays.stream(m.modulePath.split(File.pathSeparator))
+                    .map(Paths::get)
+                    .filter(Files::isDirectory)
+                    .forEachOrdered(task::addModulePath);
+        }
+        if (m.classpath == null) {
+            m.classpath = System.getenv("CLASSPATH");
+        }
+        if (m.classpath == null) {
+            m.classpath = Paths.get("").toAbsolutePath().normalize().toString();
+        }
+        Arrays.stream(m.classpath.split(File.pathSeparator))
+                .flatMap(p -> {
+                    if (p.endsWith("/*") || p.equals("*")) {
+                        try {
+                            return Files.list(Paths.get(p.substring(0, p.length() - 1)))
+                                    .filter(Files::isRegularFile)
+                                    .filter(t -> t.toAbsolutePath().getFileName().toString().toLowerCase().endsWith(".jar"));
+                        } catch (Exception e) {
+                            return Stream.empty();
+                        }
+                    }
+                    return Stream.of(Paths.get(p));
+                })
+                .filter(Files::exists)
+                .map(Path::toAbsolutePath)
+                .forEachOrdered(task::addClassPath);
+        task.setOutputDir(m.outputDir);
+        task.addClasses(m.classes);
+        task.setErrorHandle(new PrintWriter(System.err, true));
+        task.addRuntimeSearchPath();
+        task.run();
+    }
+
+    @Override
+    public String toString() {
+        return String.format("Main[modulePath='%s', classpath='%s', showVersion=%s, showHelp=%s, outputDir=%s, classes=%s]", modulePath, classpath, showVersion, showHelp, outputDir, classes);
+    }
+}