1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
4 * Copyright (C) 2008 Sergey Yanovich <ynvich@gmail.com>
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation; either version 2 of the
9 * License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
16 * You should have received a copy of the GNU General Public
17 * License along with this program; if not, write to the
18 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19 * Boston, MA 02111-1307, USA.
22 #include "xpcom-config.h"
26 #include "nsComponentManagerUtils.h"
27 #include "nsStringAPI.h"
30 #include "nsIInterfaceRequestorUtils.h"
32 /* Project includes */
33 #include "aaITimeFrame.h"
35 #include "aaBaseKeys.h"
36 #include "aaISqlRequest.h"
37 #include "aaISqlFilter.h"
38 #include "aaISqliteChannel.h"
39 #include "aaBaseLoaders.h"
40 #include "aaAccountLoaders.h"
41 #include "aaTranscript.h"
42 #include "aaTranscriptCounter.h"
43 #include "aaTranscriptDiffCounter.h"
44 #include "aaTranscriptLoader.h"
46 aaTranscriptLoader::aaTranscriptLoader()
50 NS_IMPL_ISUPPORTS1(aaTranscriptLoader
,
53 /* aaISqlTransaction */
55 aaTranscriptLoader::SetParam(nsISupports
*aParam
)
58 NS_ENSURE_ARG_POINTER(aParam
);
59 mParam
= do_QueryInterface(aParam
, &rv
);
61 mParam
= do_GetInterface(aParam
, &rv
);
66 aaTranscriptLoader::Execute(aaISqliteChannel
*aChannel
, nsISupports
* * _retval
)
68 NS_ENSURE_ARG_POINTER(_retval
);
69 NS_ENSURE_ARG_POINTER(aChannel
);
70 NS_ENSURE_TRUE(mParam
, NS_ERROR_NOT_INITIALIZED
);
74 NS_ENSURE_SUCCESS(rv
, rv
);
76 nsCOMPtr
<aaTranscript
> retval
= new aaTranscript(mParam
);
78 rv = loadOpening(aChannel, retval);
79 NS_ENSURE_SUCCESS(rv, rv);
81 rv = loadClosing(aChannel, retval);
82 NS_ENSURE_SUCCESS(rv, rv);
84 rv
= loadList(aChannel
, retval
);
85 NS_ENSURE_SUCCESS(rv
, rv
);
87 rv
= loadDiffs(aChannel
, retval
);
88 NS_ENSURE_SUCCESS(rv
, rv
);
90 return CallQueryInterface(retval
, _retval
);
95 aaTranscriptLoader::checkState()
99 mBalanceFilter
= do_CreateInstance(AA_SQLFILTER_CONTRACT
, &rv
);
100 NS_ENSURE_SUCCESS(rv
, rv
);
102 if (!mBalanceLoader
) {
103 mBalanceLoader
= do_CreateInstance(AA_LOADBALANCE_CONTRACT
, &rv
);
104 NS_ENSURE_SUCCESS(rv
, rv
);
105 rv
= mBalanceLoader
->SetFilter(mBalanceFilter
);
106 NS_ENSURE_SUCCESS(rv
, rv
);
108 if (!mIncomeLoader
) {
109 mIncomeLoader
= do_CreateInstance(AA_LOADINCOME_CONTRACT
, &rv
);
110 NS_ENSURE_SUCCESS(rv
, rv
);
111 rv
= mIncomeLoader
->SetFilter(mBalanceFilter
);
112 NS_ENSURE_SUCCESS(rv
, rv
);
115 mListFilter
= do_CreateInstance(AA_SQLFILTER_CONTRACT
, &rv
);
116 NS_ENSURE_SUCCESS(rv
, rv
);
119 mListLoader
= do_CreateInstance(AA_LOADFACTLIST_CONTRACT
, &rv
);
120 NS_ENSURE_SUCCESS(rv
, rv
);
121 rv
= mListLoader
->SetFilter(mListFilter
);
122 NS_ENSURE_SUCCESS(rv
, rv
);
128 aaTranscriptLoader::loadList(aaISqliteChannel
*aChannel
, aaTranscript
*retval
)
132 nsCOMPtr
<aaIFlow
> flow
;
133 rv
= mParam
->GetFlow(getter_AddRefs(flow
));
134 NS_ENSURE_SUCCESS(rv
, rv
);
137 rv
= mParam
->GetStart(&start
);
138 NS_ENSURE_SUCCESS(rv
, rv
);
139 rv
= mParam
->GetEnd(&end
);
140 NS_ENSURE_SUCCESS(rv
, rv
);
142 nsCOMPtr
<aaITimeFrame
> frame
143 = do_CreateInstance(AA_TIMEFRAME_CONTRACT
, &rv
);
144 NS_ENSURE_SUCCESS(rv
, rv
);
145 rv
= frame
->SetStart(start
);
146 NS_ENSURE_SUCCESS(rv
, rv
);
147 rv
= frame
->SetEnd(end
);
148 NS_ENSURE_SUCCESS(rv
, rv
);
150 rv
= mListFilter
->SetInterface(NS_GET_IID(aaIFlow
), flow
);
151 NS_ENSURE_SUCCESS(rv
, rv
);
153 rv
= mListFilter
->SetInterface(NS_GET_IID(aaITimeFrame
), frame
);
154 NS_ENSURE_SUCCESS(rv
, rv
);
156 nsCOMPtr
<aaILoadObserver
> counter
= new aaTranscriptCounter(retval
);
157 rv
= aChannel
->Open(mListLoader
, counter
);
158 NS_ENSURE_SUCCESS(rv
, rv
);
163 aaTranscriptLoader::loadDiffs(aaISqliteChannel
*aChannel
, aaTranscript
*retval
)
167 nsCOMPtr
<aaIFlow
> flow
;
168 rv
= mParam
->GetFlow(getter_AddRefs(flow
));
169 NS_ENSURE_SUCCESS(rv
, rv
);
172 rv
= mParam
->GetStart(&start
);
173 NS_ENSURE_SUCCESS(rv
, rv
);
174 rv
= mParam
->GetEnd(&end
);
175 NS_ENSURE_SUCCESS(rv
, rv
);
177 nsCOMPtr
<aaITimeFrame
> frame
178 = do_CreateInstance(AA_TIMEFRAME_CONTRACT
, &rv
);
179 NS_ENSURE_SUCCESS(rv
, rv
);
181 PR_ExplodeTime(start
, &PR_GMTParameters
, &tm
);
183 PR_NormalizeTime(&tm
, &PR_GMTParameters
);
184 rv
= frame
->SetStart(PR_ImplodeTime(&tm
));
185 NS_ENSURE_SUCCESS(rv
, rv
);
186 rv
= frame
->SetEnd(end
);
187 NS_ENSURE_SUCCESS(rv
, rv
);
189 rv
= mBalanceFilter
->SetInterface(NS_GET_IID(aaIFlow
), flow
);
190 NS_ENSURE_SUCCESS(rv
, rv
);
192 rv
= mBalanceFilter
->SetInterface(NS_GET_IID(aaITimeFrame
), frame
);
193 NS_ENSURE_SUCCESS(rv
, rv
);
195 nsCOMPtr
<aaILoadObserver
> result
= new aaTranscriptDiffCounter(retval
);
196 rv
= aChannel
->Open(flow
->PickId() ? mBalanceLoader
: mIncomeLoader
, result
);
197 NS_ENSURE_SUCCESS(rv
, rv
);