flipper/src/dispatcher/androidDevice.js

/**
 * Copyright 2018-present Facebook.
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 * @format
 */

import AndroidDevice from '../devices/AndroidDevice';
import child_process from 'child_process';
import promiseRetry from 'promise-retry';
import type {Store} from '../reducers/index.js';
import type BaseDevice from '../devices/BaseDevice';
import type Logger from '../fb-stubs/Logger.js';
const adb = require('adbkit-fb');

function createDevice(client, device): Promise<AndroidDevice> {
  return new Promise((resolve, reject) => {
    const type =
      device.type !== 'device' || device.id.startsWith('emulator')
        ? 'emulator'
        : 'physical';

    client.getProperties(device.id).then(async props => {
      let name = props['ro.product.model'];
      if (type === 'emulator') {
        name = (await getRunningEmulatorName(device.id)) || name;
      }
      const androidDevice = new AndroidDevice(device.id, type, name, client);
      androidDevice.reverse();
      resolve(androidDevice);
    });
  });
}

function getRunningEmulatorName(id: string): Promise<?string> {
  return new Promise((resolve, reject) => {
    const port = id.replace('emulator-', '');
    // The GNU version of netcat doesn't terminate after 1s when
    // specifying `-w 1`, so we kill it after a timeout. Because
    // of that, even in case of an error, there may still be
    // relevant data for us to parse.
    child_process.exec(
      `echo "avd name" | nc -w 1 localhost ${port}`,
      {timeout: 1000, encoding: 'utf-8'},
      (error: ?Error, data) => {
        if (data != null && typeof data === 'string') {
          const match = data.trim().match(/(.*)\r\nOK$/);
          resolve(match != null && match.length > 0 ? match[1] : null);
        } else {
          reject(error);
        }
      },
    );
  });
}

export default (store: Store, logger: Logger) => {
  // Using this client before adb server is started up will cause failures.
  // this gets around this by waiting for listDevices first, which ensures
  // the server is up and running before allowing any other operations.

  function createClient() {
    const unsafeClient = adb.createClient();
    return promiseRetry(
      (retry, number) => {
        return unsafeClient
          .listDevices()
          .then(() => {
            return unsafeClient;
          })
          .catch(e => {
            console.warn(`Failed to start adb client. Retrying. ${e.message}`);
            retry(e);
          });
      },
      {
        minTimeout: 200,
        retries: 5,
      },
    );
  }
  const clientPromise = createClient();

  const watchAndroidDevices = () => {
    // get emulators
    child_process.exec(
      'emulator -list-avds',
      (error: ?Error, data: ?string) => {
        if (error != null || data == null) {
          console.error(error || 'Failed to list AVDs');
          return;
        }
        const payload = data.split('\n').filter(Boolean);
        store.dispatch({
          type: 'REGISTER_ANDROID_EMULATORS',
          payload,
        });
      },
    );

    clientPromise
      .then(client => {
        client
          .trackDevices()
          .then(tracker => {
            tracker.on('error', err => {
              if (err.message === 'Connection closed') {
                // adb server has shutdown, remove all android devices
                const {connections} = store.getState();
                const deviceIDsToRemove: Array<
                  string,
                > = connections.devices
                  .filter(
                    (device: BaseDevice) => device instanceof AndroidDevice,
                  )
                  .map((device: BaseDevice) => device.serial);

                store.dispatch({
                  type: 'UNREGISTER_DEVICES',
                  payload: new Set(deviceIDsToRemove),
                });
                console.error('adb server was shutdown');
                setTimeout(watchAndroidDevices, 500);
              } else {
                throw err;
              }
            });

            tracker.on('add', async device => {
              if (device.type !== 'offline') {
                const androidDevice = await createDevice(client, device);
                store.dispatch({
                  type: 'REGISTER_DEVICE',
                  payload: androidDevice,
                });
              }
            });

            tracker.on('change', async device => {
              if (device.type === 'offline') {
                store.dispatch({
                  type: 'UNREGISTER_DEVICES',
                  payload: new Set([device.id]),
                });
              } else {
                const androidDevice = await createDevice(client, device);
                store.dispatch({
                  type: 'REGISTER_DEVICE',
                  payload: androidDevice,
                });
              }
            });

            tracker.on('remove', device => {
              store.dispatch({
                type: 'UNREGISTER_DEVICES',
                payload: new Set([device.id]),
              });
            });
          })
          .catch(err => {
            if (err.code === 'ECONNREFUSED') {
              // adb server isn't running
            } else {
              throw err;
            }
          });
      })
      .catch(e => {
        console.error(`Failed to watch for android devices: ${e.message}`);
      });
  };

  watchAndroidDevices();
};