com.spun.util.io.TabDelimitedFileReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of approvaltests-util Show documentation

There is a newer version: 24.9.0

package com.spun.util.io;

import com.spun.util.ObjectUtils;
import com.spun.util.StringUtils;

import java.io.BufferedReader;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;

/**
  * A static class of convenience functions for Files
  **/
public class TabDelimitedFileReader
{
  private BufferedReader reader   = null;
  private String         lastRead = null;
  private boolean        trim     = false;
  public TabDelimitedFileReader(String absoluteFileName, boolean trim)
  {
    this.reader = ObjectUtils
        .throwAsError(() -> Files.newBufferedReader(Paths.get(absoluteFileName), StandardCharsets.UTF_8));
    this.trim = trim;
  }
  public boolean next()
  {
    return prepNext() != null;
  }
  public String prepNext()
  {
    try
    {
      if (reader == null)
      { return null; }
      lastRead = reader.readLine();
      if (lastRead == null)
      {
        reader.close();
        reader = null;
      }
      return lastRead;
    }
    catch (IOException e)
    {
      throw ObjectUtils.throwAsError(e);
    }
  }
  public String[] readLine(int minimumIndexReturned)
  {
    if (lastRead == null)
    {
      if (prepNext() == null)
      { return null; }
    }
    String[] found = StringUtils.split(lastRead, "\t", trim);
    clean(found);
    if (found.length < minimumIndexReturned)
    {
      String[] temp = new String[minimumIndexReturned];
      System.arraycopy(found, 0, temp, 0, found.length);
      found = temp;
    }
    lastRead = null;
    return found;
  }
  private void clean(String[] found)
  {
    for (int i = 0; i < found.length; i++)
    {
      if (found[i] != null && found[i].startsWith("\"") && found[i].endsWith("\""))
      {
        found[i] = found[i].substring(1, found[i].length() - 1);
        found[i] = found[i].replaceAll("\"\"", "\"");
      }
    }
  }
}