195 lines
15 KiB
JavaScript
195 lines
15 KiB
JavaScript
const { Server } = require('@modelcontextprotocol/sdk/server/index.js');
|
|
const { StdioServerTransport } = require('@modelcontextprotocol/sdk/server/stdio.js');
|
|
const { ListToolsRequestSchema, CallToolRequestSchema } = require('@modelcontextprotocol/sdk/types.js');
|
|
|
|
const tools = {
|
|
openUrl: require('./tools/openUrl'),
|
|
click: require('./tools/click'),
|
|
type: require('./tools/type'),
|
|
screenshot: require('./tools/screenshot'),
|
|
getText: require('./tools/getText'),
|
|
getLinks: require('./tools/getLinks'),
|
|
waitForSelector: require('./tools/waitForSelector'),
|
|
extractElements: require('./tools/extractElements'),
|
|
smartClick: require('./tools/smartClick'),
|
|
smartType: require('./tools/smartType'),
|
|
getButtons: require('./tools/getButtons'),
|
|
smartWaitNavigation: require('./tools/smartWaitNavigation'),
|
|
getForms: require('./tools/getForms'),
|
|
fillFormAuto: require('./tools/fillFormAuto'),
|
|
agentFlow: require('./tools/agentFlow'),
|
|
agentFlowV2: require('./tools/agentFlowV2'),
|
|
closeBrowser: require('./tools/closeBrowser'),
|
|
getURL: require('./tools/getURL'),
|
|
getTitle: require('./tools/getTitle'),
|
|
getHTML: require('./tools/getHTML'),
|
|
goBack: require('./tools/goBack'),
|
|
goForward: require('./tools/goForward'),
|
|
reload: require('./tools/reload'),
|
|
evaluateJS: require('./tools/evaluateJS'),
|
|
evaluateJs: require('./tools/evaluateJS'),
|
|
waitForURL: require('./tools/waitForURL'),
|
|
newTab: require('./tools/newTab'),
|
|
switchTab: require('./tools/switchTab'),
|
|
listTabs: require('./tools/listTabs'),
|
|
closeTab: require('./tools/closeTab'),
|
|
setViewport: require('./tools/setViewport'),
|
|
emulateDevice: require('./tools/emulateDevice'),
|
|
hover: require('./tools/hover'),
|
|
select: require('./tools/select'),
|
|
pressKey: require('./tools/pressKey'),
|
|
scrollTo: require('./tools/scrollTo'),
|
|
pdf: require('./tools/pdf'),
|
|
getCookies: require('./tools/getCookies'),
|
|
setCookies: require('./tools/setCookies'),
|
|
clearCookies: require('./tools/clearCookies'),
|
|
getStorage: require('./tools/getStorage'),
|
|
setStorage: require('./tools/setStorage'),
|
|
clearStorage: require('./tools/clearStorage'),
|
|
getNetworkLogs: require('./tools/getNetworkLogs'),
|
|
blockResources: require('./tools/blockResources'),
|
|
uploadFile: require('./tools/uploadFile'),
|
|
getPerformanceMetrics: require('./tools/getPerformanceMetrics'),
|
|
wait: require('./tools/wait'),
|
|
enableVisualOverlay: require('./tools/enableVisualOverlay'),
|
|
showToast: require('./tools/showToast'),
|
|
highlightElement: require('./tools/highlightElement'),
|
|
dragAndDrop: require('./tools/dragAndDrop'),
|
|
annotatedScreenshot: require('./tools/annotatedScreenshot'),
|
|
getInputValues: require('./tools/getInputValues'),
|
|
waitForClickable: require('./tools/waitForClickable'),
|
|
waitForElementVisible: require('./tools/waitForElementVisible'),
|
|
waitForText: require('./tools/waitForText')
|
|
};
|
|
|
|
const server = new Server({ name: 'browser', version: '3.1.1' }, { capabilities: { tools: {} } });
|
|
|
|
const toolSchemas = [
|
|
{ name: 'open_url', description: 'Abre uma URL no navegador', inputSchema: { type: 'object', properties: { url: { type: 'string', description: 'URL para abrir' }, waitUntil: { type: 'string', enum: ['domcontentloaded', 'load', 'networkidle'] } }, required: ['url'] } },
|
|
{ name: 'click', description: 'Clica em elemento pelo seletor CSS', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, button: { type: 'string', enum: ['left', 'right', 'middle'] }, double: { type: 'boolean' } }, required: ['selector'] } },
|
|
{ name: 'type', description: 'Digita texto em campo', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, text: { type: 'string' }, clear: { type: 'boolean' }, slowly: { type: 'boolean' } }, required: ['selector', 'text'] } },
|
|
{ name: 'screenshot', description: 'Captura screenshot', inputSchema: { type: 'object', properties: { fullPage: { type: 'boolean' }, element: { type: 'string' }, path: { type: 'string' } } } },
|
|
{ name: 'get_text', description: 'Extrai texto', inputSchema: { type: 'object', properties: { selector: { type: 'string' } } } },
|
|
{ name: 'get_links', description: 'Lista links', inputSchema: { type: 'object', properties: { selector: { type: 'string' } } } },
|
|
{ name: 'wait_for_selector', description: 'Espera elemento', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, timeout: { type: 'number' }, state: { type: 'string', enum: ['attached', 'detached', 'visible', 'hidden'] } }, required: ['selector'] } },
|
|
{ name: 'extract_elements', description: 'Extrai elementos com atributos', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, attributes: { type: 'array', items: { type: 'string' } } }, required: ['selector'] } },
|
|
{ name: 'smart_click', description: 'Clica por texto visível', inputSchema: { type: 'object', properties: { text: { type: 'string' } }, required: ['text'] } },
|
|
{ name: 'smart_type', description: 'Digita por label/placeholder', inputSchema: { type: 'object', properties: { label: { type: 'string' }, text: { type: 'string' } }, required: ['label', 'text'] } },
|
|
{ name: 'get_buttons', description: 'Lista botões', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'smart_wait_navigation', description: 'Espera mudança de URL', inputSchema: { type: 'object', properties: { timeout: { type: 'number' } } } },
|
|
{ name: 'get_forms', description: 'Lista formulários', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'fill_form_auto', description: 'Preenche formulário', inputSchema: { type: 'object', properties: { data: { type: 'object' } }, required: ['data'] } },
|
|
{ name: 'agent_flow', description: 'Fluxo automático simples', inputSchema: { type: 'object', properties: { goal: { type: 'string' }, data: { type: 'object' } }, required: ['goal'] } },
|
|
{ name: 'agent_flow_v2', description: 'Fluxo com retry', inputSchema: { type: 'object', properties: { goal: { type: 'string' }, data: { type: 'object' }, retries: { type: 'number' } }, required: ['goal'] } },
|
|
{ name: 'close_browser', description: 'Fecha navegador', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'get_url', description: 'Obtém URL atual', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'get_title', description: 'Obtém título', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'get_html', description: 'Obtém HTML', inputSchema: { type: 'object', properties: { selector: { type: 'string' } } } },
|
|
{ name: 'go_back', description: 'Voltar no histórico', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'go_forward', description: 'Avançar no histórico', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'reload', description: 'Recarrega página', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'evaluate_js', description: 'Executa JavaScript', inputSchema: { type: 'object', properties: { expression: { type: 'string' }, selector: { type: 'string' } }, required: ['expression'] } },
|
|
{ name: 'wait_for_url', description: 'Espera URL específica', inputSchema: { type: 'object', properties: { pattern: { type: 'string' }, timeout: { type: 'number' } }, required: ['pattern'] } },
|
|
{ name: 'new_tab', description: 'Nova aba', inputSchema: { type: 'object', properties: { url: { type: 'string' } } } },
|
|
{ name: 'switch_tab', description: 'Muda aba', inputSchema: { type: 'object', properties: { index: { type: 'number' } }, required: ['index'] } },
|
|
{ name: 'list_tabs', description: 'Lista abas', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'close_tab', description: 'Fecha aba', inputSchema: { type: 'object', properties: { index: { type: 'number' } } } },
|
|
{ name: 'set_viewport', description: 'Define viewport', inputSchema: { type: 'object', properties: { width: { type: 'number' }, height: { type: 'number' } }, required: ['width', 'height'] } },
|
|
{ name: 'emulate_device', description: 'Emula dispositivo', inputSchema: { type: 'object', properties: { device: { type: 'string' } }, required: ['device'] } },
|
|
{ name: 'hover', description: 'Faz hover', inputSchema: { type: 'object', properties: { selector: { type: 'string' } }, required: ['selector'] } },
|
|
{ name: 'select', description: 'Seleciona opções', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, values: { type: 'array', items: { type: 'string' } } }, required: ['selector', 'values'] } },
|
|
{ name: 'press_key', description: 'Pressiona tecla', inputSchema: { type: 'object', properties: { key: { type: 'string' }, selector: { type: 'string' } }, required: ['key'] } },
|
|
{ name: 'scroll_to', description: 'Rola página', inputSchema: { type: 'object', properties: { position: { oneOf: [{ type: 'number' }, { type: 'string', enum: ['top', 'bottom'] }, { type: 'object', properties: { selector: { type: 'string' } } }] } } } },
|
|
{ name: 'pdf', description: 'Gera PDF', inputSchema: { type: 'object', properties: { path: { type: 'string' }, format: { type: 'string' }, landscape: { type: 'boolean' } } } },
|
|
{ name: 'get_cookies', description: 'Obtém cookies', inputSchema: { type: 'object', properties: { urls: { type: 'array', items: { type: 'string' } } } } },
|
|
{ name: 'set_cookies', description: 'Define cookies', inputSchema: { type: 'object', properties: { cookies: { type: 'array' } }, required: ['cookies'] } },
|
|
{ name: 'clear_cookies', description: 'Limpa cookies', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'get_storage', description: 'Obtém storage', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'set_storage', description: 'Define storage', inputSchema: { type: 'object', properties: { localStorage: { type: 'object' }, sessionStorage: { type: 'object' } } } },
|
|
{ name: 'clear_storage', description: 'Limpa storage', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'get_network_logs', description: 'Logs de rede', inputSchema: { type: 'object', properties: { type: { type: 'string' }, url: { type: 'string' } } } },
|
|
{ name: 'block_resources', description: 'Bloqueia recursos', inputSchema: { type: 'object', properties: { types: { type: 'array', items: { type: 'string' } } }, required: ['types'] } },
|
|
{ name: 'upload_file', description: 'Upload de arquivo', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, filePaths: { oneOf: [{ type: 'string' }, { type: 'array', items: { type: 'string' } }] } }, required: ['selector', 'filePaths'] } },
|
|
{ name: 'get_performance_metrics', description: 'Métricas de performance', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'wait', description: 'Aguarda ms', inputSchema: { type: 'object', properties: { ms: { type: 'number' } }, required: ['ms'] } },
|
|
{ name: 'enable_visual_overlay', description: 'Ativa cursor customizado e indicadores visuais de rolagem', inputSchema: { type: 'object', properties: {} } },
|
|
{ name: 'show_toast', description: 'Mostra notificação toast na página', inputSchema: { type: 'object', properties: { message: { type: 'string' }, duration: { type: 'number' } } } },
|
|
{ name: 'highlight_element', description: 'Destaca elemento com animação visual', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, color: { type: 'string' }, duration: { type: 'number' } }, required: ['selector'] } },
|
|
{ name: 'drag_and_drop', description: 'Arrasta e solta elemento', inputSchema: { type: 'object', properties: { sourceSelector: { type: 'string' }, targetSelector: { type: 'string' } }, required: ['sourceSelector', 'targetSelector'] } },
|
|
{ name: 'annotated_screenshot', description: 'Screenshot com elementos destacados', inputSchema: { type: 'object', properties: { fullPage: { type: 'boolean' }, path: { type: 'string' }, highlight: { oneOf: [{ type: 'string' }, { type: 'array', items: { type: 'string' } }] } } } },
|
|
{ name: 'get_input_values', description: 'Obtém valores de todos os inputs', inputSchema: { type: 'object', properties: { selector: { type: 'string' } } } },
|
|
{ name: 'wait_for_clickable', description: 'Espera elemento ser clicável', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, timeout: { type: 'number' } }, required: ['selector'] } },
|
|
{ name: 'wait_for_element_visible', description: 'Espera elemento visível', inputSchema: { type: 'object', properties: { selector: { type: 'string' }, timeout: { type: 'number' }, state: { type: 'string', enum: ['attached', 'detached', 'visible', 'hidden'] } }, required: ['selector'] } },
|
|
{ name: 'wait_for_text', description: 'Espera texto aparecer na página', inputSchema: { type: 'object', properties: { text: { type: 'string' }, timeout: { type: 'number' } }, required: ['text'] } }
|
|
];
|
|
|
|
server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: toolSchemas }));
|
|
|
|
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
const { name, arguments: args } = request.params;
|
|
let toolName = name.replace(/-/g, '').replace(/_([a-z])/g, (_, c) => c.toUpperCase());
|
|
|
|
const nameMap = {
|
|
// camelCase sem underscores
|
|
'geturl': 'getURL',
|
|
'gethtml': 'getHTML',
|
|
'goback': 'goBack',
|
|
'goforward': 'goForward',
|
|
'newtab': 'newTab',
|
|
'switchtab': 'switchTab',
|
|
'listtabs': 'listTabs',
|
|
'closetab': 'closeTab',
|
|
'setviewport': 'setViewport',
|
|
'emulatedevice': 'emulateDevice',
|
|
'waitforurl': 'waitForURL',
|
|
'waitforselector': 'waitForSelector',
|
|
'getcookies': 'getCookies',
|
|
'setcookies': 'setCookies',
|
|
'clearcookies': 'clearCookies',
|
|
'getstorage': 'getStorage',
|
|
'setstorage': 'setStorage',
|
|
'clearstorage': 'clearStorage',
|
|
'getnetworklogs': 'getNetworkLogs',
|
|
'blockresources': 'blockResources',
|
|
'getperformancemetrics': 'getPerformanceMetrics',
|
|
'waitfortimeout': 'waitForTimeout',
|
|
'enablevisualoverlay': 'enableVisualOverlay',
|
|
'showtoast': 'showToast',
|
|
'highlightelement': 'highlightElement',
|
|
'draganddrop': 'dragAndDrop',
|
|
'annotatedscreenshot': 'annotatedScreenshot',
|
|
'getinputvalues': 'getInputValues',
|
|
'waitforclickable': 'waitForClickable',
|
|
'waitforelementvisible': 'waitForElementVisible',
|
|
'waitfortext': 'waitForText',
|
|
'fillformauto': 'fillFormAuto',
|
|
'agentflow': 'agentFlow',
|
|
'agentflowv2': 'agentFlowV2',
|
|
'smartclick': 'smartClick',
|
|
'smarttype': 'smartType',
|
|
'smartwaitnavigation': 'smartWaitNavigation',
|
|
'getbuttons': 'getButtons',
|
|
'getforms': 'getForms',
|
|
'extractelements': 'extractElements',
|
|
'gettext': 'getText',
|
|
'getlinks': 'getLinks',
|
|
'openurl': 'openUrl',
|
|
'closebrowser': 'closeBrowser',
|
|
'gettitle': 'getTitle'
|
|
};
|
|
|
|
if (nameMap[toolName]) toolName = nameMap[toolName];
|
|
|
|
const tool = tools[toolName] || tools[name];
|
|
if (!tool) throw new Error(`Tool "${name}" não encontrada`);
|
|
try {
|
|
const result = await tool(args || {});
|
|
return { content: [{ type: 'text', text: JSON.stringify(result) }] };
|
|
} catch (error) {
|
|
return { content: [{ type: 'text', text: JSON.stringify({ error: error.message }) }], isError: true };
|
|
}
|
|
});
|
|
|
|
const transport = new StdioServerTransport();
|
|
server.connect(transport).catch(console.error);
|