1 files changed, 25 insertions, 11 deletions

diff --git a/app/src/main/java/org/pacien/tincapp/activities/PromptActivity.kt b/app/src/main/java/org/pacien/tincapp/activities/PromptActivity.kt
index 6906ead..6310d63 100644
--- a/app/src/main/java/org/pacien/tincapp/activities/PromptActivity.kt
+++ b/app/src/main/java/org/pacien/tincapp/activities/PromptActivity.kt
@@ -2,9 +2,14 @@ package org.pacien.tincapp.activities
 
 import android.app.Activity
 import android.content.Intent
+import android.net.Uri
 import android.net.VpnService
 import android.os.Bundle
+import org.pacien.tincapp.BuildConfig
 import org.pacien.tincapp.context.App
+import org.pacien.tincapp.intent.action.ACTION_CONNECT
+import org.pacien.tincapp.intent.action.ACTION_DISCONNECT
+import org.pacien.tincapp.intent.action.TINC_SCHEME
 import org.pacien.tincapp.service.TincVpnService
 
 /**
@@ -15,35 +20,44 @@ class PromptActivity : Activity() {
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
 
-        when (intent.getSerializableExtra(INTENT_EXTRA_ACTION) as Action) {
-            Action.REQUEST_PERMISSION -> requestVpnPermission()
+        when (intent.action) {
+            ACTION_CONNECT -> connect()
+            ACTION_DISCONNECT -> disconnect()
         }
     }
 
     override fun onActivityResult(request: Int, result: Int, data: Intent?) {
-        if (result == Activity.RESULT_OK) TincVpnService.startVpn(intent.getStringExtra(INTENT_EXTRA_NET_NAME))
+        if (result == Activity.RESULT_OK) TincVpnService.startVpn(intent.data.schemeSpecificPart)
         finish()
     }
 
-    private fun requestVpnPermission() = VpnService.prepare(this).let {
+    private fun connect() = VpnService.prepare(this).let {
         if (it != null)
             startActivityForResult(it, 0)
         else
-            onActivityResult(0, Activity.RESULT_OK, Intent())
+            onActivityResult(0, Activity.RESULT_OK, null)
+    }
+
+    private fun disconnect() {
+        TincVpnService.stopVpn()
+        finish()
     }
 
     companion object {
-        private val INTENT_EXTRA_ACTION = "action"
-        private val INTENT_EXTRA_NET_NAME = "netName"
 
-        private enum class Action { REQUEST_PERMISSION }
+        fun connect(netName: String) {
+            App.getContext().startActivity(Intent(App.getContext(), PromptActivity::class.java)
+                    .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
+                    .setAction(ACTION_CONNECT)
+                    .setData(Uri.Builder().scheme(TINC_SCHEME).opaquePart(netName).build()))
+        }
 
-        fun requestVpnPermission(netName: String) {
+        fun disconnect() {
             App.getContext().startActivity(Intent(App.getContext(), PromptActivity::class.java)
                     .addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
-                    .putExtra(INTENT_EXTRA_ACTION, Action.REQUEST_PERMISSION)
-                    .putExtra(INTENT_EXTRA_NET_NAME, netName))
+                    .setAction(ACTION_DISCONNECT))
         }
+
     }
 
 }