Add .loadUrl, support for headers, support for userAgent (#75)

* Add the ability to load a URL

* Add the ability to set a user agent

* Prepare 0.0.17 (0.1.14)

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.0.17 (binary 0.1.14) -- 2024-10-02
+
+- Add `webview.loadUrl` to load a new URL after the webview is initialized
+- Add the ability to specify headers when instantiating a webview with a URL
+- Add `userAgent` to `WebViewOptions` to construct a new webview with a custom user agent.
+
 ## 0.0.16 (binary 0.1.13) -- 2024-09-29
 
 - Add `initializationScript` to `WebViewOptions`. Allows providing JS that runs before `onLoad`.

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "deno-webview"
-version = "0.1.13"
+version = "0.1.14"
 edition = "2021"
 
 [profile.release]

deno.json

@@ -1,7 +1,7 @@
 {
   "name": "@justbe/webview",
   "exports": "./src/lib.ts",
-  "version": "0.0.16",
+  "version": "0.0.17",
   "tasks": {
     "dev": "deno run --watch main.ts",
     "gen": "deno task gen:rust && deno task gen:deno",

examples/load-url.ts

@@ -0,0 +1,25 @@
+import { createWebView } from "../src/lib.ts";
+
+using webview = await createWebView({
+  title: "Load Url Example",
+  url: "https://example.com",
+  headers: {
+    "Content-Type": "text/html",
+  },
+  userAgent: "curl/7.81.0",
+  devtools: true,
+});
+
+webview.on("started", async () => {
+  await webview.openDevTools();
+  await sleep(2000);
+  await webview.loadUrl("https://val.town/", {
+    "Content-Type": "text/html",
+  });
+});
+
+await webview.waitUntilClosed();
+
+function sleep(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}

scripts/generate-schema.ts

@@ -34,6 +34,7 @@ type NodeIR =
       value: NodeIR;
     }[];
   }
+  | { type: "record"; valueType: string }
   | { type: "boolean"; optional?: boolean }
   | { type: "string"; optional?: boolean }
   | { type: "literal"; value: string }
@@ -170,20 +171,34 @@ function jsonSchemaToIR(schema: JSONSchema): DocIR {
         },
       )
       .with(
-        { type: "object" },
-        () => ({
-          type: "object" as const,
-          properties: Object.entries(node.properties ?? {}).map((
-            [key, value],
-          ) => {
-            return ({
+        { type: P.union("object", P.when(isOptionalType("object"))) },
+        () => {
+          if (Object.keys(node.properties ?? {}).length === 0) {
+            if (
+              typeof node.additionalProperties === "object" &&
+              "type" in node.additionalProperties
+            ) {
+              return {
+                type: "record" as const,
+                valueType: typeof node.additionalProperties.type === "string"
+                  ? node.additionalProperties.type
+                  : "unknown",
+              };
+            }
+            return { type: "record" as const, valueType: "unknown" };
+          }
+          return ({
+            type: "object" as const,
+            properties: Object.entries(node.properties ?? {}).map((
+              [key, value],
+            ) => ({
               key,
               required: node.required?.includes(key) ?? false,
               description: (value as JSONSchema).description,
               value: nodeToIR(value as JSONSchema),
-            });
-          }),
-        }),
+            })),
+          });
+        },
       )
       .otherwise(() => ({ type: "unknown" }));
   };
@@ -215,6 +230,9 @@ function generateTypes(ir: DocIR) {
       .with({ type: "boolean" }, () => w("boolean"))
       .with({ type: "string" }, () => w("string"))
       .with({ type: "literal" }, (node) => w(`"${node.value}"`))
+      .with({ type: "record" }, (node) => {
+        w(`Record<string, ${node.valueType}>`);
+      })
       .with({ type: "object" }, (node) => {
         wn("{");
         for (const { key, required, description, value } of node.properties) {
@@ -300,6 +318,9 @@ function generateZodSchema(ir: DocIR) {
         { type: "literal" },
         (node) => w(`z.literal("${node.value}")`),
       )
+      .with({ type: "record" }, (node) => {
+        w(`z.record(z.string(), z.${node.valueType}())`);
+      })
       .with({ type: "object" }, (node) => {
         w("z.object({");
         for (const { key, required, value } of node.properties) {

src/lib.ts

@@ -38,7 +38,7 @@ export type { WebViewOptions } from "./schemas.ts";
 
 // Should match the cargo package version
 /** The version of the webview binary that's expected */
-export const BIN_VERSION = "0.1.13";
+export const BIN_VERSION = "0.1.14";
 
 type WebViewNotification = Extract<
   WebViewMessage,
@@ -463,6 +463,14 @@ export class WebView implements Disposable {
     return returnAck(result);
   }
 
+  /**
+   * Loads a URL in the webview.
+   */
+  async loadUrl(url: string, headers?: Record<string, string>): Promise<void> {
+    const result = await this.#send({ $type: "loadUrl", url, headers });
+    return returnAck(result);
+  }
+
   /**
    * Destroys the webview and cleans up resources.
    *

src/main.rs

@@ -1,7 +1,9 @@
 use std::borrow::Cow;
 use std::cell::RefCell;
+use std::collections::HashMap;
 use std::env;
 use std::io::{self, BufRead, Write};
+use std::str::FromStr;
 use std::sync::mpsc;
 use std::sync::Arc;
 
@@ -16,6 +18,7 @@ use tao::{
     event_loop::{ControlFlow, EventLoop},
     window::WindowBuilder,
 };
+use wry::http::header::{HeaderName, HeaderValue};
 use wry::http::Response as HttpResponse;
 use wry::WebViewBuilder;
 
@@ -92,6 +95,9 @@ struct WebViewOptions {
     #[serde(default)]
     /// Run JavaScript code when loading new pages. When the webview loads a new page, this code will be executed. It is guaranteed that the code is executed before window.onload.
     initialization_script: Option<String>,
+    /// Sets the user agent to use when loading pages.
+    #[serde(default)]
+    user_agent: Option<String>,
 }
 
 fn default_true() -> bool {
@@ -105,6 +111,8 @@ enum WebViewTarget {
     Url {
         /// Url to load in the webview. Note: Don't use data URLs here, as they are not supported. Use the `html` field instead.
         url: String,
+        /// Optional headers to send with the request.
+        headers: Option<HashMap<String, String>>,
     },
     Html {
         /// Html to load in the webview.
@@ -232,6 +240,14 @@ enum Request {
         /// If not specified, the origin will be set to the value of the `origin` field when the webview was created.
         origin: Option<String>,
     },
+    LoadUrl {
+        /// The id of the request.
+        id: String,
+        /// URL to load in the webview.
+        url: String,
+        /// Optional headers to send with the request.
+        headers: Option<HashMap<String, String>>,
+    },
 }
 
 /// Responses from the webview to the client.
@@ -310,7 +326,22 @@ fn main() -> wry::Result<()> {
 
     let html_cell_init = html_cell.clone();
     let mut webview_builder = match webview_options.target {
-        WebViewTarget::Url { url } => WebViewBuilder::new(&window).with_url(url),
+        WebViewTarget::Url { url, headers } => {
+            let mut webview_builder = WebViewBuilder::new(&window).with_url(url);
+            if let Some(headers) = headers {
+                let headers = headers
+                    .into_iter()
+                    .map(|(k, v)| {
+                        (
+                            HeaderName::from_str(&k).unwrap(),
+                            HeaderValue::from_str(&v).unwrap(),
+                        )
+                    })
+                    .collect();
+                webview_builder = webview_builder.with_headers(headers);
+            }
+            webview_builder
+        }
         WebViewTarget::Html { html, origin } => {
             origin_cell.replace(origin.clone());
             html_cell.replace(html);
@@ -346,6 +377,9 @@ fn main() -> wry::Result<()> {
         webview_builder =
             webview_builder.with_initialization_script(initialization_script.as_str());
     }
+    if let Some(user_agent) = webview_options.user_agent {
+        webview_builder = webview_builder.with_user_agent(user_agent.as_str());
+    }
     let webview = webview_builder.build()?;
 
     let notify_tx = tx.clone();
@@ -527,6 +561,30 @@ fn main() -> wry::Result<()> {
                                 .unwrap();
                             res(Response::Ack { id });
                         }
+                        Request::LoadUrl { id, url, headers } => {
+                            let resp = match headers {
+                                Some(headers) => {
+                                    let headers = headers
+                                        .into_iter()
+                                        .map(|(k, v)| {
+                                            (
+                                                HeaderName::from_str(&k).unwrap(),
+                                                HeaderValue::from_str(&v).unwrap(),
+                                            )
+                                        })
+                                        .collect();
+                                    webview.load_url_with_headers(&url, headers)
+                                }
+                                None => webview.load_url(&url),
+                            };
+                            match resp {
+                                Ok(_) => res(Response::Ack { id }),
+                                Err(err) => res(Response::Err {
+                                    id,
+                                    message: err.to_string(),
+                                }),
+                            }
+                        }
                     }
                 }
             }