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

#ifndef CHROME_BROWSER_UI_VIEWS_RELAUNCH_NOTIFICATION_RELAUNCH_REQUIRED_DIALOG_VIEW_H_
#define CHROME_BROWSER_UI_VIEWS_RELAUNCH_NOTIFICATION_RELAUNCH_REQUIRED_DIALOG_VIEW_H_

#include "base/functional/callback.h"
#include "base/time/time.h"
#include "chrome/browser/ui/views/relaunch_notification/relaunch_required_timer.h"
#include "ui/views/window/dialog_delegate.h"

class BrowserWindowInterface;
namespace views {
class Widget;
}  // namespace views

// A View for the relaunch required dialog. This is shown to users to inform
// them that Chrome will be relaunched by the RelaunchNotificationController as
// dictated by policy settings and upgrade availability.
class RelaunchRequiredDialogView : public views::DialogDelegateView {
 public:
  // Shows the dialog in |browser| for a relaunch that will be forced at
  // |deadline|. |on_accept| is run if the user accepts the prompt to restart.
  // If |ap_style|, the dialog uses Advanced Protection string and icon.
  static views::Widget* Show(BrowserWindowInterface* browser,
                             base::Time deadline,
                             bool ap_style,
                             base::RepeatingClosure on_accept);

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

  ~RelaunchRequiredDialogView() override;

  // Returns the instance hosted by |widget|. |widget| must be an instance
  // previously returned from Show().
  static RelaunchRequiredDialogView* FromWidget(views::Widget* widget);

  // Sets the relaunch deadline to |deadline| and refreshes the view's title
  // accordingly.
  void SetDeadline(base::Time deadline);

  // Returns the deadline used to derive the time-to-relaunch shown to the user.
  base::Time deadline() const { return relaunch_required_timer_.deadline(); }

  // views::DialogDelegateView:
  std::u16string GetWindowTitle() const override;
  ui::ImageModel GetWindowIcon() override;

 private:
  RelaunchRequiredDialogView(base::Time deadline,
                             bool ap_style,
                             base::RepeatingClosure on_accept);

  // Invoked when the timer fires to refresh the title text.
  void UpdateWindowTitle();

  // Timer that schedules title refreshes.
  RelaunchRequiredTimer relaunch_required_timer_;

  // Show Advanced Protection Program string and icon.
  bool ap_style_ = false;
};

#endif  // CHROME_BROWSER_UI_VIEWS_RELAUNCH_NOTIFICATION_RELAUNCH_REQUIRED_DIALOG_VIEW_H_
