/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-

 * vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ :

 * This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "nsError.h"

#include "nsMemory.h"

#include "nsString.h"

#include "mozStoragePrivateHelpers.h"

#include "mozStorageArgValueArray.h"

namespace mozilla {

namespace storage {

////////////////////////////////////////////////////////////////////////////////

//// ArgValueArray

ArgValueArray::ArgValueArray(int32_t aArgc,

                             sqlite3_value **aArgv)

: mArgc(aArgc)

, mArgv(aArgv)

{

}

NS_IMPL_ISUPPORTS(

  ArgValueArray,

  mozIStorageValueArray

)

////////////////////////////////////////////////////////////////////////////////

//// mozIStorageValueArray

NS_IMETHODIMP

ArgValueArray::GetNumEntries(uint32_t *_size)

{

  *_size = mArgc;

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetTypeOfIndex(uint32_t aIndex,

                              int32_t *_type)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  int t = ::sqlite3_value_type(mArgv[aIndex]);

  switch (t) {

    case SQLITE_INTEGER:

      *_type = VALUE_TYPE_INTEGER;

      break;

    case SQLITE_FLOAT:

      *_type = VALUE_TYPE_FLOAT;

      break;

    case SQLITE_TEXT:

      *_type = VALUE_TYPE_TEXT;

      break;

    case SQLITE_BLOB:

      *_type = VALUE_TYPE_BLOB;

      break;

    case SQLITE_NULL:

      *_type = VALUE_TYPE_NULL;

      break;

    default:

      return NS_ERROR_FAILURE;

  }

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetInt32(uint32_t aIndex,

                        int32_t *_value)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  *_value = ::sqlite3_value_int(mArgv[aIndex]);

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetInt64(uint32_t aIndex,

                        int64_t *_value)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  *_value = ::sqlite3_value_int64(mArgv[aIndex]);

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetDouble(uint32_t aIndex,

                         double *_value)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  *_value = ::sqlite3_value_double(mArgv[aIndex]);

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetUTF8String(uint32_t aIndex,

                             nsACString &_value)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  if (::sqlite3_value_type(mArgv[aIndex]) == SQLITE_NULL) {

    // NULL columns should have IsVoid set to distinguish them from an empty

    // string.

    _value.SetIsVoid(true);

  }

  else {

    _value.Assign(reinterpret_cast<const char *>(::sqlite3_value_text(mArgv[aIndex])),

                  ::sqlite3_value_bytes(mArgv[aIndex]));

  }

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetString(uint32_t aIndex,

                         nsAString &_value)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  if (::sqlite3_value_type(mArgv[aIndex]) == SQLITE_NULL) {

    // NULL columns should have IsVoid set to distinguish them from an empty

    // string.

    _value.SetIsVoid(true);

  } else {

    _value.Assign(static_cast<const char16_t *>(::sqlite3_value_text16(mArgv[aIndex])),

                  ::sqlite3_value_bytes16(mArgv[aIndex]) / 2);

  }

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetBlob(uint32_t aIndex,

                       uint32_t *_size,

                       uint8_t **_blob)

{

  ENSURE_INDEX_VALUE(aIndex, mArgc);

  int size = ::sqlite3_value_bytes(mArgv[aIndex]);

  void* blob = moz_xmemdup(::sqlite3_value_blob(mArgv[aIndex]), size);

  *_blob = static_cast<uint8_t *>(blob);

  *_size = size;

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetBlobAsString(uint32_t aIndex, nsAString& aValue)

{

  return DoGetBlobAsString(this, aIndex, aValue);

}

NS_IMETHODIMP

ArgValueArray::GetBlobAsUTF8String(uint32_t aIndex, nsACString& aValue)

{

  return DoGetBlobAsString(this, aIndex, aValue);

}

NS_IMETHODIMP

ArgValueArray::GetIsNull(uint32_t aIndex,

                         bool *_isNull)

{

  // GetTypeOfIndex will check aIndex for us, so we don't have to.

  int32_t type;

  nsresult rv = GetTypeOfIndex(aIndex, &type);

  NS_ENSURE_SUCCESS(rv, rv);

  *_isNull = (type == VALUE_TYPE_NULL);

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetSharedUTF8String(uint32_t aIndex,

                                   uint32_t *_length,

                                   const char **_string)

{

  if (_length)

    *_length = ::sqlite3_value_bytes(mArgv[aIndex]);

  *_string = reinterpret_cast<const char *>(::sqlite3_value_text(mArgv[aIndex]));

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetSharedString(uint32_t aIndex,

                               uint32_t *_length,

                               const char16_t **_string)

{

  if (_length)

    *_length = ::sqlite3_value_bytes(mArgv[aIndex]);

  *_string = static_cast<const char16_t *>(::sqlite3_value_text16(mArgv[aIndex]));

  return NS_OK;

}

NS_IMETHODIMP

ArgValueArray::GetSharedBlob(uint32_t aIndex,

                             uint32_t *_size,

                             const uint8_t **_blob)

{

  *_size = ::sqlite3_value_bytes(mArgv[aIndex]);

  *_blob = static_cast<const uint8_t *>(::sqlite3_value_blob(mArgv[aIndex]));

  return NS_OK;

}

} // namespace storage

} // namespace mozilla