add PowerSource

device.go

@@ -657,6 +657,19 @@ func (d *device) Reboot() (err error) {
 	return
 }
 
+func (d *device) PowerSource() (powerInfo interface{}, err error) {
+	if _, err = d.lockdownService(); err != nil {
+		return
+	}
+	if d.diagnosticsRelay, err = d.lockdown.DiagnosticsRelayService(); err != nil {
+		return
+	}
+	if powerInfo, err = d.diagnosticsRelay.PowerSource(); err != nil {
+		return
+	}
+	return
+}
+
 func (d *device) Shutdown() (err error) {
 	if _, err = d.lockdownService(); err != nil {
 		return

device_test.go

@@ -155,6 +155,15 @@ func Test_device_Shutdown(t *testing.T) {
 	dev.Shutdown()
 }
 
+func Test_device_PowerInfo(t *testing.T) {
+	setupDevice(t)
+	data, err := dev.PowerSource()
+	if err != nil {
+		panic(err)
+	}
+	log.Println(data)
+}
+
 func Test_device_InstallationProxyBrowse(t *testing.T) {
 	setupDevice(t)
 

diagnosticsrelay.go

@@ -1,6 +1,8 @@
 package giDevice
 
-import "github.com/SonicCloudOrg/sonic-gidevice/pkg/libimobiledevice"
+import (
+	"github.com/SonicCloudOrg/sonic-gidevice/pkg/libimobiledevice"
+)
 
 func newDiagnosticsRelay(client *libimobiledevice.DiagnosticsRelayClient) *diagnostics {
 	return &diagnostics{
@@ -37,3 +39,26 @@ func (d *diagnostics) Shutdown() (err error) {
 	}
 	return
 }
+
+func (d *diagnostics) PowerSource() (powerInfo interface{}, err error) {
+	var data map[string]string
+	data["Request"] = "IORegistry"
+	data["EntryClass"] = "IOPMPowerSource"
+	data["Label"] = "org.cloud.sonic.gidevice"
+	var pkt libimobiledevice.Packet
+	if pkt, err = d.client.NewXmlPacket(
+		d.client.NewBasicRequest("Shutdown"),
+	); err != nil {
+		return nil, err
+	}
+	if err = d.client.SendPacket(pkt); err != nil {
+		return nil, err
+	}
+	if pkt, err = d.client.ReceivePacket(); err != nil {
+		return nil, err
+	}
+	if err = pkt.Unmarshal(powerInfo); err != nil {
+		return nil, err
+	}
+	return
+}

idevice.go

@@ -67,6 +67,7 @@ type Device interface {
 
 	Reboot() error
 	Shutdown() error
+	PowerSource() (powerInfo interface{}, err error)
 
 	crashReportMoverService() (crashReportMover CrashReportMover, err error)
 	MoveCrashReport(hostDir string, opts ...CrashReportMoverOption) (err error)
@@ -234,6 +235,7 @@ type Perfd interface {
 type DiagnosticsRelay interface {
 	Reboot() error
 	Shutdown() error
+	PowerSource() (powerInfo interface{}, err error)
 }
 
 type WebInspector interface {

pkg/libimobiledevice/diagnosticsrelay.go

@@ -37,3 +37,7 @@ func (c *DiagnosticsRelayClient) NewXmlPacket(req interface{}) (Packet, error) {
 func (c *DiagnosticsRelayClient) SendPacket(pkt Packet) (err error) {
 	return c.client.SendPacket(pkt)
 }
+
+func (c *DiagnosticsRelayClient) ReceivePacket() (respPkt Packet, err error) {
+	return c.client.ReceivePacket()
+}