package com.bytezone.diskbrowser.nufx;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;

import com.bytezone.diskbrowser.prodos.write.DiskFullException;
import com.bytezone.diskbrowser.prodos.write.FileAlreadyExistsException;
import com.bytezone.diskbrowser.prodos.write.ProdosDisk;
import com.bytezone.diskbrowser.prodos.write.VolumeCatalogFullException;
import com.bytezone.diskbrowser.utilities.Utility;

// -----------------------------------------------------------------------------------//
public class Binary2
// -----------------------------------------------------------------------------------//
{
  private static final String UNDERLINE =
      "------------------------------------------------------" + "-----------------------";

  Binary2Header binary2Header;
  byte[] buffer;
  List<Binary2Header> headers = new ArrayList<> ();
  int totalBlocks;
  String fileName;

  // ---------------------------------------------------------------------------------//
  public Binary2 (Path path) throws IOException
  // ---------------------------------------------------------------------------------//
  {
    fileName = path.toFile ().getName ();
    buffer = Files.readAllBytes (path);
    read (buffer);
  }

  // ---------------------------------------------------------------------------------//
  private void read (byte[] buffer)
  // ---------------------------------------------------------------------------------//
  {
    int ptr = 0;

    do
    {
      binary2Header = new Binary2Header (buffer, ptr);
      headers.add (binary2Header);

      totalBlocks += binary2Header.totalBlocks;
      ptr += ((binary2Header.eof - 1) / 128 + 1) * 128 + 128;
    } while (binary2Header.filesToFollow > 0);

  }

  // ---------------------------------------------------------------------------------//
  public byte[] getDiskBuffer ()
      throws DiskFullException, VolumeCatalogFullException, FileAlreadyExistsException, IOException
  // ---------------------------------------------------------------------------------//
  {
    ProdosDisk disk = new ProdosDisk (800, "DiskBrowser");

    for (Binary2Header header : headers)
    {
      if (header.compressed && buffer[header.ptr + 128] == 0x76
          && buffer[header.ptr + 129] == (byte) 0xFF)
      {
        byte[] tmp = new byte[header.eof];       // this sux
        System.arraycopy (buffer, header.ptr + 128, tmp, 0, tmp.length);
        String name = Utility.getCString (tmp, 4);

        Squeeze squeeze = new Squeeze ();
        byte[] dataBuffer = squeeze.unSqueeze (tmp);
        disk.addFile (name, header.fileType, header.auxType, header.created, header.modified,
            dataBuffer, header.eof);
      }
      else
      {
        byte[] dataBuffer = new byte[header.eof];       // this sux
        System.arraycopy (buffer, header.ptr + 128, dataBuffer, 0, dataBuffer.length);
        disk.addFile (header.fileName, header.fileType, header.auxType, header.created,
            header.modified, dataBuffer, header.eof);
      }

    }
    disk.close ();

    return disk.getBuffer ();
  }

  // ---------------------------------------------------------------------------------//
  @Override
  public String toString ()
  // ---------------------------------------------------------------------------------//
  {
    StringBuilder text = new StringBuilder ();

    text.append (
        String.format (" %-15.15s                                                  Files:%5d%n%n",
            fileName, headers.size ()));

    text.append (
        " Name                              Type Auxtyp Modified" + "         Fmat   Length\n");

    text.append (String.format ("%s%n", UNDERLINE));

    for (Binary2Header header : headers)
      text.append (String.format ("%s%n", header.getLine ()));

    text.append (String.format ("%s%n", UNDERLINE));

    return text.toString ();
  }
}