my-extension/
├── manifest.json          # 扩展配置 (必需)
├── background.js          # Service Worker (MV3) / Background Page (MV2)
├── content.js             # Content Script (注入到网页)
├── popup.html             # 弹出窗口 UI
├── popup.js
├── options.html           # 设置页面
├── icons/
│   ├── icon16.png
│   ├── icon48.png
│   └── icon128.png
└── _locales/              # 国际化
    ├── zh_CN/
    │   └── messages.json
    └── en/
        └── messages.json

{
  "manifest_version": 3,
  "name": "我的扩展",
  "version": "1.0.0",
  "description": "一个示例扩展",
  "permissions": ["activeTab", "storage", "scripting"],
  "host_permissions": ["https://*.example.com/*"],
  "background": {
    "service_worker": "background.js"
  },
  "content_scripts": [
    {
      "matches": ["https://*.example.com/*"],
      "js": ["content.js"],
      "css": ["content.css"],
      "run_at": "document_end"
    }
  ],
  "action": {
    "default_popup": "popup.html",
    "default_icon": {
      "16": "icons/icon16.png",
      "48": "icons/icon48.png"
    }
  }
}

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

options = Options()

# 加载 .crx 扩展文件
options.add_extension("/path/to/extension.crx")

# 加载多个扩展
options.add_extension("/path/to/ext1.crx")
options.add_extension("/path/to/ext2.crx")

# 注意: --headless (旧版) 不支持扩展加载
# --headless=new 支持扩展加载
options.add_argument("--headless=new")

driver = webdriver.Chrome(options=options)
driver.get("https://example.com")

options = Options()

# 加载解压的扩展目录 (开发阶段常用)
options.add_argument("--load-extension=/path/to/my-extension")

# 加载多个扩展
options.add_argument(
    "--load-extension=/path/to/ext1,/path/to/ext2"
)

# 禁用其他扩展 (排除干扰)
options.add_argument("--disable-extensions-except=/path/to/my-extension")

driver = webdriver.Chrome(options=options)

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

def wait_for_extension(driver, extension_id, timeout=10):
    """等待扩展加载完成"""
    # 扩展页面可通过 chrome-extension://ID/ 访问
    driver.get(f"chrome-extension://{extension_id}/popup.html")
    WebDriverWait(driver, timeout).until(
        EC.presence_of_element_located((By.TAG_NAME, "body"))
    )

# 获取扩展 ID
# 方法 1: 手动查看 chrome://extensions
# 方法 2: 通过 CDP 获取
extensions_info = driver.execute_cdp_cmd("Management.getAll", {})
for ext in extensions_info:
    print(f"ID: {ext['id']}, Name: {ext['name']}")

const { chromium } = require('playwright');
const path = require('path');

(async () => {
  const extensionPath = path.resolve(__dirname, './my-extension');

  const context = await chromium.launchPersistentContext('', {
    headless: false,  // Playwright 扩展加载需要有头模式
    args: [
      `--disable-extensions-except=${extensionPath}`,
      `--load-extension=${extensionPath}`,
    ],
  });

  // 等待扩展背景页面
  const background = context.serviceWorkers()[0] ||
                     context.waitForEvent('serviceworker');

  const page = await context.newPage();
  await page.goto('https://example.com');

  // ... 测试

  await context.close();
})();

test('扩展弹出窗口工作正常', async ({ context }) => {
  // 获取扩展 ID
  let [background] = context.serviceWorkers();
  if (!background) {
    background = await context.waitForEvent('serviceworker');
  }
  const extensionId = background.url().split('/')[2];

  // 打开扩展弹出页面
  const page = await context.newPage();
  await page.goto(`chrome-extension://${extensionId}/popup.html`);

  // 测试弹出窗口内容
  await expect(page.locator('h1')).toHaveText('我的扩展');
  await page.getByRole('button', { name: '启用' }).click();
  await expect(page.locator('.status')).toHaveText('已启用');
});

def test_adblock():
    options = Options()
    options.add_extension("/path/to/adblock.crx")
    options.add_argument("--headless=new")

    driver = webdriver.Chrome(options=options)
    driver.get("https://example-news-site.com")

    # 验证广告被拦截
    ads = driver.find_elements(By.CSS_SELECTOR, ".ad-banner, .ad-container")
    assert len(ads) == 0, f"仍有 {len(ads)} 个广告元素未被拦截"

    # 检查网络请求 (通过 CDP)
    logs = driver.get_log("performance")
    ad_requests = [
        log for log in logs
        if "doubleclick.net" in str(log) or "googleads" in str(log)
    ]
    assert len(ad_requests) == 0, "仍有广告网络请求"

    driver.quit()

def test_with_vpn_extension():
    options = Options()
    options.add_extension("/path/to/vpn-extension.crx")
    options.add_argument("--headless=new")

    driver = webdriver.Chrome(options=options)

    # 等待 VPN 连接
    time.sleep(5)

    # 验证 IP 已改变
    driver.get("https://api.ipify.org?format=json")
    ip_data = json.loads(driver.find_element(By.TAG_NAME, "body").text)
    print(f"当前 IP: {ip_data['ip']}")

    driver.quit()

def test_with_password_manager():
    options = Options()
    options.add_argument("--load-extension=/path/to/password-manager")
    options.add_argument("--headless=new")
    # 加载已有的用户配置 (包含已保存的密码)
    options.add_argument("--user-data-dir=/path/to/profile")

    driver = webdriver.Chrome(options=options)
    driver.get("https://example.com/login")

    # 密码管理器可能自动填充
    time.sleep(3)
    username = driver.find_element(By.ID, "username")
    if username.get_attribute("value"):
        print("密码管理器已自动填充用户名")

    driver.quit()

def test_custom_extension():
    """测试自定义开发的扩展"""
    options = Options()
    options.add_argument("--load-extension=./my-extension")
    options.add_argument("--disable-extensions-except=./my-extension")
    options.add_argument("--headless=new")

    driver = webdriver.Chrome(options=options)
    driver.get("https://target-website.com")

    # 验证扩展 Content Script 注入
    result = driver.execute_script(
        "return document.querySelector('#my-extension-injected-element') !== null"
    )
    assert result, "扩展未正确注入 Content Script"

    # 验证扩展修改了页面样式
    bg_color = driver.execute_script(
        "return window.getComputedStyle(document.body).backgroundColor"
    )
    assert bg_color == "rgb(255, 255, 255)", f"扩展未正确设置背景色: {bg_color}"

    driver.quit()

# 启动 Chrome 并开启远程调试
options = Options()
options.add_argument("--load-extension=./my-extension")
options.add_argument("--remote-debugging-port=9222")
options.add_argument("--headless=new")

driver = webdriver.Chrome(options=options)

# 通过 CDP 连接到扩展的 background context
# 在 DevTools 中访问 chrome://inspect 可查看扩展

// 在 content.js 中添加调试日志
console.log('[MyExtension] Content script loaded on:', window.location.href);

// 使用 debugger 语句 (DevTools 打开时会暂停)
// debugger;

// 监听来自 background 的消息
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
  console.log('[MyExtension] Received message:', message);
  if (message.action === 'getData') {
    sendResponse({ data: document.title });
  }
});

def test_extension_messaging():
    """测试扩展内部通信"""
    options = Options()
    options.add_argument("--load-extension=./my-extension")
    options.add_argument("--headless=new")

    driver = webdriver.Chrome(options=options)
    driver.get("https://example.com")

    # 通过执行脚本触发扩展逻辑
    result = driver.execute_script("""
        return new Promise((resolve) => {
            chrome.runtime.sendMessage(
                { action: 'getStatus' },
                (response) => resolve(response)
            );
        });
    """)
    assert result["active"] == True

    driver.quit()

def audit_extension_permissions(extension_path):
    """审计扩展权限"""
    manifest_path = f"{extension_path}/manifest.json"
    with open(manifest_path) as f:
        manifest = json.load(f)

    permissions = manifest.get("permissions", [])
    host_permissions = manifest.get("host_permissions", [])

    high_risk = ["cookies", "webRequest", "webRequestBlocking", "<all_urls>"]

    print("=== 扩展权限审计 ===")
    print(f"扩展名: {manifest.get('name')}")
    print(f"版本: {manifest.get('version')}")
    print(f"Manifest 版本: {manifest.get('manifest_version')}")
    print(f"\n权限 ({len(permissions)}):")
    for p in permissions:
        risk = "⚠️ 高" if p in high_risk else "✅ 低"
        print(f"  {risk} - {p}")
    print(f"\n主机权限 ({len(host_permissions)}):")
    for h in host_permissions:
        print(f"  🔗 {h}")

    return permissions, host_permissions

def test_extension_performance():
    """测试扩展对页面性能的影响"""
    # 无扩展基准测试
    options_no_ext = Options()
    options_no_ext.add_argument("--headless=new")
    driver = webdriver.Chrome(options=options_no_ext)
    driver.get("https://example.com")
    baseline = driver.execute_script("""
        const perf = performance.getEntriesByType('navigation')[0];
        return {
            domContentLoaded: perf.domContentLoadedEventEnd - perf.startTime,
            loadComplete: perf.loadEventEnd - perf.startTime,
        };
    """)
    driver.quit()

    # 有扩展对比测试
    options_with_ext = Options()
    options_with_ext.add_argument("--load-extension=./my-extension")
    options_with_ext.add_argument("--headless=new")
    driver = webdriver.Chrome(options_with_ext)
    driver.get("https://example.com")
    with_ext = driver.execute_script("""
        const perf = performance.getEntriesByType('navigation')[0];
        return {
            domContentLoaded: perf.domContentLoadedEventEnd - perf.startTime,
            loadComplete: perf.loadEventEnd - perf.startTime,
        };
    """)
    driver.quit()

    # 比较
    dom_overhead = with_ext["domContentLoaded"] - baseline["domContentLoaded"]
    load_overhead = with_ext["loadComplete"] - baseline["loadComplete"]

    print(f"DOM ContentLoaded 开销: {dom_overhead:.0f}ms")
    print(f"页面完全加载开销: {load_overhead:.0f}ms")

    # 断言扩展对性能的影响不超过 500ms
    assert load_overhead < 500, f"扩展性能开销过大: {load_overhead:.0f}ms"