Re-add support for snapshots

jpmorganchase · Nov 7, 2024 · cecb5fa · cecb5fa
1 parent 0540260
commit cecb5fa
Show file tree

Hide file tree

Showing 6 changed files with 169 additions and 228 deletions.
diff --git a/packages/loaders/src/index.ts b/packages/loaders/src/index.ts
@@ -1,54 +1,21 @@
 import path from 'path';
-import matter from 'gray-matter';
-import type { SafeParseError } from 'zod';
-import { activeEnvSchema } from '@jpmorganchase/mosaic-schemas';
 import type { SharedConfig } from '@jpmorganchase/mosaic-store';
+import type { MosaicMode } from '@jpmorganchase/mosaic-types';
 
 import type { LoaderPage } from './types/index.js';
+import { loadActiveContent, loadActiveMosaicData } from './loadActiveContent';
+import { loadSnapshotFileContent, loadSnapshotFileMosaicData } from './loadSnapshotFileContent';
 
 export * from './types/index.js';
 
-const normalizePageUrl = (url: string): string => (/\/index$/.test(url) ? `${url}.mdx` : url);
-
-type ActiveModeUrlEnv = {
-  MOSAIC_ACTIVE_MODE_URL: string;
-};
-
-export class LoadPageError extends Error {
-  statusCode: number;
-  constructor({ message, statusCode }: { message: string; statusCode: number }) {
-    super(message);
-    this.statusCode = statusCode;
-  }
-}
-
-const getFSRootUrl = (): string => {
-  const env = activeEnvSchema.safeParse(process.env);
-  if (!env.success) {
-    const { error } = env as SafeParseError<ActiveModeUrlEnv>;
-    error.issues.forEach(issue => {
-      console.error(
-        `Missing process.env.${issue.path.join()} environment variable required to load pages`
-      );
-    });
-    throw new LoadPageError({
-      message: `Environment variables missing to load pages`,
-      statusCode: 500
-    });
-  }
-  return env.data.MOSAIC_ACTIVE_MODE_URL;
-};
-
 export const loadMosaicData = async <T>(url: string): Promise<T> => {
-  const fsRootUrl = getFSRootUrl();
-  const dataUrl = new URL(url, fsRootUrl);
-  const response = await fetch(dataUrl);
+  const mode: MosaicMode = (process.env.MOSAIC_MODE || 'active') as MosaicMode;
 
-  if (!response.ok) {
-    // This will activate the closest `error.js` Error Boundary
-    throw new Error(`Failed to fetch mosaic data @ ${dataUrl}`);
+  if (mode === 'snapshot-file') {
+    return loadSnapshotFileMosaicData(url);
   }
-  return response.json();
+
+  return loadActiveMosaicData(url);
 };
 
 export const loadSharedConfig = async (route: string): Promise<SharedConfig | undefined> => {
@@ -58,20 +25,11 @@ export const loadSharedConfig = async (route: string): Promise<SharedConfig | un
 };
 
 export const loadPage = async (route: string): Promise<LoaderPage> => {
-  const fsRootUrl = getFSRootUrl();
-  const pageUrl = normalizePageUrl(`${fsRootUrl}${route}`);
-  const response = await fetch(pageUrl);
-  if (response.status === 302) {
-    const { redirect } = await response.json();
-    return loadPage(redirect);
-  }
-  if (response.ok) {
-    const source = await response.text();
-    const { content, data } = matter(source);
-    return { source: content, data };
+  const mode: MosaicMode = (process.env.MOSAIC_MODE || 'active') as MosaicMode;
+
+  if (mode === 'snapshot-file') {
+    return loadSnapshotFileContent(route);
   }
-  throw new LoadPageError({
-    message: `Could not load page : ${pageUrl} ${response.status}/${response.statusText}`,
-    statusCode: 404
-  });
+
+  return loadActiveContent(route);
 };
diff --git a/packages/loaders/src/loadActiveContent.ts b/packages/loaders/src/loadActiveContent.ts
@@ -0,0 +1,69 @@
+import matter from 'gray-matter';
+import type { SafeParseError } from 'zod';
+import { activeEnvSchema } from '@jpmorganchase/mosaic-schemas';
+
+import type { LoaderPage } from './types/index.js';
+
+export * from './types/index.js';
+
+const normalizePageUrl = (url: string): string => (/\/index$/.test(url) ? `${url}.mdx` : url);
+
+type ActiveModeUrlEnv = {
+  MOSAIC_ACTIVE_MODE_URL: string;
+};
+
+export class LoadPageError extends Error {
+  statusCode: number;
+  constructor({ message, statusCode }: { message: string; statusCode: number }) {
+    super(message);
+    this.statusCode = statusCode;
+  }
+}
+
+const getFSRootUrl = (): string => {
+  const env = activeEnvSchema.safeParse(process.env);
+  if (!env.success) {
+    const { error } = env as SafeParseError<ActiveModeUrlEnv>;
+    error.issues.forEach(issue => {
+      console.error(
+        `Missing process.env.${issue.path.join()} environment variable required to load pages`
+      );
+    });
+    throw new LoadPageError({
+      message: `Environment variables missing to load pages`,
+      statusCode: 500
+    });
+  }
+  return env.data.MOSAIC_ACTIVE_MODE_URL;
+};
+
+export const loadActiveMosaicData = async <T>(url: string): Promise<T> => {
+  const fsRootUrl = getFSRootUrl();
+  const dataUrl = new URL(url, fsRootUrl);
+  const response = await fetch(dataUrl);
+
+  if (!response.ok) {
+    // This will activate the closest `error.js` Error Boundary
+    throw new Error(`Failed to fetch mosaic data @ ${dataUrl}`);
+  }
+  return response.json();
+};
+
+export const loadActiveContent = async (route: string): Promise<LoaderPage> => {
+  const fsRootUrl = getFSRootUrl();
+  const pageUrl = normalizePageUrl(`${fsRootUrl}${route}`);
+  const response = await fetch(pageUrl);
+  if (response.status === 302) {
+    const { redirect } = await response.json();
+    return loadActiveContent(redirect);
+  }
+  if (response.ok) {
+    const source = await response.text();
+    const { content, data } = matter(source);
+    return { source: content, data };
+  }
+  throw new LoadPageError({
+    message: `Could not load page : ${pageUrl} ${response.status}/${response.statusText}`,
+    statusCode: 404
+  });
+};
diff --git a/packages/loaders/src/loadSnapshotFileContent.ts b/packages/loaders/src/loadSnapshotFileContent.ts
@@ -0,0 +1,83 @@
+import { snapshotFileEnvSchema } from '@jpmorganchase/mosaic-schemas';
+
+import type { LoaderPage } from './types/index.js';
+import path from 'path';
+import fs from 'node:fs/promises';
+import matter from 'gray-matter';
+
+export * from './types/index.js';
+
+const normalizePageUrl = (url: string): string => (/\/index$/.test(url) ? `${url}.mdx` : url);
+
+export class LoadPageError extends Error {
+  statusCode: number;
+  constructor({ message, statusCode }: { message: string; statusCode: number }) {
+    super(message);
+    this.statusCode = statusCode;
+  }
+}
+
+const getFSRootUrl = (): string => {
+  const env = snapshotFileEnvSchema.safeParse(process.env);
+  if (!env.success) {
+    env.error.issues.forEach(issue => {
+      console.error(
+        `Missing process.env.${issue.path.join()} environment variable required to load pages`
+      );
+    });
+    throw new LoadPageError({
+      message: `Environment variables missing to load pages`,
+      statusCode: 500
+    });
+  }
+  return env.data.MOSAIC_SNAPSHOT_DIR;
+};
+
+export const loadSnapshotFileMosaicData = async <T>(url: string): Promise<T> => {
+  const matches = url.match(/(.*)[!/]/);
+  const urlPath = matches?.length ? matches[1] : '';
+
+  const fsRootUrl = getFSRootUrl();
+  const filePath = path.join(process.cwd(), fsRootUrl, urlPath);
+
+  let fileExists = false;
+  try {
+    await fs.stat(filePath);
+    fileExists = true;
+  } catch {}
+  if (fileExists) {
+    let localPath = filePath;
+    if ((await fs.stat(filePath)).isDirectory()) {
+      localPath = path.posix.join(localPath, 'index');
+    }
+    const realPath = await fs.realpath(localPath);
+    console.log(localPath, realPath);
+    const source = await fs.readFile(realPath, 'utf-8');
+    return JSON.parse(source);
+  }
+  throw new Error(`Failed to fetch mosaic data @ ${url}`);
+};
+
+export const loadSnapshotFileContent = async (route: string): Promise<LoaderPage> => {
+  const fsRootUrl = getFSRootUrl();
+  const pageUrl = normalizePageUrl(route);
+  const filePath = path.posix.join(process.cwd(), fsRootUrl, pageUrl);
+  try {
+    let localPath = filePath;
+    if ((await fs.stat(filePath)).isDirectory()) {
+      localPath = path.posix.join(localPath, 'index');
+    }
+    const realPath = await fs.realpath(localPath);
+    const source = await fs.readFile(realPath, 'utf-8');
+    const { content, data } = matter(source);
+    return { source: content, data };
+  } catch (error) {
+    if (error instanceof Error) {
+      console.error(error.message);
+    }
+    throw new LoadPageError({
+      message: `Could not read local file '${filePath}' for '${route}'`,
+      statusCode: 404
+    });
+  }
+};
diff --git a/packages/site-components/package.json b/packages/site-components/package.json
@@ -44,7 +44,6 @@
     "@jpmorganchase/mosaic-content-editor-plugin": "^0.1.0-beta.89",
     "@jpmorganchase/mosaic-labs-components": "^0.1.0-beta.89",
     "@jpmorganchase/mosaic-open-api-component": "^0.1.0-beta.89",
-    "@jpmorganchase/mosaic-site-middleware": "^0.1.0-beta.89",
     "@jpmorganchase/mosaic-store": "^0.1.0-beta.89",
     "@jpmorganchase/mosaic-theme": "^0.1.0-beta.89",
     "@salt-ds/core": "^1.33.0",

diff --git a/packages/site-preset-styles/package.json b/packages/site-preset-styles/package.json
@@ -39,6 +39,7 @@
     "@jpmorganchase/mosaic-site-components": "0.1.0-beta.89",
     "@jpmorganchase/mosaic-layouts": "0.1.0-beta.89",
     "@jpmorganchase/mosaic-theme": "0.1.0-beta.89",
+    "@jpmorganchase/mosaic-mdx-components": "0.1.0-beta.89",
     "prismjs": "^1.23.0"
   },
   "peerDependencies": {}