SvnServerWriter.java

/**
 * This file is part of git-as-svn. It is subject to the license terms
 * in the LICENSE file found in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/gpl-2.0.html. No part of git-as-svn,
 * including this file, may be copied, modified, propagated, or distributed
 * except according to the terms contained in the LICENSE file.
 */
package svnserver.parser;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import svnserver.parser.token.*;

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.util.Map;

/**
 * Интерфейс для записи данных в поток.
 *
 * @author Artem V. Navrotskiy <bozaro@users.noreply.github.com>
 */
public class SvnServerWriter {
  @NotNull
  private final OutputStream stream;
  private int depth = 0;

  public SvnServerWriter(@NotNull OutputStream stream) {
    this.stream = new BufferedOutputStream(stream);
  }

  @NotNull
  public SvnServerWriter listBegin() throws IOException {
    return write(ListBeginToken.instance);
  }

  @NotNull
  public SvnServerWriter listEnd() throws IOException {
    return write(ListEndToken.instance);
  }

  @NotNull
  public SvnServerWriter word(@NotNull String word) throws IOException {
    WordToken.write(stream, word);
    if (depth == 0) stream.flush();
    return this;
  }

  @NotNull
  public SvnServerWriter word(char c) throws IOException {
    return word(String.valueOf(c));
  }

  @SuppressWarnings("QuestionableName")
  @NotNull
  public SvnServerWriter stringNullable(@Nullable String text) throws IOException {
    return text != null ? string(text) : this;
  }

  @SuppressWarnings("QuestionableName")
  @NotNull
  public SvnServerWriter string(@NotNull String text) throws IOException {
    return binary(text.getBytes(StandardCharsets.UTF_8));
  }

  @NotNull
  public SvnServerWriter binary(@NotNull byte[] data) throws IOException {
    return binary(data, 0, data.length);
  }

  @NotNull
  public SvnServerWriter binary(@NotNull byte[] data, int offset, int length) throws IOException {
    StringToken.write(stream, data, offset, length);
    if (depth == 0) stream.flush();
    return this;
  }

  @NotNull
  public SvnServerWriter number(long number) throws IOException {
    NumberToken.write(stream, number);
    if (depth == 0) stream.flush();
    return this;
  }

  @NotNull
  public SvnServerWriter separator() throws IOException {
    stream.write('\n');
    return this;
  }

  @NotNull
  public SvnServerWriter bool(boolean value) throws IOException {
    return word(value ? "true" : "false");
  }

  @NotNull
  public SvnServerWriter write(@NotNull SvnServerToken token) throws IOException {
    token.write(stream);
    if (token.equals(ListBeginToken.instance)) {
      depth++;
    } else if (token.equals(ListEndToken.instance)) {
      depth--;
      if (depth < 0) {
        throw new IllegalStateException("Too many closed lists.");
      }
    }
    if (depth == 0) {
      separator();
      stream.flush();
    }
    return this;
  }

  @NotNull
  public SvnServerWriter writeMap(@Nullable Map<String, String> properties) throws IOException {
    listBegin();
    if (properties != null) {
      for (Map.Entry<String, String> entry : properties.entrySet()) {
        listBegin();
        string(entry.getKey());
        string(entry.getValue());
        listEnd();
      }
    }
    listEnd();
    return this;
  }
}