package edu.ufl.cise.plc.runtime;

/**
 * Class to support runtime IO with source or destination "console" in PLCLang
 * 
 */
import static javax.swing.WindowConstants.EXIT_ON_CLOSE;

import java.awt.image.BufferedImage;
import java.io.InputStream;
import java.io.PrintStream;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.InputMismatchException;
import java.util.Scanner;

import javax.swing.ImageIcon;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.SwingUtilities;

public class ConsoleIO {

	/**
	 * Destination of "console" output. Can be changed to redirect output. Generated
	 * code should use ConsoleIO.console.println(...) etc. instead of
	 * System.out.println
	 */
	public static PrintStream console = System.out;

	/**
	 * change destination of console output for non-image types
	 */
	public static void setConsole(PrintStream out) {
		console = out;
	}

	/** Default source of "console" input. */
	public static InputStream consoleInput = System.in;

	/** Change source of "console" input */
	public static void setConsoleInput(InputStream in) {
		consoleInput = in;
	}

	/*
	 * java.util.Scanner for input from "console" Implementation is a singleton.
	 */
	private static Scanner scanner;

	private static Scanner getScanner() {
		if (scanner == null) {
			scanner = new Scanner(consoleInput);
		}
		return scanner;
	}

	public static void resetScanner() {
		if (scanner != null) {
			scanner.close();
			scanner = null;
		}
	}

	/**
	 * Reads a value of the given type from the console. The type must be one of
	 * "INT", "FLOAT", "STRING", or "COLOR". If the scanner cannot convert the input
	 * token to the expected type, it will print "INVALID INPUT" and read another
	 * token.
	 * 
	 * See readImage to read an image.
	 * 
	 * @param type   type of value to read
	 * @param prompt prompt to user for value
	 * @return
	 */
	public static Object readValueFromConsole(String type, String prompt) {
		System.out.print(prompt);
		Scanner scanner = getScanner();
		try {
			return switch (type) {
			case "INT" -> {
				int val = scanner.nextInt();
				scanner.nextLine();
				yield val;
			}
			case "FLOAT" -> {
				float val = scanner.nextFloat();
				scanner.nextLine();
				yield val;
			}
			case "STRING" -> {
				yield scanner.nextLine();
			}
			case "COLOR" -> {
				int r = scanner.nextInt();
				int g = scanner.nextInt();
				int b = scanner.nextInt();
				scanner.nextLine();
				yield new ColorTuple(r, g, b);
			}
			case "BOOLEAN" -> {
				boolean val = scanner.nextBoolean();
				scanner.nextLine();
				yield val;
			}
			default -> throw new IllegalArgumentException("Compiler bug Unexpected value: " + type);
			};
		} catch (InputMismatchException e) {
			System.out.print("INVALID INPUT ");
			getScanner().next(); // throw away invalid input token
			return readValueFromConsole(type, prompt);
		}
	}

	public static boolean DISPLAY_IMAGES = true;
	public static ArrayList<BufferedImage> consoleImages = new ArrayList<>();

	public static void displayImageOnScreen(BufferedImage image) {
		consoleImages.add(image);
		if (DISPLAY_IMAGES) {
			System.err.println("Displaying image = " + image);
			JFrame frame = new JFrame();
			frame.setDefaultCloseOperation(EXIT_ON_CLOSE);
			frame.setSize(image.getWidth(), image.getHeight());
			JLabel label = new JLabel(new ImageIcon(image));
			frame.add(label);
			frame.pack();
			try {
				SwingUtilities.invokeAndWait(new Runnable() {
					public void run() {
						frame.setVisible(true);
					}
				});
			} catch (InvocationTargetException | InterruptedException e) {
				e.printStackTrace();
			}
		}

	}

	public static void resetConsoleImages() {
		consoleImages = new ArrayList<>();
	}

	/**
	 * Displays the given image on the screen. The difference between this and
	 * displayImageOnScreen is the location of the image.
	 * 
	 * @param image
	 */
	public static void displayReferenceImageOnScreen(BufferedImage image) {
		System.err.println("Displaying image = " + image);
		JFrame frame = new JFrame();
		frame.setLocationRelativeTo(null);
		frame.setDefaultCloseOperation(EXIT_ON_CLOSE);
		frame.setSize(image.getWidth(), image.getHeight());
		JLabel label = new JLabel(new ImageIcon(image));
		frame.add(label);
		frame.pack();
		try {
			SwingUtilities.invokeAndWait(new Runnable() {
				public void run() {
					frame.setVisible(true);
				}
			});
		} catch (InvocationTargetException | InterruptedException e) {
			e.printStackTrace();
		}
	}

}