alarm_manager.cc revision c2e0dbddbe15c98d52c4786dac06cb8952a8ae6d
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/api/alarms/alarm_manager.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/json/json_writer.h" 9c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/lazy_instance.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/message_loop.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/time.h" 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/time/clock.h" 13c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/time/default_clock.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/value_conversions.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/event_router.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_service.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_system.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/state_store.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_notification_types.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/notification_service.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kOnAlarmEvent[] = "alarms.onAlarm"; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A list of alarms that this extension has set. 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kRegisteredAlarms[] = "alarms"; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kAlarmGranularity[] = "granularity"; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The minimum period between polling for alarms to run. 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::TimeDelta kDefaultMinPollPeriod = base::TimeDelta::FromDays(1); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DefaultAlarmDelegate : public AlarmManager::Delegate { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit DefaultAlarmDelegate(Profile* profile) : profile_(profile) {} 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DefaultAlarmDelegate() {} 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnAlarm(const std::string& extension_id, 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const Alarm& alarm) OVERRIDE { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<ListValue> args(new ListValue()); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args->Append(alarm.js_alarm->ToValue().release()); 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<Event> event(new Event(kOnAlarmEvent, args.Pass())); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ExtensionSystem::Get(profile_)->event_router()->DispatchEventToExtension( 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) extension_id, event.Pass()); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Profile* profile_; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Creates a TimeDelta from a delay as specified in the API. 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta TimeDeltaFromDelay(double delay_in_minutes) { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return base::TimeDelta::FromMicroseconds( 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay_in_minutes * base::Time::kMicrosecondsPerMinute); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::vector<Alarm> AlarmsFromValue(const base::ListValue* list) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<Alarm> alarms; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < list->GetSize(); ++i) { 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::DictionaryValue* alarm_dict = NULL; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Alarm alarm; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list->GetDictionary(i, &alarm_dict) && 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) api::alarms::Alarm::Populate(*alarm_dict, alarm.js_alarm.get())) { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Value* time_value = NULL; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarm_dict->Get(kAlarmGranularity, &time_value)) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::GetValueAsTimeDelta(*time_value, &alarm.granularity); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.push_back(alarm); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return alarms; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_ptr<base::ListValue> AlarmsToValue(const std::vector<Alarm>& alarms) { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::ListValue> list(new ListValue()); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < alarms.size(); ++i) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::DictionaryValue> alarm = 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms[i].js_alarm->ToValue().Pass(); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm->Set(kAlarmGranularity, 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::CreateTimeDeltaValue(alarms[i].granularity)); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list->Append(alarm.release()); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return list.Pass(); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// AlarmManager 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 94c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)AlarmManager::AlarmManager(Profile* profile) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : profile_(profile), 96c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) clock_(new base::DefaultClock()), 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate_(new DefaultAlarmDelegate(profile)) { 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED, 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::Source<Profile>(profile_)); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (storage) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->RegisterKey(kRegisteredAlarms); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AlarmManager::~AlarmManager() { 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::AddAlarm(const std::string& extension_id, 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Alarm& alarm) { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddAlarmImpl(extension_id, alarm); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const Alarm* AlarmManager::GetAlarm( 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id, const std::string& name) { 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator it = GetAlarmIterator(extension_id, name); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it.first == alarms_.end()) 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &*it.second; 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const AlarmManager::AlarmList* AlarmManager::GetAllAlarms( 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id) { 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &list->second; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AlarmManager::AlarmIterator AlarmManager::GetAlarmIterator( 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id, const std::string& name) { 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(alarms_.end(), AlarmList::iterator()); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmList::iterator it = list->second.begin(); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != list->second.end(); ++it) { 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it->js_alarm->name == name) 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(list, it); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(alarms_.end(), AlarmList::iterator()); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AlarmManager::RemoveAlarm(const std::string& extension_id, 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& name) { 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator it = GetAlarmIterator(extension_id, name); 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it.first == alarms_.end()) 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(it); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::RemoveAllAlarms(const std::string& extension_id) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: I'm using indices rather than iterators here because 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // RemoveAlarmIterator will delete the list when it becomes empty. 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0, size = list->second.size(); i < size; ++i) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(AlarmIterator(list, list->second.begin())); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(alarms_.find(extension_id) == alarms_.end()); 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 171c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)void AlarmManager::SetClockForTesting(base::Clock* clock) { 172c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) clock_.reset(clock); 173c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 174c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 175c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)static base::LazyInstance<ProfileKeyedAPIFactory<AlarmManager> > 176c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)g_factory = LAZY_INSTANCE_INITIALIZER; 177c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 178c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// static 179c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)ProfileKeyedAPIFactory<AlarmManager>* AlarmManager::GetFactoryInstance() { 180c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return &g_factory.Get(); 181c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 182c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 183c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// static 184c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)AlarmManager* AlarmManager::Get(Profile* profile) { 185c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return ProfileKeyedAPIFactory<AlarmManager>::GetForProfile(profile); 186c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 187c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::RemoveAlarmIterator(const AlarmIterator& iter) { 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmList& list = iter.first->second; 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list.erase(iter.second); 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list.empty()) 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_.erase(iter.first); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel the timer if there are no more alarms. 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We don't need to reschedule the poll otherwise, because in 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the worst case we would just poll one extra time. 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarms_.empty()) 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::OnAlarm(AlarmIterator it) { 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(it.first != alarms_.end()); 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Alarm& alarm = *it.second; 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string extension_id_copy(it.first->first); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_->OnAlarm(extension_id_copy, alarm); 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Update our scheduled time for the next alarm. 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (double* period_in_minutes = 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->period_in_minutes.get()) { 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->scheduled_time = 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (last_poll_time_ + 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TimeDeltaFromDelay(*period_in_minutes)).ToJsTime(); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(it); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id_copy); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::AddAlarmImpl(const std::string& extension_id, 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Alarm& alarm) { 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override any old alarm with the same name. 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator old_alarm = GetAlarmIterator(extension_id, 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->name); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (old_alarm.first != alarms_.end()) 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(old_alarm); 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_[extension_id].push_back(alarm); 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScheduleNextPoll(); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::WriteToStorage(const std::string& extension_id) { 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!storage) 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<Value> alarms; 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list != alarms_.end()) 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.reset(AlarmsToValue(list->second).release()); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.reset(AlarmsToValue(std::vector<Alarm>()).release()); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->SetExtensionValue(extension_id, kRegisteredAlarms, alarms.Pass()); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::ReadFromStorage(const std::string& extension_id, 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Value> value) { 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::ListValue* list = NULL; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!value.get() || !value->GetAsList(&list)) 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<Alarm> alarm_states = AlarmsFromValue(list); 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < alarm_states.size(); ++i) { 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddAlarmImpl(extension_id, alarm_states[i]); 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AlarmManager::ScheduleNextPoll() { 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 0. If there are no alarms, stop the timer. 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarms_.empty()) { 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(yoz): Try not to reschedule every single time if we're adding 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a lot of alarms. 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Find the soonest alarm that is scheduled to run and the smallest 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // granularity of any alarm. 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alarms_ guarantees that none of its contained lists are empty. 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time soonest_alarm_time = base::Time::FromJsTime( 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_.begin()->second.begin()->js_alarm->scheduled_time); 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::TimeDelta min_granularity = kDefaultMinPollPeriod; 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmMap::const_iterator m_it = alarms_.begin(), m_end = alarms_.end(); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) m_it != m_end; ++m_it) { 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmList::const_iterator l_it = m_it->second.begin(); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) l_it != m_it->second.end(); ++l_it) { 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time cur_alarm_time = 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time::FromJsTime(l_it->js_alarm->scheduled_time); 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cur_alarm_time < soonest_alarm_time) 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) soonest_alarm_time = cur_alarm_time; 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (l_it->granularity < min_granularity) 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) min_granularity = l_it->granularity; 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Time next_poll(last_poll_time_ + min_granularity); 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the next alarm is more than min_granularity in the future, wait for it. 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Otherwise, only poll as often as min_granularity. 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // As a special case, if we've never checked for an alarm before 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // (e.g. during startup), let alarms fire asap. 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (last_poll_time_.is_null() || next_poll < soonest_alarm_time) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll = soonest_alarm_time; 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Schedule the poll. 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll_time_ = next_poll; 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta delay = std::max(base::TimeDelta::FromSeconds(0), 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) next_poll - clock_->Now()); 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Start(FROM_HERE, 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay, 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &AlarmManager::PollAlarms); 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::PollAlarms() { 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) last_poll_time_ = clock_->Now(); 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Run any alarms scheduled in the past. OnAlarm uses vector::erase to remove 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // elements from the AlarmList, and map::erase to remove AlarmLists from the 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AlarmMap. 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmMap::iterator m_it = alarms_.begin(), m_end = alarms_.end(); 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) m_it != m_end;) { 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator cur_extension = m_it++; 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Iterate (a) backwards so that removing elements doesn't affect 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // upcoming iterations, and (b) with indices so that if the last 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // iteration destroys the AlarmList, I'm not about to use the end 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // iterator that the destruction invalidates. 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = cur_extension->second.size(); i > 0; --i) { 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmList::iterator cur_alarm = cur_extension->second.begin() + i - 1; 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (base::Time::FromJsTime(cur_alarm->js_alarm->scheduled_time) <= 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll_time_) { 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OnAlarm(make_pair(cur_extension, cur_alarm)); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScheduleNextPoll(); 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::Observe( 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int type, 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const content::NotificationSource& source, 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const content::NotificationDetails& details) { 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (type) { 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case chrome::NOTIFICATION_EXTENSION_LOADED: { 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Extension* extension = 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::Details<const Extension>(details).ptr(); 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (storage) { 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->GetExtensionValue(extension->id(), kRegisteredAlarms, 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&AlarmManager::ReadFromStorage, 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AsWeakPtr(), extension->id())); 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// AlarmManager::Alarm 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::Alarm() 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : js_alarm(new api::alarms::Alarm()) { 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::Alarm(const std::string& name, 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const api::alarms::AlarmCreateInfo& create_info, 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta min_granularity, 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Time now) 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : js_alarm(new api::alarms::Alarm()) { 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->name = name; 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (create_info.when.get()) { 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Absolute scheduling. 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->scheduled_time = *create_info.when; 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) granularity = base::Time::FromJsTime(js_alarm->scheduled_time) - now; 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Relative scheduling. 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) double* delay_in_minutes = create_info.delay_in_minutes.get(); 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delay_in_minutes == NULL) 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay_in_minutes = create_info.period_in_minutes.get(); 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(delay_in_minutes != NULL) 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "ValidateAlarmCreateInfo in alarms_api.cc should have " 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "prevented this call."; 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta delay = TimeDeltaFromDelay(*delay_in_minutes); 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) js_alarm->scheduled_time = (now + delay).ToJsTime(); 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) granularity = delay; 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (granularity < min_granularity) 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) granularity = min_granularity; 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check for repetition. 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (create_info.period_in_minutes.get()) { 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->period_in_minutes.reset( 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new double(*create_info.period_in_minutes)); 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::~Alarm() { 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 397