src/atrus_controller.cc

// Copyright 2017 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "atrus_controller.h"

#include <base/logging.h>

#include "upgrade.h"

namespace atrusctl {

namespace {

// Default values for diagnostic intervals
const int kDefaultDiagIntervalSec = 60;
const int kDefaultExtDiagIntervalSec = 1800;

}  // namespace

AtrusController::AtrusController(const base::FilePath& firmware_path)
    : firmware_path_(firmware_path) {}

void AtrusController::OnDeviceAdded(const std::string& device_path) {
  diagnostics_.UpdateNumberOfDevices(++device_counter_);
  if (!current_device_path_.empty()) {
    LOG(INFO) << "Device " << device_path
              << " added, but it's ignored in favor of "
              << current_device_path_;
    return;
  } else {
    LOG(INFO) << "Atrus device added";
  }
  current_device_path_ = device_path;

  // Upgrade firmware (if needed) and start diagnostics
  UpgradeDeviceFirmware(firmware_path_);
}

void AtrusController::OnDeviceRemoved(const std::string& device_path) {
  diagnostics_.UpdateNumberOfDevices(--device_counter_);
  if (device_path != current_device_path_) {
    return;
  }
  LOG(INFO) << "Atrus device removed";
  DisableDiagnostics();
  current_device_path_.clear();
  LOG(INFO) << "(" << diagnostics_.number_of_connected_slaves() << ")"
            << " daisy-chained devices disconnected as a result";
}

bool AtrusController::UpgradeDeviceFirmware(const base::FilePath& firmware_path,
                                            bool force) {
  DisableDiagnostics();

  bool skipped;
  bool ret = PerformUpgrade(firmware_path, &skipped, force);
  LOG(INFO) << "Upgrade "
            << (ret ? (skipped ? "skipped" : "succeeded") : "failed");
  LOG(INFO) << "[Device is entering usage mode]";

  EnableDiagnostics(base::TimeDelta::FromSeconds(kDefaultDiagIntervalSec),
                    base::TimeDelta::FromSeconds(kDefaultExtDiagIntervalSec));

  return ret;
}

void AtrusController::EnableDiagnostics(base::TimeDelta diag_interval,
                                        base::TimeDelta ext_diag_interval) {
  LOG(INFO) << "Polling diagnostics every " << diag_interval;
  VLOG(1) << "Diagnostics specification can be found here: go/atrus-diag";

  diagnostics_.Start(diag_interval, ext_diag_interval, current_device_path_);
}

void AtrusController::DisableDiagnostics() {
  LOG(INFO) << "Diagnostics disabled";

  diagnostics_.Stop();
}

}  // namespace atrusctl