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