public/browser
1
0
Fork 0
mirror of https://github.com/lightpanda-io/browser.git synced 2026-03-22 04:34:44 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

mcp: handle errors during resource and tool streaming

Browse Source
This commit is contained in:
Adrià Arrufat
2026-03-01 20:23:23 +09:00
parent 947e672d18
commit b3d52c966d
2 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/mcp/resources.zig
View File

@@ -1,6 +1,7 @@
const std = @import("std");
const lp = @import("lightpanda");
const log = lp.log;
const protocol = @import("protocol.zig");
const Server = @import("Server.zig");
@@ -58,8 +59,12 @@ const ResourceStreamingResult = struct {
try jw.writer.writeByte('"');
var escaped = protocol.JsonEscapingWriter.init(jw.writer);
switch (self.format) {
.html => try lp.dump.root(self.server.page.document, .{}, &escaped.writer, self.server.page),
.markdown => try lp.markdown.dump(self.server.page.document.asNode(), .{}, &escaped.writer, self.server.page),
.html => lp.dump.root(self.server.page.document, .{}, &escaped.writer, self.server.page) catch |err| {
log.err(.mcp, "html dump failed", .{ .err = err });
},
.markdown => lp.markdown.dump(self.server.page.document.asNode(), .{}, &escaped.writer, self.server.page) catch |err| {
log.err(.mcp, "markdown dump failed", .{ .err = err });
},
}
try jw.writer.writeByte('"');
jw.endWriteRaw();
@@ -79,7 +84,7 @@ pub fn handleRead(server: *Server, arena: std.mem.Allocator, req: protocol.Reque
};
if (std.mem.eql(u8, params.uri, "mcp://page/html")) {
const result = ResourceStreamingResult{
const result: ResourceStreamingResult = .{
.contents = &.{.{
.uri = params.uri,
.mimeType = "text/html",
@@ -88,7 +93,7 @@ pub fn handleRead(server: *Server, arena: std.mem.Allocator, req: protocol.Reque
};
try sendResult(server, req.id.?, result);
} else if (std.mem.eql(u8, params.uri, "mcp://page/markdown")) {
const result = ResourceStreamingResult{
const result: ResourceStreamingResult = .{
.contents = &.{.{
.uri = params.uri,
.mimeType = "text/markdown",

26
src/mcp/tools.zig
View File

@@ -126,21 +126,23 @@ const ToolStreamingText = struct {
var escaped = protocol.JsonEscapingWriter.init(jw.writer);
const w = &escaped.writer;
switch (self.action) {
.markdown => try lp.markdown.dump(self.server.page.document.asNode(), .{}, w, self.server.page),
.markdown => lp.markdown.dump(self.server.page.document.asNode(), .{}, w, self.server.page) catch |err| {
log.err(.mcp, "markdown dump failed", .{ .err = err });
},
.links => {
const list = Selector.querySelectorAll(self.server.page.document.asNode(), "a[href]", self.server.page) catch |err| {
log.err(.mcp, "Error querying links: {s}", .{@errorName(err)});
return;
};
var first = true;
for (list._nodes) |node| {
if (node.is(Element)) |el| {
if (el.getAttributeSafe(String.wrap("href"))) |href| {
if (!first) try w.writeByte('\n');
try w.writeAll(href);
first = false;
if (Selector.querySelectorAll(self.server.page.document.asNode(), "a[href]", self.server.page)) |list| {
var first = true;
for (list._nodes) |node| {
if (node.is(Element)) |el| {
if (el.getAttributeSafe(String.wrap("href"))) |href| {
if (!first) try w.writeByte('\n');
try w.writeAll(href);
first = false;
}
}
}
} else |err| {
log.err(.mcp, "query links failed", .{ .err = err });
}
},
}