fix: improve regex matching (#91)

probot · May 18, 2023 · a0d7ab2 · a0d7ab2
1 parent f59878a
commit a0d7ab2
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/index.js b/index.js
@@ -1,4 +1,4 @@
-const regex = /\n\n<!-- probot = (.*) -->/
+const regex = /(\n\n|\r\n)<!-- probot = (.*) -->/
 
 module.exports = (context, issue = null) => {
   const github = context.octokit || context.github
@@ -17,7 +17,7 @@ module.exports = (context, issue = null) => {
       const match = body.match(regex)
 
       if (match) {
-        const data = JSON.parse(match[1])[prefix]
+        const data = JSON.parse(match[2])[prefix]
         return key ? data && data[key] : data
       }
     },
@@ -28,10 +28,13 @@ module.exports = (context, issue = null) => {
 
       if (!body) body = (await github.issues.get(issue)).data.body || ''
 
-      body = body.replace(regex, (_, json) => {
-        data = JSON.parse(json)
-        return ''
-      })
+      const match = body.match(regex)
+
+      if (match) {
+        data = JSON.parse(match[2])
+      }
+
+      body = body.replace(regex, '')
 
       if (!data[prefix]) data[prefix] = {}