package sig;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Arrays;
import java.util.Iterator;
import java.util.regex.Pattern;

import javax.imageio.IIOImage;
import javax.imageio.ImageIO;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;
import javax.imageio.stream.ImageOutputStream;
import javax.swing.JFrame;
import java.awt.Toolkit;
import sig.engine.Panel;

import java.awt.AWTException;
import java.awt.GraphicsEnvironment;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.awt.Robot;
import java.awt.Graphics2D;

public class SigShare {
	static Robot r;
	public static final String PROGRAM_NAME="SigShare";
	public static void main(String[] args) throws AWTException {
		r = new Robot();
		if (args.length==2&&args[1].equalsIgnoreCase("server")) {
			ServerSocket socket;
			try {
				socket = new ServerSocket(4191);
				System.out.println("Listening on port 4191.");
				try (Socket client = socket.accept()) {
					System.out.println("New client connection detected: "+client.toString());
					System.out.println("Taking screenshot...");
					System.out.println("Sending initial data...");
					BufferedReader in = new BufferedReader(new InputStreamReader(client.getInputStream(),"ISO-8859-1"));
					DataOutputStream clientOutput = new DataOutputStream(client.getOutputStream());
					double SCREEN_MULT=2;
					int SCREEN_WIDTH=(int)(GraphicsEnvironment.getLocalGraphicsEnvironment().getMaximumWindowBounds().getWidth()/SCREEN_MULT);
					int SCREEN_HEIGHT=(int)(GraphicsEnvironment.getLocalGraphicsEnvironment().getMaximumWindowBounds().getHeight()/SCREEN_MULT);
					clientOutput.write(("DESKTOP "+SCREEN_WIDTH+" "+SCREEN_HEIGHT+"\r\n").getBytes());
					System.out.println("Send initial screen");
					//char[] screen = new char[SCREEN_WIDTH*SCREEN_HEIGHT];
					/*for (int y=0;y<SCREEN_HEIGHT;y++) {
						for (int x=0;x<SCREEN_WIDTH;x++) {
							int col = pixels[y*SCREEN_WIDTH+x] = screenshot.getRGB(x, y);
							int r = ((col&0x00FF0000)>>>16)/8;
							int g = ((col&0x0000FF00)>>>8)/8;
							int b = ((col&0x000000FF))/8;
							char compressedCol=(char)((r<<10)+(g<<5)+b);
							clientOutput.writeChar(compressedCol);
							//screen[y*SCREEN_WIDTH+x]=compressedCol;
						}	
					}*/
					int frame=0;
					while (true) {
						CaptureScreen();
						FileInputStream stream = new FileInputStream(new File("screenshot.jpg"));
						while (stream.available()>0) {
							clientOutput.writeByte(stream.read());
						}
						stream.close();
						for (int i=0;i<10;i++) {
							clientOutput.writeChar('-');
						}
						System.out.println("Frame "+frame+++" processed. Waiting on client.");
						while (!in.ready()) {
						}
						System.out.println("Client no longer idle.");
						in.readLine();
					}
				} catch (IOException e) {
					e.printStackTrace();
				}
			} catch (IOException e1) {
				e1.printStackTrace();
			}
		} else 
		if (args.length==2&&args[1].equalsIgnoreCase("client")) {
			Socket socket;
			DataOutputStream out;
			DataInputStream in;
			
			JFrame f = new JFrame(PROGRAM_NAME);
			Panel p = new Panel(f);

			try {
				socket = new Socket(args[0],4191);
				out = new DataOutputStream(socket.getOutputStream());
				in=new DataInputStream(socket.getInputStream());
				
			
				while (true) {
					String line;
					if (in.available()>0) {
					line=in.readLine();
					 System.out.println(line);
					 if (line.contains("DESKTOP")) {
						 String[] split = line.split(Pattern.quote(" "));
						 
						int SCREEN_WIDTH=Integer.parseInt(split[1]);
						int SCREEN_HEIGHT=Integer.parseInt(split[2]);
						p.init(SCREEN_WIDTH,SCREEN_HEIGHT);
						
						f.add(p);
						f.setSize(SCREEN_WIDTH,SCREEN_HEIGHT);
						f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
						f.setVisible(true);

						int frame=0;
						int dashCount=0;
						while (true)
						{
							BufferedOutputStream stream = null;
							while (in.available()>0) {
								if (stream==null) {
									System.out.println("Stream opened.");
									stream=new BufferedOutputStream(new FileOutputStream(new File("screenshot_out.jpg")));
								}
								int val = in.read();
								stream.write(val);
								if (val=='-') {
									dashCount++;
								} else 
								if (val!=0) {
									dashCount=0;
								}
							}
							if (dashCount>=10&&stream!=null) {
								stream.close();
								stream=null;
								dashCount=0;
								out.writeChars("Done\r\n");
								System.out.println("Frame "+frame+++" processed.");
							}
						}
					 }
					}
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		} else {
			System.err.println("Args: <Connecting IP Address> server|client");
			return;
		}
	}
	private static void CaptureScreen() throws IOException {
		//BufferedImage screenshot = toBufferedImage(r.createScreenCapture(GraphicsEnvironment.getLocalGraphicsEnvironment().getMaximumWindowBounds()).getScaledInstance(w, h, Image.SCALE_DEFAULT));
		
		ImageOutputStream  ios =  ImageIO.createImageOutputStream(new File("screenshot.jpg"));
		Iterator<ImageWriter> iter = ImageIO.getImageWritersByFormatName("jpeg");
		ImageWriter writer = iter.next();
		ImageWriteParam iwp = writer.getDefaultWriteParam();
		iwp.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
		iwp.setCompressionQuality(0.9f);
		writer.setOutput(ios);
		writer.write(null, new IIOImage(r.createScreenCapture(GraphicsEnvironment.getLocalGraphicsEnvironment().getMaximumWindowBounds()),null,null),iwp);
		writer.dispose();
	
		//return screenshot;
	}
	public static BufferedImage toBufferedImage(Image img)
{
    if (img instanceof BufferedImage)
    {
        return (BufferedImage) img;
    }

    // Create a buffered image with transparency
    BufferedImage bimage = new BufferedImage(img.getWidth(null), img.getHeight(null), BufferedImage.TYPE_INT_ARGB);

    // Draw the image on to the buffered image
    Graphics2D bGr = bimage.createGraphics();
    bGr.drawImage(img, 0, 0, null);
    bGr.dispose();

    // Return the buffered image
    return bimage;
}
}