From 6a0d6814731ba78b3c51bfa57cc03d683385d243 Mon Sep 17 00:00:00 2001
From: Robrecht Meersman <robrecht.meersman@inthepocket.com>
Date: Tue, 1 Mar 2022 17:26:00 +0100
Subject: [PATCH] feat: support width and height options

---
 README.md    |  4 ++++
 src/index.ts | 44 ++++++++++++++++++++++++++------------------
 2 files changed, 30 insertions(+), 18 deletions(-)

diff --git a/README.md b/README.md
index 9329f93..ba9a23e 100644
--- a/README.md
+++ b/README.md
@@ -174,6 +174,10 @@ const nextApiOgImageConfig = {
   strategy: 'query',
   // 'Content-Type' HTTP header
   contentType: 'image/png',
+  // Width of the image in pixels (default 1200)
+  width: 2048,
+  // Height of the image in pixels (default 630)
+  height: 1170,
   // 'Cache-Control' HTTP header
   cacheControl: 'max-age 3600, must-revalidate',
   // NOTE: Options within 'dev' object works only when process.env.NODE_ENV === 'development'
diff --git a/src/index.ts b/src/index.ts
index 8072306..29aad88 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,6 +1,6 @@
 import type { NextApiRequest, NextApiResponse } from 'next'
 import type { Except, RequireExactlyOne } from 'type-fest'
-import type { Page } from 'puppeteer-core'
+import type { Page, Viewport } from 'puppeteer-core'
 import type { ReactElement } from 'react'
 import { renderToStaticMarkup } from 'react-dom/server'
 import deepMerge from 'deepmerge'
@@ -35,6 +35,8 @@ export type NextApiOgImageConfig<
   strategy?: StrategyOption
   contentType?: string
   cacheControl?: string
+  width?: number
+  height?: number
   dev?: Partial<{
     inspectHtml: boolean
     errorsInResponse: boolean
@@ -56,6 +58,8 @@ export function withOGImage<
     contentType: 'image/png',
     strategy: 'query',
     cacheControl: 'max-age 3600, must-revalidate',
+    width: 1200,
+    height: 630,
     dev: {
       inspectHtml: true,
       errorsInResponse: true,
@@ -69,6 +73,8 @@ export function withOGImage<
     cacheControl,
     strategy,
     contentType,
+    width,
+    height,
     dev: { inspectHtml, errorsInResponse },
   } = options
 
@@ -84,7 +90,7 @@ export function withOGImage<
 
   const createBrowserEnvironment = pipe(
     getChromiumExecutable,
-    prepareWebPage,
+    prepareWebPageFactory({ width, height }),
     createImageFactory(inspectHtml),
   )
 
@@ -203,26 +209,28 @@ function getChromiumExecutable(browserEnvironment: BrowserEnvironment) {
   return { ...browserEnvironment, executable }
 }
 
-async function prepareWebPage(browserEnvironment: BrowserEnvironment) {
-  const { page, envMode, executable } = browserEnvironment
+function prepareWebPageFactory(viewPort: Viewport) {
+  return async function (browserEnvironment: BrowserEnvironment) {
+    const { page, envMode, executable } = browserEnvironment
 
-  if (page) {
-    return { ...browserEnvironment, page }
-  }
+    if (page) {
+      return { ...browserEnvironment, page }
+    }
 
-  const chromiumOptions = !isProductionLikeMode(envMode)
-    ? { args: [], executablePath: executable, headless: true }
-    : {
-        args: chrome.args,
-        executablePath: await chrome.executablePath,
-        headless: chrome.headless,
-      }
+    const chromiumOptions = !isProductionLikeMode(envMode)
+      ? { args: [], executablePath: executable, headless: true }
+      : {
+          args: chrome.args,
+          executablePath: await chrome.executablePath,
+          headless: chrome.headless,
+        }
 
-  const browser = await core.launch(chromiumOptions)
-  const newPage = await browser.newPage()
-  await newPage.setViewport({ width: 1200, height: 630 })
+    const browser = await core.launch(chromiumOptions)
+    const newPage = await browser.newPage()
+    await newPage.setViewport(viewPort)
 
-  return { ...browserEnvironment, page: newPage }
+    return { ...browserEnvironment, page: newPage }
+  }
 }
 
 function createImageFactory(inspectHtml: boolean) {