From 70620ba6864a817320d0c65851a88bc632c92425 Mon Sep 17 00:00:00 2001
From: Ryan Rivest <ryan.rivest@gmail.com>
Date: Mon, 23 Apr 2018 11:14:14 -0400
Subject: [PATCH] feat: support for TOML front matter (#141) (#164)

---
 lib/prepare.js                | 13 ++++++-------
 lib/util/index.js             | 19 +++++++++++++------
 lib/webpack/markdownLoader.js |  9 ++++-----
 package.json                  |  4 ++--
 4 files changed, 25 insertions(+), 20 deletions(-)

diff --git a/lib/prepare.js b/lib/prepare.js
index 1f86c2e2f1..c2c7a66d1d 100644
--- a/lib/prepare.js
+++ b/lib/prepare.js
@@ -3,10 +3,9 @@ const fs = require('fs-extra')
 const globby = require('globby')
 const yamlParser = require('js-yaml')
 const tomlParser = require('toml')
-const yaml = require('yaml-front-matter')
 const createMarkdown = require('./markdown')
 const tempPath = path.resolve(__dirname, 'app/.temp')
-const { inferTitle, extractHeaders } = require('./util')
+const { inferTitle, extractHeaders, parseFrontmatter } = require('./util')
 
 fs.ensureDirSync(tempPath)
 
@@ -174,23 +173,23 @@ async function resolveOptions (sourceDir) {
 
     // extract yaml frontmatter
     const content = await fs.readFile(path.resolve(sourceDir, file), 'utf-8')
-    const frontmatter = yaml.loadFront(content)
+    const frontmatter = parseFrontmatter(content)
     // infer title
     const title = inferTitle(frontmatter)
     if (title) {
       data.title = title
     }
     const headers = extractHeaders(
-      frontmatter.__content,
+      frontmatter.content,
       ['h2', 'h3'],
       options.markdown
     )
     if (headers.length) {
       data.headers = headers
     }
-    delete frontmatter.__content
-    if (Object.keys(frontmatter).length) {
-      data.frontmatter = frontmatter
+    delete frontmatter.content
+    if (Object.keys(frontmatter.data).length) {
+      data.frontmatter = frontmatter.data
     }
     return data
   }))
diff --git a/lib/util/index.js b/lib/util/index.js
index bfa48e24cf..30f6cca9d9 100644
--- a/lib/util/index.js
+++ b/lib/util/index.js
@@ -26,21 +26,28 @@ exports.applyUserWebpackConfig = function (userConfig, config, isServer) {
 }
 
 exports.inferTitle = function (frontmatter) {
-  if (frontmatter.home) {
+  if (frontmatter.data.home) {
     return 'Home'
   }
-  if (frontmatter.title) {
-    return frontmatter.title
+  if (frontmatter.data.title) {
+    return frontmatter.data.title
   }
-  const match = frontmatter.__content.trim().match(/^#+\s+(.*)/)
+  const match = frontmatter.content.trim().match(/^#+\s+(.*)/)
   if (match) {
     return match[1]
   }
 }
 
 exports.parseFrontmatter = content => {
-  const yaml = require('yaml-front-matter')
-  return yaml.loadFront(content)
+  const matter = require('gray-matter')
+  const toml = require('toml')
+
+  return matter(content, {
+    engines: {
+      toml: toml.parse.bind(toml),
+      excerpt: false
+    }
+  })
 }
 
 const LRU = require('lru-cache')
diff --git a/lib/webpack/markdownLoader.js b/lib/webpack/markdownLoader.js
index dc296948c9..ab2c699050 100644
--- a/lib/webpack/markdownLoader.js
+++ b/lib/webpack/markdownLoader.js
@@ -3,8 +3,7 @@ const path = require('path')
 const hash = require('hash-sum')
 const { EventEmitter } = require('events')
 const { getOptions } = require('loader-utils')
-const yaml = require('yaml-front-matter')
-const { inferTitle, extractHeaders } = require('../util')
+const { inferTitle, extractHeaders, parseFrontmatter } = require('../util')
 const LRU = require('lru-cache')
 
 const cache = LRU({ max: 1000 })
@@ -25,13 +24,13 @@ module.exports = function (src) {
     return cached
   }
 
-  const frontmatter = yaml.loadFront(src)
-  const content = frontmatter.__content
+  const frontmatter = parseFrontmatter(src)
+  const content = frontmatter.content
 
   if (!isProd && !isServer) {
     const inferredTitle = inferTitle(frontmatter)
     const headers = extractHeaders(content, ['h2', 'h3'], markdown)
-    delete frontmatter.__content
+    delete frontmatter.content
 
     // diff frontmatter and title, since they are not going to be part of the
     // returned component, changes in frontmatter do not trigger proper updates
diff --git a/package.json b/package.json
index b0f72b72ea..b0bb3303c3 100644
--- a/package.json
+++ b/package.json
@@ -56,6 +56,7 @@
     "file-loader": "^1.1.11",
     "fs-extra": "^5.0.0",
     "globby": "^8.0.1",
+    "gray-matter": "^4.0.1",
     "js-yaml": "^3.11.0",
     "koa-connect": "^2.0.1",
     "koa-mount": "^3.0.0",
@@ -91,8 +92,7 @@
     "webpack-merge": "^4.1.2",
     "webpack-serve": "^0.3.1",
     "webpackbar": "^2.6.1",
-    "workbox-build": "^3.1.0",
-    "yaml-front-matter": "^4.0.0"
+    "workbox-build": "^3.1.0"
   },
   "devDependencies": {
     "conventional-changelog": "^1.1.23",