/* 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/. */

#ifndef nsArrayUtils_h_
#define nsArrayUtils_h_

#include "nsCOMPtr.h"
#include "nsIArray.h"

// helper class for do_QueryElementAt
class MOZ_STACK_CLASS nsQueryArrayElementAt final : public nsCOMPtr_helper {
 public:
  nsQueryArrayElementAt(nsIArray* aArray, uint32_t aIndex, nsresult* aErrorPtr)
      : mArray(aArray), mIndex(aIndex), mErrorPtr(aErrorPtr) {}

  virtual nsresult NS_FASTCALL operator()(const nsIID& aIID,
                                          void**) const override;

 private:
  nsIArray* MOZ_NON_OWNING_REF mArray;
  uint32_t mIndex;
  nsresult* mErrorPtr;
};

inline const nsQueryArrayElementAt do_QueryElementAt(
    nsIArray* aArray, uint32_t aIndex, nsresult* aErrorPtr = nullptr) {
  return nsQueryArrayElementAt(aArray, aIndex, aErrorPtr);
}

#endif  // nsArrayUtils_h_
