minios/screens/screen_download.h

// Copyright 2021 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.

#ifndef MINIOS_SCREENS_SCREEN_DOWNLOAD_H_
#define MINIOS_SCREENS_SCREEN_DOWNLOAD_H_

#include <memory>
#include <string>

#include "minios/recovery_installer.h"
#include "minios/screens/screen_base.h"
#include "minios/update_engine_proxy.h"

namespace minios {

class ScreenDownload : public UpdateEngineProxy::UpdaterDelegate,
                       public ScreenBase {
 public:
  ScreenDownload(std::unique_ptr<RecoveryInstallerInterface> recovery_installer,
                 std::shared_ptr<UpdateEngineProxy> update_engine_proxy,
                 std::shared_ptr<DrawInterface> draw_utils,
                 ScreenControllerInterface* screen_controller);

  ~ScreenDownload() = default;

  ScreenDownload(const ScreenDownload&) = delete;
  ScreenDownload& operator=(const ScreenDownload&) = delete;

  void Show() override;
  void Reset() override;
  void OnKeyPress(int key_changed) override;
  ScreenType GetType() override;
  std::string GetName() override;

  void SetDisplayUpdateEngineStateForTest(bool display) {
    display_update_engine_state_ = display;
  }

 private:
  FRIEND_TEST(ScreenDownloadTest, UpdateEngineError);
  FRIEND_TEST(ScreenDownloadTest, UpdateEngineProgressComplete);
  FRIEND_TEST(ScreenDownloadTest, RepartitionDisk);
  FRIEND_TEST(ScreenDownloadTest, RepartitionDiskFailed);
  FRIEND_TEST(ScreenDownloadTest, StartUpdateFailed);
  FRIEND_TEST(ScreenDownloadTest, IdleError);
  FRIEND_TEST(ScreenDownloadTest, ShowUpdateProgress);

  // Updates buttons with current selection.
  void ShowButtons();

  // Calls reboot.
  void Completed();

  // Shown when in the process of verifying update.
  void Finalizing();

  // Begin repartitioning disk, wiping data, and calling update engine to start
  // installation.
  void StartRecovery();

  // Calls corresponding MiniOs screen based on update engine status. If UE is
  // `DOWNLOADING` then shows a progress bar with percentage.
  void OnProgressChanged(const update_engine::StatusResult& status) override;

  std::unique_ptr<RecoveryInstallerInterface> recovery_installer_;
  std::shared_ptr<UpdateEngineProxy> update_engine_proxy_;

  // Determines whether we want to display the update engine state changes to
  // the UI. Only necessary after user has entered their password and connected
  // to the network.
  bool display_update_engine_state_;

  // Used to keep track of the last seen Update Engine stage to prevent
  // unnecessary screen changes.
  update_engine::Operation previous_update_state_{
      update_engine::Operation::IDLE};
};

}  // namespace minios

#endif  // MINIOS_SCREENS_SCREEN_DOWNLOAD_H_