com.github.sbt.jni.javah.JavahTask Maven / Gradle / Ivy

Go to download
package com.github.sbt.jni.javah;

import com.github.sbt.jni.javah.search.ClassPath;
import com.github.sbt.jni.javah.search.ModulePath;
import com.github.sbt.jni.javah.search.RuntimeSearchPath;
import com.github.sbt.jni.javah.search.SearchPath;
import com.github.sbt.jni.javah.util.JNIGenerator;

import java.io.PrintWriter;
import java.io.Writer;
import java.nio.file.Path;
import java.util.LinkedList;
import java.util.List;
import java.util.Objects;

public final class JavahTask {
    private final List searchPaths = new LinkedList<>();
    private Path outputDir;
    private Path outputFile;
    private PrintWriter errorHandle = new PrintWriter(System.err, true);
    private final List classes = new LinkedList<>();

    public void run() {
        if (outputDir == null && outputFile == null) {
            throw new AssertionError();
        }
        JNIGenerator generator = new JNIGenerator(outputDir, searchPaths, errorHandle);
        for (ClassName cls : classes) {
            try {
                generator.generate(cls);
            } catch (Exception ex) {
                ex.printStackTrace(errorHandle);
            }
        }
    }

    public boolean hasClasses() {
        return !classes.isEmpty();
    }

    public void addClass(ClassName name) {
        Objects.requireNonNull(name);
        classes.add(name);
    }

    public void addClass(String name) {
        Objects.requireNonNull(name);
        classes.add(ClassName.ofFullName(name));
    }

    public void addClasses(Iterable i) {
        Objects.requireNonNull(i);
        i.forEach(c -> classes.add(ClassName.ofFullName(c)));
    }

    public void addRuntimeSearchPath() {
        searchPaths.add(RuntimeSearchPath.INSTANCE);
    }

    public void addSearchPath(SearchPath searchPath) {
        Objects.requireNonNull(searchPath);
        searchPaths.add(searchPath);
    }

    public void addClassPath(Path classPath) {
        Objects.requireNonNull(classPath);
        searchPaths.add(new ClassPath(classPath));
    }

    public void addModulePath(Path modulePath) {
        Objects.requireNonNull(modulePath);
        searchPaths.add(new ModulePath(modulePath));
    }

    public Path getOutputDir() {
        return outputDir;
    }

    public void setOutputDir(Path outputDir) {
        this.outputDir = outputDir;
    }

    public Path getOutputFile() {
        return outputFile;
    }

    public void setOutputFile(Path outputFile) {
        this.outputFile = outputFile;
    }

    public PrintWriter getErrorHandle() {
        return errorHandle;
    }

    public void setErrorHandle(Writer errorHandle) {
        if (errorHandle instanceof PrintWriter || errorHandle == null) {
            this.errorHandle = (PrintWriter) errorHandle;
        } else {
            this.errorHandle = new PrintWriter(errorHandle);
        }
    }
}