com.github.cosysoft.device.android.impl.AndroidDeviceStore Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of device-api Show documentation
A set of tools for operate android device via android debug bridge
There is a newer version: 0.9.3
package com.github.cosysoft.device.android.impl;

import com.android.ddmlib.AndroidDebugBridge;
import com.android.ddmlib.IDevice;
import com.github.cosysoft.device.DeviceStore;
import com.github.cosysoft.device.android.AndroidDevice;
import com.github.cosysoft.device.exception.AndroidDeviceException;
import com.github.cosysoft.device.exception.DeviceNotFoundException;
import com.github.cosysoft.device.exception.NestedException;
import com.github.cosysoft.device.shell.AndroidSdk;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.HashMap;
import java.util.Map;
import java.util.TreeSet;

public class AndroidDeviceStore implements DeviceStore {

    protected static final Logger logger = LoggerFactory
            .getLogger(AndroidDeviceStore.class);

    private Map connectedDevices = new HashMap<>();

    private AndroidDebugBridge bridge;
    private boolean shouldKeepAdbAlive = false;

    static class DeviceStoreHolder {

        static final AndroidDeviceStore instance = init();

        static AndroidDeviceStore init() {
            AndroidDeviceStore instance;
            instance = new AndroidDeviceStore();
            instance.initAndroidDevices(false);
            return instance;
        }
    }

    public static AndroidDeviceStore getInstance() {
        return DeviceStoreHolder.instance;
    }

    /**
     * call once
     *
     * @param shouldKeepAdbAlive
     */
    public void initAndroidDevices(boolean shouldKeepAdbAlive)
            throws AndroidDeviceException {
//        DdmPreferences.setLogLevel(LogLevel.VERBOSE.getStringValue());
        this.initializeAdbConnection();
    }

    /**
     * Initializes the AndroidDebugBridge and registers the
     * DefaultHardwareDeviceManager with the AndroidDebugBridge device change
     * listener.
     */
    protected void initializeAdbConnection() {
        // Get a device bridge instance. Initialize, create and restart.
        try {
            AndroidDebugBridge.init(true);
        } catch (IllegalStateException e) {
            // When we keep the adb connection alive the AndroidDebugBridge may
            // have been already
            // initialized at this point and it generates an exception. Do not
            // print it.
            if (!shouldKeepAdbAlive) {
                logger.error(
                        "The IllegalStateException is not a show "
                                + "stopper. It has been handled. This is just debug spew. Please proceed.",
                        e);
                throw new NestedException("ADB init failed", e);
            }
        }

        bridge = AndroidDebugBridge.getBridge();

        if (bridge == null) {
            bridge = AndroidDebugBridge.createBridge(AndroidSdk.adb()
                    .getAbsolutePath(), false);
        }

        long timeout = System.currentTimeMillis() + 60000;
        while (!bridge.hasInitialDeviceList()
                && System.currentTimeMillis() < timeout) {
            try {
                Thread.sleep(50);
            } catch (InterruptedException e) {
                throw new RuntimeException(e);
            }
        }
        // Add the existing devices to the list of devices we are tracking.
        IDevice[] devices = bridge.getDevices();
        logger.info("initialDeviceList size {}", devices.length);
        for (int i = 0; i < devices.length; i++) {
            logger.info("devices state: {},{} ", devices[i].getName(),
                    devices[i].getState());
            connectedDevices.put(devices[i], new DefaultHardwareDevice(
                    devices[i]));
        }


        bridge.addDeviceChangeListener(new DeviceChangeListener(connectedDevices));
    }

    @Override
    public TreeSet getDevices() {
        return new TreeSet(connectedDevices.values());
    }

    @Override
    public AndroidDevice getDeviceBySerial(String serialID) {
        for (AndroidDevice device : getDevices()) {
            if (device.getSerialNumber().equalsIgnoreCase(serialID)) {
                return device;
            }
        }
        throw new DeviceNotFoundException(String.format("Device %s not found", serialID));
    }

    /**
     * Shutdown the AndroidDebugBridge and clean up all connected devices.
     */
    @Override
    public void shutdown() {
        if (!shouldKeepAdbAlive) {
            AndroidDebugBridge.disconnectBridge();
            AndroidDebugBridge.terminate();
        }
        logger.info("stopping Device Manager");
    }

    /**
     * used with caution or don't call this method
     */
    @Override
    public void shutdownForcely() {
        AndroidDebugBridge.disconnectBridge();
        AndroidDebugBridge.terminate();
    }


}