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

package de.mas.jnus.lib.implementations;

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

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

@Log
public class NUSDataProviderWoomy extends NUSDataProvider{
    @Getter @Setter private WoomyInfo woomyInfo;
    @Setter private WoomyZipFile woomyZipFile;
    
    @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");
            System.exit(1);
        }
        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 null;
    }

    @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");
            System.exit(1);
        }
        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");
            System.exit(1);
        }
           
        WoomyZipFile zipFile = getNewWoomyZipFile();        
        byte[] result = zipFile.getEntryAsByte(entry);
        zipFile.close();
        return  result;
    }

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

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

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

    @Override
    public byte[] getRawCert() throws IOException {
        // TODO Auto-generated method stub
        return null;
    }
}