@atlaspack/runtime-js 2.12.1-canary.3583 → 2.12.1-canary.3585

package/lib/helpers/bundle-url-common.ts

@@ -0,0 +1,10 @@
+// Get the URL without the filename (last / segment)
+function getBaseURL(url: string) {
+  return url.slice(0, url.lastIndexOf('/')) + '/';
+}
+
+const stackTraceUrlRegexp =
+  /(https?|file|ftp|(chrome|moz|safari-web)-extension):\/\/[^)\n]+/g;
+
+exports.getBaseURL = getBaseURL;
+exports.stackTraceUrlRegexp = stackTraceUrlRegexp;

package/lib/helpers/bundle-url-shards.ts

@@ -0,0 +1,95 @@
+const {getBaseURL, stackTraceUrlRegexp} = require('./bundle-url-common');
+
+const bundleURL: Record<string, string> = {};
+
+function getShardedBundleURL(
+  bundleName: string,
+  cookieName: string,
+  cookieString: string,
+  maxShards: number,
+  inputError?: string,
+): string {
+  let value = bundleURL[bundleName];
+
+  if (value) {
+    return value;
+  }
+
+  try {
+    throw inputError ?? new Error();
+  } catch (err) {
+    var matches = ('' + err.stack).match(stackTraceUrlRegexp);
+
+    if (!matches) {
+      return '/';
+    }
+
+    // The first stack frame will be this function.
+    // Use the 2nd one, which will be a runtime in the original bundle.
+    const stackUrl = matches[1];
+    const baseUrl = getBaseURL(stackUrl);
+
+    // If the cookie doesn't exist then we don't need to shard
+    if (cookieString.indexOf(cookieName) === -1) {
+      return baseUrl;
+    }
+
+    const shardNumber = getDomainShardIndex(bundleName, maxShards);
+    const url = new URL(baseUrl);
+
+    const shardedDomain = getShardedDomain(url.hostname, shardNumber);
+    url.hostname = shardedDomain;
+
+    value = url.toString();
+
+    bundleURL[bundleName] = value;
+    return value;
+  }
+}
+
+function getDomainShardIndex(str: string, maxShards: number) {
+  let shard = str.split('').reduce((a, b) => {
+    const n = (a << maxShards) - a + b.charCodeAt(0);
+
+    // The value returned by << is 64 bit, the & operator coerces to 32,
+    // prevents overflow as we iterate.
+    return n & n;
+  }, 0);
+
+  shard = shard % maxShards;
+
+  // Make number positive
+  if (shard < 0) {
+    shard += maxShards;
+  }
+
+  return shard;
+}
+
+function getShardedDomain(domain: string, shard: number) {
+  let i = domain.indexOf('.');
+
+  // Domains like localhost have no . separators
+  if (i === -1) {
+    return `${removeTrailingShard(domain)}-${shard}`;
+  }
+
+  // If this domain already has a shard number in it, strip it out before adding
+  // the new one
+  const firstSubdomain = removeTrailingShard(domain.slice(0, i));
+
+  return `${firstSubdomain}-${shard}${domain.slice(i)}`;
+}
+
+const trailingShardRegex = /-\d+$/;
+
+function removeTrailingShard(subdomain: string) {
+  if (!trailingShardRegex.test(subdomain)) {
+    return subdomain;
+  }
+
+  const shardIdx = subdomain.lastIndexOf('-');
+  return subdomain.slice(0, shardIdx);
+}
+
+exports.getShardedBundleURL = getShardedBundleURL;

package/lib/helpers/bundle-url.ts

@@ -0,0 +1,36 @@
+const {getBaseURL, stackTraceUrlRegexp} = require('./bundle-url-common');
+
+const bundleURL = {};
+
+function getBundleURLCached(id: string) {
+  let value = bundleURL[id];
+
+  if (!value) {
+    value = getBundleURL();
+    bundleURL[id] = value;
+  }
+
+  return value;
+}
+
+function getBundleURL() {
+  try {
+    throw new Error();
+  } catch (err) {
+    var matches = ('' + err.stack).match(stackTraceUrlRegexp);
+    if (matches) {
+      // The first two stack frames will be this function and getBundleURLCached.
+      // Use the 3rd one, which will be a runtime in the original bundle.
+      return getBaseURL(matches[2]);
+    }
+  }
+
+  return '/';
+}
+
+function getOrigin(url: string) {
+  return new URL(url).origin;
+}
+
+exports.getOrigin = getOrigin;
+exports.getBundleURL = getBundleURLCached;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atlaspack/runtime-js",
-  "version": "2.12.1-canary.3583+3054dda39",
+  "version": "2.12.1-canary.3585+0929ac006",
   "license": "(MIT OR Apache-2.0)",
   "publishConfig": {
     "access": "public"
@@ -12,15 +12,15 @@
   "main": "lib/JSRuntime.js",
   "source": "src/JSRuntime.js",
   "engines": {
-    "atlaspack": "2.12.1-canary.3583+3054dda39",
+    "atlaspack": "2.12.1-canary.3585+0929ac006",
     "node": ">= 16.0.0"
   },
   "dependencies": {
-    "@atlaspack/diagnostic": "2.12.1-canary.3583+3054dda39",
-    "@atlaspack/feature-flags": "2.12.1-canary.3583+3054dda39",
-    "@atlaspack/plugin": "2.12.1-canary.3583+3054dda39",
-    "@atlaspack/utils": "2.12.1-canary.3583+3054dda39",
+    "@atlaspack/diagnostic": "2.12.1-canary.3585+0929ac006",
+    "@atlaspack/feature-flags": "2.12.1-canary.3585+0929ac006",
+    "@atlaspack/plugin": "2.12.1-canary.3585+0929ac006",
+    "@atlaspack/utils": "2.12.1-canary.3585+0929ac006",
     "nullthrows": "^1.1.1"
   },
-  "gitHead": "3054dda39b54666caba268f09cadf801b43ce3a5"
+  "gitHead": "0929ac006bd11b9b91c2ef27f90a272d2355b3a1"
 }