storage/aaSaveResource.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 "nsStringAPI.h"
  26 #include "nsEmbedString.h"
  27
  28 /* Unfrozen API */
  29 #include "unstable/mozIStorageConnection.h"
  30 #include "unstable/mozIStorageStatement.h"
  31
  32 /* Project includes */
  33 #include <abstract/base/aaIResource.h>
  34 #include <abstract/storage/aaISaveQuery.h>
  35 #include "aaSaveResource.h"
  36
  37 aaSaveResource::aaSaveResource(nsISupports *aOuter)
  38 {
  39   nsresult rv;
  40   mConnection = do_QueryInterface(aOuter);
  41   PRBool isReady;
  42   rv = mConnection->GetConnectionReady(&isReady);
  43   if (NS_FAILED(rv))
  44     mConnection = nsnull;
  45 }
  46
  47 aaSaveResource::~aaSaveResource()
  48 {
  49 }
  50
  51 NS_IMPL_ISUPPORTS1(aaSaveResource,
  52                    aaISaveQuery)
  53 /* aaISaveQuery */
  54 NS_IMETHODIMP
  55 aaSaveResource::Save(aaIDataNode *aNode, aaIDataNode *aOldNode)
  56 {
  57   nsresult rv;
  58
  59   nsCOMPtr<aaIResource> resource(do_QueryInterface(aNode, &rv));
  60   NS_ENSURE_SUCCESS(rv, rv);
  61
  62   nsEmbedString wTag;
  63   rv = resource->GetTag(wTag);
  64   NS_ENSURE_SUCCESS(rv, rv);
  65   nsEmbedCString newTag = NS_ConvertUTF16toUTF8(wTag);
  66
  67   if (! aOldNode) {
  68     PRInt64 id = insert(newTag);
  69     if (! id)
  70       return NS_ERROR_FAILURE;
  71     resource->SetId(id);
  72     return NS_OK;
  73   }
  74
  75   nsCOMPtr<aaIResource> old(do_QueryInterface(aOldNode, &rv));
  76   NS_ENSURE_SUCCESS(rv, rv);
  77   rv = old->GetTag(wTag);
  78   NS_ENSURE_SUCCESS(rv, rv);
  79   nsEmbedCString oldTag = NS_ConvertUTF16toUTF8(wTag);
  80
  81   return replace(newTag, oldTag);
  82 }
  83
  84 /* Private methods */
  85 PRInt64
  86 aaSaveResource::insert(const nsACString &newTag)
  87 {
  88   nsresult rv;
  89   NS_ENSURE_TRUE(mConnection, NS_ERROR_NOT_INITIALIZED);
  90
  91   nsCOMPtr<mozIStorageStatement> statement;
  92   rv = mConnection->CreateStatement(NS_LITERAL_CSTRING(
  93         "INSERT INTO resource (tag) VALUES (?)"), getter_AddRefs(statement));
  94   NS_ENSURE_SUCCESS(rv, 0);
  95
  96   rv = statement->BindUTF8StringParameter(0, newTag);
  97   NS_ENSURE_SUCCESS(rv, 0);
  98
  99   rv =  statement->Execute();
 100   NS_ENSURE_SUCCESS(rv, 0);
 101
 102   PRInt64 id;
 103   rv = mConnection->GetLastInsertRowID(&id);
 104   NS_ENSURE_SUCCESS(rv, 0);
 105
 106   return id;
 107 }
 108
 109 nsresult
 110 aaSaveResource::replace(const nsACString &newTag, const nsACString &oldTag)
 111 {
 112   nsresult rv;
 113   NS_ENSURE_TRUE(mConnection, NS_ERROR_NOT_INITIALIZED);
 114
 115   nsCOMPtr<mozIStorageStatement> statement;
 116   rv = mConnection->CreateStatement(NS_LITERAL_CSTRING(
 117         "UPDATE resource SET tag=? WHERE tag=?"), getter_AddRefs(statement));
 118   NS_ENSURE_SUCCESS(rv, rv);
 119
 120   rv = statement->BindUTF8StringParameter(0, newTag);
 121   rv = statement->BindUTF8StringParameter(1, oldTag);
 122   NS_ENSURE_SUCCESS(rv, rv);
 123   return statement->Execute();
 124 }
 125