mirror of
https://github.com/storybookjs/storybook.git
synced 2025-03-17 05:02:23 +08:00
114 lines
2.9 KiB
JavaScript
114 lines
2.9 KiB
JavaScript
// From: https://gist.github.com/ivanoats/8d01d9e934fdc17bae9090147f1e799b
|
|
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const sm = require('sitemap');
|
|
|
|
function pagesToSitemap(pages) {
|
|
return pages
|
|
.filter((p) => !!p.path)
|
|
.map((p) => ({
|
|
url: p.path,
|
|
changefreq: 'daily',
|
|
priority: 0.7,
|
|
}));
|
|
}
|
|
|
|
function getVersionData(distTag) {
|
|
const versionFile = `${__dirname}/src/versions/${distTag}.json`;
|
|
if (!fs.existsSync(versionFile)) {
|
|
return null;
|
|
}
|
|
const data = {
|
|
[distTag]: JSON.parse(fs.readFileSync(versionFile)),
|
|
};
|
|
return data;
|
|
}
|
|
|
|
function generateVersionsFile() {
|
|
const latest = getVersionData('latest');
|
|
const next = getVersionData('next');
|
|
const data = { ...latest, ...next };
|
|
fs.writeFileSync(`${__dirname}/public/versions.json`, JSON.stringify(data));
|
|
}
|
|
|
|
function generateSitemap(pages) {
|
|
const sitemap = sm.createSitemap({
|
|
hostname: 'https://storybook.js.org',
|
|
cacheTime: '60000',
|
|
urls: pagesToSitemap(pages),
|
|
});
|
|
fs.writeFileSync(`${__dirname}/public/sitemap.xml`, sitemap.toString());
|
|
}
|
|
|
|
module.exports = {
|
|
async onPostBuild({ graphql }) {
|
|
const result = await graphql(`
|
|
{
|
|
allSitePage {
|
|
edges {
|
|
node {
|
|
path
|
|
}
|
|
}
|
|
}
|
|
}
|
|
`);
|
|
generateVersionsFile();
|
|
generateSitemap(result.data.allSitePage.edges.map(({ node }) => node));
|
|
},
|
|
onCreateNode({ node, boundActionCreators, getNode }) {
|
|
const { createNodeField } = boundActionCreators;
|
|
let slug;
|
|
if (node.internal.type === 'MarkdownRemark') {
|
|
const fileNode = getNode(node.parent);
|
|
const parsedFilePath = path.parse(fileNode.relativePath);
|
|
if (parsedFilePath.name !== 'index' && parsedFilePath.dir !== '') {
|
|
slug = `/${parsedFilePath.dir}/${parsedFilePath.name}/`;
|
|
} else if (parsedFilePath.dir === '') {
|
|
slug = `/${parsedFilePath.name}/`;
|
|
} else {
|
|
slug = `/${parsedFilePath.dir}/`;
|
|
}
|
|
|
|
// Add slug as a field on the node.
|
|
createNodeField({ node, name: 'slug', value: slug });
|
|
}
|
|
},
|
|
async createPages({ graphql, boundActionCreators }) {
|
|
const { createPage } = boundActionCreators;
|
|
|
|
const template = path.resolve('src/templates/_docstemplate.jsx');
|
|
// Query for all markdown "nodes" and for the slug we previously created.
|
|
const result = await graphql(
|
|
`
|
|
{
|
|
allMarkdownRemark {
|
|
edges {
|
|
node {
|
|
fields {
|
|
slug
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
`
|
|
);
|
|
if (result.errors) {
|
|
throw result.errors;
|
|
}
|
|
|
|
// Create pages.
|
|
result.data.allMarkdownRemark.edges.forEach((edge) => {
|
|
createPage({
|
|
path: edge.node.fields.slug, // required
|
|
component: template,
|
|
context: {
|
|
slug: edge.node.fields.slug,
|
|
},
|
|
});
|
|
});
|
|
},
|
|
};
|