aswi2020horky-gitlab/demo_mh/Deserializer/src/deserialize/ToJSON.java @ fe28c174

package deserialize;

import java.io.BufferedInputStream;
import java.io.BufferedWriter;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import org.json.simple.JSONObject;

public class ToJSON extends Thread {
	
	private File input;
	private File output;
	
	private byte buffer[];
	
	public ToJSON(File inputFilename, File outputFilename) {
		this.input = inputFilename;
		this.output = outputFilename;
	}
	
	@Override
	public void run() {
		super.run();
		
		buffer = readByteArray();
		if (buffer == null) {
			// TODO Chyba na??t?n? dat
		} else if (deserializeClassically()) {
			// TODO Chyba klasick? deserializace
		} else {
			// Univerzalni deserializer.
			convert();
		}
	}
	
	private void convert() {
		System.out.println("Spoustim univerzalni deserializer.");
		
		Grammar g = new Grammar(buffer);
		JSONObject results = g.process();
		
		save(results);
		
		System.out.println("JSON byl vytvoren.");
	}
	
	private boolean deserializeClassically() {
		try {
			
			// Try to deserialize classically.
			ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(buffer));
			@SuppressWarnings("unused")
			Deserializer d = (Deserializer) ois.readObject();
			System.out.println("Klasicka deserializace probehla v poradku.");
			ois.close();
			return true;
			
		} catch (IOException e) {
			
			// Stream error.
			System.out.println("Chyba pri otevirani (uzavirani) streamu pri pokusu o klasickou deserializaci.");
			e.printStackTrace();
			
		} catch (Exception e) {
			
			// Others: ClassNotFoundException, ClassCastException, ...
			// Usually due to problems with classical deserialization.
			
		}
		return false;
	}
	
	private byte[] readByteArray() {
		byte array[] = null;
		
		/*
		 * Note.
		 * For ZIP - the first four bytes should be one of the following combinations:
		 *     50 4B 03 04
		 *     50 4B 05 06 (empty archive)
		 *     50 4B 07 08 (spanned archive)
		 * Source: https://en.wikipedia.org/wiki/List_of_file_signatures
		 */
		ZipFile zipFile = null;
		ZipEntry entry = null;
	    try {
	    	zipFile = new ZipFile(input);
	        entry = (ZipEntry) zipFile.entries().nextElement();
	        System.out.println("Velikost ZIP dat je " + entry.getSize() + "B.");
		} catch (Exception e) {
			System.out.println("Soubor neni zazipovany...");
		}
	    
	    try {
	    	if (entry != null) {
	    		array = new byte[(int) entry.getSize()];
	    		zipFile.getInputStream(entry).read(array);
		    } else {
		    	array = new byte[(int) input.length()];
				BufferedInputStream bis = new BufferedInputStream(new FileInputStream(input));
				bis.read(array);
				bis.close();
		    }
	    	System.out.println("Byty ze souboru byly v poradku nacteny.");
	    } catch (IOException e) {
	    	System.out.println("Pri nacitani bytu doslo k chybe.");
	    	return null;
	    }
	    
	    return array;
	}
	
	private void save(JSONObject obj) {
        try {
            FileWriter fw = new FileWriter(output);
            BufferedWriter bw = new BufferedWriter(fw);
            bw.write(obj.toJSONString());
            bw.flush();
            bw.close();
            Report.info("JSON", null, "Ulo?en? souboru prob?hlo v po??dku.");
        } catch (IOException e) {
            e.printStackTrace();
            Report.info("JSON", null, "P?i ukl?d?n? souboru nastala chyba.");
        }
	}
	
}