using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading;
using System.Windows.Automation;
using System.Windows.Forms;
using WebSocketSharp;
using WebSocketSharp.Server;

namespace ChromeAutomation
{
    public struct CursorPoint
    {
        public int X;

        public int Y;
    }
    partial class Form1
    {

        [DllImport("gdi32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int CreatePen(int nPenStyle, int nWidth, int crColor);

        [DllImport("gdi32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int SelectObject(int hdc, int hObject);

        [DllImport("gdi32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int SetROP2(int hdc, int nDrawMode);

        [DllImport("gdi32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int Rectangle(int hdc, int X1, int Y1, int X2, int Y2);

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int GetDC(int hwnd);

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int ReleaseDC(int hwnd, int hdc);

        [DllImport("gdi32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int DeleteObject(int hObject);

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int ReleaseCapture();

        [DllImport("user32.dll")]
        public static extern bool SetForegroundWindow(IntPtr hWnd);

        [DllImport("user32.dll")]
        internal static extern bool GetPhysicalCursorPos(ref ChromeAutomation.CursorPoint lpPoint);

        [DllImport("user32.dll", CallingConvention = CallingConvention.Cdecl)]
        public static extern IntPtr WindowFromPoint(System.Windows.Point point);

        [DllImport("user32.dll")]
        private static extern int GetClassName(IntPtr hWnd, StringBuilder lpString, int nMaxCount);

        public delegate void RS_DATA_CALLBACK(int left, int top, int width, int height, Dictionary<string, object> dictionary);

        private static RS_DATA_CALLBACK MessageCallback = null;

        private static AutomationElement selectedElement = null;

        public static Form1.SocketBehavior CureentSocketBehavior = null;

        public static void RegisterMessageCallback(RS_DATA_CALLBACK callback)
        {
            Form1.MessageCallback = callback;
        }

        public class SocketBehavior : WebSocketBehavior {
            protected override void OnOpen()
            {
                base.OnOpen();
                Console.WriteLine("WebSocket OnOpen");
                Form1.CureentSocketBehavior = this;
                Send("{\"event\":\"UI\"}");
            }

            

            protected override void OnMessage(MessageEventArgs e)
            {
                Console.WriteLine("WebSocket OnMessage:" + e.Data);
                JObject jo = new JObject();
                try {
                    jo = (JObject)JsonConvert.DeserializeObject(e.Data);
                }
                catch (Exception) { }
                if (jo.ContainsKey("event") && jo["event"].ToString().Equals("UI") && jo.ContainsKey("data"))
                {
                    Console.WriteLine("来自客户端的xpath消息:" + e.Data.ToString());
                    try
                    {
                        IHTML iHTML = JsonConvert.DeserializeObject<IHTML>(jo["data"].ToString());
                        int left = iHTML.left;
                        int top = iHTML.top;
                        int width = iHTML.width;
                        int height = iHTML.height;
                        Dictionary<string, object> dictionary = new Dictionary<string, object>();
                        dictionary.Add("id", iHTML.id);
                        dictionary.Add("tag", iHTML.tag);
                        dictionary.Add("text", iHTML.text);
                        dictionary.Add("value", iHTML.value);
                        dictionary.Add("xpath", iHTML.xpath);
                        dictionary.Add("full_xpath", iHTML.full_xpath);
                        Console.WriteLine("OnMessage:" + (left, top, width, height));
                        MessageCallback(left, top, width, height, dictionary);
                    }
                    catch (Exception)
                    {
                    }
                    bool flag2 = ((int)Form1.GetAsyncKeyState(17) & 32768) != 0;
                    if (flag2)
                    {
                        Application.Exit();
                    }
                    
                }
                else if(jo.ContainsKey("event") && jo["event"].ToString().Equals("CHECK"))
                {
                    System.Drawing.Point point = new System.Drawing.Point(-1, -1);
                    ChromeAutomation.CursorPoint cursorPoint = default(ChromeAutomation.CursorPoint);
                    Form1.GetPhysicalCursorPos(ref cursorPoint);
                    System.Drawing.Point point2 = new System.Drawing.Point(cursorPoint.X, cursorPoint.Y);
                    Console.WriteLine("cursorPoint：x:" + cursorPoint.X.ToString() + ",y:" + cursorPoint.Y.ToString());
                    bool flag4 = Form1.CureentSocketBehavior != null;
                    System.Windows.Point point3 = new System.Windows.Point((double)point2.X, (double)point2.Y);
                    Form1.selectedElement = AutomationElement.FromPoint(point3);
                    AutomationElement.AutomationElementInformation current = Form1.selectedElement.Current;
                    IntPtr hWnd = Form1.WindowFromPoint(point3);
                    StringBuilder stringBuilder = new StringBuilder(256);
                    Form1.GetClassName(hWnd, stringBuilder, stringBuilder.Capacity);
                    AutomationElement.AutomationElementInformation current2 = Form1.selectedElement.Current;
                    Console.WriteLine("Form1.CureentSocketBehavior:" + flag4);
                    if (flag4)
                    {
                        int x = point2.X;
                        int y = point2.Y;
                        Dictionary<string, object> dictionary = new Dictionary<string, object>();
                        dictionary.Add("event", "UI");
                        dictionary.Add("data", new Dictionary<string, object>
                            {
                                {
                                    "x",
                                    x
                                },
                                {
                                    "y",
                                    y
                                }
                            });
                        Console.WriteLine("鼠标坐标：x:" + x.ToString() + ",y:" + y.ToString());
                        SendMessage(JsonConvert.SerializeObject(dictionary));
                    }
                }
            }

            public void SendMessage(string msg)
            {
                base.Send(msg);
            }
        }

        /// <summary>
        /// 必需的设计器变量。
        /// </summary>
        private System.ComponentModel.IContainer components = null;

        /// <summary>
        /// 清理所有正在使用的资源。
        /// </summary>
        /// <param name="disposing">如果应释放托管资源，为 true；否则为 false。</param>
        protected override void Dispose(bool disposing)
        {
            if (disposing && (components != null))
            {
                components.Dispose();
            }
            base.Dispose(disposing);
        }

        private int hDC;

        private int left = 0, top = 0, width = 0, height = 0;

        Dictionary<string, object> dictionary;

        public void DrawRect(int left, int top, int width, int height, Dictionary<string, object> dictionary)
        {
            // 回调实现方法
            Console.WriteLine("DrawRect:" + (left, top, width, height));
            this.left = left;
            this.top = top;
            this.width = width;
            this.height = height;
            this.dictionary = dictionary;
            // this.loadRectangle();
        }

        [DebuggerBrowsable(DebuggerBrowsableState.Never), AccessedThroughProperty("Timer1"), CompilerGenerated]
        private System.Windows.Forms.Timer _Timer1;

        internal virtual System.Windows.Forms.Timer Timer1
        {
            [CompilerGenerated]
            get
            {
                return this._Timer1;
            }
            [CompilerGenerated]
            [MethodImpl(MethodImplOptions.Synchronized)]
            set
            {
                EventHandler value2 = new EventHandler(this.Timer1_Tick);
                System.Windows.Forms.Timer timer = this._Timer1;
                if (timer != null)
                {
                    timer.Tick -= value2;
                }
                this._Timer1 = value;
                timer = this._Timer1;
                if (timer != null)
                {
                    timer.Tick += value2;
                }
            }
        }

        private void loadRectangle()
        {
            Form1.Rectangle(this.hDC, left, top + height, left + width, top);
            Thread.Sleep(300);
            Form1.Rectangle(this.hDC, left, top + height, left + width, top);
            Thread.Sleep(500);
        }

        private void Timer1_Tick(object sender, EventArgs e)
        {
            if (left > 0 && top > 0)
            {
                Form1.Rectangle(this.hDC, left, top + height, left + width, top);
                Thread.Sleep(300);
                Form1.Rectangle(this.hDC, left, top + height, left + width, top);
                Thread.Sleep(500);
            }
        }


        #region Windows 窗体设计器生成的代码

        /// <summary>
        /// 设计器支持所需的方法 - 不要修改
        /// 使用代码编辑器修改此方法的内容。
        /// </summary>
        /// 

        private int Pen;

        private int PreviousPen;

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int RegisterHotKey(int hwnd, int id, int fsModifiers, int vk);

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern int UnregisterHotKey(int hwnd, int id);

        [DllImport("user32.dll", CharSet = CharSet.Ansi, ExactSpelling = true, SetLastError = true)]
        public static extern short GetAsyncKeyState(int vKey);

        private KeyEventHandler myKeyEventHandeler = null;//按键钩子
        private KeyboardHook k_hook = new KeyboardHook();

        // private Object o = new Object();

        private void hook_KeyDown(object sender, KeyEventArgs e)
        {
            try {
                lock (this)
                {
                    //  这里写具体实现
                    Console.WriteLine("按下按键" + e.KeyValue);
                    if (e.KeyValue == 162)
                    {
                        
                        // 先移开鼠标截图，然后还原鼠标位置
                        Point screenPoint = Control.MousePosition;
                        MouseHookHelper.SetCursorPos(0, 0);
                        // MouseHookHelper.mouse_event(MouseHookHelper.MOUSEEVENTF_MOVE, 0, 1000, 0, 0);
                        Thread.Sleep(400);
                        String imagePath = this.SaveImage(left, top, width, height);
                        MouseHookHelper.SetCursorPos(screenPoint.X, screenPoint.Y);
                        // string base64FromImage = ImageUtil.GetBase64FromImage(image);
                        dictionary.Add("screenShot", imagePath);
                        dictionary.Add("x", left);
                        dictionary.Add("y", top);
                        dictionary.Add("w", width + left);
                        dictionary.Add("h", height + top);
                        Console.WriteLine("imagePath", imagePath);
                        DeleteObject(this.Pen);
                        DeleteObject(this.PreviousPen);
                        Socket socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
                        //连接服务器
                        socket.Connect(new IPEndPoint(IPAddress.Parse("127.0.0.1"), 10083));
                        dictionary.Add("dataType", "CHROME_UIA");
                        socket.Send(Encoding.UTF8.GetBytes(JsonConvert.SerializeObject(dictionary)));
                        socket.Close();
                        try
                        {

                            // 关闭
                            stopListen();
                            Thread.Sleep(400);
                            this.Timer1.Stop();
                            this.Close();
                            Application.Exit();
                        }
                        catch (Exception)
                        {
                            // 关闭
                            stopListen();
                            Thread.Sleep(400);
                            this.Timer1.Stop();
                            this.Close();
                            Application.Exit();
                        }
                    }
                    else if (e.KeyValue == 27)
                    {
                        // 关闭
                        stopListen();
                        this.Timer1.Stop();
                        this.Close();
                        Application.Exit();
                    }
                }
            } catch (Exception)
            { 

            }
        }

        // 保存截图
        private String SaveImage(int x, int y, int width, int height)
        {
            Bitmap bitmap = new Bitmap(width, height);
            Graphics graphics = Graphics.FromImage(bitmap);
            graphics.CopyFromScreen(x, y, 0, 0, new System.Drawing.Size(width, height));
            String imagePath = Directory.GetCurrentDirectory() + "\\img.png";
            bitmap.Save(imagePath, ImageFormat.Png);
            return imagePath;
        }

        public void stopListen()
        {
            if (myKeyEventHandeler != null)
            {
                k_hook.KeyDownEvent -= myKeyEventHandeler;//取消按键事件
                myKeyEventHandeler = null;
                k_hook.Stop();//关闭键盘钩子
            }
        }

        private void InitializeComponent()
        {
            this.components = new System.ComponentModel.Container();
            this.AutoScaleMode = System.Windows.Forms.AutoScaleMode.Font;
            this.ClientSize = new System.Drawing.Size(800, 450);
            this.Text = "谷歌浏览器拾取插件";
            CheckForIllegalCrossThreadCalls = false;
            this.hDC = Form1.GetDC(0);
            this.Pen = Form1.CreatePen(0, 3, 255);
            this.PreviousPen = Form1.SelectObject(this.hDC, this.Pen);
            Form1.SetROP2(this.hDC, 10);
            this.Timer1 = new System.Windows.Forms.Timer(this.components);
            this.Timer1.Start();
            base.Name = "CINDATA AUTOMATION";
            base.ClientSize = new System.Drawing.Size(200, 30);
            base.StartPosition = FormStartPosition.Manual;
            Rectangle rect = Screen.GetWorkingArea(this);
            System.Drawing.Point p = new System.Drawing.Point(rect.Width - 250, rect.Height - 40);
            this.Location = p;
            this.TopMost = true;
            this.ControlBox = false;
            base.ShowInTaskbar = false;

            // websocket
            WebSocketServer webSocketServer = new WebSocketServer("ws://127.0.0.1:63360");
            // WebSocketServer webSocketServer = new WebSocketServer("ws://127.0.0.1:63366");
            string uri = "/chrome";
            webSocketServer.AddWebSocketService<SocketBehavior>(uri);
            webSocketServer.Start();
            RegisterMessageCallback(new RS_DATA_CALLBACK(this.DrawRect));

            myKeyEventHandeler = new KeyEventHandler(hook_KeyDown);
            k_hook.KeyDownEvent += myKeyEventHandeler;//钩住键按下
            k_hook.Start();//安装键盘钩子
        }

        #endregion
    }
}