michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "nsProperties.h"
michael@0: 
michael@0: ////////////////////////////////////////////////////////////////////////////////
michael@0: 
michael@0: NS_IMPL_AGGREGATED(nsProperties)
michael@0: NS_INTERFACE_MAP_BEGIN_AGGREGATED(nsProperties)
michael@0:     NS_INTERFACE_MAP_ENTRY(nsIProperties)
michael@0: NS_INTERFACE_MAP_END
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsProperties::Get(const char* prop, const nsIID & uuid, void* *result)
michael@0: {
michael@0:     if (NS_WARN_IF(!prop))
michael@0:         return NS_ERROR_INVALID_ARG;
michael@0: 
michael@0:     nsCOMPtr<nsISupports> value;
michael@0:     if (!nsProperties_HashBase::Get(prop, getter_AddRefs(value))) {
michael@0:         return NS_ERROR_FAILURE;
michael@0:     }
michael@0:     return (value) ? value->QueryInterface(uuid, result) : NS_ERROR_NO_INTERFACE;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsProperties::Set(const char* prop, nsISupports* value)
michael@0: {
michael@0:     if (NS_WARN_IF(!prop))
michael@0:         return NS_ERROR_INVALID_ARG;
michael@0:     Put(prop, value);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsProperties::Undefine(const char* prop)
michael@0: {
michael@0:     if (NS_WARN_IF(!prop))
michael@0:         return NS_ERROR_INVALID_ARG;
michael@0: 
michael@0:     nsCOMPtr<nsISupports> value;
michael@0:     if (!nsProperties_HashBase::Get(prop, getter_AddRefs(value)))
michael@0:         return NS_ERROR_FAILURE;
michael@0: 
michael@0:     Remove(prop);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsProperties::Has(const char* prop, bool *result)
michael@0: {
michael@0:     if (NS_WARN_IF(!prop))
michael@0:         return NS_ERROR_INVALID_ARG;
michael@0: 
michael@0:     nsCOMPtr<nsISupports> value;
michael@0:     *result = nsProperties_HashBase::Get(prop,
michael@0:                                          getter_AddRefs(value));
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: struct GetKeysEnumData
michael@0: {
michael@0:     char **keys;
michael@0:     uint32_t next;
michael@0:     nsresult res;
michael@0: };
michael@0: 
michael@0:  PLDHashOperator
michael@0: GetKeysEnumerate(const char *key, nsISupports* data,
michael@0:                  void *arg)
michael@0: {
michael@0:     GetKeysEnumData *gkedp = (GetKeysEnumData *)arg;
michael@0:     gkedp->keys[gkedp->next] = strdup(key);
michael@0: 
michael@0:     if (!gkedp->keys[gkedp->next]) {
michael@0:         gkedp->res = NS_ERROR_OUT_OF_MEMORY;
michael@0:         return PL_DHASH_STOP;
michael@0:     }
michael@0: 
michael@0:     gkedp->next++;
michael@0:     return PL_DHASH_NEXT;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP 
michael@0: nsProperties::GetKeys(uint32_t *count, char ***keys)
michael@0: {
michael@0:     if (NS_WARN_IF(!count) || NS_WARN_IF(!keys))
michael@0:         return NS_ERROR_INVALID_ARG;
michael@0: 
michael@0:     uint32_t n = Count();
michael@0:     char ** k = (char **) nsMemory::Alloc(n * sizeof(char *));
michael@0: 
michael@0:     GetKeysEnumData gked;
michael@0:     gked.keys = k;
michael@0:     gked.next = 0;
michael@0:     gked.res = NS_OK;
michael@0: 
michael@0:     EnumerateRead(GetKeysEnumerate, &gked);
michael@0: 
michael@0:     nsresult rv = gked.res;
michael@0:     if (NS_FAILED(rv)) {
michael@0:         // Free 'em all
michael@0:         for (uint32_t i = 0; i < gked.next; i++)
michael@0:             nsMemory::Free(k[i]);
michael@0:         nsMemory::Free(k);
michael@0:         return rv;
michael@0:     }
michael@0: 
michael@0:     *count = n;
michael@0:     *keys = k;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: ////////////////////////////////////////////////////////////////////////////////