usbtotcp.go

package main

import (
	"fmt"

	"github.com/codeskyblue/iosbridge/usbmuxd"
)

type UsbToTcpHandler struct {
	*usbmuxd.MuxRelayHandler
	udid string
	port int
}

func NewUsbToTcpHandler(udid string, port int) (*UsbToTcpHandler, error) {
	handler := &UsbToTcpHandler{
		MuxRelayHandler: usbmuxd.NewMuxRelayHandlerSimple(),
		udid:            udid,
		port:            port,
	}
	_, err := handler.MuxRelayHandler.GetDeviceByUDID(udid)
	return handler, err
}

func (h *UsbToTcpHandler) HandleListDevices(muxConn *usbmuxd.UsbMuxConnection) {
	deviceList := []usbmuxd.USBDeviceAttachedDetachedFrame{}
	deviceFrame, err := h.MuxRelayHandler.GetDeviceByUDID(h.udid)
	if err == nil {
		deviceList = append(deviceList, deviceFrame)
	}
	muxConn.SendPlistMessage(map[string]interface{}{
		"DeviceList": deviceList,
	})
}

func startUsbToTcp(udid string, port int) error {
	handler, err := NewUsbToTcpHandler(udid, port)
	if err != nil {
		return err
	}
	server := usbmuxd.NewServer(fmt.Sprintf("tcp://:%d", port), handler)
	return server.ListenAndServe()
}