Refactor context handling and update package exports

Author: AbanoubGhadban

node_package/src/ReactOnRails.node.ts

@@ -1,7 +1,10 @@
 import ReactOnRails from './ReactOnRails';
+import { setReactOnRails } from './context';
 import streamServerRenderedReactComponent from './streamServerRenderedReactComponent';
 
 ReactOnRails.streamServerRenderedReactComponent = streamServerRenderedReactComponent;
 
-export * from './ReactOnRails';
-export { default } from './ReactOnRails';
+setReactOnRails(ReactOnRails);
+
+export default ReactOnRails;
+export * from './types';

node_package/src/ReactOnRails.ts

@@ -8,7 +8,6 @@ import serverRenderReactComponent from './serverRenderReactComponent';
 import buildConsoleReplay from './buildConsoleReplay';
 import createReactOutput from './createReactOutput';
 import Authenticity from './Authenticity';
-import context from './context';
 import type {
   RegisteredComponent,
   RenderParams,
@@ -19,31 +18,16 @@ import type {
   AuthenticityHeaders,
   Store,
   StoreGenerator,
+  ReactOnRails as ReactOnRailsType,
 } from './types';
 import reactHydrateOrRender from './reactHydrateOrRender';
 
-const ctx = context();
-
-if (ctx === undefined) {
-  throw new Error("The context (usually Window or NodeJS's Global) is undefined.");
-}
-
-if (ctx.ReactOnRails !== undefined) {
-  throw new Error(`
-    The ReactOnRails value exists in the ${ctx} scope, it may not be safe to overwrite it.
-    
-    This could be caused by setting Webpack's optimization.runtimeChunk to "true" or "multiple," rather than "single." Check your Webpack configuration.
-    
-    Read more at https://github.com/shakacode/react_on_rails/issues/1558.
-  `);
-}
-
 const DEFAULT_OPTIONS = {
   traceTurbolinks: false,
   turbo: false,
 };
 
-ctx.ReactOnRails = {
+const ReactOnRails: ReactOnRailsType = {
   options: {},
   /**
    * Main entry point to using the react-on-rails npm package. This is how Rails will be able to
@@ -299,9 +283,4 @@ ctx.ReactOnRails = {
   },
 };
 
-ctx.ReactOnRails.resetOptions();
-
-ClientStartup.clientStartup(ctx);
-
-export * from "./types";
-export default ctx.ReactOnRails;
+export default ReactOnRails;

node_package/src/ReactOnRails.web.ts

@@ -0,0 +1,7 @@
+import ReactOnRails from './ReactOnRails';
+import { setReactOnRails } from './context';
+
+setReactOnRails(ReactOnRails);
+
+export default ReactOnRails;
+export * from './types';

node_package/src/context.ts

@@ -1,9 +1,36 @@
+import { ReactOnRails as ReactOnRailsType } from './types';
+import * as ClientStartup from './clientStartup';
+
 /**
  * Get the context, be it window or global
  * @returns {boolean|Window|*|context}
  */
-export default function context(this: void): Window | NodeJS.Global | void {
+export function context(this: void): Window | NodeJS.Global | void {
   return ((typeof window !== 'undefined') && window) ||
     ((typeof global !== 'undefined') && global) ||
     this;
 }
+
+export const setReactOnRails = (value: ReactOnRailsType) => {
+  const ctx = context();
+
+  if (ctx === undefined) {
+    throw new Error("The context (usually Window or NodeJS's Global) is undefined.");
+  }
+
+  if (ctx.ReactOnRails !== undefined) {
+    throw new Error(`
+      The ReactOnRails value exists in the ${ctx} scope, it may not be safe to overwrite it.
+      
+      This could be caused by setting Webpack's optimization.runtimeChunk to "true" or "multiple," rather than "single." Check your Webpack configuration.
+      
+      Read more at https://github.com/shakacode/react_on_rails/issues/1558.
+    `);
+  }
+
+  ctx.ReactOnRails = value;
+
+  ctx.ReactOnRails.resetOptions();
+
+  ClientStartup.clientStartup(ctx);
+};

package.json

@@ -5,7 +5,7 @@
   "exports": {
     ".": {
       "node": "./node_package/lib/ReactOnRails.node.js",
-      "default": "./node_package/lib/ReactOnRails.js"
+      "default": "./node_package/lib/ReactOnRails.web.js"
     }
   },
   "directories": {