Fixes split-view issue

Author: akaalias

main.js

@@ -5,10 +5,10 @@
   "main": "ExtractHighlightsPlugin.js",
   "type": "module",
   "scripts": {
-      "dev": "rollup --config rollup.config.js -w",
-      "build": "rollup --config rollup.config.js",
-      "test": "cross-env TS_NODE_COMPILER_OPTIONS='{ \"module\": \"commonjs\" }' mocha -r ts-node/register -r ignore-styles -r jsdom-global/register test/**/*.ts",
-      "test:watch": "cross-env TS_NODE_COMPILER_OPTIONS='{ \"module\": \"commonjs\" }' mocha -r ts-node/register -r ignore-styles -r jsdom-global/register --watch --watch-files src, test/**/*.ts"
+    "dev": "rollup --config rollup.config.js -w",
+    "build": "rollup --config rollup.config.js",
+    "test": "cross-env TS_NODE_COMPILER_OPTIONS='{ \"module\": \"commonjs\" }' mocha -r ts-node/register -r ignore-styles -r jsdom-global/register test/**/*.ts",
+    "test:watch": "cross-env TS_NODE_COMPILER_OPTIONS='{ \"module\": \"commonjs\" }' mocha -r ts-node/register -r ignore-styles -r jsdom-global/register --watch --watch-files src, test/**/*.ts"
   },
   "keywords": [],
   "author": "",
@@ -18,21 +18,21 @@
     "@rollup/plugin-node-resolve": "^9.0.0",
     "@rollup/plugin-typescript": "^6.0.0",
     "@types/chai": "^4.2.14",
-    "@types/mocha": "^8.0.4",
-    "@types/node": "^14.14.9",
+    "@types/mocha": "^8.2.0",
+    "@types/node": "^14.14.14",
     "chai": "^4.2.0",
     "cross-env": "^7.0.2",
     "ignore-styles": "^5.0.1",
     "jsdom": "^16.4.0",
     "jsdom-global": "^3.0.2",
     "mocha": "^8.2.1",
     "obsidian": "https://github.com/obsidianmd/obsidian-api/tarball/master",
-    "rollup": "^2.32.1",
-    "ts-node": "^9.1.0",
+    "rollup": "^2.35.1",
+    "ts-node": "^9.1.1",
     "tslib": "^1.14.1",
-    "typescript": "^4.0.3"
+    "typescript": "^4.1.3"
   },
   "dependencies": {
-    "electron": "^10.1.5"
+    "electron": "^10.2.0"
   }
 }

package.json

@@ -1,4 +1,4 @@
-import {Plugin, Notice, addIcon, View, MarkdownView} from "obsidian"
+import {Plugin, Notice, addIcon, View, MarkdownView, Workspace} from "obsidian"
 import ExtractHighlightsPluginSettings from "./ExtractHighlightsPluginSettings"
 import ExtractHighlightsPluginSettingsTab from "./ExtractHighlightsPluginSettingsTab"
 import {Position} from "codemirror";
@@ -24,16 +24,6 @@ export default class ExtractHighlightsPlugin extends Plugin {
 			this.extractHighlights();
 		});
 
-		this.registerEvent(
-			this.app.on('codemirror', (cm: CodeMirror.Editor | MarkdownView) => {
-				if ('sourceMode' in cm) {
-					this.editor = cm.sourceMode.cmEditor;
-				} else {
-					this.editor = cm;
-				}
-			}),
-		);
-
 		this.addCommand({
 			id: "shortcut-extract-highlights",
 			name: "Shortcut for extracting highlights",
@@ -173,6 +163,10 @@ export default class ExtractHighlightsPlugin extends Plugin {
 	}
 
 	createHighlight() {
+		const mdView = this.app.workspace.activeLeaf.view as MarkdownView;
+		const doc = mdView.sourceMode.cmEditor;
+		this.editor = doc;
+
 		const cursorPosition = this.editor.getCursor();
 		let lineText = this.editor.getLine(cursorPosition.line);