Fix NPE with initial user state
[aquarium] / src / main / scala / gr / grnet / aquarium / computation / state / parts / AgreementHistory.scala
1 /*
2  * Copyright 2011-2012 GRNET S.A. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or
5  * without modification, are permitted provided that the following
6  * conditions are met:
7  *
8  *   1. Redistributions of source code must retain the above
9  *      copyright notice, this list of conditions and the following
10  *      disclaimer.
11  *
12  *   2. Redistributions in binary form must reproduce the above
13  *      copyright notice, this list of conditions and the following
14  *      disclaimer in the documentation and/or other materials
15  *      provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28  * POSSIBILITY OF SUCH DAMAGE.
29  *
30  * The views and conclusions contained in the software and
31  * documentation are those of the authors and should not be
32  * interpreted as representing official policies, either expressed
33  * or implied, of GRNET S.A.
34  */
35
36 package gr.grnet.aquarium.computation
37 package state
38 package parts
39
40 import scala.collection.immutable
41 import gr.grnet.aquarium.logic.accounting.dsl.Timeslot
42 import gr.grnet.aquarium.policy.UserAgreementModel
43 import gr.grnet.aquarium.charging.state.WorkingAgreementHistory
44
45 /**
46  * The whole history of of a user's agreements.
47  *
48  * @author Christos KK Loverdos <loverdos@gmail.com>
49  */
50
51 case class AgreementHistory(agreements: List[UserAgreementModel]) {
52   def toWorkingAgreementHistory = {
53     (new WorkingAgreementHistory) ++ agreements
54   }
55
56   def agreementByTimeslot: immutable.SortedMap[Timeslot, UserAgreementModel] = {
57     immutable.TreeMap(agreements.map(ag ⇒ (ag.timeslot, ag)): _*)
58   }
59
60   /**
61    * Get the user agreement at the specified timestamp
62    */
63   def findForTime(at: Long): Option[UserAgreementModel] = {
64     // FIXME: Refactor and do not make this static call to Policy
65     None
66   }
67
68   /**
69    * Returns the first, chronologically, agreement.
70    */
71   def firstAgreement: Option[UserAgreementModel] = {
72     agreementByTimeslot.valuesIterator.toList.lastOption
73   }
74
75   /**
76    * Returns the last, chronologically, agreement.
77    */
78   def lastAgreement: Option[UserAgreementModel] = {
79     agreementByTimeslot.valuesIterator.toList.headOption
80   }
81 }
82
83 object AgreementHistory {
84   final val Empty = AgreementHistory(Nil)
85
86   def initial(userAgreement: UserAgreementModel): AgreementHistory ={
87     AgreementHistory(userAgreement :: Nil)
88   }
89 }