/* 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 "nsDialogParamBlock.h"
#include "nsString.h"
#include "nsReadableUtils.h"

NS_IMPL_ISUPPORTS(nsDialogParamBlock, nsIDialogParamBlock)

nsDialogParamBlock::~nsDialogParamBlock() { delete[] mString; }

NS_IMETHODIMP
nsDialogParamBlock::SetNumberStrings(int32_t aNumStrings) {
  if (mString) {
    return NS_ERROR_ALREADY_INITIALIZED;
  }

  mString = new nsString[aNumStrings];
  if (!mString) {
    return NS_ERROR_OUT_OF_MEMORY;
  }
  mNumStrings = aNumStrings;
  return NS_OK;
}

NS_IMETHODIMP
nsDialogParamBlock::GetInt(int32_t aIndex, int32_t* aResult) {
  nsresult rv = InBounds(aIndex, kNumInts);
  if (rv == NS_OK) {
    *aResult = mInt[aIndex];
  }
  return rv;
}

NS_IMETHODIMP
nsDialogParamBlock::SetInt(int32_t aIndex, int32_t aInt) {
  nsresult rv = InBounds(aIndex, kNumInts);
  if (rv == NS_OK) {
    mInt[aIndex] = aInt;
  }
  return rv;
}

NS_IMETHODIMP
nsDialogParamBlock::GetString(int32_t aIndex, char16_t** aResult) {
  if (mNumStrings == 0) {
    SetNumberStrings(kNumStrings);
  }
  nsresult rv = InBounds(aIndex, mNumStrings);
  if (rv == NS_OK) {
    *aResult = ToNewUnicode(mString[aIndex]);
  }
  return rv;
}

NS_IMETHODIMP
nsDialogParamBlock::SetString(int32_t aIndex, const char16_t* aString) {
  if (mNumStrings == 0) {
    SetNumberStrings(kNumStrings);
  }
  nsresult rv = InBounds(aIndex, mNumStrings);
  if (rv == NS_OK) {
    mString[aIndex] = aString;
  }
  return rv;
}

NS_IMETHODIMP
nsDialogParamBlock::GetObjects(nsIMutableArray** aObjects) {
  NS_ENSURE_ARG_POINTER(aObjects);
  NS_IF_ADDREF(*aObjects = mObjects);
  return NS_OK;
}

NS_IMETHODIMP
nsDialogParamBlock::SetObjects(nsIMutableArray* aObjects) {
  mObjects = aObjects;
  return NS_OK;
}
