@echo off
:: 将当前RDP会话无缝切换到服务器本地控制台
:: 保持会话后台锁定但活动，图形渲染不中断
for /f "skip=1 tokens=3" %%s in ('query user %USERNAME%') do (
  %windir%\System32\tscon.exe %%s /dest:console
)

using System;
using System.Drawing;
using System.Drawing.Imaging;
using System.Threading;
using System.Windows.Forms;
using System.Runtime.InteropServices;

namespace ConsoleApp1
{
    class Program
    {
        // 保存选择区域时的信息
        private static RECT _selectedRect;
        private static Size _selectTimeScreenSize;

        [STAThread] // 必须标记，否则WinForms窗口无法正常显示
        static void Main(string[] args)
        {
            Console.WriteLine("=== 腾讯云RDP区域截屏工具 ===");
            Console.WriteLine("提示：按ESC可取消区域选择");
            Console.WriteLine("按任意键开始选择区域...");
            Console.ReadKey();

            // 1. 记录选择区域时的屏幕分辨率
            _selectTimeScreenSize = Screen.PrimaryScreen.Bounds.Size;

            // 2. 打开区域选择窗口
            Console.WriteLine("\n正在打开区域选择窗口...");
            using (RegionSelectForm selectForm = new RegionSelectForm())
            {
                if (selectForm.ShowDialog() != DialogResult.OK)
                {
                    Console.WriteLine("已取消选择，程序退出。");
                    return;
                }

                _selectedRect = selectForm.SelectedRect;
                if (_selectedRect.Right <= _selectedRect.Left || _selectedRect.Bottom <= _selectedRect.Top)
                {
                    Console.WriteLine("错误：选择的区域无效");
                    return;
                }
            }

            Console.WriteLine($"\n已选择区域：({_selectedRect.Left},{_selectedRect.Top})-({_selectedRect.Right},{_selectedRect.Bottom})");
            
            // 3. 5秒倒计时
            Console.WriteLine("倒计时开始：");
            for (int i = 5; i > 0; i--)
            {
                Console.Write($"{i}... ");
                Thread.Sleep(1000);
            }
            Console.WriteLine("\n正在截屏...");

            // 4. 执行截屏
            CaptureWithAutoScale();

            Console.WriteLine("\n按任意键退出...");
            Console.ReadKey();
        }

        /// <summary>
        /// 自动适配分辨率的截屏方法（核心）
        /// </summary>
        private static void CaptureWithAutoScale()
        {
            IntPtr hScreenDC = IntPtr.Zero;
            IntPtr hMemDC = IntPtr.Zero;
            IntPtr hBitmap = IntPtr.Zero;
            IntPtr hOldBitmap = IntPtr.Zero;

            try
            {
                // 1. 获取截屏时（控制台）的实际物理分辨率
                Size captureTimeScreenSize = GetPhysicalScreenSize();
                
                // 2. 计算分辨率缩放比例
                float scaleX = (float)captureTimeScreenSize.Width / _selectTimeScreenSize.Width;
                float scaleY = (float)captureTimeScreenSize.Height / _selectTimeScreenSize.Height;

                // 3. 自动修正坐标
                int scaledLeft = (int)Math.Round(_selectedRect.Left * scaleX);
                int scaledTop = (int)Math.Round(_selectedRect.Top * scaleY);
                int scaledRight = (int)Math.Round(_selectedRect.Right * scaleX);
                int scaledBottom = (int)Math.Round(_selectedRect.Bottom * scaleY);

                int width = scaledRight - scaledLeft;
                int height = scaledBottom - scaledTop;

                // 4. 纯Win32 API截屏
                hScreenDC = NativeMethods.CreateDC("DISPLAY", null, null, IntPtr.Zero);
                hMemDC = NativeMethods.CreateCompatibleDC(hScreenDC);
                hBitmap = NativeMethods.CreateCompatibleBitmap(hScreenDC, width, height);
                hOldBitmap = NativeMethods.SelectObject(hMemDC, hBitmap);

                NativeMethods.BitBlt(hMemDC, 0, 0, width, height, hScreenDC, scaledLeft, scaledTop, 0x00CC0020);

                // 5. 保存图片
                string fileName = $"screenshot_{DateTime.Now:yyyyMMdd_HHmmss}.jpg";
                string path = System.IO.Path.Combine(AppDomain.CurrentDomain.BaseDirectory, fileName);
                
                using (Bitmap bmp = Image.FromHbitmap(hBitmap))
                {
                    bmp.Save(path, ImageFormat.Jpeg);
                }

                // 6. 输出结果
                Console.WriteLine("\n========== 截屏成功 ==========");
                Console.WriteLine($"保存路径：{path}");
                Console.WriteLine($"\n【调试信息】");
                Console.WriteLine($"选择时分辨率：{_selectTimeScreenSize.Width}×{_selectTimeScreenSize.Height}");
                Console.WriteLine($"截屏时分辨率：{captureTimeScreenSize.Width}×{captureTimeScreenSize.Height}");
                Console.WriteLine($"缩放比例：X={scaleX:F2}, Y={scaleY:F2}");
                Console.WriteLine($"原始坐标：({_selectedRect.Left},{_selectedRect.Top})-({_selectedRect.Right},{_selectedRect.Bottom})");
                Console.WriteLine($"修正后坐标：({scaledLeft},{scaledTop})-({scaledRight},{scaledBottom})");
                Console.WriteLine("==============================");
            }
            catch (Exception ex)
            {
                Console.WriteLine($"\n错误：截屏失败 - {ex.Message}");
            }
            finally
            {
                // 严格释放GDI资源
                if (hOldBitmap != IntPtr.Zero) NativeMethods.SelectObject(hMemDC, hOldBitmap);
                if (hBitmap != IntPtr.Zero) NativeMethods.DeleteObject(hBitmap);
                if (hMemDC != IntPtr.Zero) NativeMethods.DeleteDC(hMemDC);
                if (hScreenDC != IntPtr.Zero) NativeMethods.DeleteDC(hScreenDC);
            }
        }

        /// <summary>
        /// 获取物理屏幕的实际分辨率
        /// </summary>
        private static Size GetPhysicalScreenSize()
        {
            IntPtr hDC = NativeMethods.CreateDC("DISPLAY", null, null, IntPtr.Zero);
            int width = NativeMethods.GetDeviceCaps(hDC, 8); // HORZRES
            int height = NativeMethods.GetDeviceCaps(hDC, 10); // VERTRES
            NativeMethods.DeleteDC(hDC);
            return new Size(width, height);
        }

        #region 区域选择窗口（独立类，复用原逻辑）
        private class RegionSelectForm : Form
        {
            private Point _startPoint;
            private Point _endPoint;
            private bool _isSelecting;
            public RECT SelectedRect { get; private set; }

            public RegionSelectForm()
            {
                this.FormBorderStyle = FormBorderStyle.None;
                this.Bounds = GetAllScreensBounds();
                this.BackColor = Color.Black;
                this.Opacity = 0.3;
                this.TopMost = true;
                this.ShowInTaskbar = false;
                this.DoubleBuffered = true;
                this.Cursor = Cursors.Cross;
                this.Text = "拖动鼠标选择区域，ESC取消";

                this.MouseDown += RegionSelectForm_MouseDown;
                this.MouseMove += RegionSelectForm_MouseMove;
                this.MouseUp += RegionSelectForm_MouseUp;
                this.Paint += RegionSelectForm_Paint;
                this.KeyDown += RegionSelectForm_KeyDown;
            }

            private Rectangle GetAllScreensBounds()
            {
                int left = int.MaxValue, top = int.MaxValue;
                int right = int.MinValue, bottom = int.MinValue;
                foreach (Screen screen in Screen.AllScreens)
                {
                    left = Math.Min(left, screen.Bounds.Left);
                    top = Math.Min(top, screen.Bounds.Top);
                    right = Math.Max(right, screen.Bounds.Right);
                    bottom = Math.Max(bottom, screen.Bounds.Bottom);
                }
                return new Rectangle(left, top, right - left, bottom - top);
            }

            private void RegionSelectForm_MouseDown(object sender, MouseEventArgs e)
            {
                if (e.Button == MouseButtons.Left)
                {
                    _isSelecting = true;
                    _startPoint = Control.MousePosition;
                    _endPoint = _startPoint;
                }
            }

            private void RegionSelectForm_MouseMove(object sender, MouseEventArgs e)
            {
                if (_isSelecting)
                {
                    _endPoint = Control.MousePosition;
                    this.Invalidate();
                }
            }

            private void RegionSelectForm_MouseUp(object sender, MouseEventArgs e)
            {
                if (e.Button == MouseButtons.Left && _isSelecting)
                {
                    _isSelecting = false;
                    _endPoint = Control.MousePosition;

                    int left = Math.Min(_startPoint.X, _endPoint.X);
                    int top = Math.Min(_startPoint.Y, _endPoint.Y);
                    int right = Math.Max(_startPoint.X, _endPoint.X);
                    int bottom = Math.Max(_endPoint.Y, _startPoint.Y);
                    
                    SelectedRect = new RECT(left, top, right, bottom);
                    this.DialogResult = DialogResult.OK;
                    this.Close();
                }
            }

            private void RegionSelectForm_Paint(object sender, PaintEventArgs e)
            {
                if (_isSelecting)
                {
                    Point clientStart = this.PointToClient(_startPoint);
                    Point clientEnd = this.PointToClient(_endPoint);

                    int x = Math.Min(clientStart.X, clientEnd.X);
                    int y = Math.Min(clientStart.Y, clientEnd.Y);
                    int w = Math.Abs(clientEnd.X - clientStart.X);
                    int h = Math.Abs(clientEnd.Y - clientStart.Y);

                    using (Pen pen = new Pen(Color.Lime, 3))
                    {
                        e.Graphics.DrawRectangle(pen, x, y, w, h);
                    }
                }
            }

            private void RegionSelectForm_KeyDown(object sender, KeyEventArgs e)
            {
                if (e.KeyCode == Keys.Escape)
                {
                    this.DialogResult = DialogResult.Cancel;
                    this.Close();
                }
            }
        }
        #endregion

        #region Win32 API
        [StructLayout(LayoutKind.Sequential)]
        public struct RECT
        {
            public int Left;
            public int Top;
            public int Right;
            public int Bottom;

            public RECT(int left, int top, int right, int bottom)
            {
                Left = left;
                Top = top;
                Right = right;
                Bottom = bottom;
            }
        }

        private static class NativeMethods
        {
            [DllImport("gdi32.dll")]
            public static extern IntPtr CreateDC(string lpszDriver, string lpszDevice, string lpszOutput, IntPtr lpInitData);
            
            [DllImport("gdi32.dll")]
            public static extern IntPtr CreateCompatibleDC(IntPtr hdc);
            
            [DllImport("gdi32.dll")]
            public static extern IntPtr CreateCompatibleBitmap(IntPtr hdc, int nWidth, int nHeight);
            
            [DllImport("gdi32.dll")]
            public static extern IntPtr SelectObject(IntPtr hdc, IntPtr hgdiobj);
            
            [DllImport("gdi32.dll")]
            public static extern bool BitBlt(IntPtr hdcDest, int nXDest, int nYDest, int nWidth, int nHeight, IntPtr hdcSrc, int nXSrc, int nYSrc, uint dwRop);
            
            [DllImport("gdi32.dll")]
            public static extern bool DeleteObject(IntPtr hObject);
            
            [DllImport("gdi32.dll")]
            public static extern bool DeleteDC(IntPtr hdc);
            
            [DllImport("gdi32.dll")]
            public static extern int GetDeviceCaps(IntPtr hdc, int nIndex);
        }
        #endregion
    }
}