ash/system/chromeos/session/tray_session_length_limit.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef ASH_SYSTEM_CHROMEOS_SESSION_SESSION_LENGTH_LIMIT_H_
   6 #define ASH_SYSTEM_CHROMEOS_SESSION_SESSION_LENGTH_LIMIT_H_
   7
   8 #include "ash/system/chromeos/session/session_length_limit_observer.h"
   9 #include "ash/system/tray/system_tray_item.h"
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/time/time.h"
  14 #include "base/timer/timer.h"
  15
  16 namespace ash {
  17 namespace test {
  18 class TraySessionLengthLimitTest;
  19 }
  20
  21 namespace tray {
  22 class RemainingSessionTimeTrayView;
  23 }
  24
  25 // Adds a countdown timer to the system tray if the session length is limited.
  26 class ASH_EXPORT TraySessionLengthLimit : public SystemTrayItem,
  27                                           public SessionLengthLimitObserver {
  28  public:
  29   enum LimitState {
  30     LIMIT_NONE,
  31     LIMIT_SET,
  32     LIMIT_EXPIRING_SOON
  33   };
  34
  35   explicit TraySessionLengthLimit(SystemTray* system_tray);
  36   virtual ~TraySessionLengthLimit();
  37
  38   // SystemTrayItem:
  39   virtual views::View* CreateTrayView(user::LoginStatus status) OVERRIDE;
  40   virtual void DestroyTrayView() OVERRIDE;
  41   virtual void UpdateAfterShelfAlignmentChange(
  42       ShelfAlignment alignment) OVERRIDE;
  43
  44   // SessionLengthLimitObserver:
  45   virtual void OnSessionStartTimeChanged() OVERRIDE;
  46   virtual void OnSessionLengthLimitChanged() OVERRIDE;
  47
  48   LimitState GetLimitState() const;
  49   base::TimeDelta GetRemainingSessionTime() const;
  50
  51  private:
  52   friend class test::TraySessionLengthLimitTest;
  53
  54   static const char kNotificationId[];
  55
  56   void Update();
  57
  58   bool IsTrayViewVisibleForTest();
  59
  60   tray::RemainingSessionTimeTrayView* tray_view_;
  61
  62   LimitState limit_state_;
  63   base::TimeTicks session_start_time_;
  64   base::TimeDelta limit_;
  65   base::TimeDelta remaining_session_time_;
  66   scoped_ptr<base::RepeatingTimer<TraySessionLengthLimit> > timer_;
  67
  68   DISALLOW_COPY_AND_ASSIGN(TraySessionLengthLimit);
  69 };
  70
  71 }  // namespace ash
  72
  73 #endif  // ASH_SYSTEM_CHROMEOS_SESSION_SESSION_LENGTH_LIMIT_H_