@mrtkrcm/mcp-puppeteer
Version:
Model Context Protocol server for browser automation using Puppeteer
167 lines • 7.05 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (g && (g = 0, op[0] && (_ = 0)), _) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
/**
* Generate a YAML-formatted accessibility snapshot from the current page
* @param page Puppeteer page
* @returns YAML string representation of the accessibility tree
*/
export function generateAccessibilitySnapshot(page) {
return __awaiter(this, void 0, void 0, function () {
var snapshot, processedSnapshot;
return __generator(this, function (_a) {
switch (_a.label) {
case 0: return [4 /*yield*/, page.accessibility.snapshot()];
case 1:
snapshot = _a.sent();
if (!snapshot) {
return [2 /*return*/, "- document [ref=s1e1]: No accessibility data available"];
}
processedSnapshot = processSnapshot(snapshot, 's1');
return [2 /*return*/, formatAsYaml(processedSnapshot)];
}
});
});
}
/**
* Process a raw accessibility node into a structured node with references
* @param node Raw accessibility node from Puppeteer
* @param refPrefix Reference prefix for this node
* @param index Index for generating unique references
* @returns Processed accessibility node
*/
function processSnapshot(node, refPrefix, index) {
if (index === void 0) { index = 0; }
// Process node and its children
var ref = "".concat(refPrefix, "e").concat(index + 1);
var result = {
role: node.role || 'unknown',
ref: ref
};
// Add properties if they exist
if (node.name)
result.name = node.name;
if (node.value)
result.value = node.value;
if (node.description)
result.description = node.description;
if (node.selected)
result.selected = node.selected;
if (node.checked)
result.checked = node.checked;
if (node.disabled)
result.disabled = node.disabled;
if (node.required)
result.required = node.required;
if (node.focused)
result.focused = node.focused;
// Handle heading level as a special case
if (node.role === 'heading' && node.level) {
result.level = node.level;
}
// Process children recursively with increased indentation
if (node.children && node.children.length > 0) {
result.children = node.children.map(function (child, idx) {
return processSnapshot(child, ref, idx);
});
}
return result;
}
/**
* Format a processed accessibility node as YAML
* @param node Processed accessibility node
* @param indent Current indentation level
* @returns YAML string representation
*/
function formatAsYaml(node, indent) {
if (indent === void 0) { indent = 0; }
var indentStr = ' '.repeat(indent);
var lines = [];
// Start with node role and name
var line = "".concat(indentStr, "- ").concat(node.role);
// Add name if present (quoted)
if (node.name) {
line += " \"".concat(escapeString(node.name), "\"");
}
// Add attributes in brackets
var attributes = [];
if (node.value)
attributes.push("value=\"".concat(escapeString(node.value), "\""));
if (node.selected)
attributes.push('selected');
if (node.checked)
attributes.push('checked');
if (node.disabled)
attributes.push('disabled');
if (node.required)
attributes.push('required');
if (node.focused)
attributes.push('focused');
if (node.level !== undefined)
attributes.push("level=".concat(node.level));
// Add ref at the end
attributes.push("ref=".concat(node.ref));
if (attributes.length > 0) {
line += " [".concat(attributes.join('] ['), "]");
}
// Add colon if there are children
if (node.children && node.children.length > 0) {
line += ':';
}
lines.push(line);
// Process children with increased indentation
if (node.children && node.children.length > 0) {
for (var _i = 0, _a = node.children; _i < _a.length; _i++) {
var child = _a[_i];
lines.push(formatAsYaml(child, indent + 1));
}
}
return lines.join('\n');
}
/**
* Escape special characters in strings for YAML
* @param str String to escape
* @returns Escaped string
*/
function escapeString(str) {
return str
.replace(/\\/g, '\\\\')
.replace(/"/g, '\\"')
.replace(/\n/g, '\\n')
.replace(/\r/g, '\\r')
.replace(/\t/g, '\\t');
}
//# sourceMappingURL=accessibilitySnapshot.js.map