From bf288fdaeb4dc64d4390edf9d35f38c3a38f624a Mon Sep 17 00:00:00 2001
From: Baptiste Augrain <daiyam@zokugun.org>
Date: Sun, 25 Nov 2018 14:55:29 +0100
Subject: [PATCH] fix pasting into fenced code block

---
 browser/components/CodeEditor.js | 84 +++++++++++++++++++++++++-------
 1 file changed, 66 insertions(+), 18 deletions(-)

diff --git a/browser/components/CodeEditor.js b/browser/components/CodeEditor.js
index 130cc86e..0193d5cb 100644
--- a/browser/components/CodeEditor.js
+++ b/browser/components/CodeEditor.js
@@ -562,14 +562,17 @@ export default class CodeEditor extends React.Component {
   }
 
   handlePaste (editor, e) {
+    const { storageKey, noteKey, fetchUrlTitle } = this.props
+
     const clipboardData = e.clipboardData
-    const { storageKey, noteKey } = this.props
     const dataTransferItem = clipboardData.items[0]
     const pastedTxt = clipboardData.getData('text')
+
     const isURL = str => {
       const matcher = /^(?:\w+:)?\/\/([^\s\.]+\.\S{2}|localhost[\:?\d]*)\S*$/
       return matcher.test(str)
     }
+
     const isInLinkTag = editor => {
       const startCursor = editor.getCursor('start')
       const prevChar = editor.getRange(
@@ -584,30 +587,69 @@ export default class CodeEditor extends React.Component {
       return prevChar === '](' && nextChar === ')'
     }
 
-    const pastedHtml = clipboardData.getData('text/html')
-    if (pastedHtml !== '') {
-      this.handlePasteHtml(e, editor, pastedHtml)
-    } else if (dataTransferItem.type.match('image')) {
-      attachmentManagement.handlePastImageEvent(
-        this,
-        storageKey,
-        noteKey,
-        dataTransferItem
-      )
-    } else if (
-      this.props.fetchUrlTitle &&
-      isURL(pastedTxt) &&
-      !isInLinkTag(editor)
-    ) {
-      this.handlePasteUrl(e, editor, pastedTxt)
+    const isInFencedCodeBlock = editor => {
+      const cursor = editor.getCursor()
+
+      let token = editor.getTokenAt(cursor)
+      if (token.state.fencedState) {
+        return true
+      }
+
+      let line = line = cursor.line - 1
+      while (line >= 0) {
+        token = editor.getTokenAt({
+          ch: 3,
+          line
+        })
+
+        if (token.start === token.end) {
+          --line
+        } else if (token.type === 'comment') {
+          if (line > 0) {
+            token = editor.getTokenAt({
+              ch: 3,
+              line: line - 1
+            })
+
+            return token.type !== 'comment'
+          } else {
+            return true
+          }
+          return true
+        } else {
+          return false
+        }
+      }
+
+      return false
     }
+
+    if (isInFencedCodeBlock(editor)) {
+      this.handlePasteText(e, editor, pastedTxt)
+    } else {
+      const pastedHtml = clipboardData.getData('text/html')
+      if (pastedHtml !== '') {
+        this.handlePasteHtml(e, editor, pastedHtml)
+      } else if (dataTransferItem.type.match('image')) {
+        attachmentManagement.handlePastImageEvent(
+          this,
+          storageKey,
+          noteKey,
+          dataTransferItem
+        )
+      } else if (fetchUrlTitle && isURL(pastedTxt) && !isInLinkTag(editor)) {
+        this.handlePasteUrl(e, editor, pastedTxt)
+      }
+    }
+
     if (attachmentManagement.isAttachmentLink(pastedTxt)) {
+      e.preventDefault()
+
       attachmentManagement
         .handleAttachmentLinkPaste(storageKey, noteKey, pastedTxt)
         .then(modifiedText => {
           this.editor.replaceSelection(modifiedText)
         })
-      e.preventDefault()
     }
   }
 
@@ -663,6 +705,12 @@ export default class CodeEditor extends React.Component {
     editor.replaceSelection(markdown)
   }
 
+  handlePasteText (e, editor, pastedTxt) {
+    e.preventDefault()
+
+    editor.replaceSelection(pastedTxt)
+  }
+
   mapNormalResponse (response, pastedTxt) {
     return this.decodeResponse(response).then(body => {
       return new Promise((resolve, reject) => {