com.dropchop.snakejar.impl.LibraryLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snakejar Show documentation

A Java library with embedded Python (CPython) which supports native image compile and usage in Java Application Servers, targeting native Python Machine Learning frameworks and libraries.

There is a newer version: 1.35.7

Show newest version

package com.dropchop.snakejar.impl;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;

/**
 * @author Nikola Ivačič  on 6. 11. 21.
 */
public class LibraryLoader {
  private static final String OS = System.getProperty("os.name").toLowerCase();
  private static final String ARCH = System.getProperty("os.arch").toLowerCase();

  public static boolean isWindows() {
    return (OS.contains("win"));
  }

  public static boolean isFreeBSD() {
    return (OS.contains("freebsd"));
  }

  public static boolean isMac() {
    return (OS.contains("mac"));
  }

  public static boolean isUnix() {
    return OS.contains("nix") ||
      OS.contains("nux");
  }

  public static boolean isOpenBSD() {
    return (OS.contains("openbsd"));
  }

  public static boolean is64Bit() {
    return (ARCH.indexOf("64") > 0);
  }

  public static String osArchSuffix(final String name, final String pythonVersion) {
    if (isUnix()) {
      return String.format("%s-linux-py%s-%s", name, pythonVersion, is64Bit() ? "x64" : "");
    } /*else if (isMac()) {
      return String.format("%s-osx-py%s", name, pythonVersion);
    } else if (isFreeBSD()) {
      return String.format("%s-freebsd%s", name, is64Bit() ? "x64" : "");
    } else if (isWindows() && is64Bit()) {
      return String.format("%s-win64", name);
    } else if (isOpenBSD()) {
      return String.format("%s-openbsd%s", name, is64Bit() ? "x64" : "");
    } */

    throw new UnsupportedOperationException(String.format("Cannot determine JNI library name for ARCH='%s' OS='%s' name='%s'", ARCH, OS, name));
  }

  public static String getLibraryBaseName(final String name, final String pythonVersion) {
    return osArchSuffix(name, pythonVersion);
  }

  private static String osExtension() {
    if (isUnix() || isFreeBSD() || isOpenBSD()) {
      return  ".so";
    } else if (isMac()) {
      return ".dylib";
    } else if (isWindows()) {
      return ".dll";
    }
    throw new UnsupportedOperationException();
  }

  public static String getLibraryFileName(final String name, final String pythonVersion) {
    return  "lib" + getLibraryBaseName(name, pythonVersion) + osExtension();
  }

  static File fromJarToTemp(final String baseName, final String pythonVersion, final ClassLoader classLoader) throws IOException {
    final File temp;
    temp = File.createTempFile("lib" + getLibraryBaseName(baseName, pythonVersion), osExtension());

    if (!temp.exists()) {
      throw new RuntimeException("File " + temp.getAbsolutePath() + " does not exist.");
    } else {
      temp.deleteOnExit();
    }
    String libFileName = "lib" + getLibraryBaseName(baseName, pythonVersion) + osExtension();
    try (final InputStream is = classLoader.getResourceAsStream(libFileName)) {
      if (is == null) {
        throw new RuntimeException("Missing [" + libFileName + "] in jar.");
      } else {
        Files.copy(is, temp.toPath(), StandardCopyOption.REPLACE_EXISTING);
      }
    }

    return temp;
  }
}