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