nuxt-og-image 2.0.0-beta.2 → 2.0.0-beta.22

package/dist/module.mjs

@@ -1,29 +1,33 @@
 import { readFile, writeFile } from 'node:fs/promises';
-import { defineNuxtModule, createResolver, addTemplate, addServerHandler, addImports, addComponent } from '@nuxt/kit';
+import { defineNuxtModule, useLogger, createResolver, addTemplate, addServerHandler, addImports, addComponent, addServerPlugin } from '@nuxt/kit';
 import { execa } from 'execa';
 import chalk from 'chalk';
 import defu from 'defu';
 import { toRouteMatcher, createRouter } from 'radix3';
-import { joinURL } from 'ufo';
-import { resolve, relative } from 'pathe';
+import { withBase, joinURL } from 'ufo';
+import { resolve, relative, dirname } from 'pathe';
 import { tinyws } from 'tinyws';
 import sirv from 'sirv';
 import { pathExists, copy, mkdirp } from 'fs-extra';
-import { provider } from 'std-env';
+import { globby } from 'globby';
 import playwrightCore from 'playwright-core';
 import { existsSync } from 'node:fs';
 import { createBirpcGroup } from 'birpc';
 import { stringify, parse } from 'flatted';
+import { addDependency } from 'nypm';
+import { provider } from 'std-env';
 
 async function createBrowser() {
-  try {
-    const { Launcher } = await import(String("chrome-launcher"));
-    const chromePath = Launcher.getFirstInstallation();
-    return await playwrightCore.chromium.launch({
-      headless: true,
-      executablePath: chromePath
-    });
-  } catch (e) {
+  if (process.dev || process.env.prerender) {
+    try {
+      const { Launcher } = await import(String("chrome-launcher"));
+      const chromePath = Launcher.getFirstInstallation();
+      return await playwrightCore.chromium.launch({
+        headless: true,
+        executablePath: chromePath
+      });
+    } catch (e) {
+    }
   }
   try {
     return await playwrightCore.chromium.launch({
@@ -54,9 +58,9 @@ async function screenshot(browser, options) {
     width: options.width || 1200,
     height: options.height || 630
   });
-  const isHtml = options.path.startsWith("html:") || options.html;
+  const isHtml = options.html || options.path?.startsWith("html:");
   if (isHtml) {
-    const html = options.html || options.path.substring(5);
+    const html = options.html || options.path?.substring(5);
     await page.evaluate((html2) => {
       document.open("text/html");
       document.write(html2);
@@ -65,10 +69,13 @@ async function screenshot(browser, options) {
     await page.waitForLoadState("networkidle");
   } else {
     await page.goto(`${options.host}${options.path}`, {
-      timeout: 1e4,
+      timeout: process.env.prerender || process.dev ? 1e4 : 2e3,
       waitUntil: "networkidle"
     });
   }
+  const screenshotOptions = {
+    timeout: process.env.prerender || process.dev ? 1e4 : 2e3
+  };
   if (options.delay)
     await page.waitForTimeout(options.delay);
   if (options.mask) {
@@ -78,8 +85,10 @@ async function screenshot(browser, options) {
     }, options.mask);
   }
   if (options.selector)
-    return await page.locator(options.selector).screenshot();
-  return await page.screenshot();
+    return await page.locator(options.selector).screenshot(screenshotOptions);
+  const screenshot2 = await page.screenshot(screenshotOptions);
+  await page.close();
+  return screenshot2;
 }
 
 function setupPlaygroundRPC(nuxt, config) {
@@ -113,7 +122,7 @@ function setupPlaygroundRPC(nuxt, config) {
   const birpc = createBirpcGroup(serverFunctions, []);
   nuxt.hook("builder:watch", (e, path) => {
     if (e === "change")
-      birpc.boardcast.refresh.asEvent(path);
+      birpc.broadcast.refresh.asEvent(path);
   });
   const middleware = async (req, res) => {
     if (req.ws) {
@@ -164,11 +173,15 @@ function getBodyJson(req) {
   });
 }
 
-function decodeHtmlEntities(obj) {
+function decodeHtml(html) {
+  return html.replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&amp;/g, "&").replace(/&cent;/g, "\xA2").replace(/&pound;/g, "\xA3").replace(/&yen;/g, "\xA5").replace(/&euro;/g, "\u20AC").replace(/&copy;/g, "\xA9").replace(/&reg;/g, "\xAE").replace(/&quot;/g, '"').replace(/&#39;/g, "'").replace(/&#x27;/g, "'").replace(/&#x2F;/g, "/").replace(/&#([0-9]+);/g, (full, int) => {
+    return String.fromCharCode(parseInt(int));
+  });
+}
+function decodeObjectHtmlEntities(obj) {
   Object.entries(obj).forEach(([key, value]) => {
-    if (typeof value === "string") {
-      obj[key] = value.replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&amp;/g, "&").replace(/&quot;/g, '"').replace(/&#x27;/g, "'").replace(/&#x2F;/g, "/");
-    }
+    if (typeof value === "string")
+      obj[key] = decodeHtml(value);
   });
   return obj;
 }
@@ -185,27 +198,88 @@ function extractOgImageOptions(html) {
       console.warn("Failed to parse #nuxt-og-image-options", e, options);
   }
   if (options) {
-    const description = html.match(/<meta property="og:description" content="(.*?)">/)?.[1];
-    if (description)
-      options.description = description;
-    else
-      options.description = html.match(/<meta name="description" content="(.*?)">/)?.[1];
-    return decodeHtmlEntities(options);
+    if (!options.description) {
+      const description = html.match(/<meta property="og:description" content="(.*?)">/)?.[1];
+      if (description)
+        options.description = description;
+      else
+        options.description = html.match(/<meta name="description" content="(.*?)">/)?.[1];
+    }
+    return decodeObjectHtmlEntities(options);
   }
   return false;
 }
-function stripOgImageOptions(html) {
-  return html.replace(/<script id="nuxt-og-image-options" type="application\/json">(.*?)<\/script>/, "");
+
+const SVG2PNGWasmPlaceholder = '"/* NUXT_OG_IMAGE_SVG2PNG_WASM */"';
+const YogaWasmPlaceholder = '"/* NUXT_OG_IMAGE_YOGA_WASM */"';
+const ReSVGWasmPlaceholder = '"/* NUXT_OG_IMAGE_RESVG_WASM */"';
+const Wasms = [
+  {
+    placeholder: SVG2PNGWasmPlaceholder,
+    path: "svg2png/svg2png.wasm",
+    file: "svg2png.wasm"
+  },
+  {
+    placeholder: ReSVGWasmPlaceholder,
+    path: "resvg/resvg.wasm",
+    file: "resvg.wasm"
+  },
+  {
+    placeholder: YogaWasmPlaceholder,
+    path: "yoga/yoga.wasm",
+    file: "yoga.wasm"
+  }
+];
+const DefaultRuntimeCompatibility = {
+  browser: "playwright",
+  satori: "default",
+  wasm: "import",
+  png: "resvg"
+};
+const RuntimeCompatibility = {
+  "stackblitz": {
+    browser: false,
+    satori: "yoga-wasm",
+    wasm: "inline"
+  },
+  "netlify": {
+    browser: "lambda",
+    wasm: "inline"
+  },
+  "cloudflare-pages": {
+    browser: false
+  },
+  "cloudflare": {
+    browser: false
+  },
+  "vercel-edge": {
+    browser: false,
+    wasmImportQuery: "?module"
+  }
+};
+
+function getNitroProviderCompatibility(nuxt) {
+  if (nuxt.options.dev || nuxt.options._prepare) {
+    return defu({
+      wasm: "fetch",
+      browser: "universal"
+    }, DefaultRuntimeCompatibility);
+  }
+  if (provider === "stackblitz")
+    return defu(RuntimeCompatibility.stackblitz, DefaultRuntimeCompatibility);
+  const target = process.env.NITRO_PRESET || nuxt.options.nitro.preset;
+  const compatibility = RuntimeCompatibility[target];
+  if (compatibility === false)
+    return false;
+  return defu(compatibility || {}, DefaultRuntimeCompatibility);
+}
+function ensureDependency(nuxt, dep) {
+  return addDependency(dep, { cwd: nuxt.options.rootDir });
 }
 
 const PATH = "/__nuxt_og_image__";
 const PATH_ENTRY = `${PATH}/entry`;
 const PATH_PLAYGROUND = `${PATH}/client`;
-const edgeProvidersSupported = [
-  "cloudflare",
-  "vercel-edge",
-  "netlify-edge"
-];
 const module = defineNuxtModule({
   meta: {
     name: "nuxt-og-image",
@@ -218,26 +292,47 @@ const module = defineNuxtModule({
   defaults(nuxt) {
     const siteUrl = process.env.NUXT_PUBLIC_SITE_URL || process.env.NUXT_SITE_URL || nuxt.options.runtimeConfig.public?.siteUrl || nuxt.options.runtimeConfig.siteUrl;
     return {
-      // when we run `nuxi generate` we need to force prerendering
-      forcePrerender: !nuxt.options.dev && nuxt.options._generate,
       siteUrl,
       defaults: {
         component: "OgImageBasic",
         width: 1200,
-        height: 630
+        height: 630,
+        cacheTtl: 24 * 60 * 60 * 1e3
+        // default is to cache the image for 24 hours
       },
-      satoriProvider: true,
-      browserProvider: true,
+      runtimeSatori: true,
+      runtimeBrowser: nuxt.options.dev,
       fonts: [],
+      runtimeCacheStorage: false,
       satoriOptions: {},
-      experimentalInlineWasm: process.env.NITRO_PRESET === "netlify-edge" || nuxt.options.nitro.preset === "netlify-edge" || false,
-      experimentalRuntimeBrowser: false,
-      playground: process.env.NODE_ENV === "development" || nuxt.options.dev
+      playground: process.env.NODE_ENV === "development" || nuxt.options.dev,
+      debug: false
     };
   },
   async setup(config, nuxt) {
+    const logger = useLogger("nuxt-og-image");
     const { resolve } = createResolver(import.meta.url);
+    const nitroCompatibility = getNitroProviderCompatibility(nuxt);
+    if (!nitroCompatibility) {
+      const target = process.env.NITRO_PRESET || nuxt.options.nitro.preset;
+      logger.warn(`It looks like the nitro target ${target} doesn't support \`nuxt-og-image\`.`);
+      return;
+    }
+    if (!nitroCompatibility.browser && config.runtimeBrowser) {
+      config.runtimeBrowser = false;
+      logger.warn("It looks like you're using a nitro target that does not support the browser provider, disabling `runtimeBrowser`.");
+    }
+    if (nitroCompatibility.browser === "lambda") {
+      logger.info("It looks like you're deploying to an environment which requires `chrome-aws-lambda`, checking for dependency...");
+      await ensureDependency(nuxt, "chrome-aws-lambda");
+    }
     config.siteUrl = config.siteUrl || config.host;
+    if (!nuxt.options.dev && nuxt.options._generate && !config.siteUrl)
+      logger.warn("Nuxt is being prerendered but no siteUrl is set. This will result in broken images.");
+    if (config.runtimeCacheStorage && !nuxt.options.dev) {
+      nuxt.options.nitro.storage = nuxt.options.nitro.storage || {};
+      nuxt.options.nitro.storage["og-image"] = config.runtimeCacheStorage;
+    }
     if (!config.fonts.length)
       config.fonts = ["Inter:400", "Inter:700"];
     const distResolve = (p) => {
@@ -269,12 +364,14 @@ export {}
       lazy: true,
       handler: resolve("./runtime/nitro/middleware/og.png")
     });
-    ["html", "options", "svg", "vnode", "font"].forEach((type) => {
-      addServerHandler({
-        lazy: true,
-        route: `/api/og-image-${type}`,
-        handler: resolve(`./runtime/nitro/routes/${type}`)
-      });
+    ["html", "options", "svg", "vnode", "font", "debug"].forEach((type) => {
+      if (type !== "debug" || config.debug) {
+        addServerHandler({
+          lazy: true,
+          route: `/api/og-image-${type}`,
+          handler: resolve(`./runtime/nitro/routes/${type}`)
+        });
+      }
     });
     nuxt.hook("devtools:customTabs", (iframeTabs) => {
       iframeTabs.push({
@@ -323,92 +420,118 @@ export {}
     });
     const runtimeDir = resolve("./runtime");
     nuxt.options.build.transpile.push(runtimeDir);
-    const moduleAssetDir = resolve("./runtime/public-assets");
-    const assetDirs = [
-      resolve(nuxt.options.rootDir, nuxt.options.dir.public),
-      moduleAssetDir
+    addServerPlugin(resolve("./runtime/nitro/plugins/prerender"));
+    const customAssetDirs = [
+      // allows us to show custom error images
+      resolve("./runtime/public-assets")
     ];
+    if (config.runtimeSatori) {
+      if (config.fonts.includes("Inter:400"))
+        customAssetDirs.push(resolve("./runtime/public-assets-optional/inter-font"));
+      if (nitroCompatibility.png === "resvg" && nitroCompatibility.wasm === "fetch")
+        customAssetDirs.push(resolve("./runtime/public-assets-optional/resvg"));
+      else if (nitroCompatibility.png === "svg2png" && nitroCompatibility.wasm === "fetch")
+        customAssetDirs.push(resolve("./runtime/public-assets-optional/svg2png"));
+      if (nitroCompatibility.satori === "yoga-wasm")
+        customAssetDirs.push(resolve("./runtime/public-assets-optional/yoga"));
+    }
     nuxt.hooks.hook("modules:done", async () => {
       nuxt.hooks.callHook("og-image:config", config);
-      nuxt.options.runtimeConfig["nuxt-og-image"] = { ...config, assetDirs };
+      nuxt.options.runtimeConfig["nuxt-og-image"] = {
+        ...config,
+        // avoid adding credentials
+        runtimeCacheStorage: Boolean(config.runtimeCacheStorage),
+        assetDirs: [
+          resolve(nuxt.options.rootDir, nuxt.options.dir.public),
+          ...customAssetDirs
+        ]
+      };
     });
-    const useSatoriWasm = provider === "stackblitz";
     nuxt.hooks.hook("nitro:config", async (nitroConfig) => {
       nitroConfig.externals = defu(nitroConfig.externals || {}, {
         inline: [runtimeDir]
       });
-      if (config.experimentalRuntimeBrowser) {
+      if (config.runtimeBrowser) {
         nitroConfig.alias = nitroConfig.alias || {};
         nitroConfig.alias.electron = "unenv/runtime/mock/proxy-cjs";
         nitroConfig.alias.bufferutil = "unenv/runtime/mock/proxy-cjs";
         nitroConfig.alias["utf-8-validate"] = "unenv/runtime/mock/proxy-cjs";
       }
       nitroConfig.publicAssets = nitroConfig.publicAssets || [];
-      nitroConfig.publicAssets.push({ dir: moduleAssetDir, maxAge: 31536e3 });
+      customAssetDirs.forEach((dir) => {
+        nitroConfig.publicAssets.push({ dir, maxAge: 31536e3 });
+      });
       const providerPath = `${runtimeDir}/nitro/providers`;
-      if (config.browserProvider) {
-        nitroConfig.virtual["#nuxt-og-image/browser"] = nuxt.options.dev || config.experimentalRuntimeBrowser ? `
-import node from '${providerPath}/browser/node'
-
+      if (config.runtimeBrowser) {
+        nitroConfig.virtual["#nuxt-og-image/browser"] = `
+let browser
 export default async function() {
-  return node
-}
-` : `export default async function() {
- return () => {}
+  browser = browser || await import('${providerPath}/browser/${nitroCompatibility.browser}').then((m) => m.default || m)
+  return browser
 }
 `;
       }
-      if (config.satoriProvider) {
-        nitroConfig.virtual["#nuxt-og-image/satori"] = `import satori from '${providerPath}/satori/${useSatoriWasm ? "webworker" : "node"}'
+      if (config.runtimeSatori) {
+        nitroConfig.virtual["#nuxt-og-image/satori"] = `import satori from '${providerPath}/satori/${nitroCompatibility.satori}'
 export default async function() {
   return satori
 }`;
-        nitroConfig.virtual["#nuxt-og-image/svg2png"] = `
-import svg2png from '${providerPath}/svg2png/universal'
+        nitroConfig.virtual["#nuxt-og-image/png"] = `import png from '${providerPath}/png/${nitroCompatibility.png}'
 export default async function() {
- return svg2png
-}`;
+ return png
+}
+`;
       }
       nitroConfig.virtual["#nuxt-og-image/provider"] = `
-import satori from '${relative(nuxt.options.rootDir, resolve("./runtime/nitro/renderers/satori"))}'
-import browser from '${relative(nuxt.options.rootDir, resolve("./runtime/nitro/renderers/browser"))}'
+${config.runtimeSatori ? `import satori from '${relative(nuxt.options.rootDir, resolve("./runtime/nitro/renderers/satori"))}'` : ""}
+${config.runtimeBrowser ? `import browser from '${relative(nuxt.options.rootDir, resolve("./runtime/nitro/renderers/browser"))}'` : ""}
 
 export async function useProvider(provider) {
   if (provider === 'satori')
-    return satori
+    return ${config.runtimeSatori ? "satori" : "null"}
   if (provider === 'browser')
-    return browser
+    return ${config.runtimeBrowser ? "browser" : "null"}
+  return null
 }
       `;
     });
     nuxt.hooks.hook("nitro:init", async (nitro) => {
       let screenshotQueue = [];
       nitro.hooks.hook("compiled", async (_nitro) => {
-        if (edgeProvidersSupported.includes(_nitro.options.preset)) {
-          await copy(resolve("./runtime/public-assets/inter-latin-ext-400-normal.woff"), resolve(_nitro.options.output.publicDir, "inter-latin-ext-400-normal.woff"));
-          await copy(resolve("./runtime/public-assets/inter-latin-ext-700-normal.woff"), resolve(_nitro.options.output.publicDir, "inter-latin-ext-700-normal.woff"));
-          if (!config.experimentalInlineWasm) {
-            await copy(resolve("./runtime/public-assets/svg2png.wasm"), resolve(_nitro.options.output.serverDir, "svg2png.wasm"));
-            if (useSatoriWasm)
-              await copy(resolve("./runtime/public-assets/yoga.wasm"), resolve(_nitro.options.output.serverDir, "yoga.wasm"));
+        if (!config.runtimeSatori || nuxt.options.dev)
+          return;
+        if (config.fonts.includes("Inter:400"))
+          await copy(resolve("./runtime/public-assets-optional/inter-font/inter-latin-ext-400-normal.woff"), resolve(_nitro.options.output.publicDir, "inter-latin-ext-400-normal.woff"));
+        if (config.fonts.includes("Inter:700"))
+          await copy(resolve("./runtime/public-assets-optional/inter-font/inter-latin-ext-700-normal.woff"), resolve(_nitro.options.output.publicDir, "inter-latin-ext-700-normal.woff"));
+        const configuredEntry = nitro.options.rollupConfig?.output.entryFileNames;
+        const wasmProviderPath = resolve(_nitro.options.output.serverDir, typeof configuredEntry === "string" ? configuredEntry : "index.mjs");
+        const paths = [wasmProviderPath];
+        const chunks = await globby([`${_nitro.options.output.serverDir}/chunks/**/*.mjs`], { absolute: true });
+        paths.push(...chunks);
+        for (const path of paths) {
+          if (!await pathExists(path))
+            continue;
+          let contents = await readFile(path, "utf-8");
+          let updated = false;
+          if (_nitro.options.preset.includes("vercel") && path === wasmProviderPath) {
+            contents = contents.replace(".cwd(),", '?.cwd || "/",');
+            updated = true;
           }
-          const indexFile = resolve(_nitro.options.output.serverDir, _nitro.options.preset === "netlify-edge" ? "server.mjs" : "index.mjs");
-          if (await pathExists(indexFile)) {
-            const indexContents = (await readFile(indexFile, "utf-8")).replace(".cwd(),", '?.cwd || "/",');
-            if (!config.experimentalInlineWasm) {
-              await writeFile(
-                indexFile,
-                indexContents.replace('"/* NUXT_OG_IMAGE_SVG2PNG_WASM */"', 'import("./svg2png.wasm").then(m => m.default || m)').replace('"/* NUXT_OG_IMAGE_YOGA_WASM */"', 'import("./yoga.wasm").then(m => m.default || m)')
-              );
-            } else {
-              const svg2pngWasm = await readFile(resolve("./runtime/public-assets/svg2png.wasm"), "base64");
-              const yogaWasm = await readFile(resolve("./runtime/public-assets/yoga.wasm"), "base64");
-              await writeFile(
-                indexFile,
-                indexContents.replace('"/* NUXT_OG_IMAGE_SVG2PNG_WASM */"', `Buffer.from("${svg2pngWasm}", "base64")`).replace('"/* NUXT_OG_IMAGE_YOGA_WASM */"', `Buffer.from("${yogaWasm}", "base64")`)
-              );
+          for (const wasm of Wasms) {
+            if (contents.includes(wasm.placeholder)) {
+              if (nitroCompatibility.wasm === "import") {
+                contents = contents.replace(wasm.placeholder, `import("./${wasm.file}${nitroCompatibility.wasmImportQuery || ""}").then(m => m.default || m)`);
+                await copy(resolve(`./runtime/public-assets-optional/${wasm.path}`), resolve(dirname(path), wasm.file));
+              } else if (nitroCompatibility.wasm === "inline") {
+                const wasmBuffer = await readFile(resolve(`./runtime/public-assets-optional/${wasm.path}`), "base64");
+                contents = contents.replace(wasm.placeholder, `Buffer.from("${wasmBuffer}", "base64")`);
+              }
+              updated = true;
             }
           }
+          if (updated)
+            await writeFile(path, contents, { encoding: "utf-8" });
         }
       });
       const _routeRulesMatcher = toRouteMatcher(
@@ -421,7 +544,6 @@ export async function useProvider(provider) {
         if (!html)
           return;
         const extractedOptions = extractOgImageOptions(html);
-        ctx.contents = stripOgImageOptions(html);
         const routeRules = defu({}, ..._routeRulesMatcher.matchAll(ctx.route).reverse());
         if (!extractedOptions || routeRules.ogImage === false)
           return;
@@ -440,21 +562,6 @@ export async function useProvider(provider) {
         await nuxt.callHook("og-image:prerenderScreenshots", screenshotQueue);
         if (screenshotQueue.length === 0)
           return;
-        for (const entry of screenshotQueue) {
-          if (entry.route && Object.keys(entry).length === 1) {
-            const html = await $fetch(entry.route);
-            const extractedOptions = extractOgImageOptions(html);
-            const routeRules = defu({}, ..._routeRulesMatcher.matchAll(entry.route).reverse());
-            Object.assign(entry, {
-              // @ts-expect-error runtime
-              path: extractedOptions.component ? `/api/og-image-html?path=${entry.route}` : entry.route,
-              ...extractedOptions,
-              ...routeRules.ogImage || {}
-            });
-          }
-          if (entry.component)
-            entry.html = await $fetch(entry.path);
-        }
         nitro.logger.info("Ensuring chromium install for og:image generation...");
         const installChromeProcess = execa("npx", ["playwright", "install", "chromium"], {
           stdio: "inherit"
@@ -481,12 +588,27 @@ export async function useProvider(provider) {
           browser = await createBrowser();
           if (browser) {
             nitro.logger.info(`Prerendering ${screenshotQueue.length} og:image screenshots...`);
+            for (const entry of screenshotQueue) {
+              if (entry.route && Object.keys(entry).length === 1) {
+                const html = await $fetch(entry.route, { baseURL: withBase(nuxt.options.app.baseURL, host) });
+                const extractedOptions = extractOgImageOptions(html);
+                const routeRules = defu({}, ..._routeRulesMatcher.matchAll(entry.route).reverse());
+                Object.assign(entry, {
+                  // @ts-expect-error runtime
+                  path: extractedOptions.component ? `/api/og-image-html?path=${entry.route}` : entry.route,
+                  ...extractedOptions,
+                  ...routeRules.ogImage || {}
+                });
+              }
+              if (entry.component)
+                entry.html = await globalThis.$fetch(entry.path);
+            }
             for (const k in screenshotQueue) {
               const entry = screenshotQueue[k];
               const start = Date.now();
               let hasError = false;
-              const dirname = joinURL(nitro.options.output.publicDir, entry.route, "/__og_image__/");
-              const filename = joinURL(dirname, "/og.png");
+              const dirname2 = joinURL(nitro.options.output.publicDir, entry.route, "/__og_image__/");
+              const filename = joinURL(dirname2, "/og.png");
               try {
                 const imgBuffer = await screenshot(browser, {
                   ...config.defaults || {},
@@ -494,7 +616,7 @@ export async function useProvider(provider) {
                   host
                 });
                 try {
-                  await mkdirp(dirname);
+                  await mkdirp(dirname2);
                 } catch (e) {
                 }
                 await writeFile(filename, imgBuffer);
@@ -518,7 +640,7 @@ export async function useProvider(provider) {
         }
         screenshotQueue = [];
       };
-      if (!nuxt.options._prepare) {
+      if (nuxt.options._generate) {
         nitro.hooks.hook("rollup:before", async () => {
           await captureScreenshots();
         });

package/dist/runtime/browserUtil.d.ts

@@ -1,4 +1,5 @@
 /// <reference types="node" />
+import type { Buffer } from 'node:buffer';
 import type { Browser } from 'playwright-core';
 import type { ScreenshotOptions } from '../types';
 export declare function screenshot(browser: Browser, options: Partial<ScreenshotOptions> & Record<string, any>): Promise<Buffer>;

package/dist/runtime/browserUtil.mjs

@@ -6,9 +6,9 @@ export async function screenshot(browser, options) {
     width: options.width || 1200,
     height: options.height || 630
   });
-  const isHtml = options.path.startsWith("html:") || options.html;
+  const isHtml = options.html || options.path?.startsWith("html:");
   if (isHtml) {
-    const html = options.html || options.path.substring(5);
+    const html = options.html || options.path?.substring(5);
     await page.evaluate((html2) => {
       document.open("text/html");
       document.write(html2);
@@ -17,10 +17,13 @@ export async function screenshot(browser, options) {
     await page.waitForLoadState("networkidle");
   } else {
     await page.goto(`${options.host}${options.path}`, {
-      timeout: 1e4,
+      timeout: process.env.prerender || process.dev ? 1e4 : 2e3,
       waitUntil: "networkidle"
     });
   }
+  const screenshotOptions = {
+    timeout: process.env.prerender || process.dev ? 1e4 : 2e3
+  };
   if (options.delay)
     await page.waitForTimeout(options.delay);
   if (options.mask) {
@@ -30,6 +33,8 @@ export async function screenshot(browser, options) {
     }, options.mask);
   }
   if (options.selector)
-    return await page.locator(options.selector).screenshot();
-  return await page.screenshot();
+    return await page.locator(options.selector).screenshot(screenshotOptions);
+  const screenshot2 = await page.screenshot(screenshotOptions);
+  await page.close();
+  return screenshot2;
 }

package/dist/runtime/composables/defineOgImage.mjs

@@ -1,7 +1,7 @@
-import { useServerHead } from "@vueuse/head";
 import { withBase } from "ufo";
 import { useRequestEvent } from "#app";
-import { useRouter, useRuntimeConfig } from "#imports";
+import { useHostname } from "../nitro/util-hostname.mjs";
+import { useRouter, useRuntimeConfig, useServerHead } from "#imports";
 export function defineOgImageScreenshot(options = {}) {
   const router = useRouter();
   const route = router?.currentRoute?.value?.path || "";
@@ -14,29 +14,31 @@ export function defineOgImageScreenshot(options = {}) {
   });
 }
 export function defineOgImageDynamic(options = {}) {
-  const { satoriProvider, forcePrerender } = useRuntimeConfig()["nuxt-og-image"];
+  const { runtimeSatori } = useRuntimeConfig()["nuxt-og-image"];
   defineOgImage({
-    provider: satoriProvider ? "satori" : "browser",
-    static: !!forcePrerender,
+    provider: runtimeSatori ? "satori" : "browser",
+    static: false,
     ...options
   });
 }
 export function defineOgImageStatic(options = {}) {
-  const { satoriProvider } = useRuntimeConfig()["nuxt-og-image"];
+  const { runtimeSatori } = useRuntimeConfig()["nuxt-og-image"];
   defineOgImage({
-    provider: satoriProvider ? "satori" : "browser",
+    provider: runtimeSatori ? "satori" : "browser",
     static: true,
     ...options
   });
 }
 export function defineOgImage(options = {}) {
   if (process.server) {
-    const { forcePrerender, defaults, host } = useRuntimeConfig()["nuxt-og-image"];
+    let resolveSrc = function() {
+      return withBase(`${route === "/" ? "" : route}/__og_image__/og.png`, baseUrl);
+    };
+    const { defaults, siteUrl } = useRuntimeConfig()["nuxt-og-image"];
     const router = useRouter();
     const route = router?.currentRoute?.value?.path || "";
     const e = useRequestEvent();
-    if ((forcePrerender || options.static) && options.provider === "satori")
-      e.res.setHeader("x-nitro-prerender", `${route === "/" ? "" : route}/__og_image__/og.png`);
+    const baseUrl = process.env.prerender ? siteUrl : useHostname(e);
     const meta = [
       {
         name: "twitter:card",
@@ -44,11 +46,11 @@ export function defineOgImage(options = {}) {
       },
       {
         name: "twitter:image:src",
-        content: () => withBase(`${route === "/" ? "" : route}/__og_image__/og.png`, host)
+        content: resolveSrc
       },
       {
         property: "og:image",
-        content: () => withBase(`${route === "/" ? "" : route}/__og_image__/og.png`, host)
+        content: resolveSrc
       },
       {
         property: "og:image:width",