50 lines
1.3 KiB
TypeScript
50 lines
1.3 KiB
TypeScript
import type { VFileData as Data, VFile } from 'vfile';
|
|
import type { MarkdownAstroData } from './types.js';
|
|
|
|
function isValidAstroData(obj: unknown): obj is MarkdownAstroData {
|
|
if (typeof obj === 'object' && obj !== null && obj.hasOwnProperty('frontmatter')) {
|
|
const { frontmatter } = obj as any;
|
|
try {
|
|
// ensure frontmatter is JSON-serializable
|
|
JSON.stringify(frontmatter);
|
|
} catch {
|
|
return false;
|
|
}
|
|
return typeof frontmatter === 'object' && frontmatter !== null;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
export class InvalidAstroDataError extends TypeError {}
|
|
|
|
export function safelyGetAstroData(vfileData: Data): MarkdownAstroData | InvalidAstroDataError {
|
|
const { astro } = vfileData;
|
|
|
|
if (!astro || !isValidAstroData(astro)) {
|
|
return new InvalidAstroDataError();
|
|
}
|
|
|
|
return astro;
|
|
}
|
|
|
|
export function setVfileFrontmatter(vfile: VFile, frontmatter: Record<string, any>) {
|
|
vfile.data ??= {};
|
|
vfile.data.astro ??= {};
|
|
(vfile.data.astro as any).frontmatter = frontmatter;
|
|
}
|
|
|
|
/**
|
|
* @deprecated Use `setVfileFrontmatter` instead
|
|
*/
|
|
export function toRemarkInitializeAstroData({
|
|
userFrontmatter,
|
|
}: {
|
|
userFrontmatter: Record<string, any>;
|
|
}) {
|
|
return () =>
|
|
function (tree: any, vfile: VFile) {
|
|
if (!vfile.data.astro) {
|
|
vfile.data.astro = { frontmatter: userFrontmatter };
|
|
}
|
|
};
|
|
}
|