import json
import urllib.request
import websocket

# ========== 第一步：获取页面的 WebSocket URL ==========

CDP_HTTP = 'http://localhost:9222'

def get_page_ws(pattern=''):
    """获取第一个匹配 pattern 的页面的 WebSocket URL"""
    data = json.loads(
        urllib.request.urlopen(f'{CDP_HTTP}/json', timeout=5).read()
    )
    for page in data:
        if pattern in page.get('url', ''):
            return page['webSocketDebuggerUrl']
    # 如果没有匹配，默认取第一个页面
    return data[0]['webSocketDebuggerUrl'] if data else None

ws_url = get_page_ws()
print(f'Connecting to: {ws_url}')

# ========== 第二步：建立 WebSocket 连接 ==========

ws = websocket.create_connection(ws_url, timeout=30)

# ========== 第三步：封装 send/receive ==========

_request_id = 1

def send_cmd(ws, method, params=None):
    """发送 CDP 命令并等待返回结果"""
    global _request_id
    if params is None:
        params = {}
    _request_id += 1
    request = {'id': _request_id, 'method': method, 'params': params}
    ws.send(json.dumps(request))
    
    while True:
        response = json.loads(ws.recv())
        if response.get('id') == _request_id:
            return response.get('result', {})

# ========== 第四步：启用必要域 ==========

send_cmd(ws, 'Page.enable')       # 启用页面域
send_cmd(ws, 'Runtime.enable')    # 启用运行时域

# ========== 第五步：开始操控浏览器 ==========

# 导航到目标页面
result = send_cmd(ws, 'Page.navigate', {'url': 'https://www.example.com'})
print(f'Navigation started, frameId: {result.get("frameId")}')

# 在当前页面执行 JavaScript
result = send_cmd(ws, 'Runtime.evaluate', {
    'expression': 'document.title',
    'returnByValue': True
})
print(f'Page title: {result["result"]["value"]}')

# 关闭连接
ws.close()

import base64

# 全页截图
result = send_cmd(ws, 'Page.captureScreenshot', {
    'format': 'png',
    'quality': 80,
    'fromSurface': True
})

with open('screenshot.png', 'wb') as f:
    f.write(base64.b64decode(result['data']))
print('Screenshot saved as screenshot.png')


# 指定区域的截图（裁剪）
# clip = x, y, width, height
result = send_cmd(ws, 'Page.captureScreenshot', {
    'format': 'png',
    'clip': {'x': 0, 'y': 0, 'width': 800, 'height': 600, 'scale': 1}
})

# 获取页面信息
result = send_cmd(ws, 'Runtime.evaluate', {
    'expression': 'JSON.stringify({title: document.title, url: location.href, cookies: document.cookie})',
    'returnByValue': True
})
page_info = json.loads(result['result']['value'])
print(page_info)

# 获取元素的文本内容
result = send_cmd(ws, 'Runtime.evaluate', {
    'expression': 'document.querySelector("h1").innerText',
    'returnByValue': True
})
print(f'H1 text: {result["result"]["value"]}')

# 修改页面（可以执行任何 JS 操作）
send_cmd(ws, 'Runtime.evaluate', {
    'expression': 'document.title = "被 CDP 修改的标题"',
    'returnByValue': True
})

# 获取文档根节点
result = send_cmd(ws, 'DOM.getDocument')
root_node_id = result['root']['nodeId']

# 通过选择器查找元素
result = send_cmd(ws, 'DOM.querySelector', {
    'nodeId': root_node_id,
    'selector': 'div.content'
})
content_node_id = result['nodeId']

# 获取元素的 HTML
result = send_cmd(ws, 'DOM.getOuterHTML', {
    'nodeId': content_node_id
})
print(f'Element HTML: {result["outerHTML"][:200]}...')

# 修改元素的属性
send_cmd(ws, 'DOM.setAttributeValue', {
    'nodeId': content_node_id,
    'name': 'style',
    'value': 'background-color: yellow;'
})

# 启用网络域
send_cmd(ws, 'Network.enable')

# 设置请求拦截的回调
def on_request(event_data):
    """每次有网络请求时被调用"""
    request = event_data['params']['request']
    url = request['url']
    method = request['method']
    print(f'[{method}] {url}')
    
    # 可以修改请求头
    # 返回 {'continue': True} 表示继续请求

# 注册请求事件监听
# CDP 的事件通过 WebSocket 主动推送，需要单独处理
import threading

def event_listener(ws):
    """后台线程：持续接收 CDP 事件"""
    while True:
        try:
            msg = json.loads(ws.recv())
            if 'method' in msg:
                if msg['method'] == 'Network.requestWillBeSent':
                    on_request(msg)
                # 可以添加更多事件处理
        except Exception as e:
            print(f'Event listener error: {e}')
            break

# 启动事件监听线程
threading.Thread(target=event_listener, args=(ws,), daemon=True).start()

# 导航到页面
send_cmd(ws, 'Page.navigate', {'url': 'https://example.com'})

# ... 页面加载中，事件监听器会输出所有请求 ...
import time
time.sleep(5)  # 等待页面加载

# 拦截特定 URL 模式
send_cmd(ws, 'Network.setBlockedURLs', {
    'urls': ['*.jpg', '*.png', '*.gif']   # 拦截所有图片
})

# 模拟弱网环境
send_cmd(ws, 'Network.emulateNetworkConditions', {
    'offline': False,
    'latency': 300,          # 延迟 300ms
    'downloadThroughput': 500 * 1024,   # 下载 500 KB/s
    'uploadThroughput': 100 * 1024      # 上传 100 KB/s
})

# 获取响应体
# 首先在 Network.responseReceived 事件中拿到 requestId
# 然后：
result = send_cmd(ws, 'Network.getResponseBody', {
    'requestId': request_id
})
print(f'Response body: {result["body"][:500]}')
print(f'Base64 encoded: {result["base64Encoded"]}')

def click(ws, x, y, button='left'):
    """在指定坐标点击"""
    send_cmd(ws, 'Input.dispatchMouseEvent', {
        'type': 'mousePressed',
        'x': x, 'y': y,
        'button': button,
        'clickCount': 1
    })
    send_cmd(ws, 'Input.dispatchMouseEvent', {
        'type': 'mouseReleased',
        'x': x, 'y': y,
        'button': button,
        'clickCount': 1
    })

def type_text(ws, text):
    """输入文本"""
    send_cmd(ws, 'Input.insertText', {'text': text})

def press_enter(ws):
    """按 Enter 键"""
    send_cmd(ws, 'Input.dispatchKeyEvent', {
        'type': 'rawKeyDown',
        'windowsVirtualKeyCode': 13,
        'key': 'Enter'
    })
    send_cmd(ws, 'Input.dispatchKeyEvent', {
        'type': 'keyUp',
        'windowsVirtualKeyCode': 13,
        'key': 'Enter'
    })

# 使用示例：自动填写表单
click(ws, 500, 300)          # 点击输入框
type_text(ws, 'hello@example.com')  # 输入邮箱
press_enter(ws)              # 提交

# 在页面加载前注入脚本，覆盖自动化特征
send_cmd(ws, 'Page.addScriptToEvaluateOnNewDocument', {
    'source': '''
        // 覆盖 webdriver 属性
        Object.defineProperty(navigator, 'webdriver', {
            get: () => undefined
        });
        
        // 覆盖 chrome 对象
        window.chrome = {
            runtime: {},
            loadTimes: function() {},
            csi: function() {},
            app: {}
        };
        
        // 覆盖权限查询
        const originalQuery = navigator.permissions.query;
        navigator.permissions.query = (params) => (
            params.name === 'notifications' ?
                Promise.resolve({state: Notification.permission}) :
                originalQuery(params)
        );
        
        // 覆盖 plugins
        Object.defineProperty(navigator, 'plugins', {
            get: () => [1, 2, 3, 4, 5]
        });
        
        // 覆盖 languages
        Object.defineProperty(navigator, 'languages', {
            get: () => ['zh-CN', 'zh', 'en']
        });
    '''
})

# 以上脚本会在每个新页面上自动执行
# 然后再导航
send_cmd(ws, 'Page.navigate', {'url': 'https://bot.sannysoft.com/'})

# 监听 Target.targetCreated 事件
# 当新窗口打开时，自动获取它的 WebSocket URL

def on_target_created(event_data):
    target_info = event_data['params']['targetInfo']
    print(f'新标签页: {target_info["url"]}')
    # 可以通过 CDP_HTTP/json 获取新页面的 WebSocket URL

# 也可以用 --remote-debugging-pipe 参数使用管道而非 WebSocket
# 或者用 Target.attachToTarget 命令

# 开始性能追踪
send_cmd(ws, 'Performance.enable')

# 导航
send_cmd(ws, 'Page.navigate', {'url': 'https://example.com'})
time.sleep(3)

# 获取性能指标
result = send_cmd(ws, 'Performance.getMetrics')
metrics = {m['name']: m['value'] for m in result['metrics']}

print(f'DOMContentLoaded: {metrics.get("DomContentLoaded", "N/A")} ms')
print(f'首次绘制: {metrics.get("FirstPaint", "N/A")} ms')
print(f'JS 堆大小: {metrics.get("JSHeapUsedSize", "N/A")} bytes')
print(f'布局次数: {metrics.get("LayoutCount", "N/A")}')
print(f'重绘次数: {metrics.get("RecalcStyleCount", "N/A")}')

result = send_cmd(ws, 'Page.printToPDF', {
    'paperWidth': 8.27,       # A4 宽度（英寸）
    'paperHeight': 11.69,     # A4 高度
    'marginTop': 0.4,
    'marginBottom': 0.4,
    'marginLeft': 0.4,
    'marginRight': 0.4,
    'printBackground': True,
    'displayHeaderFooter': True,
    'headerTemplate': '<span style="font-size:10px;margin-left:10px;">标题</span>',
    'footerTemplate': '<span style="font-size:10px;margin-right:10px;">第 <span class="pageNumber"></span> 页 / 共 <span class="totalPages"></span> 页</span>'
})

with open('output.pdf', 'wb') as f:
    f.write(base64.b64decode(result['data']))

#!/usr/bin/env python3
"""
CDP 命令行截图工具

用法：
    python cdp_screenshooter.py https://example.com -o screenshot.png -w 1920 -h 1080
"""
import json
import urllib.request
import websocket
import base64
import argparse
import time

# ========== 工具函数 ==========

def find_page_ws(cdp_url, pattern=''):
    data = json.loads(urllib.request.urlopen(f'{cdp_url}/json', timeout=5).read())
    for page in data:
        if pattern.lower() in page.get('url', '').lower():
            return page['webSocketDebuggerUrl']
    return data[0]['webSocketDebuggerUrl'] if data else None

class CDPConnection:
    """CDP 连接封装"""
    
    def __init__(self, ws_url):
        self.ws = websocket.create_connection(ws_url, timeout=30)
        self._id = 0
        # 启用核心域
        self._cmd('Page.enable')
        self._cmd('Runtime.enable')
    
    def _cmd(self, method, params=None):
        if params is None:
            params = {}
        self._id += 1
        self.ws.send(json.dumps({'id': self._id, 'method': method, 'params': params}))
        while True:
            r = json.loads(self.ws.recv())
            if r.get('id') == self._id:
                return r.get('result', {})
    
    def navigate(self, url):
        """导航到 URL 并等待页面加载完成"""
        self._cmd('Page.navigate', {'url': url})
        # 等待页面加载（生产环境应监听 Page.loadEventFired 事件）
        time.sleep(3)
    
    def screenshot(self, output_path, width=1920, height=1080):
        """截取页面截图"""
        # 设置视口大小
        self._cmd('Emulation.setDeviceMetricsOverride', {
            'width': width,
            'height': height,
            'deviceScaleFactor': 1,
            'mobile': False
        })
        time.sleep(0.5)
        
        # 截图
        result = self._cmd('Page.captureScreenshot', {
            'format': 'png',
            'fromSurface': True
        })
        
        with open(output_path, 'wb') as f:
            f.write(base64.b64decode(result['data']))
        print(f'✅ 截图已保存: {output_path} ({width}x{height})')
    
    def close(self):
        self.ws.close()


def main():
    parser = argparse.ArgumentParser(description='CDP 命令行截图工具')
    parser.add_argument('url', help='目标 URL')
    parser.add_argument('-o', '--output', default='screenshot.png', help='输出文件路径')
    parser.add_argument('-w', '--width', type=int, default=1920, help='视口宽度')
    parser.add_argument('-H', '--height', type=int, default=1080, help='视口高度')
    parser.add_argument('--cdp', default='http://localhost:9222', help='CDP HTTP 地址')
    parser.add_argument('--pattern', default='', help='匹配特定标签页')
    args = parser.parse_args()
    
    print(f'🔍 连接 Chrome: {args.cdp}')
    ws_url = find_page_ws(args.cdp, args.pattern)
    if not ws_url:
        print('❌ 未找到可用的页面')
        return
    
    print(f'🔗 WebSocket: {ws_url[:60]}...')
    cdp = CDPConnection(ws_url)
    
    print(f'🌐 导航到: {args.url}')
    cdp.navigate(args.url)
    
    cdp.screenshot(args.output, args.width, args.height)
    cdp.close()
    print('🎉 完成！')


if __name__ == '__main__':
    main()

result = send_cmd(ws, 'Runtime.evaluate', {
    'expression': '''
        (() => {
            const ta = document.querySelector('.xterm-helper-textarea');
            if (!ta) return false;
            ta.value = '要输入的命令';
            ta.dispatchEvent(new Event('input', {bubbles: true}));
            return true;
        })()
    ''',
    'returnByValue': True
})