main.go

package main

import (
	"fmt"
	"iconsole/frames"
	"iconsole/tunnel"
	"os"

	"github.com/urfave/cli"
)

var globalFlags = []cli.Flag{
	cli.StringFlag{
		Name:   "UDID, u",
		Usage:  "device serialNumber UDID",
		EnvVar: "DEVICE_UDID",
		Value:  "",
	},
}

func session(udid string, cb func(*tunnel.LockdownConnection) error) error {
	device, err := getDevice(udid)
	if err != nil {
		return err
	}

	conn, err := tunnel.LockdownDial(device)
	if err != nil {
		return err
	}

	defer conn.Close()

	if err := conn.StartSession(); err != nil {
		return err
	}

	defer conn.StopSession()

	return cb(conn)
}

func getDevice(udid string) (frames.Device, error) {
	devices, err := tunnel.Devices()
	if err != nil {
		return nil, err
	}

	var ds []frames.Device

	for i, d := range devices {
		if udid == "" && d.GetConnectionType() == "USB" {
			return d, nil
		}
		if d.GetSerialNumber() == udid {
			ds = append(ds, devices[i])
		}
	}

	if len(ds) > 0 {
		for _, d := range ds {
			if d.GetConnectionType() == "USB" {
				return d, nil
			}
		}
		return ds[0], nil
	}

	return nil, fmt.Errorf("device %s was not found", udid)
}

func main() {
	app := cli.NewApp()
	app.Name = "iConsole"
	app.Usage = "iOS device tools"
	app.Version = "1.0.0"
	app.Authors = []cli.Author{
		{
			Name:  "anonymous5l",
			Email: "wxdxfg@hotmail.com",
		},
	}
	app.Commands = []cli.Command{
		initDevices(),
		initSyslogCommond(),
		initSimCommond(),
		initScreenShotCommond(),
		initSyncCommond(),
		initValueCommond(),
		initTransportCommand(),
		initMountCommand(),
		initAFCCommand(),
		initArrest(),
		initProcessCommond(),
	}

	if err := app.Run(os.Args); err != nil {
		fmt.Println(err.Error())
		return
	}
}