Make .buffer() return a Uint8Array instead of Buffer

index.d.ts

@@ -1,4 +1,3 @@
-import type {Buffer} from 'node:buffer';
 import type {
 	PuppeteerLaunchOptions,
 	Page,
@@ -466,7 +465,7 @@ declare const captureWebsite: {
 	@param input - The URL, file URL, data URL, local file path to the website, or HTML.
 	@returns The screenshot as binary.
 	*/
-	buffer: (input: string, options?: Options) => Promise<Buffer>;
+	buffer: (input: string, options?: Options) => Promise<Uint8Array>;
 
 	/**
 	Capture a screenshot of the given `input`.

index.js

@@ -452,7 +452,7 @@ captureWebsite.file = async (url, filePath, options = {}) => {
 	});
 };
 
-captureWebsite.buffer = async (url, options) => internalCaptureWebsite(url, options);
+captureWebsite.buffer = async (url, options) => new Uint8Array(await internalCaptureWebsite(url, options));
 
 captureWebsite.base64 = async (url, options) => {
 	const screenshot = await internalCaptureWebsite(url, options);

index.test-d.ts

@@ -1,4 +1,3 @@
-import type {Buffer} from 'node:buffer';
 import {expectType} from 'tsd';
 import captureWebsite, {devices} from './index.js';
 
@@ -18,7 +17,7 @@ expectType<Promise<string>>(
 	captureWebsite.base64('https://github.com/sindresorhus/capture-website#readme'),
 );
 
-expectType<Promise<Buffer>>(
+expectType<Promise<Uint8Array>>(
 	captureWebsite.buffer('https://github.com/sindresorhus/capture-website#readme'),
 );
 

package.json

@@ -59,6 +59,7 @@
 		"png-js": "^1.0.0",
 		"tempy": "^3.1.0",
 		"tsd": "^0.29.0",
+		"uint8array-extras": "^0.5.0",
 		"xo": "^0.56.0"
 	}
 }

test.js

@@ -1,5 +1,4 @@
 /* global document */
-import {Buffer} from 'node:buffer';
 import fs from 'node:fs';
 import test from 'ava';
 import {imageDimensionsFromData} from 'image-dimensions';
@@ -13,6 +12,7 @@ import delay from 'delay';
 import toughCookie from 'tough-cookie';
 import fileUrl from 'file-url';
 import {KnownDevices} from 'puppeteer';
+import {base64ToUint8Array} from 'uint8array-extras';
 import captureWebsite from './index.js';
 
 const defaultResponse = (() => {
@@ -109,7 +109,7 @@ test('captureWebsite.base64()', async t => {
 	});
 
 	t.is(typeof screenshot, 'string');
-	t.true(isPng(Buffer.from(screenshot, 'base64')));
+	t.true(isPng(base64ToUint8Array(screenshot)));
 });
 
 test('`type` option', async t => {