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 <abstract/aacore.h>
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsComponentManagerUtils.h"
  26 #include "nsIArray.h"
  27 #include "nsStringAPI.h"
  28
  29 /* Unfrozen API */
  30 #include "unstable/mozIStorageConnection.h"
  31
  32 /* Project includes */
  33 #include <abstract/base/aaIFlow.h>
  34 #include <abstract/base/aaIBalance.h>
  35 #include <abstract/storage/aaIFilter.h>
  36 #include <abstract/storage/aaIStateFilter.h>
  37 #include <abstract/storage/aaBaseLoaders.h>
  38 #include <abstract/storage/aaAccountLoaders.h>
  39 #include <abstract/report/aaITranscript.h>
  40 #include <abstract/storage/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   mList->SetFilter(this);
 106
 107   rv = mOpening->Load();
 108   NS_ENSURE_SUCCESS(rv, rv);
 109
 110   rv = mClosing->Load();
 111   NS_ENSURE_SUCCESS(rv, rv);
 112
 113   rv = mList->Load();
 114   NS_ENSURE_SUCCESS(rv, rv);
 115
 116   mLoadedFlow = mFlow;
 117   return NS_OK;
 118 }
 119
 120 NS_IMETHODIMP
 121 aaTranscript::GetFilter(aaIFilter * * aFilter)
 122 {
 123   return NS_ERROR_NOT_IMPLEMENTED;
 124 }
 125 NS_IMETHODIMP
 126 aaTranscript::SetFilter(aaIFilter * aFilter)
 127 {
 128   return NS_ERROR_NOT_IMPLEMENTED;
 129 }
 130 aaIFilter *
 131 aaTranscript::PickFilter()
 132 {
 133   return nsnull;
 134 }
 135
 136 /* aaITranscript */
 137 NS_IMETHODIMP
 138 aaTranscript::GetFlow(aaIFlow * *aFlow)
 139 {
 140   NS_ENSURE_ARG_POINTER(aFlow);
 141   NS_IF_ADDREF(*aFlow = mFlow);
 142   return NS_OK;
 143 }
 144 NS_IMETHODIMP
 145 aaTranscript::SetFlow(aaIFlow * aFlow)
 146 {
 147   mFlow = aFlow;
 148   return NS_OK;
 149 }
 150 aaIFlow *
 151 aaTranscript::PickLoadedFlow()
 152 {
 153   return mLoadedFlow;
 154 }
 155
 156 NS_IMETHODIMP
 157 aaTranscript::GetStart(PRTime *aStart)
 158 {
 159   NS_ENSURE_ARG_POINTER(aStart);
 160   *aStart = mStart;
 161   return NS_OK;
 162 }
 163 NS_IMETHODIMP
 164 aaTranscript::SetStart(PRTime aStart)
 165 {
 166   mStart = aStart;
 167   return NS_OK;
 168 }
 169
 170 NS_IMETHODIMP
 171 aaTranscript::GetEnd(PRTime *aEnd)
 172 {
 173   NS_ENSURE_ARG_POINTER(aEnd);
 174   *aEnd = mEnd;
 175   return NS_OK;
 176 }
 177 NS_IMETHODIMP
 178 aaTranscript::SetEnd(PRTime aEnd)
 179 {
 180   mEnd = aEnd;
 181   return NS_OK;
 182 }
 183
 184 NS_IMETHODIMP
 185 aaTranscript::GetOpening(aaIBalance * *aOpen)
 186 {
 187   nsresult rv;
 188   NS_ENSURE_ARG_POINTER(aOpen);
 189   NS_ENSURE_TRUE(mOpening, NS_ERROR_UNEXPECTED);
 190
 191   PRUint32 count;
 192   rv = mOpening->GetLength(&count);
 193   NS_ENSURE_SUCCESS(rv, rv);
 194   if (count) {
 195     return mOpening->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aOpen);
 196   }
 197
 198   *aOpen = nsnull;
 199   return NS_OK;
 200 }
 201
 202 NS_IMETHODIMP
 203 aaTranscript::GetClosing(aaIBalance * *aClose)
 204 {
 205   nsresult rv;
 206   NS_ENSURE_ARG_POINTER(aClose);
 207   NS_ENSURE_TRUE(mClosing, NS_ERROR_UNEXPECTED);
 208
 209   PRUint32 count;
 210   rv = mClosing->GetLength(&count);
 211   NS_ENSURE_SUCCESS(rv, rv);
 212   if (count) {
 213     return mClosing->QueryElementAt(0, NS_GET_IID(aaIBalance), (void **) aClose);
 214   }
 215
 216   *aClose = nsnull;
 217   return NS_OK;
 218 }
 219
 220 /* aaIFilter */
 221 NS_IMETHODIMP
 222 aaTranscript::GetExpression(nsACString & aExpression)
 223 {
 224   NS_ENSURE_TRUE(mFlow, NS_ERROR_NOT_INITIALIZED);
 225
 226   if (mFlow->PickId()) {
 227     aExpression.Assign(" WHERE txn.day>");
 228     aExpression.AppendInt(aa_julian_day(mOpeningFilter->PickDate()));
 229     aExpression.Append(" AND txn.day<=");
 230     aExpression.AppendInt(aa_julian_day(mClosingFilter->PickDate()));
 231     aExpression.Append(" AND (toF.flow_id=");
 232     aExpression.AppendInt(mOpeningFilter->PickFlow()->PickId());
 233     aExpression.Append(" OR fromF.flow_id=");
 234     aExpression.AppendInt(mOpeningFilter->PickFlow()->PickId());
 235     aExpression.Append(")");
 236   } else {
 237     aExpression.Assign(" WHERE txn.day>");
 238     aExpression.AppendInt(aa_julian_day(mOpeningFilter->PickDate()));
 239     aExpression.Append(" AND txn.day<=");
 240     aExpression.AppendInt(aa_julian_day(mClosingFilter->PickDate()));
 241     aExpression.Append(" AND txn.status=1");
 242   }
 243   return NS_OK;
 244 }