com.github.ejahns.Main Maven / Gradle / Ivy

Go to download
package com.github.ejahns;

import java.io.File;
import java.io.FileNotFoundException;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.List;

import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter;
import com.github.ejahns.model.Feature;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

public class Main {

	@Parameter(names = {"-h", "--help"}, description = "Display help", help = true)
	private boolean help;

	@Parameter(names = {"-l", "--log"}, description = "Log and report errors rather than throwing exceptions")
	private boolean logErrors;

	@Parameter(names = {"-p", "--pretty"}, description = "Pretty print json")
	private boolean pretty;

	@Parameter(names = {"-s", "--suppress"}, description = "Suppress json output, only report feature validity")
	private boolean suppress;

	@Parameter(description = "files")
	private List files;

	public static void main(String[] args) throws URISyntaxException, FileNotFoundException, MalformedURLException {
		Main main = new Main();
		JCommander commander = JCommander.newBuilder()
			.addObject(main)
			.programName("brine")
			.build();
		commander.parse(args);
		if (args.length == 0 || main.help) {
			commander.usage();
			return;
		}
		main.run();
	}

	private void run() throws FileNotFoundException {
		ArrayList fileList = new ArrayList<>();
		for (String s : files) {
			findFeatures(new File(s), fileList);
		}
		Gson gson;
		if (!pretty) {
			gson = new Gson();
		}
		else {
			gson = new GsonBuilder().setPrettyPrinting().create();
		}
		for (File f : fileList) {
			if (!logErrors) {
				Writers.out.println("------------------------------------------------------------------------------------");
				Writers.out.println(f.toString());
				Writers.out.println("------------------------------------------------------------------------------------");
				Feature cure = PickleJar.cure(f);
				Writers.out.println(gson.toJson(cure));
				Writers.out.println();
			}
			else {
				Writers.out.println("------------------------------------------------------------------------------------");
				Writers.out.println(f.toString());
				Writers.out.println("------------------------------------------------------------------------------------");
				List errors = new ArrayList<>();
				Feature cure = PickleJar.cureCollectErrors(f, errors);
				if (errors.size() > 0) {
					Writers.out.println("\terrors:");
					for (String s : errors) {
						Writers.out.println("\t\t" + s);
					}
					if (!suppress) {
						Writers.out.println(gson.toJson(cure));
					}
				}
				else {
					if (!suppress) {
						Writers.out.println(gson.toJson(cure));
					}
					else {
						Writers.out.println("\tvalid");
					}
				}
				Writers.out.println();
			}
		}
	}

	private void findFeatures(File f, List features) {
		if (f.isFile()) {
			if (f.getName().endsWith(".feature")) {
				features.add(f);
			}
			return;
		}
		File[] files = f.listFiles();
		if (files != null) {
			for (File f2 : files) {
				findFeatures(f2, features);
			}
		}
	}
}