storage/aaSession.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 "nsServiceManagerUtils.h"
  27 #include "nsDirectoryServiceDefs.h"
  28 #include "nsIProperties.h"
  29 #include "nsIFile.h"
  30 #include "nsStringAPI.h"
  31 #include "nsIArray.h"
  32
  33 /* Unfrozen API */
  34 #include "unstable/nsAppDirectoryServiceDefs.h"
  35 #include "unstable/mozStorageCID.h"
  36 #include "unstable/mozIStorageService.h"
  37 #include "unstable/mozIStorageConnection.h"
  38
  39 /* Project includes */
  40 #include <abstract/storage/aaISession.h>
  41 #include <abstract/storage/aaILoadQuery.h>
  42 #include "aaSession.h"
  43
  44 aaSession::aaSession()
  45 {
  46 }
  47
  48 aaSession::~aaSession()
  49 {
  50 }
  51
  52 NS_IMPL_ISUPPORTS2(aaSession,
  53                    aaISaveQuery,
  54                    aaISession)
  55
  56 /* aaISession */
  57 NS_IMETHODIMP
  58 aaSession::CreateQuery(const char *aContractID, nsISupports * *_retval)
  59 {
  60   NS_ENSURE_ARG_POINTER(_retval);
  61   nsresult rv;
  62   nsCOMPtr<nsISupports> query (do_CreateInstance(aContractID, mConnection,
  63         &rv));
  64   NS_ENSURE_SUCCESS(rv, rv);
  65   NS_IF_ADDREF(*_retval = query);
  66   return NS_OK;
  67 }
  68
  69 NS_IMETHODIMP
  70 aaSession::GetConnection(mozIStorageConnection * * aConnection)
  71 {
  72   NS_ENSURE_TRUE(mConnection, NS_ERROR_NOT_INITIALIZED);
  73   NS_IF_ADDREF(*aConnection = mConnection);
  74   return NS_OK;
  75 }
  76
  77 /* aaISaveQuery */
  78 NS_IMETHODIMP
  79 aaSession::Save(aaIDataNode *aNode, aaIDataNode *aOldNode)
  80 {
  81   nsresult rv;
  82   if (! mDispatcher) {
  83     mDispatcher = do_CreateInstance("@aasii.org/storage/save-dispatcher;1",
  84         this, &rv);
  85     NS_ENSURE_SUCCESS(rv, rv);
  86   }
  87   return mDispatcher->Save(aNode, aOldNode);
  88 }
  89
  90 /* Public methods */
  91
  92 /* Private methods */
  93 #define AA_TABLE_ENTITY "entity"
  94 #define AA_TABLE_ENTITY_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
  95   tag CHAR (47) UNIQUE"
  96
  97 #define AA_TABLE_RESOURCE "resource"
  98 #define AA_TABLE_RESOURCE_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
  99   tag CHAR (47) UNIQUE"
 100
 101 #define AA_TABLE_FLOW "flow"
 102 #define AA_TABLE_FLOW_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
 103   tag CHAR (47) UNIQUE, entity_id INTEGER NOT NULL, rate DOUBLE NOT NULL,\
 104   cap DOUBLE NOT NULL"
 105
 106 #define AA_TABLE_TERM "term"
 107 #define AA_TABLE_TERM_CREATE "flow_id INTEGER NOT NULL, side BOOLEAN NOT\
 108   NULL, resource_id INTEGER NOT NULL, PRIMARY KEY (flow_id, side)"
 109
 110 #define AA_TABLE_FACT "fact"
 111 #define AA_TABLE_FACT_CREATE "day INTEGER NOT NULL, event_id INTEGER NOT NULL,\
 112   transfer_id INTEGER NOT NULL, side BOOLEAN NOT NULL, flow_id INTEGER\
 113   NOT NULL, PRIMARY KEY (day, event_id, transfer_id, side)"
 114
 115 #define AA_TABLE_STATE "state"
 116 #define AA_TABLE_STATE_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
 117   flow_id INTEGER NOT NULL, side INTEGER NOT NULL,\
 118   amount DOUBLE NOT NULL, start INTEGER NOT NULL, paid INTEGER"
 119
 120 /*
 121 #define AA_TABLE_JOB "job"
 122 #define AA_TABLE_JOB_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
 123   flow_id INTEGER NOT NULL, resource_id INTEGER NOT NULL"
 124 */
 125
 126 #define AA_TABLE_EVENT "event"
 127 #define AA_TABLE_EVENT_CREATE "day INTEGER NOT NULL, id INTEGER NOT NULL,\
 128   PRIMARY KEY (day, id)"
 129
 130 #define AA_TABLE_TRANSFER "transfer"
 131 #define AA_TABLE_TRANSFER_CREATE "day INTEGER NOT NULL, event_id INTEGER\
 132   NOT NULL, id INTEGER NOT NULL, amount DOUBLE NOT NULL,\
 133   PRIMARY KEY (day, event_id, id)"
 134
 135 #define AA_TABLE_TRANSACTION "txn"
 136 #define AA_TABLE_TRANSACTION_CREATE "day INTEGER NOT NULL, event_id INTEGER\
 137   NOT NULL, transfer_id INTEGER NOT NULL, value DOUBLE NOT NULL,\
 138   status INTEGER NOT NULL DEFAULT 0, earnings DOUBLE,\
 139   PRIMARY KEY (day, event_id, transfer_id)"
 140
 141 /* XXX _acc This belongs to a separate module */
 142 #define AA_TABLE_QUOTE "quote"
 143 #define AA_TABLE_QUOTE_CREATE "resource_id INTEGER NOT NULL,\
 144   day INTEGER NOT NULL, rate DOUBLE NOT NULL, PRIMARY KEY (resource_id, day)"
 145
 146 #define AA_TABLE_BALANCE "balance"
 147 #define AA_TABLE_BALANCE_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
 148   flow_id INTEGER NOT NULL, side INTEGER NOT NULL,\
 149   amount DOUBLE NOT NULL, value DOUBLE NOT NULL, start INTEGER NOT NULL,\
 150   paid INTEGER, UNIQUE (flow_id, start)"
 151
 152 #define AA_TABLE_INCOME "income"
 153 #define AA_TABLE_INCOME_CREATE "start INTEGER PRIMARY KEY NOT NULL,\
 154   side INTEGER NOT NULL,\
 155   value DOUBLE NOT NULL,\
 156   paid INTEGER"
 157
 158 nsresult
 159 aaSession::getDatabase()
 160 {
 161   nsresult rv;
 162
 163   nsCOMPtr<nsIProperties> dsp;
 164   dsp = do_GetService("@mozilla.org/file/directory_service;1", &rv);
 165   NS_ENSURE_SUCCESS(rv, rv);
 166
 167   nsCOMPtr<nsIFile> file;
 168   rv = dsp->Get(NS_APP_USER_PROFILE_50_DIR, NS_GET_IID(nsIFile)
 169       , getter_AddRefs( file ));
 170   NS_ENSURE_SUCCESS(rv, rv);
 171
 172   rv = file->AppendNative(nsDependentCString("default.sqlite"));
 173   NS_ENSURE_SUCCESS(rv, rv);
 174
 175   nsCOMPtr<mozIStorageService> storageSvc;
 176   storageSvc = do_GetService(MOZ_STORAGE_SERVICE_CONTRACTID, &rv);
 177   NS_ENSURE_SUCCESS(rv, rv);
 178
 179   nsCOMPtr<mozIStorageConnection> connection;
 180   rv = storageSvc->OpenDatabase(file, getter_AddRefs( connection ));
 181   NS_ENSURE_SUCCESS(rv, rv);
 182
 183   rv = checkTable(connection, AA_TABLE_ENTITY, AA_TABLE_ENTITY_CREATE);
 184   NS_ENSURE_SUCCESS(rv, rv);
 185   rv = checkTable(connection, AA_TABLE_RESOURCE, AA_TABLE_RESOURCE_CREATE);
 186   NS_ENSURE_SUCCESS(rv, rv);
 187   rv = checkTable(connection, AA_TABLE_FLOW, AA_TABLE_FLOW_CREATE);
 188   NS_ENSURE_SUCCESS(rv, rv);
 189   rv = checkTable(connection, AA_TABLE_TERM, AA_TABLE_TERM_CREATE);
 190   NS_ENSURE_SUCCESS(rv, rv);
 191   rv = checkTable(connection, AA_TABLE_EVENT, AA_TABLE_EVENT_CREATE);
 192   NS_ENSURE_SUCCESS(rv, rv);
 193   rv = checkTable(connection, AA_TABLE_TRANSFER, AA_TABLE_TRANSFER_CREATE);
 194   NS_ENSURE_SUCCESS(rv, rv);
 195   rv = checkTable(connection, AA_TABLE_FACT, AA_TABLE_FACT_CREATE);
 196   NS_ENSURE_SUCCESS(rv, rv);
 197   rv = checkTable(connection, AA_TABLE_STATE, AA_TABLE_STATE_CREATE);
 198   NS_ENSURE_SUCCESS(rv, rv);
 199   rv = checkTable(connection, AA_TABLE_QUOTE, AA_TABLE_QUOTE_CREATE);
 200   NS_ENSURE_SUCCESS(rv, rv);
 201   rv = checkTable(connection, AA_TABLE_TRANSACTION, AA_TABLE_TRANSACTION_CREATE);
 202   NS_ENSURE_SUCCESS(rv, rv);
 203   rv = checkTable(connection, AA_TABLE_BALANCE, AA_TABLE_BALANCE_CREATE);
 204   NS_ENSURE_SUCCESS(rv, rv);
 205   rv = checkTable(connection, AA_TABLE_INCOME, AA_TABLE_INCOME_CREATE);
 206   NS_ENSURE_SUCCESS(rv, rv);
 207   mConnection = connection;
 208
 209   return NS_OK;
 210 }
 211
 212 nsresult
 213 aaSession::checkTable(mozIStorageConnection *aConnection,
 214     const char *tableName, const char *createString)
 215 {
 216   nsresult rv;
 217   PRBool hasTable;
 218
 219   rv = aConnection->TableExists(nsDependentCString(tableName), &hasTable);
 220   NS_ENSURE_SUCCESS(rv, rv);
 221   if (! hasTable ) {
 222     rv = aConnection->CreateTable(tableName, createString);
 223     NS_ENSURE_SUCCESS(rv, rv);
 224   }
 225
 226   return NS_OK;
 227 }
 228