!7

set flag LeanMode.

do not include function load.

p {

  noRegularGC();

  try {

    go(asList(args));

  } catch e {

    //print(exceptionToStringShort(e));

    printStackTrace(e);

  }

}

svoid go(LS args) {

  bool verbose;

  //print(+args); // javax still swallowing -v!?

  S cmd = popFirst(args);

  bool byteMode = contains(cmd, "b");

  cmd = replace(cmd, "b", "");

  bool toUpper = contains(cmd, "u");

  cmd = replace(cmd, "u", "");

  bool balancing = contains(cmd, "B");

  cmd = replace(cmd, "B", "");

  if (eqOneOf(first(args), "-v", "-verbose", "-V")) {

    args = dropFirst(args);

    set verbose;

  }

  S archiveName = first(args);

  if (eqic(cmd, "x")) {

    print("Extracting");

    if (empty(archiveName)) fail("Need archive name as second argument");

    File f = newFile(archiveName);

    if (!fileExists(f)) fail("File not found: " + f.getAbsolutePath());

    // TODO: byteMode

    LineCompReader lcr = new(f);

    LS filesToExtract = dropFirst(args);

    if (empty(filesToExtract)) {

      for (S version : lcr.versions())

        extractVersion(lcr, version);

    } else {

      for (S name : filesToExtract)

        if (!lcr.versions().contains(name))

          print("Warning: File " + name + " not found in archive");

        else

          extractVersion(lcr, name);

    }

  } else if (eqic(cmd, "c")) {

    //print("Creating");

    if (empty(archiveName)) fail("Need archive name as second argument");

    LS filesToAdd = dropFirst(args);

    if (empty(filesToAdd)) fail("Please specify files to add to archive");

    new LinkedHashMap<S> data;

    for (S file : filesToAdd) {

      File f = newFile(file);

      if (!isFile(f))

        print("Warning: File " + f.getAbsolutePath() + " not found, skipping");

      else {

        print("Adding file " + file);

        data.put(file, bytesToString_raw(loadBinaryFile(f)));

      }

    }

    print("Compressing a total of " + nBytes(totalStringLength(values(data))) + "...");

    time "Compression done" {

      new LineCompCompressor compressor;

      compressor.byteMode = byteMode;

      compressor.toUpper = toUpper;

      if (byteMode) print("byte mode");

      if (toUpper) print("toUpper mode");

      compressor.verboseCompressionSteps = print(+verbose);

      compressor.balancing = balancing;

      if (balancing) print("balancing mode");

      compressor.verify = false;

      compressor.verboseStats = true;

      compressor.loadTexts(data);

      data = null;

      compressor.run();

      File archiveFile = newFile(archiveName);

      lcSaveToFile(compressor, archiveFile);

    }

    listArchive(archiveFile, false);

  } else if (eqic(cmd, "t")) {

    //print("Listing");

    if (empty(archiveName)) fail("Need archive name as second argument");

    File f = newFile(archiveName);

    if (!fileExists(f)) fail("File not found: " + f.getAbsolutePath());

    listArchive(f, true);

  } else {

    print(autoUnindent_mls([[

      This is LINECOMP v0.3, a supercompressor for a bunch of similar text files with fast extraction of any file.

      (c) Stefan Reich 2020, info@BotCompany.de. Source: https://code.BotCompany.de/1028231

      Commands:

        c archive.lc textFile1 textFile2 ... - create LINECOMP archive archive.lc from files

        cb archive.lc file1 file2 ... - create BYTECOMP archive [compress byte-by-byte]

        x archive.lc - extract archive.lc to current directory

        x archive.lc textFile1 - extract textFile1 from archive.lc to current directory

        t archive.lc - show contents of archive.lc

    ]]));

  }

}

svoid extractVersion(LineCompReader lcr, S version) {

  File outFile = newFile(version);

  saveBinaryFileVerbose(outFile, stringToBytes_raw(lcr.getText(version)));

}

svoid listArchive(File f, bool listFiles) ctex {

  LineCompReader lcr = new(f);

  if (listFiles)

    for (S version : lcr.versions())

      print("  Found file: " + version

        //+ " (" + nLines(lcr.lineCountForVersion(version)) + ")");

        + " (" + nBytes(lcr.byteCountForVersion(version)) + ")");

  printWithPrecedingNL("Archive " + f.getCanonicalPath() + " stats:");

    print("  " + str_toK_noSpace(lcr.totalByteCount()) + " of text compressed into " + str_toK_noSpace(l(f)) + " (" + nFiles(l(lcr.versions())) + ")");

}