setCookie

src/cdp/domains/network.zig

@@ -29,6 +29,7 @@ pub fn processMessage(cmd: anytype) !void {
         setCacheDisabled,
         setExtraHTTPHeaders,
         deleteCookies,
+        setCookie,
         setCookies,
     }, cmd.input.action) orelse return error.UnknownMethod;
 
@@ -38,6 +39,7 @@ pub fn processMessage(cmd: anytype) !void {
         .setCacheDisabled => return cmd.sendResult(null, .{}),
         .setExtraHTTPHeaders => return setExtraHTTPHeaders(cmd),
         .deleteCookies => return deleteCookies(cmd),
+        .setCookie => return setCookie(cmd),
         .setCookies => return setCookies(cmd),
     }
 }
@@ -168,9 +170,7 @@ fn percentEncodedDomain(allocator: Allocator, default_url: ?[]const u8, domain:
     } else return null;
 }
 
-fn setCookies(cmd: anytype) !void {
-    const params = (try cmd.params(struct {
-        cookies: []const struct {
+const CdpCookie = struct {
     name: []const u8,
     value: []const u8,
     url: ?[]const u8 = null,
@@ -185,18 +185,40 @@ fn setCookies(cmd: anytype) !void {
     sourceScheme: ?CookieSourceScheme = null,
     // sourcePort: Temporary ability and it will be removed from CDP
     partitionKey: ?CookiePartitionKey = null,
-        },
+};
+
+fn setCookie(cmd: anytype) !void {
+    const params = (try cmd.params(
+        CdpCookie,
+    )) orelse return error.InvalidParams;
+
+    const bc = cmd.browser_context orelse return error.BrowserContextNotLoaded;
+    try setCdpCookie(&bc.session.cookie_jar, params);
+
+    try cmd.sendResult(.{ .success = true }, .{});
+}
+
+fn setCookies(cmd: anytype) !void {
+    const params = (try cmd.params(struct {
+        cookies: []const CdpCookie,
     })) orelse return error.InvalidParams;
 
     const bc = cmd.browser_context orelse return error.BrowserContextNotLoaded;
     for (params.cookies) |param| {
+        try setCdpCookie(&bc.session.cookie_jar, param);
+    }
+
+    try cmd.sendResult(null, .{});
+}
+
+fn setCdpCookie(cookie_jar: *CookieJar, param: CdpCookie) !void {
     if (param.priority != .Medium or param.sameParty != null or param.sourceScheme != null or param.partitionKey != null) {
         return error.NotYetImplementedParams;
     }
     if (param.name.len == 0) return error.InvalidParams;
     if (param.value.len == 0) return error.InvalidParams;
 
-        var arena = std.heap.ArenaAllocator.init(bc.session.cookie_jar.allocator);
+    var arena = std.heap.ArenaAllocator.init(cookie_jar.allocator);
     errdefer arena.deinit();
     const a = arena.allocator();
 
@@ -218,10 +240,7 @@ fn setCookies(cmd: anytype) !void {
             .None => .none,
         },
     };
-        try bc.session.cookie_jar.add(cookie, std.time.timestamp());
-    }
-
-    return cmd.sendResult(null, .{});
+    try cookie_jar.add(cookie, std.time.timestamp());
 }
 
 // Upsert a header into the headers array.