aswi2020horky-gitlab/project/Deserializer/src/Converter.java @ 81ff7da4

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.FileWorker;

public class Converter extends Thread {

	private IConversionResults ui;
	private AtomicBoolean active;

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

	private File inputFile;
	private String inputString;
	private boolean testing;

	// 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, false);
	}

	public void setInput(File inputFile, String inputString, boolean testing) {
		entryLock.lock();
		this.inputFile = inputFile;
		this.inputString = inputString;
		this.testing = testing;
		entryLock.unlock();

		if (!inputAvailable.tryAcquire())
			inputAvailable.release(); // MAX VALUE = 1!
	}

	public boolean getInput(AtomicReference<File> inputFile, AtomicReference<String> inputString,
			AtomicBoolean testing) {
		try {
			inputAvailable.acquire();
		} catch (InterruptedException e) {
			return false;
		}

		entryLock.lock();
		inputFile.set(this.inputFile);
		inputString.set(this.inputString);
		testing.set(this.testing);

		this.inputFile = null;
		this.inputString = null;
		entryLock.unlock();

		return true;
	}

	@Override
	public void run() {
		super.run();
		while (active.get()) {
			AtomicReference<File> inputFile = new AtomicReference<File>();
			AtomicReference<String> inputString = new AtomicReference<String>();
			AtomicBoolean testing = new AtomicBoolean();

			if (!getInput(inputFile, inputString, testing)) {
				continue;
			}

			if (inputFile.get() != null) {
				processA(inputFile.get(), testing.get());
			} else if (inputString.get() != null) {
				processB(inputString.get().getBytes(), testing.get());
			}
		}
	}

	private void processA(File input, boolean testing) {
		byte array[];
		try {
			array = FileWorker.loadByteArray(input);
		} catch (IOException e) {
			ui.loadingInputFileError();
			return;
		}
		processB(array, testing);
	}

	private void processB(byte array[], boolean testing) {
		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());
			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(), testing);
	}

	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));
			}
		}
	}

}