screenshot/test_1

node_modules/playwright-core/lib/server/fileUploadUtils.js

@@ -0,0 +1,78 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.fileUploadSizeLimit = void 0;
+exports.prepareFilesForUpload = prepareFilesForUpload;
+var _fs = _interopRequireDefault(require("fs"));
+var _path = _interopRequireDefault(require("path"));
+var _assert = require("../utils/isomorphic/assert");
+var _utilsBundle = require("../utilsBundle");
+function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+/**
+ * Copyright (c) Microsoft Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Keep in sync with the client.
+const fileUploadSizeLimit = exports.fileUploadSizeLimit = 50 * 1024 * 1024;
+async function filesExceedUploadLimit(files) {
+  const sizes = await Promise.all(files.map(async file => (await _fs.default.promises.stat(file)).size));
+  return sizes.reduce((total, size) => total + size, 0) >= fileUploadSizeLimit;
+}
+async function prepareFilesForUpload(frame, params) {
+  var _fileBuffers;
+  const {
+    payloads,
+    streams,
+    directoryStream
+  } = params;
+  let {
+    localPaths,
+    localDirectory
+  } = params;
+  if ([payloads, localPaths, localDirectory, streams, directoryStream].filter(Boolean).length !== 1) throw new Error('Exactly one of payloads, localPaths and streams must be provided');
+  if (streams) localPaths = streams.map(c => c.path());
+  if (directoryStream) localDirectory = directoryStream.path();
+  if (localPaths) {
+    for (const p of localPaths) (0, _assert.assert)(_path.default.isAbsolute(p) && _path.default.resolve(p) === p, 'Paths provided to localPaths must be absolute and fully resolved.');
+  }
+  let fileBuffers = payloads;
+  if (!frame._page._browserContext._browser._isCollocatedWithServer) {
+    // If the browser is on a different machine read files into buffers.
+    if (localPaths) {
+      if (await filesExceedUploadLimit(localPaths)) throw new Error('Cannot transfer files larger than 50Mb to a browser not co-located with the server');
+      fileBuffers = await Promise.all(localPaths.map(async item => {
+        return {
+          name: _path.default.basename(item),
+          buffer: await _fs.default.promises.readFile(item),
+          lastModifiedMs: (await _fs.default.promises.stat(item)).mtimeMs
+        };
+      }));
+      localPaths = undefined;
+    }
+  }
+  const filePayloads = (_fileBuffers = fileBuffers) === null || _fileBuffers === void 0 ? void 0 : _fileBuffers.map(payload => ({
+    name: payload.name,
+    mimeType: payload.mimeType || _utilsBundle.mime.getType(payload.name) || 'application/octet-stream',
+    buffer: payload.buffer.toString('base64'),
+    lastModifiedMs: payload.lastModifiedMs
+  }));
+  return {
+    localPaths,
+    localDirectory,
+    filePayloads
+  };
+}