io.github.eliux.mega.MegaUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of megacmd4j Show documentation
Java client library that works on top of MEGAcmd to provide access to the services of Mega.nz
The newest version!
package io.github.eliux.mega;

import static io.github.eliux.mega.Mega.CMD_TTL_ENV_VAR;

import io.github.eliux.mega.error.MegaConfirmationRequiredException;
import io.github.eliux.mega.error.MegaIOException;
import io.github.eliux.mega.error.MegaInvalidEmailException;
import io.github.eliux.mega.error.MegaInvalidStateException;
import io.github.eliux.mega.error.MegaInvalidTypeException;
import io.github.eliux.mega.error.MegaLoginRequiredException;
import io.github.eliux.mega.error.MegaNodesNotFetchedException;
import io.github.eliux.mega.error.MegaOperationNotAllowedException;
import io.github.eliux.mega.error.MegaResourceAlreadyExistsException;
import io.github.eliux.mega.error.MegaResourceNotFoundException;
import io.github.eliux.mega.error.MegaUnexpectedFailureException;
import io.github.eliux.mega.error.MegaWrongArgumentsException;
import java.io.IOException;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.NoSuchElementException;
import java.util.Optional;
import java.util.Scanner;
import java.util.concurrent.TimeUnit;
import java.util.regex.Pattern;

public interface MegaUtils {

  DateTimeFormatter MEGA_FILE_DATE_TIME_FORMATTER = DateTimeFormatter
      .ofPattern("ddMMMyyyy HH:mm:ss", Locale.US);

  DateTimeFormatter MEGA_EXPORT_EXPIRE_DATE_FORMATTER = DateTimeFormatter
      .ofPattern("EEE, dd MMM yyyy HH:mm:ss Z", Locale.US);

  Pattern EMAIL_PATTERN = Pattern.compile(
      "^[\\w]+@[\\w]+\\.[a-zA-Z]{2,6}$",
      Pattern.CASE_INSENSITIVE
  );

  Pattern DIRECTORY_PATTERN = Pattern.compile(
      "[\\/]?[\\p{Alnum}]+(\\/[\\p{Alnum}]+)*",
      Pattern.CASE_INSENSITIVE
  );

  Integer MEGA_TTL = Optional.ofNullable(System.getenv(CMD_TTL_ENV_VAR))
      .map(Integer::new)
      .orElse(20000);


  Pattern BANNER_PATTERN = Pattern.compile(
      "^\\s?(-){5,}(.+)(-){5,}\\R",
      Pattern.MULTILINE | Pattern.DOTALL
  );

  static LocalDateTime parseFileDate(String dateStr) {
    return LocalDateTime.parse(dateStr, MEGA_FILE_DATE_TIME_FORMATTER);
  }

  static LocalDate parseBasicISODate(String dateStr) {
    return LocalDateTime.parse(dateStr, MEGA_EXPORT_EXPIRE_DATE_FORMATTER).toLocalDate();
  }

  static void handleResult(int code) {
    int posixExitStatus = Optional.ofNullable(code)
        .map(Math::abs)
        .orElse(-1);
    switch (posixExitStatus) {
      case 0:
        //Its Ok
        break;
      case 51:  //Win
      case 2:   //Unix
        throw new MegaWrongArgumentsException();
      case 12:
        throw new MegaResourceAlreadyExistsException();
      case 52:
        throw new MegaInvalidEmailException();
      case 53:
        throw new MegaResourceNotFoundException();
      case 54:
        throw new MegaInvalidStateException();
      case 55:
        throw new MegaInvalidTypeException();
      case 56:
        throw new MegaOperationNotAllowedException();
      case 57:
        throw new MegaLoginRequiredException();
      case 58:
        throw new MegaNodesNotFetchedException();
      case 59:
        throw new MegaUnexpectedFailureException();
      case 60:
        throw new MegaConfirmationRequiredException();
      default:
        throw new MegaUnexpectedFailureException(code);
    }
  }

  static String[] convertInstructionsToExecParams(String cmdInstructions) {
    return cmdInstructions.split("\\s+");
  }

  static int execCmd(String... cmd)
      throws java.io.IOException, InterruptedException {
    final Process process = Runtime.getRuntime().exec(cmd);

    Boolean succeeded = process.waitFor(MEGA_TTL, TimeUnit.MILLISECONDS);

    return succeeded ? process.exitValue() : -1;
  }

  static List handleCmdWithOutput(String... cmd)
      throws java.io.IOException {
    ProcessBuilder pb = new ProcessBuilder(cmd);
    pb.redirectErrorStream(true);

    final Process process = pb.start();

    final Scanner inputScanner = new Scanner(process.getInputStream())
        .useDelimiter(System.getProperty("line.separator"));

    final List result = new ArrayList<>();

    while (inputScanner.hasNext()) {
      result.add(inputScanner.next());
    }

    process.destroy();

    try {
      handleResult(process.waitFor());
    } catch (InterruptedException ex) {
      throw new MegaIOException(
          "The execution of '%s' was interrupted", cmd
      );
    }

    return result;
  }

  static String execCmdWithSingleOutput(String... cmd) throws IOException {
    try {
      return handleCmdWithOutput(cmd).get(0);
    } catch (IndexOutOfBoundsException ex) {
      return "";
    }
  }

  static boolean isEmail(String token) {
    return EMAIL_PATTERN.matcher(token).find();
  }

  static boolean isDirectory(String token) {
    return !isEmail(token) && DIRECTORY_PATTERN.matcher(token).find();
  }

  static List collectValidCmdOutput(Scanner inputScanner) {
    final List result = new ArrayList<>();

    try {
      inputScanner.skip(BANNER_PATTERN);
    } catch (NoSuchElementException ex) { }

    while (inputScanner.hasNext()) {
      result.add(inputScanner.next());
    }

    return result;
  }
}