report/aaTranscript.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) 2007 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 "nsComponentManagerUtils.h"
  26 #include "nsIArray.h"
  27 #include "nsStringAPI.h"
  28
  29 /* Unfrozen API */
  30 #include "mozIStorageConnection.h"
  31
  32 /* Project includes */
  33 #include "nsStringUtils.h"
  34 #include "aaIFlow.h"
  35 #include "aaIBalance.h"
  36 #include "aaIFilter.h"
  37 #include "aaIStateFilter.h"
  38 #include "aaIDiffFilter.h"
  39 #include "aaBaseLoaders.h"
  40 #include "aaAccountLoaders.h"
  41 #include "aaStorageUtils.h"
  42 #include "aaTranscriptCounter.h"
  43 #include "aaDiffCounter.h"
  44
  45 #include "aaTranscript.h"
  46
  47 aaTranscript::aaTranscript(nsISupports *aOuter)
  48 {
  49   nsCOMPtr<mozIStorageConnection> connection = do_QueryInterface(aOuter);
  50
  51   mOpeningBalance = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  52   mOpeningIncome = do_CreateInstance(AA_LOADINCOME_CONTRACT_ID, connection);
  53   mOpening = mOpeningBalance;
  54   if (!mOpening)
  55     return;
  56
  57   mClosingBalance = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  58   mClosingIncome = do_CreateInstance(AA_LOADINCOME_CONTRACT_ID, connection);
  59   mClosing = mClosingBalance;
  60   if (!mClosing)
  61     return;
  62
  63   mOpeningFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  64   if (!mOpeningFilter)
  65     return;
  66
  67   mClosingFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  68   if (!mClosingFilter)
  69     return;
  70
  71   mDiffFilter = do_CreateInstance(AA_DIFFFILTER_CONTRACT_ID);
  72   if (!mDiffFilter)
  73     return;
  74
  75   mList = do_CreateInstance(AA_LOADFACTLIST_CONTRACT_ID, connection);
  76   mListInternal = do_QueryInterface(mList);
  77   if (!mListInternal)
  78     return;
  79
  80   mDiffList = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  81   mDiffListInternal = do_QueryInterface(mDiffList);
  82   if (!mDiffListInternal)
  83     return;
  84
  85   mConnection = connection;
  86 }
  87
  88 aaTranscript::~aaTranscript()
  89 {
  90 }
  91
  92 NS_IMPL_ISUPPORTS4(aaTranscript,
  93                    nsIArray,
  94                    aaILoadQuery,
  95                    aaIFilter,
  96                    aaITranscript)
  97
  98 /* aaILoadQuery */
  99 NS_IMETHODIMP
 100 aaTranscript::Load()
 101 {
 102   NS_ENSURE_TRUE(mFlow, NS_ERROR_NOT_INITIALIZED);
 103   nsresult rv;
 104   if (mFlow->PickId()) {
 105     mOpening = mOpeningBalance;
 106     mClosing = mClosingBalance;
 107   } else {
 108     mOpening = mOpeningIncome;
 109     mClosing = mClosingIncome;
 110   }
 111   mOpeningFilter->SetFlow( mFlow );
 112   mOpeningFilter->SetDate( mStart );
 113   mOpening->SetFilter( mOpeningFilter );
 114
 115   mClosingFilter->SetFlow( mFlow );
 116   mClosingFilter->SetDate( mEnd );
 117   mClosing->SetFilter( mClosingFilter );
 118
 119   rv = mOpening->Load();
 120   NS_ENSURE_SUCCESS(rv, rv);
 121
 122   rv = mClosing->Load();
 123   NS_ENSURE_SUCCESS(rv, rv);
 124
 125   mDiffFilter->SetFlow(mFlow);
 126   mDiffFilter->SetStartDate(mStart);
 127   mDiffFilter->SetEndDate(mEnd);
 128   mDiffList->SetFilter(mDiffFilter);
 129
 130   aaDiffCounter diffCounter;
 131   mDiffListInternal->SetLoadObserver(&diffCounter);
 132   rv = mDiffList->Load();
 133   NS_ENSURE_SUCCESS(rv, rv);
 134
 135   aaTranscriptCounter counter(mFlow->PickId());
 136   mListInternal->SetLoadObserver(&counter);
 137   mList->SetFilter(this);
 138   rv = mList->Load();
 139   mList->SetFilter(nsnull);
 140   NS_ENSURE_SUCCESS(rv, rv);
 141   mTotalDebits = counter.GetDebits();
 142   mTotalCredits = counter.GetCredits();
 143   mTotalDebitsValue = counter.GetDebitsValue() + diffCounter.GetDebits();
 144   mTotalCreditsValue = counter.GetCreditsValue() + diffCounter.GetCredits();
 145   nsCOMPtr<aaIBalance> balance;
 146   GetOpening(getter_AddRefs(balance));
 147   if (balance && balance->PickSide()) {
 148     mTotalDebitsValue -= balance->PickDiff();
 149   } else if (balance) {
 150     mTotalCredits -= balance->PickDiff();
 151   }
 152   if (balance && balance->PickSide()) {
 153     mTotalDebitsValue += balance->PickDiff();
 154   } else if (balance) {
 155     mTotalCredits += balance->PickDiff();
 156   }
 157
 158   mLoadedFlow = mFlow;
 159   return NS_OK;
 160 }
 161
 162 NS_IMETHODIMP
 163 aaTranscript::GetFilter(aaIFilter * * aFilter)
 164 {
 165   return NS_ERROR_NOT_IMPLEMENTED;
 166 }
 167 NS_IMETHODIMP
 168 aaTranscript::SetFilter(aaIFilter * aFilter)
 169 {
 170   return NS_ERROR_NOT_IMPLEMENTED;
 171 }
 172 aaIFilter *
 173 aaTranscript::PickFilter()
 174 {
 175   return nsnull;
 176 }
 177
 178 /* aaITranscript */
 179 NS_IMETHODIMP
 180 aaTranscript::GetFlow(aaIFlow * *aFlow)
 181 {
 182   NS_ENSURE_ARG_POINTER(aFlow);
 183   NS_IF_ADDREF(*aFlow = mFlow);
 184   return NS_OK;
 185 }
 186 NS_IMETHODIMP
 187 aaTranscript::SetFlow(aaIFlow * aFlow)
 188 {
 189   mFlow = aFlow;
 190   return NS_OK;
 191 }
 192 aaIFlow *
 193 aaTranscript::PickLoadedFlow()
 194 {
 195   return mLoadedFlow;
 196 }
 197
 198 NS_IMETHODIMP
 199 aaTranscript::GetStart(PRTime *aStart)
 200 {
 201   NS_ENSURE_ARG_POINTER(aStart);
 202   *aStart = mStart;
 203   return NS_OK;
 204 }
 205 NS_IMETHODIMP
 206 aaTranscript::SetStart(PRTime aStart)
 207 {
 208   mStart = aStart;
 209   return NS_OK;
 210 }
 211
 212 NS_IMETHODIMP
 213 aaTranscript::GetEnd(PRTime *aEnd)
 214 {
 215   NS_ENSURE_ARG_POINTER(aEnd);
 216   *aEnd = mEnd;
 217   return NS_OK;
 218 }
 219 NS_IMETHODIMP
 220 aaTranscript::SetEnd(PRTime aEnd)
 221 {
 222   mEnd = aEnd;
 223   return NS_OK;
 224 }
 225
 226 NS_IMETHODIMP
 227 aaTranscript::GetOpening(aaIBalance * *aOpen)
 228 {
 229   nsresult rv;
 230   NS_ENSURE_ARG_POINTER(aOpen);
 231   NS_ENSURE_TRUE(mOpening, NS_ERROR_UNEXPECTED);
 232
 233   PRUint32 count;
 234   rv = mOpening->GetLength(&count);
 235   NS_ENSURE_SUCCESS(rv, rv);
 236   if (count) {
 237     return mOpening->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aOpen);
 238   }
 239
 240   *aOpen = nsnull;
 241   return NS_OK;
 242 }
 243
 244 NS_IMETHODIMP
 245 aaTranscript::GetClosing(aaIBalance * *aClose)
 246 {
 247   nsresult rv;
 248   NS_ENSURE_ARG_POINTER(aClose);
 249   NS_ENSURE_TRUE(mClosing, NS_ERROR_UNEXPECTED);
 250
 251   PRUint32 count;
 252   rv = mClosing->GetLength(&count);
 253   NS_ENSURE_SUCCESS(rv, rv);
 254   if (count) {
 255     return mClosing->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aClose);
 256   }
 257
 258   *aClose = nsnull;
 259   return NS_OK;
 260 }
 261
 262 NS_IMETHODIMP
 263 aaTranscript::GetTotalDebits(double *aTotalDebits)
 264 {
 265   NS_ENSURE_ARG_POINTER(aTotalDebits);
 266
 267   *aTotalDebits = mTotalDebits;
 268   return NS_OK;
 269 }
 270
 271 NS_IMETHODIMP
 272 aaTranscript::GetTotalCredits(double *aTotalCredits)
 273 {
 274   NS_ENSURE_ARG_POINTER(aTotalCredits);
 275
 276   *aTotalCredits = mTotalCredits;
 277   return NS_OK;
 278 }
 279
 280 NS_IMETHODIMP
 281 aaTranscript::GetTotalDebitsValue(double *aTotalDebitsValue)
 282 {
 283   NS_ENSURE_ARG_POINTER(aTotalDebitsValue);
 284
 285   *aTotalDebitsValue = mTotalDebitsValue;
 286   return NS_OK;
 287 }
 288
 289 NS_IMETHODIMP
 290 aaTranscript::GetTotalCreditsValue(double *aTotalCreditsValue)
 291 {
 292   NS_ENSURE_ARG_POINTER(aTotalCreditsValue);
 293
 294   *aTotalCreditsValue = mTotalCreditsValue;
 295   return NS_OK;
 296 }
 297
 298 /* aaIFilter */
 299 NS_IMETHODIMP
 300 aaTranscript::GetExpression(nsACString & aExpression)
 301 {
 302   NS_ENSURE_TRUE(mFlow, NS_ERROR_NOT_INITIALIZED);
 303
 304   if (mFlow->PickId()) {
 305     aExpression.Assign(" WHERE txn.day>");
 306     AppendInt(aExpression, aa_julian_day(mOpeningFilter->PickDate()));
 307     aExpression.Append(" AND txn.day<=");
 308     AppendInt(aExpression, aa_julian_day(mClosingFilter->PickDate()));
 309     aExpression.Append(" AND (toF.flow_id=");
 310     AppendInt(aExpression, mOpeningFilter->PickFlow()->PickId());
 311     aExpression.Append(" OR fromF.flow_id=");
 312     AppendInt(aExpression, mOpeningFilter->PickFlow()->PickId());
 313     aExpression.Append(")");
 314   } else {
 315     aExpression.Assign(" WHERE txn.day>");
 316     AppendInt(aExpression, aa_julian_day(mOpeningFilter->PickDate()));
 317     aExpression.Append(" AND txn.day<=");
 318     AppendInt(aExpression, aa_julian_day(mClosingFilter->PickDate()));
 319     aExpression.Append(" AND txn.status=1");
 320   }
 321   return NS_OK;
 322 }