storage/base/aaSession.h

   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,2008 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 #ifndef AASESSION_H
  23 #define AASESSION_H 1
  24
  25 #define AA_SESSION_CID \
  26 {0xc7eb3168, 0x85f9, 0x4fa2, {0x9e, 0x0f, 0xec, 0xd4, 0x21, 0xf3, 0x6e, 0x98}}
  27
  28 #include "nsAutoPtr.h"
  29 #include "aaISession.h"
  30
  31 class mozIStorageConnection;
  32 #ifdef DEBUG
  33 #include "mozIStorageConnection.h"
  34 #endif
  35
  36 class aaSession : public aaISession
  37 {
  38 public:
  39   aaSession();
  40   virtual ~aaSession();
  41   NS_DECL_ISUPPORTS
  42   NS_DECL_AAISESSION
  43   NS_DECL_AAISAVEQUERY
  44
  45   nsresult Init(nsISupports *dbFile = nsnull) { return getDatabase(dbFile); }
  46
  47   mozIStorageConnection* GetConnection() const { return mConnection; }
  48 private:
  49   friend class aaStorageTest;
  50   /* XXX _acc This has to be removed after accounting is separated*/
  51   friend class aaAccountTest;
  52
  53   nsCOMPtr<mozIStorageConnection> mConnection;
  54   nsCOMPtr<aaISaveQuery> mDispatcher;
  55   nsCOMPtr<aaISqlTransaction> mLoader;
  56
  57   nsresult getDatabase(nsISupports *dbFile);
  58   nsresult checkTable(mozIStorageConnection *aConnection,
  59       const char *tableName, const char *createString);
  60 };
  61
  62 #endif /* AASESSION_H */