JNUSLib/src/de/mas/wiiu/jnus/implementations/NUSDataProviderWoomy.java

package de.mas.wiiu.jnus.implementations;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipException;

import de.mas.wiiu.jnus.NUSTitle;
import de.mas.wiiu.jnus.Settings;
import de.mas.wiiu.jnus.entities.content.Content;
import de.mas.wiiu.jnus.implementations.woomy.WoomyInfo;
import de.mas.wiiu.jnus.implementations.woomy.WoomyZipFile;
import lombok.AccessLevel;
import lombok.Getter;
import lombok.NonNull;
import lombok.Setter;
import lombok.extern.java.Log;

@Log
public class NUSDataProviderWoomy extends NUSDataProvider {
    @Getter private final WoomyInfo woomyInfo;
    @Setter(AccessLevel.PRIVATE) private WoomyZipFile woomyZipFile;

    public NUSDataProviderWoomy(NUSTitle title, WoomyInfo woomyInfo) {
        super(title);
        this.woomyInfo = woomyInfo;
    }

    @Override
    public InputStream getInputStreamFromContent(@NonNull Content content, long fileOffsetBlock) throws IOException {
        WoomyZipFile zipFile = getSharedWoomyZipFile();
        ZipEntry entry = getWoomyInfo().getContentFiles().get(content.getFilename().toLowerCase());
        if (entry == null) {
            log.warning("Inputstream for " + content.getFilename() + " not found");
            throw new FileNotFoundException("Inputstream for " + content.getFilename() + " not found");
        }
        return zipFile.getInputStream(entry);
    }

    @Override
    public byte[] getContentH3Hash(Content content) throws IOException {
        ZipEntry entry = getWoomyInfo().getContentFiles().get(content.getFilename().toLowerCase());
        if (entry != null) {
            WoomyZipFile zipFile = getNewWoomyZipFile();
            byte[] result = zipFile.getEntryAsByte(entry);
            zipFile.close();
            return result;
        }
        return new byte[0];
    }

    @Override
    public byte[] getRawTMD() throws IOException {
        ZipEntry entry = getWoomyInfo().getContentFiles().get(Settings.TMD_FILENAME);
        if (entry == null) {
            log.warning(Settings.TMD_FILENAME + " not found in woomy file");
            throw new FileNotFoundException(Settings.TMD_FILENAME + " not found in woomy file");
        }
        WoomyZipFile zipFile = getNewWoomyZipFile();
        byte[] result = zipFile.getEntryAsByte(entry);
        zipFile.close();
        return result;
    }

    @Override
    public byte[] getRawTicket() throws IOException {
        ZipEntry entry = getWoomyInfo().getContentFiles().get(Settings.TICKET_FILENAME);
        if (entry == null) {
            log.warning(Settings.TICKET_FILENAME + " not found in woomy file");
            throw new FileNotFoundException(Settings.TICKET_FILENAME + " not found in woomy file");
        }

        WoomyZipFile zipFile = getNewWoomyZipFile();
        byte[] result = zipFile.getEntryAsByte(entry);
        zipFile.close();
        return result;
    }

    public WoomyZipFile getSharedWoomyZipFile() throws ZipException, IOException {
        if (this.woomyZipFile == null || this.woomyZipFile.isClosed()) {
            this.woomyZipFile = getNewWoomyZipFile();
        }
        return this.woomyZipFile;
    }

    private WoomyZipFile getNewWoomyZipFile() throws ZipException, IOException {
        return new WoomyZipFile(getWoomyInfo().getWoomyFile());
    }

    @Override
    public void cleanup() throws IOException {
        if (this.woomyZipFile != null && this.woomyZipFile.isClosed()) {
            this.woomyZipFile.close();
        }
    }

    @Override
    public byte[] getRawCert() throws IOException {
        return new byte[0];
    }
}