storage/base/aaInsertTerm.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) 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 #include "xpcom-config.h"
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsStringAPI.h"
  26 #include "nsComponentManagerUtils.h"
  27 #include "nsXPCOMCID.h"
  28 #include "nsISupportsPrimitives.h"
  29 #include "nsIMutableArray.h"
  30
  31 /* Unfrozen API */
  32
  33 /* Project includes */
  34 #include "aaIResource.h"
  35 #include "aaIFlow.h"
  36 #include "aaIState.h"
  37 #include "aaInsertTerm.h"
  38
  39 static const char *sql = "INSERT INTO term (flow_id, side, resource_type,\
  40                           resource_id) VALUES (?,?,?,?)";
  41
  42 aaInsertTerm::aaInsertTerm()
  43 {
  44   mSql = sql;
  45 }
  46
  47 NS_IMPL_ISUPPORTS_INHERITED0(aaInsertTerm,
  48                              aaSaveRequest)
  49
  50 /* aaSaveRequest */
  51 NS_IMETHODIMP
  52 aaInsertTerm::SetFilter(nsISupports * aFilter)
  53 {
  54   nsresult rv;
  55   mTerm = do_QueryInterface(aFilter, &rv);
  56   NS_ENSURE_SUCCESS(rv, rv);
  57
  58   return NS_OK;
  59 }
  60
  61 NS_IMETHODIMP
  62 aaInsertTerm::GetParams(nsIArray * * aParams)
  63 {
  64   NS_ENSURE_ARG_POINTER(aParams);
  65   NS_ENSURE_TRUE(mTerm, NS_ERROR_NOT_INITIALIZED);
  66   NS_ENSURE_ARG_POINTER(mTerm->PickFlow());
  67   nsresult rv;
  68
  69   nsCOMPtr<aaIResource> resource;
  70   rv = mTerm->GetResource(getter_AddRefs(resource));
  71   NS_ENSURE_SUCCESS(rv, rv);
  72   NS_ENSURE_ARG_POINTER(resource);
  73
  74   nsCOMPtr<nsIMutableArray> result
  75     = do_CreateInstance(NS_ARRAY_CONTRACTID, &rv);
  76   NS_ENSURE_SUCCESS(rv, rv);
  77
  78   nsCOMPtr<nsISupportsPRInt64> valInt64
  79     = do_CreateInstance(NS_SUPPORTS_PRINT64_CONTRACTID, &rv);
  80   NS_ENSURE_SUCCESS(rv, rv);
  81   rv = valInt64->SetData(mTerm->PickFlow()->PickId());
  82   NS_ENSURE_SUCCESS(rv, rv);
  83   rv = result->InsertElementAt(valInt64, 0, PR_FALSE);
  84   NS_ENSURE_SUCCESS(rv, rv);
  85
  86   nsCOMPtr<nsISupportsPRInt32> valInt32
  87     = do_CreateInstance(NS_SUPPORTS_PRINT32_CONTRACTID, &rv);
  88   NS_ENSURE_SUCCESS(rv, rv);
  89   rv = valInt32->SetData(mTerm->PickSide());
  90   NS_ENSURE_SUCCESS(rv, rv);
  91   rv = result->InsertElementAt(valInt32, 1, PR_FALSE);
  92   NS_ENSURE_SUCCESS(rv, rv);
  93
  94   valInt32 = do_CreateInstance(NS_SUPPORTS_PRINT32_CONTRACTID, &rv);
  95   NS_ENSURE_SUCCESS(rv, rv);
  96   rv = valInt32->SetData(resource->PickType());
  97   NS_ENSURE_SUCCESS(rv, rv);
  98   rv = result->InsertElementAt(valInt32, 2, PR_FALSE);
  99   NS_ENSURE_SUCCESS(rv, rv);
 100
 101   valInt64 = do_CreateInstance(NS_SUPPORTS_PRINT64_CONTRACTID, &rv);
 102   NS_ENSURE_SUCCESS(rv, rv);
 103   rv = valInt64->SetData(resource->PickId());
 104   NS_ENSURE_SUCCESS(rv, rv);
 105   rv = result->InsertElementAt(valInt64, 3, PR_FALSE);
 106   NS_ENSURE_SUCCESS(rv, rv);
 107
 108   *aParams = result;
 109   result.forget();
 110
 111   return NS_OK;
 112 }