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: */
4 * Copyright (C) 2007 Sergey Yanovich <ynvich@gmail.com>
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.
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.
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.
22 #include <abstract/aacore.h>
25 #include "nsComponentManagerUtils.h"
26 #include "nsServiceManagerUtils.h"
27 #include "nsDirectoryServiceDefs.h"
28 #include "nsIProperties.h"
30 #include "nsStringAPI.h"
34 #include "unstable/nsAppDirectoryServiceDefs.h"
35 #include "unstable/mozStorageCID.h"
36 #include "unstable/mozIStorageService.h"
37 #include "unstable/mozIStorageConnection.h"
39 /* Project includes */
40 #include <abstract/storage/aaISession.h>
41 #include <abstract/storage/aaILoadQuery.h>
42 #include "aaSession.h"
44 aaSession::aaSession()
48 aaSession::~aaSession()
52 NS_IMPL_ISUPPORTS2(aaSession
,
58 aaSession::CreateQuery(const char *aContractID
, nsISupports
* *_retval
)
60 NS_ENSURE_ARG_POINTER(_retval
);
62 nsCOMPtr
<nsISupports
> query (do_CreateInstance(aContractID
, mConnection
,
64 NS_ENSURE_SUCCESS(rv
, rv
);
65 NS_IF_ADDREF(*_retval
= query
);
70 aaSession::GetConnection(mozIStorageConnection
* * aConnection
)
72 NS_ENSURE_TRUE(mConnection
, NS_ERROR_NOT_INITIALIZED
);
73 NS_IF_ADDREF(*aConnection
= mConnection
);
79 aaSession::Save(aaIDataNode
*aNode
, aaIDataNode
*aOldNode
)
83 mDispatcher
= do_CreateInstance("@aasii.org/storage/save-dispatcher;1",
85 NS_ENSURE_SUCCESS(rv
, rv
);
87 return mDispatcher
->Save(aNode
, aOldNode
);
93 #define AA_TABLE_ENTITY "entity"
94 #define AA_TABLE_ENTITY_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
97 #define AA_TABLE_RESOURCE "resource"
98 #define AA_TABLE_RESOURCE_CREATE "id INTEGER PRIMARY KEY AUTOINCREMENT,\
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,\
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)"
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)"
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"
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"
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)"
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)"
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)"
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)"
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)"
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,\
159 aaSession::getDatabase()
163 nsCOMPtr
<nsIProperties
> dsp
;
164 dsp
= do_GetService("@mozilla.org/file/directory_service;1", &rv
);
165 NS_ENSURE_SUCCESS(rv
, rv
);
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
);
172 rv
= file
->AppendNative(nsDependentCString("default.sqlite"));
173 NS_ENSURE_SUCCESS(rv
, rv
);
175 nsCOMPtr
<mozIStorageService
> storageSvc
;
176 storageSvc
= do_GetService(MOZ_STORAGE_SERVICE_CONTRACTID
, &rv
);
177 NS_ENSURE_SUCCESS(rv
, rv
);
179 nsCOMPtr
<mozIStorageConnection
> connection
;
180 rv
= storageSvc
->OpenDatabase(file
, getter_AddRefs( connection
));
181 NS_ENSURE_SUCCESS(rv
, rv
);
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
;
213 aaSession::checkTable(mozIStorageConnection
*aConnection
,
214 const char *tableName
, const char *createString
)
219 rv
= aConnection
->TableExists(nsDependentCString(tableName
), &hasTable
);
220 NS_ENSURE_SUCCESS(rv
, rv
);
222 rv
= aConnection
->CreateTable(tableName
, createString
);
223 NS_ENSURE_SUCCESS(rv
, rv
);