com.databricks.sdk.core.utils.OSUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of databricks-sdk-java Show documentation
There is a newer version: 0.35.0
package com.databricks.sdk.core.utils;

import com.databricks.sdk.core.DatabricksException;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * OSUtils is an interface that provides utility methods for determining the current operating
 * system and returning executable command based on the operating system.
 */
public class OSUtils {
  private static final Logger LOG = LoggerFactory.getLogger(OSUtils.class);

  /**
   * Returns the name of the current operating system.
   *
   * @return a String representing the name of the current operating system. The value will be "win"
   *     for Windows, "mac" for macOS, and "linux" for Linux-based operating systems.
   */
  static String getOS(Environment env) {
    String systemName = env.getSystemName().toLowerCase();
    if (systemName.startsWith("win")) {
      return "win";
    } else if (systemName.startsWith("mac")) {
      return "mac";
    }
    return "linux";
  }

  /** Returns the OS-specific utilities for the current operating system. */
  public static OSUtilities get(Environment env) {
    String os = getOS(env);
    switch (os) {
      case "win":
        return new WindowsUtilities(env);
      case "mac":
        return new MacOSUtilities(env);
      case "linux":
        return new LinuxUtilities(env);
    }
    throw new DatabricksException("Unsupported OS: " + os);
  }

  public static String findExecutable(List paths, String name) {
    LOG.debug("Searching for executable named '" + name + "' in : " + String.join(", ", paths));
    for (String dir : paths) {
      Path path = Paths.get(dir, name).toAbsolutePath().normalize();
      if (!Files.isRegularFile(path)) {
        continue;
      }
      long size;
      try {
        size = Files.size(path);
      } catch (IOException e) {
        LOG.debug("Unable to get size of databricks cli, continuing", e);
        continue;
      }
      if (size < 1024 * 1024) {
        LOG.debug("Databricks CLI version <0.100.0 detected");
        continue;
      }
      String pathStr = path.toString();
      LOG.debug("Found executable named '" + name + "' in PATH: " + pathStr);
      return pathStr;
    }
    LOG.debug("Failed to find executable named '" + name + "' in PATH");
    return null;
  }
}