kz.greetgo.msoffice.xlsx.parse.SharedStrings Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of greetgo.msoffice Show documentation

greetgo library to generate or parse MS Office files

There is a newer version: 0.5.9

package kz.greetgo.msoffice.xlsx.parse;

import java.io.File;
import java.io.FileOutputStream;
import java.io.PrintStream;
import java.io.RandomAccessFile;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.nio.charset.StandardCharsets;

public class SharedStrings {
  private static final int SPACE_FOR_HEADER = 200;

  private final File file;

  public SharedStrings(String workDir) throws Exception {

    file = new File(workDir + "/xl/sharedStrings.xml");
    prepareFile();

  }

  private PrintStream printStream;

  private final void prepareFile() throws Exception {
    file.getParentFile().mkdirs();

    printStream = new PrintStream(new FileOutputStream(file), false, "UTF-8");

    StringBuilder sb = new StringBuilder();
    for (int i = 0; i < SPACE_FOR_HEADER; i++) {
      sb.append(' ');
    }
    sb.append("\r\n");

    printStream.print(sb.toString());
  }

  private int index = 0;

  public int index(String str) {
    StringBuilder sb = new StringBuilder();
    sb.append("");
    if (str == null) str = "";
    str = str.replaceAll("&", "&");
    str = str.replaceAll("<", "<");
    str = str.replaceAll(">", ">");
    sb.append(str);
    sb.append("\r\n");

    printStream.print(sb.toString());

    return index++;
  }

  public void close() throws Exception {
    printStream.print("");
    printStream.flush();
    printStream.close();
    printStream = null;

    RandomAccessFile raf = new RandomAccessFile(file, "rw");
    FileChannel fc = raf.getChannel();

    {
      fc.position(0);
      StringBuilder sb = new StringBuilder();
      sb.append("\r\n");
      sb.append("");

      {
        ByteBuffer buf = ByteBuffer.wrap(sb.toString().getBytes(StandardCharsets.UTF_8));
        while (buf.hasRemaining()) {
          fc.write(buf);
        }
      }
    }

    fc.close();
    raf.close();
  }
}