From 707400b7bed213cda12f8cc7ef526b46ad5fdacb Mon Sep 17 00:00:00 2001
From: CzBiX <gliuwr@gmail.com>
Date: Tue, 31 Mar 2020 22:46:42 +0800
Subject: [PATCH] Register protocol handler

close #11
---
 src/App.vue                | 15 +++++++++++++++
 src/components/AddForm.vue |  2 +-
 src/protocolHandler.ts     | 29 +++++++++++++++++++++++++++++
 3 files changed, 45 insertions(+), 1 deletion(-)
 create mode 100644 src/protocolHandler.ts

diff --git a/src/App.vue b/src/App.vue
index 387fa32..a39e0eb 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -74,6 +74,8 @@
 <script lang="ts">
 import Vue from 'vue';
 import { mapGetters, mapState, mapMutations } from 'vuex';
+import { registerProtocolHandler, checkDownloadUrl } from './protocolHandler';
+
 import GlobalDialog from './components/GlobalDialog.vue';
 import GlobalSnackBar from './components/GlobalSnackBar.vue';
 
@@ -145,7 +147,20 @@ export default class App extends Vue {
     return this.$vuetify.breakpoint.xsOnly;
   }
 
+  initProtocolHandler() {
+    registerProtocolHandler();
+    const url = checkDownloadUrl();
+
+    if (url) {
+      this.setPasteUrl({
+        url,
+      });
+    }
+  }
+
   async created() {
+    this.initProtocolHandler();
+
     await this.getInitData();
     appWrapEl = (this.$refs.app as any).$el.querySelector('.v-application--wrap');
     appWrapEl.addEventListener('paste', this.onPaste);
diff --git a/src/components/AddForm.vue b/src/components/AddForm.vue
index a46f507..762e28f 100644
--- a/src/components/AddForm.vue
+++ b/src/components/AddForm.vue
@@ -317,7 +317,7 @@ export default class AddForm extends Vue {
     this.files = files;
   }
 
-  @Watch('pasteUrl')
+  @Watch('pasteUrl', {immediate: true})
   onPasteUrl(v: string) {
     if (!v) {
       return;
diff --git a/src/protocolHandler.ts b/src/protocolHandler.ts
new file mode 100644
index 0000000..39e2d63
--- /dev/null
+++ b/src/protocolHandler.ts
@@ -0,0 +1,29 @@
+function registerProtocolHandler() {
+  if (!('registerProtocolHandler' in navigator)) {
+    return;
+  }
+
+  try {
+    navigator.registerProtocolHandler('magnet', location.origin + '#url=%s', document.title);
+  } catch (e) {
+    console.log('Register protocol handler failed.', e);
+  }
+}
+
+function checkDownloadUrl() {
+  if (!location.hash) {
+    return null
+  }
+
+  const params = new URLSearchParams(location.hash.substring(1));
+  const url = params.get('url')
+  if (!url) {
+    return null;
+  }
+
+  params.delete('url');
+  location.hash = '#' + params.toString()
+  return url
+}
+
+export { registerProtocolHandler, checkDownloadUrl };
\ No newline at end of file