qiskit-documentation/scripts/js/lib/markdownReader.ts

45 lines
1.3 KiB
TypeScript

// This code is a Qiskit project.
//
// (C) Copyright IBM 2024.
//
// This code is licensed under the Apache License, Version 2.0. You may
// obtain a copy of this license in the LICENSE file in the root directory
// of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
//
// Any modifications or derivative works of this code must retain this
// copyright notice, and modified files need to carry a notice indicating
// that they have been altered from the originals.
import { readFile } from "fs/promises";
import path from "node:path";
export async function readMarkdown(
filePath: string,
options: { includeCodeCellSourceCode?: boolean } = {},
): Promise<string> {
const source = await readFile(filePath, { encoding: "utf8" });
return path.extname(filePath) === ".ipynb"
? markdownFromNotebook(source, options)
: source;
}
interface JupyterCell {
cell_type: string;
source: string[];
}
export function markdownFromNotebook(
rawContent: string,
options: { includeCodeCellSourceCode?: boolean },
): string {
const cells = JSON.parse(rawContent).cells as JupyterCell[];
return cells
.filter(
(cell) =>
cell.cell_type === "markdown" ||
(options.includeCodeCellSourceCode && cell.cell_type === "code"),
)
.map((cell) => cell.source.join(""))
.join("\n\n");
}