WOL网络唤醒

/**
 * 发送幻数据包唤醒局域网中的设备
 * @param macAddress 目标设备的MAC地址（格式："00:11:22:33:44:55"）
 * @param ipAddress 广播地址，通常是 "255.255.255.255" 或具体子网的广播地址
 * @param port 发送端口，通常是 7 或 9，这里默认使用 4000
 */
suspend fun wakeOnLan(macAddress: String, ipAddress: String = "255.255.255.255", port: Int = 4000, onResult: ((Boolean) -> Unit)? = null) {
    withContext (Dispatchers.IO) {
        try {
            // 6字节的目标 MAC 地址
            val bytes = ByteArray(6)
            val mac = macAddress.split(":")
            for (i in 0 until 6) {
                bytes[i] = mac[i].toInt(16).toByte()
            }

            // 构造幻数据包（Magic Packet）
            // 1. 6 字节的 FF
            val buffer = ByteArray(6 + 16 * 6)
            for (i in 0 until 6) {
                buffer[i] = 0xFF.toByte()
            }
            // 2. 重复目标 MAC 地址 16 次
            for (i in 0 until 16) {
                System.arraycopy(bytes, 0, buffer, 6 + i * 6, 6)
            }

            // 发送 UDP 数据包
            val socket = DatagramSocket()
            socket.setBroadcast(true)
            val packet = DatagramPacket(
                buffer,
                buffer.size,
                InetAddress.getByName(ipAddress),
                port
            )
            socket.send(packet)
            socket.close()

            withContext(Dispatchers.Main) {
                onResult?.invoke(true)
            }
        } catch (e: CancellationException) {
            throw e
        } catch (e: Exception) {
            e.printStackTrace()
            withContext(Dispatchers.Main) {
                onResult?.invoke(false)
            }
        }
    }
}

/**
 * 外部设备控制器
 */
class WOLControlService : ControlsProviderService() {

    private val TAG = WOLControlService::class.java.simpleName
    private var mGlobalJob: Job? = null

    /**
     * 添加控制器时系统会调用 createPublisherForAllAvailable 方法来获取可用的设备列表
     * 使用 flowPublish 需要导入 implementation("org.jetbrains.kotlinx:kotlinx-coroutines-jdk9:1.6.4")
     */
    override fun createPublisherForAllAvailable(): Flow.Publisher<Control> = flowPublish {
        createWolDeviceControls().forEach { send(it) }
    }

    /**
     * 返回给定 controlId 的有效 Publisher
     * 添加控制器后系统会调用 createPublisherFor 方法来创建指定设备的控制器
     */
    override fun createPublisherFor(controlIds: List<String?>): Flow.Publisher<Control> = flowPublish {
        controlIds.forEach {
            it?.takeIf { it.isNotBlank() }?.let { controlId ->
                try {
                    val devices = SharedPreferencesUtils.getWOLDeviceById(this@WOLControlService, controlId.toInt())
                    createControlByDevice(devices)?.let { send(it) }
                } catch (e: Exception) {
                    e.printStackTrace()
                }
            }
        }
    }

    @OptIn(DelicateCoroutinesApi::class)
    override fun performControlAction(controlId: String, action: ControlAction, consumer: Consumer<Int?>) {
        SharedPreferencesUtils.getWOLDeviceById(this@WOLControlService, controlId.toInt())?.let {
            mGlobalJob = GlobalScope.launch {
                WOLUtils.wakeOnLan(it.macAddress, it.ipAddress, it.port) { result ->
                    if (result) {
                        Log.e(TAG, "唤醒成功: ${it.deviceName}")
                    }
                }
                consumer.accept(ControlAction.RESPONSE_OK)
            }
        } ?: run {
            consumer.accept(ControlAction.RESPONSE_FAIL)
        }
    }

    /**
     * 创建设备管理列表, 用于告知系统可用的设备列表
     */
    private fun createWolDeviceControls(): List<Control> {
        val controls = mutableListOf<Control>()
        val devices = SharedPreferencesUtils.getWOLDevices(this)
        for (device in devices) {
            createControlByDevice(device)?.let { controls.add(it) }
        }
        return controls
    }

    private fun createControlByDevice(device: WolDevices?): Control? {
        return if (device == null) null else Control.StatefulBuilder(
            device.id.toString(),
            PendingIntent.getActivity(this, 0, Intent(this, MainActivity::class.java), PendingIntent.FLAG_IMMUTABLE)
        ).setTitle(device.deviceName)
            .setSubtitle(device.macAddress)
            .setDeviceType(DeviceTypes.TYPE_TV)
            .setStatus(Control.STATUS_OK)
            .setControlTemplate(StatelessTemplate(device.id.toString()))
            .build()
    }

    override fun onDestroy() {
        super.onDestroy()
        mGlobalJob?.cancel()
    }

}