aswi2020horky-gitlab/project/Deserializer/src/io/FileWorker.java @ e608f492

package io;

import java.io.BufferedInputStream;
import java.io.BufferedWriter;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.Properties;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

import javafx.collections.ObservableList;

public class FileWorker {
	
	private static final String CONFIG = "app.config";
	public static final String INPUT_DIR = "input_dir";
	public static final String ACE_MODE = "ace_mode";
	public static final String ACE_THEME = "ace_theme";
	public static final String DB_TYPE = "db_type";
	public static final String DB_URL = "db_url";
	public static final String DB_USER = "db_user";
	
	public static String getResource(String path) {
		return FileWorker.class.getResource(path).toString();
	}
	
	public static Properties getConfig() {
		Properties properties = new Properties();
		
		try {
			File config = new File(CONFIG);
			config.createNewFile();
			properties.load(new FileInputStream(config));
		} catch (IOException e) { }
		
		return properties;
	}
	
	public static void saveConfig(Properties properties) {
		try {
			properties.store(new FileOutputStream(CONFIG), null);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	public static void getAceModesAndThemes(ObservableList<String> modes, ObservableList<String> themes) {
		final String aceFolder = "ace";
		List<String> names = new ArrayList<String>();
		
		
		String path = FileWorker.class.getResource(FileWorker.class.getSimpleName() + ".class").getFile();
		if (!path.startsWith("/")) {
			path = ClassLoader.getSystemClassLoader().getResource(path).getFile();
			File jarFile = new File(path.substring(6, path.lastIndexOf('!')));
			
			try {
				JarFile jar = new JarFile(jarFile);
				Enumeration<JarEntry> entries = jar.entries();
				while (entries.hasMoreElements()) {
					String name = entries.nextElement().getName();
					if (name.startsWith(aceFolder + "/")) {
						name = name.substring(name.lastIndexOf("/") + 1);
						if (name.length() > 0) {
							names.add(name);
						}
					}
				}
				jar.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		} else {
			URL url = FileWorker.class.getResource("/" + aceFolder);
			if (url != null) {
				try {
					File dir = new File(url.toURI());
					for (File file : dir.listFiles()) {
						names.add(file.getName());
					}
				} catch (URISyntaxException e) {
					e.printStackTrace();
				}
			}
		}
		
		
		for (int i = 0; i < names.size(); i++) {
			if (names.get(i).startsWith("mode-")) {
				modes.add(names.get(i).substring(5, names.get(i).lastIndexOf('.')));
			} else if (names.get(i).startsWith("theme-")) {
				themes.add(names.get(i).substring(6, names.get(i).lastIndexOf('.')));
			}
		}
	}
	
	public static PrintStream createRedirectStream(boolean error) throws FileNotFoundException {
		return new PrintStream(new File(error ? "jdeserialize_err.txt" : "jdeserialize_out.txt"));
	}
	
	public static byte[] loadFileContent(File file) throws IOException {
		byte buffer[] = new byte[(int) file.length()];
		BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
		bis.read(buffer);
		bis.close();
		return buffer;
	}
	
	public static byte[] extractFileContent(byte rawContent[]) throws IOException {
		byte buffer[];
		
		IZipReader input;
		try {
			input = new ZipInStream(new ByteArrayInputStream(rawContent));
		} catch (NoSuchElementException e) {
			System.out.println("Note: the file may be a special type of ZIP archive.");
			input = new InflaterInStream(new ByteArrayInputStream(rawContent));
		}
		
		try {
			List<Byte> _buffer = new ArrayList<Byte>();
			while (true) {
				int value = input.read();
				if (value < 0) {
					break;
				} else {
					_buffer.add((byte) value);
				}
			}

			buffer = new byte[_buffer.size()];
			for (int i = 0; i < buffer.length; i++) {
				buffer[i] = _buffer.get(i);
			}
			
			System.out.println(String.format("The file is a%s ZIP archive...", input.getClass() == InflaterInStream.class ? " special" : ""));
		} catch (Exception e) {
			System.out.println("The file is not a ZIP archive...");
			buffer = rawContent;
		} finally {
			input.close();
		}

		System.out.println("Bytes from the file were loaded correctly.");
		return buffer;
	}

	public static void saveJson(File file, String json) throws IOException {
		FileWriter fw = new FileWriter(file);
		BufferedWriter bw = new BufferedWriter(fw);
		bw.write(json);
		bw.flush();
		bw.close();
	}
	
}