alarm_manager.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
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" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/message_loop.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/time.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/time/clock.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/value_conversions.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/event_router.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_service.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_system.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/state_store.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_notification_types.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/notification_service.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kOnAlarmEvent[] = "alarms.onAlarm"; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A list of alarms that this extension has set. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kRegisteredAlarms[] = "alarms"; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kAlarmGranularity[] = "granularity"; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The minimum period between polling for alarms to run. 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const base::TimeDelta kDefaultMinPollPeriod = base::TimeDelta::FromDays(1); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DefaultAlarmDelegate : public AlarmManager::Delegate { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit DefaultAlarmDelegate(Profile* profile) : profile_(profile) {} 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~DefaultAlarmDelegate() {} 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnAlarm(const std::string& extension_id, 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const Alarm& alarm) OVERRIDE { 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<ListValue> args(new ListValue()); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) args->Append(alarm.js_alarm->ToValue().release()); 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<Event> event(new Event(kOnAlarmEvent, args.Pass())); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ExtensionSystem::Get(profile_)->event_router()->DispatchEventToExtension( 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) extension_id, event.Pass()); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Profile* profile_; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Creates a TimeDelta from a delay as specified in the API. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::TimeDelta TimeDeltaFromDelay(double delay_in_minutes) { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return base::TimeDelta::FromMicroseconds( 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay_in_minutes * base::Time::kMicrosecondsPerMinute); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::vector<Alarm> AlarmsFromValue(const base::ListValue* list) { 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<Alarm> alarms; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < list->GetSize(); ++i) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::DictionaryValue* alarm_dict = NULL; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Alarm alarm; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list->GetDictionary(i, &alarm_dict) && 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) api::alarms::Alarm::Populate(*alarm_dict, alarm.js_alarm.get())) { 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Value* time_value = NULL; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarm_dict->Get(kAlarmGranularity, &time_value)) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::GetValueAsTimeDelta(*time_value, &alarm.granularity); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.push_back(alarm); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return alarms; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_ptr<base::ListValue> AlarmsToValue(const std::vector<Alarm>& alarms) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::ListValue> list(new ListValue()); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < alarms.size(); ++i) { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::DictionaryValue> alarm = 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms[i].js_alarm->ToValue().Pass(); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm->Set(kAlarmGranularity, 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::CreateTimeDeltaValue(alarms[i].granularity)); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list->Append(alarm.release()); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return list.Pass(); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// AlarmManager 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)AlarmManager::AlarmManager(Profile* profile, base::Clock* clock) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : profile_(profile), 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) clock_(clock), 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate_(new DefaultAlarmDelegate(profile)) { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED, 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::Source<Profile>(profile_)); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (storage) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->RegisterKey(kRegisteredAlarms); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AlarmManager::~AlarmManager() { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::AddAlarm(const std::string& extension_id, 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Alarm& alarm) { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddAlarmImpl(extension_id, alarm); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const Alarm* AlarmManager::GetAlarm( 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id, const std::string& name) { 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator it = GetAlarmIterator(extension_id, name); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it.first == alarms_.end()) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &*it.second; 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const AlarmManager::AlarmList* AlarmManager::GetAllAlarms( 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id) { 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &list->second; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AlarmManager::AlarmIterator AlarmManager::GetAlarmIterator( 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& extension_id, const std::string& name) { 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(alarms_.end(), AlarmList::iterator()); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmList::iterator it = list->second.begin(); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != list->second.end(); ++it) { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it->js_alarm->name == name) 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(list, it); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return make_pair(alarms_.end(), AlarmList::iterator()); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AlarmManager::RemoveAlarm(const std::string& extension_id, 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& name) { 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator it = GetAlarmIterator(extension_id, name); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it.first == alarms_.end()) 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(it); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::RemoveAllAlarms(const std::string& extension_id) { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list == alarms_.end()) 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note: I'm using indices rather than iterators here because 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // RemoveAlarmIterator will delete the list when it becomes empty. 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0, size = list->second.size(); i < size; ++i) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(AlarmIterator(list, list->second.begin())); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(alarms_.find(extension_id) == alarms_.end()); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::RemoveAlarmIterator(const AlarmIterator& iter) { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmList& list = iter.first->second; 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list.erase(iter.second); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list.empty()) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_.erase(iter.first); 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel the timer if there are no more alarms. 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We don't need to reschedule the poll otherwise, because in 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the worst case we would just poll one extra time. 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarms_.empty()) 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::OnAlarm(AlarmIterator it) { 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(it.first != alarms_.end()); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Alarm& alarm = *it.second; 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string extension_id_copy(it.first->first); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate_->OnAlarm(extension_id_copy, alarm); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Update our scheduled time for the next alarm. 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (double* period_in_minutes = 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->period_in_minutes.get()) { 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->scheduled_time = 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (last_poll_time_ + 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TimeDeltaFromDelay(*period_in_minutes)).ToJsTime(); 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(it); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToStorage(extension_id_copy); 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::AddAlarmImpl(const std::string& extension_id, 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Alarm& alarm) { 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Override any old alarm with the same name. 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmIterator old_alarm = GetAlarmIterator(extension_id, 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarm.js_alarm->name); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (old_alarm.first != alarms_.end()) 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveAlarmIterator(old_alarm); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_[extension_id].push_back(alarm); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScheduleNextPoll(); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::WriteToStorage(const std::string& extension_id) { 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!storage) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<Value> alarms; 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator list = alarms_.find(extension_id); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (list != alarms_.end()) 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.reset(AlarmsToValue(list->second).release()); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms.reset(AlarmsToValue(std::vector<Alarm>()).release()); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->SetExtensionValue(extension_id, kRegisteredAlarms, alarms.Pass()); 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::ReadFromStorage(const std::string& extension_id, 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<base::Value> value) { 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::ListValue* list = NULL; 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!value.get() || !value->GetAsList(&list)) 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<Alarm> alarm_states = AlarmsFromValue(list); 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < alarm_states.size(); ++i) { 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddAlarmImpl(extension_id, alarm_states[i]); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void AlarmManager::ScheduleNextPoll() { 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // 0. If there are no alarms, stop the timer. 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (alarms_.empty()) { 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(yoz): Try not to reschedule every single time if we're adding 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a lot of alarms. 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Find the soonest alarm that is scheduled to run and the smallest 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // granularity of any alarm. 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alarms_ guarantees that none of its contained lists are empty. 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time soonest_alarm_time = base::Time::FromJsTime( 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) alarms_.begin()->second.begin()->js_alarm->scheduled_time); 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::TimeDelta min_granularity = kDefaultMinPollPeriod; 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmMap::const_iterator m_it = alarms_.begin(), m_end = alarms_.end(); 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) m_it != m_end; ++m_it) { 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmList::const_iterator l_it = m_it->second.begin(); 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) l_it != m_it->second.end(); ++l_it) { 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time cur_alarm_time = 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time::FromJsTime(l_it->js_alarm->scheduled_time); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cur_alarm_time < soonest_alarm_time) 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) soonest_alarm_time = cur_alarm_time; 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (l_it->granularity < min_granularity) 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) min_granularity = l_it->granularity; 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Time next_poll(last_poll_time_ + min_granularity); 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the next alarm is more than min_granularity in the future, wait for it. 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Otherwise, only poll as often as min_granularity. 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // As a special case, if we've never checked for an alarm before 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // (e.g. during startup), let alarms fire asap. 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (last_poll_time_.is_null() || next_poll < soonest_alarm_time) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll = soonest_alarm_time; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Schedule the poll. 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll_time_ = next_poll; 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta delay = std::max(base::TimeDelta::FromSeconds(0), 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) next_poll - clock_->Now()); 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Start(FROM_HERE, 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay, 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &AlarmManager::PollAlarms); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::PollAlarms() { 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) last_poll_time_ = clock_->Now(); 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Run any alarms scheduled in the past. OnAlarm uses vector::erase to remove 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // elements from the AlarmList, and map::erase to remove AlarmLists from the 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AlarmMap. 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (AlarmMap::iterator m_it = alarms_.begin(), m_end = alarms_.end(); 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) m_it != m_end;) { 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmMap::iterator cur_extension = m_it++; 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Iterate (a) backwards so that removing elements doesn't affect 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // upcoming iterations, and (b) with indices so that if the last 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // iteration destroys the AlarmList, I'm not about to use the end 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // iterator that the destruction invalidates. 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = cur_extension->second.size(); i > 0; --i) { 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AlarmList::iterator cur_alarm = cur_extension->second.begin() + i - 1; 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (base::Time::FromJsTime(cur_alarm->js_alarm->scheduled_time) <= 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) next_poll_time_) { 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OnAlarm(make_pair(cur_extension, cur_alarm)); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScheduleNextPoll(); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AlarmManager::Observe( 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int type, 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const content::NotificationSource& source, 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const content::NotificationDetails& details) { 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (type) { 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case chrome::NOTIFICATION_EXTENSION_LOADED: { 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Extension* extension = 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::Details<const Extension>(details).ptr(); 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StateStore* storage = ExtensionSystem::Get(profile_)->state_store(); 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (storage) { 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) storage->GetExtensionValue(extension->id(), kRegisteredAlarms, 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&AlarmManager::ReadFromStorage, 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AsWeakPtr(), extension->id())); 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// AlarmManager::Alarm 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::Alarm() 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : js_alarm(new api::alarms::Alarm()) { 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::Alarm(const std::string& name, 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const api::alarms::AlarmCreateInfo& create_info, 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta min_granularity, 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Time now) 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : js_alarm(new api::alarms::Alarm()) { 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->name = name; 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (create_info.when.get()) { 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Absolute scheduling. 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->scheduled_time = *create_info.when; 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) granularity = base::Time::FromJsTime(js_alarm->scheduled_time) - now; 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Relative scheduling. 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) double* delay_in_minutes = create_info.delay_in_minutes.get(); 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delay_in_minutes == NULL) 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delay_in_minutes = create_info.period_in_minutes.get(); 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(delay_in_minutes != NULL) 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "ValidateAlarmCreateInfo in alarms_api.cc should have " 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "prevented this call."; 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta delay = TimeDeltaFromDelay(*delay_in_minutes); 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) js_alarm->scheduled_time = (now + delay).ToJsTime(); 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) granularity = delay; 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (granularity < min_granularity) 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) granularity = min_granularity; 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check for repetition. 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (create_info.period_in_minutes.get()) { 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) js_alarm->period_in_minutes.reset( 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new double(*create_info.period_in_minutes)); 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)Alarm::~Alarm() { 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 378