Skip to content

feat(mcp): add browser adblock feature #229

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Mar 19, 2025
Merged

feat(mcp): add browser adblock feature #229

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4166888
feat(mcp): add browser ad block feature
ycjcl868 Mar 19, 2025
db59807
chore: unused
ycjcl868 Mar 19, 2025
83c657d
chore: server
ycjcl868 Mar 19, 2025
9ca5133
Merge branch 'feat/agent-tars' into feat/ad-block
ycjcl868 Mar 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 29 additions & 1 deletion apps/agent-tars/src/renderer/index.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,37 @@
http-equiv="Content-Security-Policy"
content="default-src 'self' 'unsafe-inline' file: data: blob:; img-src 'self' file: data: blob:; style-src 'self' 'unsafe-inline' https://cdn.jsdelivr.net https://cdnjs.cloudflare.com https://fonts.googleapis.com; script-src 'self' 'unsafe-inline' 'unsafe-eval' https://cdn.jsdelivr.net/npm/ https://cdn.jsdelivr.net https://cdnjs.cloudflare.com; font-src 'self' https://cdnjs.cloudflare.com https://fonts.googleapis.com; worker-src 'self' blob:; connect-src 'self' https://cdn.jsdelivr.net/npm/ blob:"
/>
<style>
@keyframes spin {
0% { transform: rotate(0deg); }
100% { transform: rotate(360deg); }
}

.loading-spinner {
display: inline-block;
width: 30px;
height: 30px;
border: 3px solid rgba(0, 0, 0, 0.1);
border-radius: 50%;
border-top-color: #767676;
animation: spin 1s ease-in-out infinite;
margin: 20px;
}

.loading-container {
display: flex;
justify-content: center;
align-items: center;
height: 100vh;
}
</style>
</head>
<body>
<div id="root"></div>
<div id="root">
<div class="loading-container">
<div class="loading-spinner"></div>
</div>
</div>
<script type="module" src="/src/main.tsx"></script>
<!-- DATA -->
</body>
Expand Down
2 changes: 1 addition & 1 deletion packages/agent-infra/browser/package.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,4 +38,4 @@
"@vitest/coverage-v8": "3.0.7",
"@rslib/core": "0.5.3"
}
}
}
2 changes: 2 additions & 0 deletions packages/agent-infra/mcp-servers/browser/package.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@
"@modelcontextprotocol/sdk": "^1.6.1"
},
"devDependencies": {
"@ghostery/adblocker-puppeteer": "2.5.0",
"cross-fetch": "4.1.0",
"@rslib/core": "0.5.3",
"turndown": "^7.2.0",
"turndown-plugin-gfm": "^1.0.2",
Expand Down
2 changes: 1 addition & 1 deletion packages/agent-infra/mcp-servers/browser/src/index.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ declare global {

setConfig({
launchOptions: {
headless: true,
headless: false,
},
logger: {
info: (...args: any[]) => {
Expand Down
10 changes: 10 additions & 0 deletions packages/agent-infra/mcp-servers/browser/src/server.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@ import { z } from 'zod';
import { ToolSchema } from '@modelcontextprotocol/sdk/types.js';
import { zodToJsonSchema } from 'zod-to-json-schema';
import { LaunchOptions, LocalBrowser, Page } from '@agent-infra/browser';
import { PuppeteerBlocker } from '@ghostery/adblocker-puppeteer';
import fetch from 'cross-fetch';
import {
getBuildDomTreeScript,
parseNode,
Expand Down Expand Up @@ -461,6 +463,14 @@ const handleToolCall: Client['callTool'] = async ({
},
browser_navigate: async (args) => {
try {
try {
const blocker =
await PuppeteerBlocker.fromPrebuiltAdsAndTracking(fetch);
await blocker.enableBlockingInPage(page as any);
} catch (e) {
logger.error('Error enabling adblocker:', e);
}

await Promise.all([
waitForPageAndFramesLoad(page),
page.goto(args.url),
Expand Down
Loading