aswi2020horky-gitlab/project/Deserializer/src/Converter.java @ 7e0b699a

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileDescriptor;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintStream;
import java.util.HashMap;
import java.util.List;
import java.util.concurrent.Semaphore;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

import jdeserialize.content;
import jdeserialize.jdeserialize;
import io.Database.DB_Messenger;
import io.Database;
import io.FileWorker;

public class Converter extends Thread {

	private IConversionResults ui;
	private AtomicBoolean active;

	// P-K without E semaphore.
	private Lock entryLock;
	private Semaphore inputAvailable;

	// Input.
	private File inputFile;
	private DB_Messenger query;
	
	// Key as enum?
	private HashMap<String, String> htmlFormatting = new HashMap<String, String>();
	private HashMap<String, String> jsonFormatting = new HashMap<String, String>();

	public Converter(IConversionResults ui) {
		this.ui = ui;
		active = new AtomicBoolean(true);
		entryLock = new ReentrantLock();
		inputAvailable = new Semaphore(0);
		fillInTestingDictionaries();
	}

	private void fillInTestingDictionaries() {
		htmlFormatting.put("indent", "<span style=\"margin-left:2em;\">");
		htmlFormatting.put("lineBreak", "<br/>");
		htmlFormatting.put("classCol", "<span style=\"color:blue;font-weight:bold;\">");
		htmlFormatting.put("fieldCol", "<span style=\"color:purple;\">");
		htmlFormatting.put("valCol", "<span style=\"color:orange;\">");
		htmlFormatting.put("keywordCol", "<span style=\"color:green;\">");
		htmlFormatting.put("closeTagCol", "</span>");

		jsonFormatting.put("indent", "\t");
		jsonFormatting.put("lineBreak", "\n");
		jsonFormatting.put("classCol", "");
		jsonFormatting.put("fieldCol", "");
		jsonFormatting.put("valCol", "");
		jsonFormatting.put("keywordCol", "");
		jsonFormatting.put("closeTagCol", "");
	}

	public void end() {
		active.set(false);
		setInput(null, null);
	}
	
	public void setInput(File inputFile, DB_Messenger query) {
		entryLock.lock();
		this.inputFile = inputFile;
		this.query = query;
		entryLock.unlock();

		if (!inputAvailable.tryAcquire())
			inputAvailable.release(); // MAX VALUE = 1!
	}
	
	public boolean getInput(AtomicReference<File> inputFile, AtomicReference<DB_Messenger> query) {
		try {
			inputAvailable.acquire();
		} catch (InterruptedException e) {
			return false;
		}

		entryLock.lock();
		inputFile.set(this.inputFile);
		query.set(this.query);
		
		this.inputFile = null;
		this.query = null;
		entryLock.unlock();

		return inputFile.get() != null || query.get() != null;
	}

	@Override
	public void run() {
		super.run();
		while (active.get()) {
			AtomicReference<File> inputFile = new AtomicReference<File>();
			AtomicReference<DB_Messenger> query = new AtomicReference<DB_Messenger>();

			if (!getInput(inputFile, query)) {
				continue;
			}

			byte array[] = inputFile.get() != null ? getBytes(inputFile.get()) : getBytes(query.get());
			StringBuilder html = new StringBuilder();
			StringBuilder json = new StringBuilder();
			
			try {
				// Redirectovany system.out do null streamu. Mozno redirect do souboru
				System.setOut(FileWorker.createRedirectStream());
				convert(array, html, json);
			} catch (Exception e) {
				json.setLength(0);
				html.setLength(0);
			} finally {
				System.setOut(new PrintStream(new FileOutputStream(FileDescriptor.out)));
			}

			//test file output
			try {
				BufferedWriter writer = new BufferedWriter(new FileWriter("html.txt"));
				writer.write(html.toString().replace("<br/>", "\n")); // TODO pridan replace pro odsazovani - lepsi orientace v souboru.
				writer.close();

				BufferedWriter writer2 = new BufferedWriter(new FileWriter("json.txt"));
				writer2.write(json.toString());
				writer2.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
			//test file output
			
			ui.completed(html.toString(), json.toString(), new String(array));
		}
	}

	private byte[] getBytes(File input) {
		byte array[];
		try {
			array = FileWorker.loadByteArray(input);
		} catch (IOException e) {
			ui.loadingInputFileError();
			return null;
		}
		return array;
	}
	
	private byte[] getBytes(DB_Messenger query) {
		return Database.getBlobBytes(query);
	}

	private void convert(byte[] buffer, StringBuilder html, StringBuilder json) throws Exception {
		jdeserialize deserializer = new jdeserialize(buffer);

		// gets the "contents" into an array - returnes the deserialization of all
		// 'writes' into serialized object via objectOutputStream
		List<content> cntnts = deserializer.getContent();

		for (content cnt : cntnts) {
			if (cnt != null) {
				// Parametrizovany toJson pomoci dictionary
				// Ciste HTML / Cisty JSON
				json.append(cnt.toJson("", null, this.jsonFormatting, false));
				html.append(cnt.toJson("", null, this.htmlFormatting, false));
			}
		}
	}

}