content: /*
  	This file is part of the OdinMS Maple Story Server
      Copyright (C) 2008 Patrick Huy <patrick.huy@frz.cc>
  		       Matthias Butz <matze@odinms.de>
  		       Jan Christian Meyer <vimes@odinms.de>
  
      This program is free software: you can redistribute it and/or modify
      it under the terms of the GNU Affero General Public License as
      published by the Free Software Foundation version 3 as published by
      the Free Software Foundation. You may not use, modify or distribute
      this program under any other version of the GNU Affero General Public
      License.
  
      This program is distributed in the hope that it will be useful,
      but WITHOUT ANY WARRANTY; without even the implied warranty of
      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      GNU Affero General Public License for more details.
  
      You should have received a copy of the GNU Affero General Public License
      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
  package provider.wz;
  
  import java.io.BufferedInputStream;
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.IOException;
  import java.io.RandomAccessFile;
  import provider.MapleData;
  import provider.MapleDataDirectoryEntry;
  import provider.MapleDataFileEntry;
  import provider.MapleDataProvider;
  import tools.data.input.GenericLittleEndianAccessor;
  import tools.data.input.GenericSeekableLittleEndianAccessor;
  import tools.data.input.InputStreamByteStream;
  import tools.data.input.LittleEndianAccessor;
  import tools.data.input.RandomAccessByteStream;
  import tools.data.input.SeekableLittleEndianAccessor;
  
  public class WZFile implements MapleDataProvider {
      static {
          ListWZFile.init();
      }
      private File wzfile;
      private LittleEndianAccessor lea;
      private SeekableLittleEndianAccessor slea;
      private int headerSize;
      private WZDirectoryEntry root;
      private boolean provideImages;
      private int cOffset;
  
      public WZFile(File wzfile, boolean provideImages) throws IOException {
          this.wzfile = wzfile;
          lea = new GenericLittleEndianAccessor(new InputStreamByteStream(new BufferedInputStream(new FileInputStream(wzfile))));
          RandomAccessFile raf = new RandomAccessFile(wzfile, "r");
          slea = new GenericSeekableLittleEndianAccessor(new RandomAccessByteStream(raf));
          root = new WZDirectoryEntry(wzfile.getName(), 0, 0, null);
          this.provideImages = provideImages;
          load();
      }
  
      private void load() throws IOException {
          lea.readAsciiString(4);
          lea.readInt();
          lea.readInt();
          headerSize = lea.readInt();
          lea.readNullTerminatedAsciiString();
          lea.readShort();
          parseDirectory(root);
          cOffset = (int) lea.getBytesRead();
          getOffsets(root);
      }
  
      private void getOffsets(MapleDataDirectoryEntry dir) {
          for (MapleDataFileEntry file : dir.getFiles()) {
              file.setOffset(cOffset);
              cOffset += file.getSize();
          }
          for (MapleDataDirectoryEntry sdir : dir.getSubdirectories()) {
              getOffsets(sdir);
          }
      }
  
      private void parseDirectory(WZDirectoryEntry dir) {
          int entries = WZTool.readValue(lea);
          for (int i = 0; i < entries; i++) {
              byte marker = lea.readByte();
              String name = null;
              int size, checksum;
              switch (marker) {
                  case 0x02:
                      name = WZTool.readDecodedStringAtOffsetAndReset(slea, lea.readInt() + this.headerSize + 1);
                      size = WZTool.readValue(lea);
                      checksum = WZTool.readValue(lea);
                      lea.readInt(); //dummy int
                      dir.addFile(new WZFileEntry(name, size, checksum, dir));
                      break;
                  case 0x03:
                  case 0x04:
                      name = WZTool.readDecodedString(lea);
                      size = WZTool.readValue(lea);
                      checksum = WZTool.readValue(lea);
                      lea.readInt(); //dummy int
                      if (marker == 3) {
                          dir.addDirectory(new WZDirectoryEntry(name, size, checksum, dir));
                      } else {
                          dir.addFile(new WZFileEntry(name, size, checksum, dir));
                      }
                      break;
                  default:
              }
          }
          for (MapleDataDirectoryEntry idir : dir.getSubdirectories()) {
              parseDirectory((WZDirectoryEntry) idir);
          }
      }
  
      public WZIMGFile getImgFile(String path) throws IOException {
          String segments[] = path.split("/");
          WZDirectoryEntry dir = root;
          for (int x = 0; x < segments.length - 1; x++) {
              dir = (WZDirectoryEntry) dir.getEntry(segments[x]);
              if (dir == null) {
                  return null;
              }
          }
          WZFileEntry entry = (WZFileEntry) dir.getEntry(segments[segments.length - 1]);
          if (entry == null) {
              return null;
          }
          String fullPath = wzfile.getName().substring(0, wzfile.getName().length() - 3).toLowerCase() + "/" + path;
          return new WZIMGFile(this.wzfile, entry, provideImages, ListWZFile.isModernImgFile(fullPath));
      }
  
      @Override
      public synchronized MapleData getData(String path) {
          try {
              WZIMGFile imgFile = getImgFile(path);
              if (imgFile == null) {
                  return null;
              }
              MapleData ret = imgFile.getRoot();
              return ret;
          } catch (IOException e) {
              e.printStackTrace();
          }
          return null;
      }
  
      @Override
      public MapleDataDirectoryEntry getRoot() {
          return root;
      }
  }
  