1
internal class NativeApi
2
{
3
    public const int STD_OUTPUT_HANDLE = -11;
4
    public const uint ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004;
5
    public const uint ENABLE_MOUSE_INPUT = 0x0010;
6
    public const int STD_INPUT_HANDLE = -10;
7
    public const uint ENABLE_QUICK_EDIT_MODE = 0x0040;
8

9
    [DllImport("kernel32.dll", SetLastError = true)]
10
    public static extern IntPtr GetStdHandle(int nStdHandle);
11

12
    [DllImport("kernel32.dll")]
13
    public static extern bool GetConsoleMode(IntPtr hConsoleHandle, out uint lpMode);
14

15
    [DllImport("kernel32.dll")]
16
    public static extern bool SetConsoleMode(IntPtr hConsoleHandle, uint dwMode);
17
    public static void EnableANSI()
18
    {
19
        var handle = GetStdHandle(STD_OUTPUT_HANDLE);
20
        GetConsoleMode(handle, out var mode);
21
        mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
22
        SetConsoleMode(handle, mode);
23
    }
24
    public static void EnableMouseInput()
25
    {
26
        var handle = GetStdHandle(STD_INPUT_HANDLE);
27
        GetConsoleMode(handle, out var mode);
28
        mode |= ENABLE_MOUSE_INPUT;
29
        mode &= ~ENABLE_QUICK_EDIT_MODE; //关闭快速编辑
30
        SetConsoleMode(handle, mode);
31
    }
32
    [DllImport("kernel32.dll", CharSet = CharSet.Unicode, SetLastError = true, EntryPoint = "ReadConsoleInputW")]
33
    public static extern bool ReadConsoleInput(
34
    IntPtr hConsoleInput,
35
    [Out] INPUT_RECORD[] lpBuffer,
36
    uint nLength,
37
    out uint lpNumberOfEventsRead);
38
    public enum InputRecordEventType : uint
39
    {
40
        KEY_EVENT = 0x1,
41
        MOUSE_EVENT = 0x2,
42
        WINDOW_BUFFER_SIZE_EVENT = 0x4,
43
        MENU_EVENT = 0x8,
44
        FOCUS_EVENT = 0x10,
45
    }
46

47
    [StructLayout(LayoutKind.Explicit)]
48
    public struct INPUT_RECORD
49
    {
50
        [FieldOffset(0)]
51
        public InputRecordEventType EventType;
52
        [FieldOffset(4)]
53
        public KEY_EVENT_RECORD KeyEvent;
54
        [FieldOffset(4)]
55
        public MOUSE_EVENT_RECORD MouseEvent;
56
        [FieldOffset(4)]
57
        public WINDOW_BUFFER_SIZE_RECORD WindowBufferSizeEvent;
58
        [FieldOffset(4)]
59
        public MENU_EVENT_RECORD MenuEvent;
60
        [FieldOffset(4)]
61
        public FOCUS_EVENT_RECORD FocusEvent;
62
    };
63
    [StructLayout(LayoutKind.Explicit, CharSet = CharSet.Unicode)]
64
    public struct KEY_EVENT_RECORD
65
    {
66
        [FieldOffset(0), MarshalAs(UnmanagedType.Bool)]
67
        public bool bKeyDown;
68
        [FieldOffset(4), MarshalAs(UnmanagedType.U2)]
69
        public ushort wRepeatCount;
70
        // ConsoleKey
71
        [FieldOffset(6), MarshalAs(UnmanagedType.U2)]
72
        public ushort wVirtualKeyCode;
73
        [FieldOffset(8), MarshalAs(UnmanagedType.U2)]
74
        public ushort wVirtualScanCode;
75
        [FieldOffset(10)]
76
        public char UnicodeChar;
77
        [FieldOffset(12), MarshalAs(UnmanagedType.U4)]
78
        public ControlKeyState dwControlKeyState;
79
    }
80

81
    // dwControlKeyState bitmask
82
    [Flags]
83
    public enum ControlKeyState
84
    {
85
        RIGHT_ALT_PRESSED = 0x1,
86
        LEFT_ALT_PRESSED = 0x2,
87
        RIGHT_CTRL_PRESSED = 0x4,
88
        LEFT_CTRL_PRESSED = 0x8,
89
        SHIFT_PRESSED = 0x10,
90
        NUMLOCK_ON = 0x20,
91
        SCROLLLOCK_ON = 0x40,
92
        CAPSLOCK_ON = 0x80,
93
        ENHANCED_KEY = 0x100
94
    }
95
    [StructLayout(LayoutKind.Sequential)]
96
    public struct MOUSE_EVENT_RECORD
97
    {
98
        public COORD dwMousePosition;
99
        public MouseButtonState dwButtonState;
100
        public short MouseDelta;
101
        public ControlKeyState dwControlKeyState;
102
        public MouseEventFlags dwEventFlags;
103
    }
104

105
    [Flags]
106
    public enum MouseButtonState : short
107
    {
108
        FROM_LEFT_1ST_BUTTON_PRESSED = 0x1,
109
        RIGHTMOST_BUTTON_PRESSED = 0x2,
110
        FROM_LEFT_2ND_BUTTON_PRESSED = 0x4,
111
        FROM_LEFT_3RD_BUTTON_PRESSED = 0x8,
112
        FROM_LEFT_4TH_BUTTON_PRESSED = 0x10
113
    }
114
    [StructLayout(LayoutKind.Sequential)]
115
    public struct COORD
116
    {
117
        public short X;
118
        public short Y;
119

120
        public COORD(short X, short Y)
121
        {
122
            this.X = X;
123
            this.Y = Y;
124
        }
125
    };
126
    [Flags]
127
    public enum MouseEventFlags
128
    {
129
        MOUSE_MOVED = 0x1,
130
        DOUBLE_CLICK = 0x2,
131
        MOUSE_WHEELED = 0x4,
132
        MOUSE_HWHEELED = 0x8
133
    }
134
    public struct WINDOW_BUFFER_SIZE_RECORD
135
    {
136
        public COORD dwSize;
137

138
        public WINDOW_BUFFER_SIZE_RECORD(short x, short y)
139
        {
140
            dwSize = new COORD(x, y);
141
        }
142
    }
143
    [StructLayout(LayoutKind.Sequential)]
144
    public struct MENU_EVENT_RECORD
145
    {
146
        public uint dwCommandId;
147
    }
148
    [StructLayout(LayoutKind.Sequential)]
149
    public struct FOCUS_EVENT_RECORD
150
    {
151
        public uint bSetFocus;
152
    }
153
}