report/aaTranscriptLoader.cpp

   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: */
   3 /*
   4  * Copyright (C) 2008 Sergey Yanovich <ynvich@gmail.com>
   5  *
   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.
  10  *
  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.
  15  *
  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.
  20  */
  21
  22 #include "xpcom-config.h"
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsIArray.h"
  26 #include "nsComponentManagerUtils.h"
  27 #include "nsStringAPI.h"
  28
  29 /* Unfrozen API */
  30 #include "nsIInterfaceRequestorUtils.h"
  31
  32 /* Project includes */
  33 #include "aaITimeFrame.h"
  34 #include "aaIFlow.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"
  45
  46 aaTranscriptLoader::aaTranscriptLoader()
  47 {
  48 }
  49
  50 NS_IMPL_ISUPPORTS1(aaTranscriptLoader,
  51                    aaISqlTransaction)
  52
  53 /* aaISqlTransaction */
  54 NS_IMETHODIMP
  55 aaTranscriptLoader::SetParam(nsISupports *aParam)
  56 {
  57   nsresult rv;
  58   NS_ENSURE_ARG_POINTER(aParam);
  59   mParam = do_QueryInterface(aParam, &rv);
  60   if (!mParam)
  61     mParam = do_GetInterface(aParam, &rv);
  62   return rv;
  63 }
  64
  65 NS_IMETHODIMP
  66 aaTranscriptLoader::Execute(aaISqliteChannel *aChannel, nsISupports * * _retval)
  67 {
  68   NS_ENSURE_ARG_POINTER(_retval);
  69   NS_ENSURE_ARG_POINTER(aChannel);
  70   NS_ENSURE_TRUE(mParam, NS_ERROR_NOT_INITIALIZED);
  71
  72   nsresult rv;
  73   rv = checkState();
  74   NS_ENSURE_SUCCESS(rv, rv);
  75
  76   nsCOMPtr<aaTranscript> retval = new aaTranscript(mParam);
  77 /*
  78   rv = loadOpening(aChannel, retval);
  79   NS_ENSURE_SUCCESS(rv, rv);
  80
  81   rv = loadClosing(aChannel, retval);
  82   NS_ENSURE_SUCCESS(rv, rv);
  83 */
  84   rv = loadList(aChannel, retval);
  85   NS_ENSURE_SUCCESS(rv, rv);
  86
  87   rv = loadDiffs(aChannel, retval);
  88   NS_ENSURE_SUCCESS(rv, rv);
  89
  90   return CallQueryInterface(retval, _retval);
  91 }
  92
  93 /* private methods */
  94 nsresult
  95 aaTranscriptLoader::checkState()
  96 {
  97   nsresult rv;
  98   if (!mBalanceFilter){
  99     mBalanceFilter = do_CreateInstance(AA_SQLFILTER_CONTRACT, &rv);
 100     NS_ENSURE_SUCCESS(rv, rv);
 101   }
 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);
 107   }
 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);
 113   }
 114   if (!mListFilter){
 115     mListFilter = do_CreateInstance(AA_SQLFILTER_CONTRACT, &rv);
 116     NS_ENSURE_SUCCESS(rv, rv);
 117   }
 118   if (!mListLoader) {
 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);
 123   }
 124   return NS_OK;
 125 }
 126
 127 nsresult
 128 aaTranscriptLoader::loadList(aaISqliteChannel *aChannel, aaTranscript *retval)
 129 {
 130   nsresult rv;
 131
 132   nsCOMPtr<aaIFlow> flow;
 133   rv = mParam->GetFlow(getter_AddRefs(flow));
 134   NS_ENSURE_SUCCESS(rv, rv);
 135
 136   PRTime start, end;
 137   rv = mParam->GetStart(&start);
 138   NS_ENSURE_SUCCESS(rv, rv);
 139   rv = mParam->GetEnd(&end);
 140   NS_ENSURE_SUCCESS(rv, rv);
 141
 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);
 149
 150   rv = mListFilter->SetInterface(NS_GET_IID(aaIFlow), flow);
 151   NS_ENSURE_SUCCESS(rv, rv);
 152
 153   rv = mListFilter->SetInterface(NS_GET_IID(aaITimeFrame), frame);
 154   NS_ENSURE_SUCCESS(rv, rv);
 155
 156   nsCOMPtr<aaILoadObserver> counter = new aaTranscriptCounter(retval);
 157   rv = aChannel->Open(mListLoader, counter);
 158   NS_ENSURE_SUCCESS(rv, rv);
 159   return NS_OK;
 160 }
 161
 162 nsresult
 163 aaTranscriptLoader::loadDiffs(aaISqliteChannel *aChannel, aaTranscript *retval)
 164 {
 165   nsresult rv;
 166
 167   nsCOMPtr<aaIFlow> flow;
 168   rv = mParam->GetFlow(getter_AddRefs(flow));
 169   NS_ENSURE_SUCCESS(rv, rv);
 170
 171   PRTime start, end;
 172   rv = mParam->GetStart(&start);
 173   NS_ENSURE_SUCCESS(rv, rv);
 174   rv = mParam->GetEnd(&end);
 175   NS_ENSURE_SUCCESS(rv, rv);
 176
 177   nsCOMPtr<aaITimeFrame> frame
 178     = do_CreateInstance(AA_TIMEFRAME_CONTRACT, &rv);
 179   NS_ENSURE_SUCCESS(rv, rv);
 180   PRExplodedTime tm;
 181   PR_ExplodeTime(start, &PR_GMTParameters, &tm);
 182   tm.tm_mday -= 1;
 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);
 188
 189   rv = mBalanceFilter->SetInterface(NS_GET_IID(aaIFlow), flow);
 190   NS_ENSURE_SUCCESS(rv, rv);
 191
 192   rv = mBalanceFilter->SetInterface(NS_GET_IID(aaITimeFrame), frame);
 193   NS_ENSURE_SUCCESS(rv, rv);
 194
 195   nsCOMPtr<aaILoadObserver> result = new aaTranscriptDiffCounter(retval);
 196   rv = aChannel->Open(flow->PickId() ? mBalanceLoader : mIncomeLoader, result);
 197   NS_ENSURE_SUCCESS(rv, rv);
 198   return NS_OK;
 199 }