Merge pull request pbreault#120 from tiwiz/toggle_wifi_mobile

Toggle WiFi and Mobile data on target device or emulator

Author: pbreault

src/main/kotlin/com/developerphil/adbidea/action/DisableMobileAction.kt

@@ -0,0 +1,9 @@
+package com.developerphil.adbidea.action
+
+import com.developerphil.adbidea.adb.AdbFacade
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+
+class DisableMobileAction : AdbAction() {
+    override fun actionPerformed(e: AnActionEvent, project: Project) = AdbFacade.disableMobile(project)
+}

src/main/kotlin/com/developerphil/adbidea/action/DisableWifiAction.kt

@@ -0,0 +1,9 @@
+package com.developerphil.adbidea.action
+
+import com.developerphil.adbidea.adb.AdbFacade
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+
+class DisableWifiAction : AdbAction() {
+    override fun actionPerformed(e: AnActionEvent, project: Project) = AdbFacade.disableWifi(project)
+}

src/main/kotlin/com/developerphil/adbidea/action/EnableMobileAction.kt

@@ -0,0 +1,9 @@
+package com.developerphil.adbidea.action
+
+import com.developerphil.adbidea.adb.AdbFacade
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+
+class EnableMobileAction : AdbAction() {
+    override fun actionPerformed(e: AnActionEvent, project: Project) = AdbFacade.enableMobile(project)
+}

src/main/kotlin/com/developerphil/adbidea/action/EnableWifiAction.kt

@@ -0,0 +1,9 @@
+package com.developerphil.adbidea.action
+
+import com.developerphil.adbidea.adb.AdbFacade
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+
+class EnableWifiAction : AdbAction() {
+    override fun actionPerformed(e: AnActionEvent, project: Project) = AdbFacade.enableWifi(project)
+}

src/main/kotlin/com/developerphil/adbidea/adb/AdbFacade.kt

@@ -2,6 +2,8 @@ package com.developerphil.adbidea.adb
 
 import com.developerphil.adbidea.ObjectGraph
 import com.developerphil.adbidea.adb.command.*
+import com.developerphil.adbidea.adb.command.SvcCommand.MOBILE
+import com.developerphil.adbidea.adb.command.SvcCommand.WIFI
 import com.developerphil.adbidea.ui.NotificationHelper
 import com.google.common.util.concurrent.ThreadFactoryBuilder
 import com.intellij.openapi.project.Project
@@ -22,6 +24,10 @@ object AdbFacade {
     fun clearData(project: Project) = executeOnDevice(project, ClearDataCommand())
     fun clearDataAndRestart(project: Project) = executeOnDevice(project, ClearDataAndRestartCommand())
     fun clearDataAndRestartWithDebugger(project: Project) = executeOnDevice(project, ClearDataAndRestartWithDebuggerCommand())
+    fun enableWifi(project: Project) = executeOnDevice(project, ToggleSvcCommand(WIFI, true))
+    fun disableWifi(project: Project) = executeOnDevice(project, ToggleSvcCommand(WIFI, false))
+    fun enableMobile(project: Project) = executeOnDevice(project, ToggleSvcCommand(MOBILE, true))
+    fun disableMobile(project: Project) = executeOnDevice(project, ToggleSvcCommand(MOBILE, false))
 
     private fun executeOnDevice(project: Project, runnable: Command) {
         if (AdbUtil.isGradleSyncInProgress(project)) {
@@ -41,4 +47,4 @@ object AdbFacade {
             NotificationHelper.error("No Device found")
         }
     }
-}
+}

src/main/kotlin/com/developerphil/adbidea/adb/command/ToggleSvcCommand.kt

@@ -0,0 +1,34 @@
+package com.developerphil.adbidea.adb.command
+
+import com.android.ddmlib.IDevice
+import com.developerphil.adbidea.adb.command.receiver.GenericReceiver
+import com.developerphil.adbidea.ui.NotificationHelper.error
+import com.developerphil.adbidea.ui.NotificationHelper.info
+import com.intellij.openapi.project.Project
+import org.jetbrains.android.facet.AndroidFacet
+import java.util.concurrent.TimeUnit
+
+enum class SvcCommand(val parameter: String, val description: String) {
+    WIFI("wifi", "Wi-Fi"),
+    MOBILE("data", "Mobile data")
+}
+
+class ToggleSvcCommand(
+        private val command: SvcCommand,
+        private val enable: Boolean) : Command {
+
+    private val shellCommand = "svc ${command.parameter} ${enable.toState()}"
+
+    override fun run(project: Project, device: IDevice, facet: AndroidFacet, packageName: String): Boolean {
+        try {
+            device.executeShellCommand(shellCommand, GenericReceiver(), 30L, TimeUnit.SECONDS)
+            info(String.format("<b>%s</b> %s%s on %s", command.description, enable.toState(), "d", device.name))
+            return true
+        } catch (e: Exception) {
+            error("Failure while attempting to ${enable.toState()} ${command.description} on ${device.name}: " + e.message)
+        }
+        return false
+    }
+
+    private fun Boolean.toState() = if (this) "enable" else "disable"
+}

src/main/resources/META-INF/plugin.xml

@@ -208,6 +208,32 @@
                     description="Clears the private storage of the app and restarts it, and attach the debugger">
             </action>
             <add-to-group group-id="AndroidToolsGroup" anchor="first"/>
+            <separator/>
+
+            <action id="com.developerphil.adbidea.action.EnableWifiAction"
+                    class="com.developerphil.adbidea.action.EnableWifiAction"
+                    text="ADB Enable Wi-Fi"
+                    description="Enable Wi-Fi on device or emulator">
+            </action>
+
+            <action id="com.developerphil.adbidea.action.DisableWifiAction"
+                    class="com.developerphil.adbidea.action.DisableWifiAction"
+                    text="ADB Disable Wi-Fi"
+                    description="Disable Wi-Fi on device or emulator">
+            </action>
+
+            <action id="com.developerphil.adbidea.action.EnableMobileAction"
+                    class="com.developerphil.adbidea.action.EnableMobileAction"
+                    text="ADB Enable Mobile Data"
+                    description="Enable mobile data on device or emulator">
+            </action>
+
+            <action id="com.developerphil.adbidea.action.DisableMobileAction"
+                    class="com.developerphil.adbidea.action.DisableMobileAction"
+                    text="ADB Disable Mobile Data"
+                    description="Disable mobile data on device or emulator">
+            </action>
+
         </group>
     </actions>
 
@@ -216,4 +242,4 @@
             <implementation-class>com.developerphil.adbidea.ObjectGraph</implementation-class>
         </component>
     </project-components>
-</idea-plugin>
+</idea-plugin>