Merge pull request #856 from tangly1024/fix/file-write

Fix/file write

lib/robots.txt.js

@@ -3,17 +3,23 @@ import fs from 'fs'
 import BLOG from '@/blog.config'
 
 export async function generateRobotsTxt() {
-  fs.mkdirSync('./public', { recursive: true })
-  fs.writeFileSync('./public/robots.txt', `
-  # *
-  User-agent: *
-  Allow: /
-
-  # Host
-  Host: ${BLOG.LINK}
-
-  # Sitemaps
-  Sitemap: ${BLOG.LINK}/sitemap.xml
-
-  `)
+  const content = `
+    # *
+    User-agent: *
+    Allow: /
+  
+    # Host
+    Host: ${BLOG.LINK}
+  
+    # Sitemaps
+    Sitemap: ${BLOG.LINK}/sitemap.xml
+  
+    `
+  try {
+    fs.mkdirSync('./public', { recursive: true })
+    fs.writeFileSync('./public/robots.txt', content)
+  } catch (error) {
+    // 在vercel运行环境是只读的，这里会报错；
+    // 但在vercel编译阶段、或VPS等其他平台这行代码会成功执行
+  }
 }

lib/rss.js

@@ -5,6 +5,26 @@ import ReactDOMServer from 'react-dom/server'
 import { getPostBlocks } from './notion'
 import NotionPage from '@/components/NotionPage'
 
+/**
+     * 生成RSS内容
+     * @param {*} post
+     * @returns
+     */
+const createFeedContent = async post => {
+  // 加密的文章内容只返回摘要
+  if (post.password && post.password !== '') {
+    return post.summary
+  }
+  const blockMap = await getPostBlocks(post.id, 'rss-content')
+  if (blockMap) {
+    post.blockMap = blockMap
+    const content = ReactDOMServer.renderToString(<NotionPage post={post} />)
+    const regexExp =
+                /<div class="notion-collection-row"><div class="notion-collection-row-body"><div class="notion-collection-row-property"><div class="notion-collection-column-title"><svg.*?class="notion-collection-column-title-icon">.*?<\/svg><div class="notion-collection-column-title-body">.*?<\/div><\/div><div class="notion-collection-row-value">.*?<\/div><\/div><\/div><\/div>/g
+    return content.replace(regexExp, '')
+  }
+}
+
 export async function generateRss(posts) {
   const year = new Date().getFullYear()
   const feed = new Feed({
@@ -30,28 +50,14 @@ export async function generateRss(posts) {
     })
   }
 
-  fs.mkdirSync('./public/rss', { recursive: true })
-  fs.writeFileSync('./public/rss/feed.xml', feed.rss2())
-  fs.writeFileSync('./public/rss/atom.xml', feed.atom1())
-  fs.writeFileSync('./public/rss/feed.json', feed.json1())
-}
-
-/**
- * 生成RSS内容
- * @param {*} post
- * @returns
- */
-const createFeedContent = async post => {
-  // 加密的文章内容只返回摘要
-  if (post.password && post.password !== '') {
-    return post.summary
-  }
-  const blockMap = await getPostBlocks(post.id, 'rss-content')
-  if (blockMap) {
-    post.blockMap = blockMap
-    const content = ReactDOMServer.renderToString(<NotionPage post={post} />)
-    const regexExp =
-      /<div class="notion-collection-row"><div class="notion-collection-row-body"><div class="notion-collection-row-property"><div class="notion-collection-column-title"><svg.*?class="notion-collection-column-title-icon">.*?<\/svg><div class="notion-collection-column-title-body">.*?<\/div><\/div><div class="notion-collection-row-value">.*?<\/div><\/div><\/div><\/div>/g
-    return content.replace(regexExp, '')
+  try {
+    fs.mkdirSync('./public/rss', { recursive: true })
+    fs.writeFileSync('./public/rss/feed.xml', feed.rss2())
+    fs.writeFileSync('./public/rss/atom.xml', feed.atom1())
+    fs.writeFileSync('./public/rss/feed.json', feed.json1())
+  } catch (error) {
+    // 在vercel运行环境是只读的，这里会报错；
+    // 但在vercel编译阶段、或VPS等其他平台这行代码会成功执行
+    // RSS被高频词访问将大量消耗服务端资源，故作为静态文件
   }
 }

lib/sitemap.xml.js

@@ -3,8 +3,6 @@ import fs from 'fs'
 import BLOG from '@/blog.config'
 
 export async function generateSitemapXml({ allPages }) {
-  fs.mkdirSync('./public', { recursive: true })
-
   const urls = [{
     loc: `${BLOG.LINK}`,
     lastmod: new Date().toISOString().split('T')[0],
@@ -31,9 +29,13 @@ export async function generateSitemapXml({ allPages }) {
     })
   })
   const xml = createSitemapXml(urls)
-  fs.writeFileSync('./public/sitemap.xml', xml)
+  try {
+    fs.writeFileSync('sitemap.xml', xml)
+    fs.writeFileSync('./public/sitemap.xml', xml)
+  } catch (error) {
+    console.warn('无法写入文件', error)
+  }
 }
-
 function createSitemapXml(urls) {
   let urlsXml = ''
   urls.forEach(u => {
@@ -46,7 +48,7 @@ function createSitemapXml(urls) {
   })
 
   return `
-  <urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9"
+    <urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9"
     xmlns:news="http://www.google.com/schemas/sitemap-news/0.9"
     xmlns:xhtml="http://www.w3.org/1999/xhtml"
     xmlns:mobile="http://www.google.com/schemas/sitemap-mobile/1.0"

pages/index.js

@@ -5,7 +5,6 @@ import * as ThemeMap from '@/themes'
 import { useGlobal } from '@/lib/global'
 import { generateRss } from '@/lib/rss'
 import { generateRobotsTxt } from '@/lib/robots.txt'
-import { generateSitemapXml } from '@/lib/sitemap.xml'
 const Index = props => {
   const { theme } = useGlobal()
   const ThemeComponents = ThemeMap[theme]
@@ -46,8 +45,6 @@ export async function getStaticProps() {
 
   // 生成robotTxt
   generateRobotsTxt()
-  // 生成sitemap.xml
-  generateSitemapXml({ allPages: props.allPages })
   // 生成Feed订阅
   if (JSON.parse(BLOG.ENABLE_RSS)) {
     generateRss(props?.latestPosts || [])

pages/sitemap.xml.js

@@ -0,0 +1,60 @@
+// pages/sitemap.xml.js
+import { getServerSideSitemap } from 'next-sitemap'
+import { getGlobalNotionData } from '@/lib/notion/getNotionData'
+import BLOG from '@/blog.config'
+
+export const getServerSideProps = async (ctx) => {
+  const { allPages } = await getGlobalNotionData({ from: 'rss' })
+  const defaultFields = [
+    {
+      loc: `${BLOG.LINK}`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }, {
+      loc: `${BLOG.LINK}/archive`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }, {
+      loc: `${BLOG.LINK}/category`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }, {
+      loc: `${BLOG.LINK}/feed`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }, {
+      loc: `${BLOG.LINK}/search`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }, {
+      loc: `${BLOG.LINK}/tag`,
+      lastmod: new Date().toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }
+  ]
+  const postFields = allPages?.map(post => {
+    return {
+      loc: `${BLOG.LINK}/${post.slug}`,
+      lastmod: new Date(post?.date?.start_date || post?.createdTime).toISOString().split('T')[0],
+      changefreq: 'daily',
+      priority: '0.7'
+    }
+  })
+  const fields = defaultFields.concat(postFields)
+
+  // 缓存
+  ctx.res.setHeader(
+    'Cache-Control',
+    'public, max-age=3600, stale-while-revalidate=59'
+  )
+
+  return getServerSideSitemap(ctx, fields)
+}
+
+export default () => { }