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 "aaBaseLoaders.h"
  39 #include "aaAccountLoaders.h"
  40 #include "aaStorageUtils.h"
  41 #include "aaTranscript.h"
  42
  43 aaTranscript::aaTranscript(nsISupports *aOuter)
  44 {
  45   nsCOMPtr<mozIStorageConnection> connection = do_QueryInterface(aOuter);
  46
  47   mOpeningBalance = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  48   mOpeningIncome = do_CreateInstance(AA_LOADINCOME_CONTRACT_ID, connection);
  49   mOpening = mOpeningBalance;
  50   if (!mOpening)
  51     return;
  52
  53   mClosingBalance = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  54   mClosingIncome = do_CreateInstance(AA_LOADINCOME_CONTRACT_ID, connection);
  55   mClosing = mClosingBalance;
  56   if (!mClosing)
  57     return;
  58
  59   mOpeningFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  60   if (!mOpeningFilter)
  61     return;
  62
  63   mClosingFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  64   if (!mClosingFilter)
  65     return;
  66
  67   mList = do_CreateInstance(AA_LOADFACTLIST_CONTRACT_ID, connection);
  68   if (!mList)
  69     return;
  70
  71   mConnection = connection;
  72 }
  73
  74 aaTranscript::~aaTranscript()
  75 {
  76 }
  77
  78 NS_IMPL_ISUPPORTS4(aaTranscript,
  79                    nsIArray,
  80                    aaILoadQuery,
  81                    aaIFilter,
  82                    aaITranscript)
  83
  84 /* aaILoadQuery */
  85 NS_IMETHODIMP
  86 aaTranscript::Load()
  87 {
  88   NS_ENSURE_TRUE(mFlow, NS_ERROR_NOT_INITIALIZED);
  89   nsresult rv;
  90   if (mFlow->PickId()) {
  91     mOpening = mOpeningBalance;
  92     mClosing = mClosingBalance;
  93   } else {
  94     mOpening = mOpeningIncome;
  95     mClosing = mClosingIncome;
  96   }
  97   mOpeningFilter->SetFlow( mFlow );
  98   mOpeningFilter->SetDate( mStart );
  99   mOpening->SetFilter( mOpeningFilter );
 100
 101   mClosingFilter->SetFlow( mFlow );
 102   mClosingFilter->SetDate( mEnd );
 103   mClosing->SetFilter( mClosingFilter );
 104
 105
 106   rv = mOpening->Load();
 107   NS_ENSURE_SUCCESS(rv, rv);
 108
 109   rv = mClosing->Load();
 110   NS_ENSURE_SUCCESS(rv, rv);
 111
 112   mList->SetFilter(this);
 113   rv = mList->Load();
 114   mList->SetFilter(nsnull);
 115   NS_ENSURE_SUCCESS(rv, rv);
 116
 117   mLoadedFlow = mFlow;
 118   return NS_OK;
 119 }
 120
 121 NS_IMETHODIMP
 122 aaTranscript::GetFilter(aaIFilter * * aFilter)
 123 {
 124   return NS_ERROR_NOT_IMPLEMENTED;
 125 }
 126 NS_IMETHODIMP
 127 aaTranscript::SetFilter(aaIFilter * aFilter)
 128 {
 129   return NS_ERROR_NOT_IMPLEMENTED;
 130 }
 131 aaIFilter *
 132 aaTranscript::PickFilter()
 133 {
 134   return nsnull;
 135 }
 136
 137 /* aaITranscript */
 138 NS_IMETHODIMP
 139 aaTranscript::GetFlow(aaIFlow * *aFlow)
 140 {
 141   NS_ENSURE_ARG_POINTER(aFlow);
 142   NS_IF_ADDREF(*aFlow = mFlow);
 143   return NS_OK;
 144 }
 145 NS_IMETHODIMP
 146 aaTranscript::SetFlow(aaIFlow * aFlow)
 147 {
 148   mFlow = aFlow;
 149   return NS_OK;
 150 }
 151 aaIFlow *
 152 aaTranscript::PickLoadedFlow()
 153 {
 154   return mLoadedFlow;
 155 }
 156
 157 NS_IMETHODIMP
 158 aaTranscript::GetStart(PRTime *aStart)
 159 {
 160   NS_ENSURE_ARG_POINTER(aStart);
 161   *aStart = mStart;
 162   return NS_OK;
 163 }
 164 NS_IMETHODIMP
 165 aaTranscript::SetStart(PRTime aStart)
 166 {
 167   mStart = aStart;
 168   return NS_OK;
 169 }
 170
 171 NS_IMETHODIMP
 172 aaTranscript::GetEnd(PRTime *aEnd)
 173 {
 174   NS_ENSURE_ARG_POINTER(aEnd);
 175   *aEnd = mEnd;
 176   return NS_OK;
 177 }
 178 NS_IMETHODIMP
 179 aaTranscript::SetEnd(PRTime aEnd)
 180 {
 181   mEnd = aEnd;
 182   return NS_OK;
 183 }
 184
 185 NS_IMETHODIMP
 186 aaTranscript::GetOpening(aaIBalance * *aOpen)
 187 {
 188   nsresult rv;
 189   NS_ENSURE_ARG_POINTER(aOpen);
 190   NS_ENSURE_TRUE(mOpening, NS_ERROR_UNEXPECTED);
 191
 192   PRUint32 count;
 193   rv = mOpening->GetLength(&count);
 194   NS_ENSURE_SUCCESS(rv, rv);
 195   if (count) {
 196     return mOpening->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aOpen);
 197   }
 198
 199   *aOpen = nsnull;
 200   return NS_OK;
 201 }
 202
 203 NS_IMETHODIMP
 204 aaTranscript::GetClosing(aaIBalance * *aClose)
 205 {
 206   nsresult rv;
 207   NS_ENSURE_ARG_POINTER(aClose);
 208   NS_ENSURE_TRUE(mClosing, NS_ERROR_UNEXPECTED);
 209
 210   PRUint32 count;
 211   rv = mClosing->GetLength(&count);
 212   NS_ENSURE_SUCCESS(rv, rv);
 213   if (count) {
 214     return mClosing->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aClose);
 215   }
 216
 217   *aClose = nsnull;
 218   return NS_OK;
 219 }
 220
 221 /* aaIFilter */
 222 NS_IMETHODIMP
 223 aaTranscript::GetExpression(nsACString & aExpression)
 224 {
 225   NS_ENSURE_TRUE(mFlow, NS_ERROR_NOT_INITIALIZED);
 226
 227   if (mFlow->PickId()) {
 228     aExpression.Assign(" WHERE txn.day>");
 229     AppendInt(aExpression, aa_julian_day(mOpeningFilter->PickDate()));
 230     aExpression.Append(" AND txn.day<=");
 231     AppendInt(aExpression, aa_julian_day(mClosingFilter->PickDate()));
 232     aExpression.Append(" AND (toF.flow_id=");
 233     AppendInt(aExpression, mOpeningFilter->PickFlow()->PickId());
 234     aExpression.Append(" OR fromF.flow_id=");
 235     AppendInt(aExpression, mOpeningFilter->PickFlow()->PickId());
 236     aExpression.Append(")");
 237   } else {
 238     aExpression.Assign(" WHERE txn.day>");
 239     AppendInt(aExpression, aa_julian_day(mOpeningFilter->PickDate()));
 240     aExpression.Append(" AND txn.day<=");
 241     AppendInt(aExpression, aa_julian_day(mClosingFilter->PickDate()));
 242     aExpression.Append(" AND txn.status=1");
 243   }
 244   return NS_OK;
 245 }