report/aaBalanceSheet.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 #include "nsWeakReference.h"
  29
  30 /* Unfrozen API */
  31 #include "unstable/mozIStorageConnection.h"
  32
  33 /* Project includes */
  34 #include <abstract/base/aaIFlow.h>
  35 #include <abstract/storage/aaIFilter.h>
  36 #include <abstract/storage/aaIStateFilter.h>
  37 #include <abstract/storage/aaILoadQuery.h>
  38 #include <abstract/storage/aaBaseLoaders.h>
  39 #include <abstract/storage/aaAccountLoaders.h>
  40 #include <abstract/report/aaIBalanceSheet.h>
  41 #include "aaBalanceCounter.h"
  42 #include "aaBalanceSheet.h"
  43
  44 aaBalanceSheet::aaBalanceSheet(nsISupports *aOuter)
  45   :mBalanceCount(0), mIncomeCount(0), mHasDate(PR_FALSE)
  46   ,mAssets(0.0), mLiabilities(0.0)
  47 {
  48   nsCOMPtr<mozIStorageConnection> connection = do_QueryInterface(aOuter);
  49
  50   mBalances = do_CreateInstance(AA_LOADBALANCE_CONTRACT_ID, connection);
  51   mBalancesInternal = do_QueryInterface(mBalances);
  52   if (!mBalancesInternal)
  53     return;
  54
  55   mIncomes = do_CreateInstance(AA_LOADINCOME_CONTRACT_ID, connection);
  56   mIncomesInternal = do_QueryInterface(mIncomes);
  57   if (!mIncomesInternal)
  58     return;
  59
  60   mBalanceFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  61   if (!mBalanceFilter)
  62     return;
  63
  64   nsCOMPtr<aaIFlow> finres =do_CreateInstance("@aasii.org/base/income-flow;1");
  65   mIncomeFilter = do_CreateInstance(AA_STATEFILTER_CONTRACT_ID);
  66   mIncomeFilter->SetFlow(finres);
  67   if (!mIncomeFilter)
  68     return;
  69
  70   mConnection = connection;
  71 }
  72
  73 aaBalanceSheet::~aaBalanceSheet()
  74 {
  75 }
  76
  77 NS_IMPL_ISUPPORTS4(aaBalanceSheet,
  78                    nsIArray,
  79                    aaILoadQuery,
  80                    aaIBalanceSheet,
  81                    aaIFilter)
  82
  83 /* nsIArray */
  84 NS_IMETHODIMP
  85 aaBalanceSheet::GetLength(PRUint32 *aLength)
  86 {
  87   NS_ENSURE_ARG_POINTER(aLength);
  88   *aLength = mBalanceCount + mIncomeCount;
  89   return NS_OK;
  90 }
  91
  92 NS_IMETHODIMP
  93 aaBalanceSheet::QueryElementAt(PRUint32 index, const nsIID & uuid,
  94     void * *result)
  95 {
  96   if (index < mBalanceCount)
  97     return mBalances->QueryElementAt(index, uuid, result);
  98   else if (index < mBalanceCount + mIncomeCount)
  99     return mIncomes->QueryElementAt(index - mBalanceCount, uuid, result);
 100   return NS_ERROR_INVALID_ARG;
 101 }
 102
 103 NS_IMETHODIMP
 104 aaBalanceSheet::IndexOf(PRUint32 startIndex, nsISupports *element,
 105     PRUint32 *_retval)
 106 {
 107   if (startIndex < mBalanceCount)
 108     return mBalances->IndexOf(startIndex, element, _retval);
 109   else if (startIndex < mBalanceCount + mIncomeCount)
 110     return mIncomes->IndexOf(startIndex - mBalanceCount, element, _retval);
 111   return NS_ERROR_INVALID_ARG;
 112 }
 113
 114 NS_IMETHODIMP
 115 aaBalanceSheet::Enumerate(nsISimpleEnumerator **_retval)
 116 {
 117   return NS_ERROR_NOT_IMPLEMENTED;
 118 }
 119
 120 /* aaILoadQuery */
 121 NS_IMETHODIMP
 122 aaBalanceSheet::Load()
 123 {
 124   nsresult rv;
 125   if (mHasDate) {
 126     rv = mBalanceFilter->SetDate(mDate);
 127     NS_ENSURE_SUCCESS(rv, rv);
 128     rv = mIncomeFilter->SetDate(mDate);
 129     NS_ENSURE_SUCCESS(rv, rv);
 130     rv = mBalances->SetFilter(mBalanceFilter);
 131     NS_ENSURE_SUCCESS(rv, rv);
 132     rv = mIncomes->SetFilter(mIncomeFilter);
 133     NS_ENSURE_SUCCESS(rv, rv);
 134   }
 135   aaBalanceCounter counter;
 136   rv = mBalancesInternal->SetLoadObserver(&counter);
 137   NS_ENSURE_SUCCESS(rv, rv);
 138   rv = mBalances->Load();
 139   NS_ENSURE_SUCCESS(rv, rv);
 140   rv = mBalances->GetLength(&mBalanceCount);
 141   NS_ENSURE_SUCCESS(rv, rv);
 142   mAssets = counter.GetAssets();
 143   mLiabilities = counter.GetLiabilities();
 144
 145   counter.Reset();
 146   rv = mIncomesInternal->SetLoadObserver(&counter);
 147   NS_ENSURE_SUCCESS(rv, rv);
 148   rv = mIncomes->Load();
 149   NS_ENSURE_SUCCESS(rv, rv);
 150   rv = mIncomes->GetLength(&mIncomeCount);
 151   NS_ENSURE_SUCCESS(rv, rv);
 152   mAssets += counter.GetAssets();
 153   mLiabilities += counter.GetLiabilities();
 154
 155
 156   return NS_OK;
 157 }
 158
 159 NS_IMETHODIMP
 160 aaBalanceSheet::GetFilter(aaIFilter * * aFilter)
 161 {
 162   NS_ENSURE_ARG_POINTER(aFilter);
 163   NS_IF_ADDREF(*aFilter = this);
 164   return NS_OK;
 165 }
 166 NS_IMETHODIMP
 167 aaBalanceSheet::SetFilter(aaIFilter * aFilter)
 168 {
 169   return NS_ERROR_NOT_AVAILABLE;
 170 }
 171 aaIFilter *
 172 aaBalanceSheet::PickFilter()
 173 {
 174   return this;
 175 }
 176
 177 /* aaIFilter */
 178 NS_IMETHODIMP
 179 aaBalanceSheet::GetExpression(nsACString & aExpression)
 180 {
 181   aExpression.Truncate();
 182   return NS_OK;
 183 }
 184
 185 /* aaIBalanceSheet */
 186 NS_IMETHODIMP
 187 aaBalanceSheet::GetDate(PRTime *aDate)
 188 {
 189   return NS_OK;
 190 }
 191 NS_IMETHODIMP
 192 aaBalanceSheet::SetDate(PRTime aDate)
 193 {
 194   mHasDate =PR_TRUE;
 195   mDate = aDate;
 196   return NS_OK;
 197 }
 198 PRTime
 199 aaBalanceSheet::PickDate()
 200 {
 201   return 0;
 202 }
 203
 204 NS_IMETHODIMP
 205 aaBalanceSheet::GetTotalAssets(double *aTotalAssets)
 206 {
 207   NS_ENSURE_ARG_POINTER(aTotalAssets);
 208   *aTotalAssets = mAssets;
 209   return NS_OK;
 210 }
 211
 212 NS_IMETHODIMP
 213 aaBalanceSheet::GetTotalLiabilities(double *aTotalLiabilities)
 214 {
 215   NS_ENSURE_ARG_POINTER(aTotalLiabilities);
 216   *aTotalLiabilities = mLiabilities;
 217   return NS_OK;
 218 }