tunnel: add ConfigDetail and PeerDetail

5 files changed, 231 insertions, 7 deletions

diff --git a/ui/src/main/java/com/wireguard/android/fragment/TunnelDetailFragment.kt b/ui/src/main/java/com/wireguard/android/fragment/TunnelDetailFragment.kt
index 81d8e8c6..fce14d20 100644
--- a/ui/src/main/java/com/wireguard/android/fragment/TunnelDetailFragment.kt
+++ b/ui/src/main/java/com/wireguard/android/fragment/TunnelDetailFragment.kt
@@ -5,6 +5,7 @@
 package com.wireguard.android.fragment
 
 import android.os.Bundle
+import android.util.Log;
 import android.view.LayoutInflater
 import android.view.Menu
 import android.view.MenuInflater
@@ -21,6 +22,7 @@ import com.wireguard.android.databinding.TunnelDetailFragmentBinding
 import com.wireguard.android.databinding.TunnelDetailPeerBinding
 import com.wireguard.android.model.ObservableTunnel
 import com.wireguard.android.util.QuantityFormatter
+import com.wireguard.android.viewmodel.ConfigDetail
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.launch
 
@@ -79,7 +81,9 @@ class TunnelDetailFragment : BaseFragment(), MenuProvider {
         } else {
             lifecycleScope.launch {
                 try {
-                    binding.config = newTunnel.getConfigAsync()
+                    var config = newTunnel.getConfigDetailAsync()
+                    binding.config = config
+                    Log.i(TAG, "onSelectedTunnelChanged " + config + ", " + config.config)
                 } catch (_: Throwable) {
                     binding.config = null
                 }
@@ -147,4 +151,8 @@ class TunnelDetailFragment : BaseFragment(), MenuProvider {
             }
         }
     }
+
+    companion object {
+        private const val TAG = "WireGuard/TunnelDetailFragment"
+    }
 }
diff --git a/ui/src/main/java/com/wireguard/android/model/ObservableTunnel.kt b/ui/src/main/java/com/wireguard/android/model/ObservableTunnel.kt
index aa237aee..03280b2f 100644
--- a/ui/src/main/java/com/wireguard/android/model/ObservableTunnel.kt
+++ b/ui/src/main/java/com/wireguard/android/model/ObservableTunnel.kt
@@ -7,12 +7,19 @@ package com.wireguard.android.model
 import android.util.Log
 import androidx.databinding.BaseObservable
 import androidx.databinding.Bindable
+import com.wireguard.android.Application
 import com.wireguard.android.BR
 import com.wireguard.android.backend.Statistics
 import com.wireguard.android.backend.Tunnel
 import com.wireguard.android.databinding.Keyed
 import com.wireguard.android.util.applicationScope
+import com.wireguard.android.viewmodel.ConfigDetail
+import com.wireguard.android.viewmodel.PeerDetail
 import com.wireguard.config.Config
+import com.wireguard.config.InetEndpoint
+import com.wireguard.config.InetNetwork
+import com.wireguard.crypto.Key
+import java.util.Optional
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.withContext
@@ -55,7 +62,17 @@ class ObservableTunnel internal constructor(
     }
 
     fun onStateChanged(state: Tunnel.State): Tunnel.State {
-        if (state != Tunnel.State.UP) onStatisticsChanged(null)
+        if (state != Tunnel.State.UP) {
+            onStatisticsChanged(null)
+            Application.getCoroutineScope().launch {
+                onPeersReset()
+            }
+        } else {
+            configDetail?.peers?.forEach {
+                var endpoint: InetEndpoint? = it.peer?.endpoint?.orElse(null)
+                it.endpoint = Optional.ofNullable(endpoint?.getResolved()?.orElse(null));
+            }
+        }
         this.state = state
         notifyPropertyChanged(BR.state)
         return state
@@ -68,6 +85,7 @@ class ObservableTunnel internal constructor(
             this@ObservableTunnel.state
     }
 
+    private var configDetail: ConfigDetail? = if (config != null) ConfigDetail(config) else null
 
     @get:Bindable
     var config = config
@@ -86,7 +104,11 @@ class ObservableTunnel internal constructor(
         private set
 
     suspend fun getConfigAsync(): Config = withContext(Dispatchers.Main.immediate) {
-        config ?: manager.getTunnelConfig(this@ObservableTunnel)
+        config ?: manager.getTunnelConfig(this@ObservableTunnel).config!!
+    }
+
+    suspend fun getConfigDetailAsync(): ConfigDetail = withContext(Dispatchers.Main.immediate) {
+        configDetail ?: manager.getTunnelConfig(this@ObservableTunnel)
     }
 
     suspend fun setConfigAsync(config: Config): Config = withContext(Dispatchers.Main.immediate) {
@@ -98,10 +120,11 @@ class ObservableTunnel internal constructor(
         }
     }
 
-    fun onConfigChanged(config: Config?): Config? {
+    fun onConfigChanged(config: Config?): ConfigDetail? {
+        this.configDetail = ConfigDetail(config)
         this.config = config
         notifyPropertyChanged(BR.config)
-        return config
+        return configDetail
     }
 
 
@@ -137,6 +160,91 @@ class ObservableTunnel internal constructor(
     }
 
 
+    // Remove dynamic peers, and reset static peers
+    fun onPeersReset() {
+        Log.i(TAG, "ObservableTunnel onPeersReset")
+        var toRemove: MutableList<PeerDetail> = ArrayList()
+
+        configDetail?.peers?.forEach {
+            if (it.peer == null) {
+                toRemove.add(it)
+            } else {
+                it.endpoint = Optional.empty()
+            }
+        }
+
+        toRemove.forEach {
+            Log.i(TAG, "ObservableTunnel remove " + it)
+            configDetail?.peers?.remove(it)
+        }
+    }
+
+    override fun onEndpointChange(publicKey: Key, newEndpoint: InetEndpoint?) {
+        Application.getCoroutineScope().launch {
+            onEndpointChanged(publicKey, newEndpoint)
+        }
+    }
+
+    private fun onEndpointChanged(publicKey: Key, newEndpoint: InetEndpoint?) {
+
+        Log.i(TAG, "ObservableTunnel onEndpointChange " + newEndpoint)
+        var peer: PeerDetail? = null
+
+        configDetail?.peers?.forEach {
+            if (it.publicKey.equals(publicKey) == true) {
+                Log.i(TAG, "ObservableTunnel peer " + it + ", " + it.peer)
+                peer = it;
+            }
+        }
+
+        if (peer == null) {
+            Log.i(TAG, "ObservableTunnel create peer " + publicKey)
+            peer = PeerDetail(publicKey)
+            configDetail?.peers?.add(peer)
+        }
+
+        var peer2: PeerDetail = peer!!
+
+        if (newEndpoint != null) {
+            peer2.endpoint = newEndpoint.getResolved()
+        } else {
+            var peer3 = peer2.peer
+            peer2.endpoint = if (peer3 != null) peer3.endpoint else Optional.empty()
+        }
+    }
+
+    fun lookupPeer(publicKey: Key): PeerDetail {
+        configDetail?.peers?.forEach {
+            if (it.publicKey.equals(publicKey) == true) {
+                Log.i(TAG, "ObservableTunnel peer " + it + ", " + it.peer)
+                return it
+            }
+        }
+
+        Log.i(TAG, "ObservableTunnel create peer " + publicKey)
+        var peer: PeerDetail = PeerDetail(publicKey)
+        configDetail?.peers?.add(peer)
+
+        return peer
+    }
+
+    override fun onAllowedIpsChange(publicKey: Key, addNetworks: List<InetNetwork>?, removeNetworks: List<InetNetwork>?) {
+        Application.getCoroutineScope().launch {
+            onAllowedIpsChanged(publicKey, addNetworks, removeNetworks)
+        }
+    }
+
+    private fun onAllowedIpsChanged(publicKey: Key, addNetworks: List<InetNetwork>?, removeNetworks: List<InetNetwork>?) {
+        var peer: PeerDetail = lookupPeer(publicKey)
+
+        removeNetworks?.let() {
+            peer.allowedIps.removeAll(removeNetworks)
+        }
+        addNetworks?.let() {
+            peer.allowedIps.addAll(addNetworks)
+        }
+    }
+
     suspend fun deleteAsync() = manager.delete(this)
 
 
diff --git a/ui/src/main/java/com/wireguard/android/model/TunnelManager.kt b/ui/src/main/java/com/wireguard/android/model/TunnelManager.kt
index ba873fa6..960adcaa 100644
--- a/ui/src/main/java/com/wireguard/android/model/TunnelManager.kt
+++ b/ui/src/main/java/com/wireguard/android/model/TunnelManager.kt
@@ -24,6 +24,7 @@ import com.wireguard.android.databinding.ObservableSortedKeyedArrayList
 import com.wireguard.android.util.ErrorMessages
 import com.wireguard.android.util.UserKnobs
 import com.wireguard.android.util.applicationScope
+import com.wireguard.android.viewmodel.ConfigDetail
 import com.wireguard.config.Config
 import kotlinx.coroutines.CompletableDeferred
 import kotlinx.coroutines.Dispatchers
@@ -94,7 +95,7 @@ class TunnelManager(private val configStore: ConfigStore) : BaseObservable() {
             applicationScope.launch { UserKnobs.setLastUsedTunnel(value?.name) }
         }
 
-    suspend fun getTunnelConfig(tunnel: ObservableTunnel): Config = withContext(Dispatchers.Main.immediate) {
+    suspend fun getTunnelConfig(tunnel: ObservableTunnel): ConfigDetail = withContext(Dispatchers.Main.immediate) {
         tunnel.onConfigChanged(withContext(Dispatchers.IO) { configStore.load(tunnel.name) })!!
     }
 
@@ -156,7 +157,7 @@ class TunnelManager(private val configStore: ConfigStore) : BaseObservable() {
         tunnel.onConfigChanged(withContext(Dispatchers.IO) {
             getBackend().setState(tunnel, tunnel.state, config)
             configStore.save(tunnel.name, config)
-        })!!
+        })!!.config!!
     }
 
     suspend fun setTunnelName(tunnel: ObservableTunnel, name: String): String = withContext(Dispatchers.Main.immediate) {
diff --git a/ui/src/main/java/com/wireguard/android/viewmodel/ConfigDetail.kt b/ui/src/main/java/com/wireguard/android/viewmodel/ConfigDetail.kt
new file mode 100644
index 00000000..af95a86a
--- /dev/null
+++ b/ui/src/main/java/com/wireguard/android/viewmodel/ConfigDetail.kt
@@ -0,0 +1,22 @@
+package com.wireguard.android.viewmodel
+
+import androidx.databinding.ObservableArrayList
+import androidx.databinding.ObservableList
+
+import com.wireguard.config.Config
+
+class ConfigDetail {
+    val config: Config?
+    val peers: ObservableList<PeerDetail> = ObservableArrayList()
+
+    constructor(other: Config?) {
+        config = other
+        if (other != null) {
+            other.peers.forEach {
+                val detail = PeerDetail(it)
+                peers.add(detail)
+                detail.bind(this)
+            }
+        }
+    }
+}
diff --git a/ui/src/main/java/com/wireguard/android/viewmodel/PeerDetail.kt b/ui/src/main/java/com/wireguard/android/viewmodel/PeerDetail.kt
new file mode 100644
index 00000000..80b32fd5
--- /dev/null
+++ b/ui/src/main/java/com/wireguard/android/viewmodel/PeerDetail.kt
@@ -0,0 +1,85 @@
+package com.wireguard.android.viewmodel
+
+import android.util.Log
+import androidx.databinding.BaseObservable
+import androidx.databinding.Bindable
+import androidx.databinding.Observable
+import androidx.databinding.ObservableList
+import androidx.databinding.ObservableArrayList
+
+import com.wireguard.android.BR
+import com.wireguard.config.InetEndpoint
+import com.wireguard.config.InetNetwork
+import com.wireguard.config.Peer
+import com.wireguard.crypto.Key;
+
+import java.util.Optional;
+
+import kotlin.collections.LinkedHashSet
+
+
+class PeerDetail : BaseObservable {
+    var peer: Peer?
+    private var owner: ConfigDetail? = null
+
+    @get:Bindable
+    var publicKey: Key
+
+    @get:Bindable
+    var allowedIps: ObservableList<InetNetwork> = ObservableArrayList<InetNetwork>()
+
+    @get:Bindable
+    var endpoint: Optional<InetEndpoint> = Optional.empty()
+        get() {
+            if (!field.isEmpty()) {
+                return field
+            } else {
+                return Optional.ofNullable(peer?.endpoint?.get())
+            }
+        }
+
+        set(value) {
+            Log.i(TAG, "notifyPropertyChanged endpoint " + this + ", " + value)
+            field = value
+            notifyPropertyChanged(BR.endpoint)
+        }
+
+    @get:Bindable
+    var persistentKeepalive: Optional<Int> = Optional.empty()
+
+    constructor(other: Peer) {
+        peer = other
+        publicKey = other.getPublicKey()
+        allowedIps.addAll(other.getAllowedIps())
+        endpoint = other.getEndpoint();
+        persistentKeepalive = other.getPersistentKeepalive()
+    }
+
+    constructor(publicKey: Key) {
+        peer = null
+        this.publicKey = publicKey
+    }
+
+    fun bind(owner: ConfigDetail) {
+        this.owner = owner
+    }
+
+    override fun addOnPropertyChangedCallback (callback: Observable.OnPropertyChangedCallback) {
+        Log.i(TAG, "addOnPropertyChangedCallback " + this + ", " + callback)
+        super.addOnPropertyChangedCallback(callback)
+    }
+    
+    /**
+     * Converts the {@code Peer} into a string suitable for debugging purposes. The {@code Peer} is
+     * identified by its public key and (if known) its endpoint.
+     *
+     * @return a concise single-line identifier for the {@code Peer}
+     */
+    override fun toString(): String {
+        return "(Peer " + publicKey.toBase64() + ")"
+    }
+
+    companion object {
+        private const val TAG = "WireGuard/PeerDetail"
+    }
+}