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

/* vim: set ts=8 sts=2 et sw=2 tw=80: */

/* 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 "nsString.h"

#include "nsIControllerCommand.h"

#include "nsControllerCommandTable.h"

#include "nsGlobalWindowCommands.h"

#include "mozilla/EditorController.h"

#include "mozilla/HTMLEditorController.h"

nsresult NS_NewControllerCommandTable(nsIControllerCommandTable** aResult);

// this value is used to size the hash table. Just a sensible upper bound

#define NUM_COMMANDS_LENGTH 32

nsControllerCommandTable::nsControllerCommandTable()

  : mCommandsTable(NUM_COMMANDS_LENGTH)

  , mMutable(true)

{

}

nsControllerCommandTable::~nsControllerCommandTable()

{

}

NS_IMPL_ISUPPORTS(nsControllerCommandTable, nsIControllerCommandTable,

                  nsISupportsWeakReference)

NS_IMETHODIMP

nsControllerCommandTable::MakeImmutable(void)

{

  mMutable = false;

  return NS_OK;

}

NS_IMETHODIMP

nsControllerCommandTable::RegisterCommand(const char* aCommandName,

                                          nsIControllerCommand* aCommand)

{

  NS_ENSURE_TRUE(mMutable, NS_ERROR_FAILURE);

  mCommandsTable.Put(nsDependentCString(aCommandName), aCommand);

  return NS_OK;

}

NS_IMETHODIMP

nsControllerCommandTable::UnregisterCommand(const char* aCommandName,

                                            nsIControllerCommand* aCommand)

{

  NS_ENSURE_TRUE(mMutable, NS_ERROR_FAILURE);

  nsDependentCString commandKey(aCommandName);

  if (!mCommandsTable.Get(commandKey, nullptr)) {

    return NS_ERROR_FAILURE;

  }

  mCommandsTable.Remove(commandKey);

  return NS_OK;

}

NS_IMETHODIMP

nsControllerCommandTable::FindCommandHandler(const char* aCommandName,

                                             nsIControllerCommand** aResult)

{

  NS_ENSURE_ARG_POINTER(aResult);

  *aResult = nullptr;

  nsCOMPtr<nsIControllerCommand> foundCommand;

  mCommandsTable.Get(nsDependentCString(aCommandName),

                     getter_AddRefs(foundCommand));

  if (!foundCommand) {

    return NS_ERROR_FAILURE;

  }

  foundCommand.forget(aResult);

  return NS_OK;

}

NS_IMETHODIMP

nsControllerCommandTable::IsCommandEnabled(const char* aCommandName,

                                           nsISupports* aCommandRefCon,

                                           bool* aResult)

{

  NS_ENSURE_ARG_POINTER(aResult);

  *aResult = false;

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  if (!commandHandler) {

    NS_WARNING("Controller command table asked about a command that it does "

               "not handle");

    return NS_OK;

  }

  return commandHandler->IsCommandEnabled(aCommandName, aCommandRefCon,

                                          aResult);

}

NS_IMETHODIMP

nsControllerCommandTable::UpdateCommandState(const char* aCommandName,

                                             nsISupports* aCommandRefCon)

{

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  if (!commandHandler) {

    NS_WARNING("Controller command table asked to update the state of a "

               "command that it does not handle");

    return NS_OK;

  }

  return NS_ERROR_NOT_IMPLEMENTED;

}

NS_IMETHODIMP

nsControllerCommandTable::SupportsCommand(const char* aCommandName,

                                          nsISupports* aCommandRefCon,

                                          bool* aResult)

{

  NS_ENSURE_ARG_POINTER(aResult);

  // XXX: need to check the readonly and disabled states

  *aResult = false;

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  *aResult = (commandHandler.get() != nullptr);

  return NS_OK;

}

NS_IMETHODIMP

nsControllerCommandTable::DoCommand(const char* aCommandName,

                                    nsISupports* aCommandRefCon)

{

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  if (!commandHandler) {

    NS_WARNING("Controller command table asked to do a command that it does "

               "not handle");

    return NS_OK;

  }

  return commandHandler->DoCommand(aCommandName, aCommandRefCon);

}

NS_IMETHODIMP

nsControllerCommandTable::DoCommandParams(const char* aCommandName,

                                          nsICommandParams* aParams,

                                          nsISupports* aCommandRefCon)

{

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  if (!commandHandler) {

    NS_WARNING("Controller command table asked to do a command that it does "

               "not handle");

    return NS_OK;

  }

  return commandHandler->DoCommandParams(aCommandName, aParams, aCommandRefCon);

}

NS_IMETHODIMP

nsControllerCommandTable::GetCommandState(const char* aCommandName,

                                          nsICommandParams* aParams,

                                          nsISupports* aCommandRefCon)

{

  nsCOMPtr<nsIControllerCommand> commandHandler;

  FindCommandHandler(aCommandName, getter_AddRefs(commandHandler));

  if (!commandHandler) {

    NS_WARNING("Controller command table asked to do a command that it does "

               "not handle");

    return NS_OK;

  }

  return commandHandler->GetCommandStateParams(aCommandName, aParams,

                                               aCommandRefCon);

}

NS_IMETHODIMP

nsControllerCommandTable::GetSupportedCommands(uint32_t* aCount,

                                               char*** aCommands)

{

  char** commands =

    static_cast<char**>(moz_xmalloc(sizeof(char*) * mCommandsTable.Count()));

  *aCount = mCommandsTable.Count();

  *aCommands = commands;

  for (auto iter = mCommandsTable.Iter(); !iter.Done(); iter.Next()) {

    *commands = ToNewCString(iter.Key());

    commands++;

  }

  return NS_OK;

}

typedef nsresult (*CommandTableRegistrar)(nsIControllerCommandTable*);

static already_AddRefed<nsIControllerCommandTable>

CreateCommandTableWithCommands(CommandTableRegistrar aRegistrar)

{

  nsCOMPtr<nsIControllerCommandTable> commandTable =

      new nsControllerCommandTable();

  nsresult rv = aRegistrar(commandTable);

  if (NS_FAILED(rv)) return nullptr;

  // we don't know here whether we're being created as an instance,

  // or a service, so we can't become immutable

  return commandTable.forget();

}

// static

already_AddRefed<nsIControllerCommandTable>

nsControllerCommandTable::CreateEditorCommandTable()

{

  return CreateCommandTableWithCommands(

      EditorController::RegisterEditorCommands);

}

// static

already_AddRefed<nsIControllerCommandTable>

nsControllerCommandTable::CreateEditingCommandTable()

{

  return CreateCommandTableWithCommands(

      EditorController::RegisterEditingCommands);

}

// static

already_AddRefed<nsIControllerCommandTable>

nsControllerCommandTable::CreateHTMLEditorCommandTable()

{

  return CreateCommandTableWithCommands(

      HTMLEditorController::RegisterHTMLEditorCommands);

}

// static

already_AddRefed<nsIControllerCommandTable>

nsControllerCommandTable::CreateHTMLEditorDocStateCommandTable()

{

  return CreateCommandTableWithCommands(

      HTMLEditorController::RegisterEditorDocStateCommands);

}

// static

already_AddRefed<nsIControllerCommandTable>

nsControllerCommandTable::CreateWindowCommandTable()

{

  return CreateCommandTableWithCommands(

      nsWindowCommandRegistration::RegisterWindowCommands);

}

nsresult

NS_NewControllerCommandTable(nsIControllerCommandTable** aResult)

{

  MOZ_ASSERT(aResult != nullptr, "null ptr");

  if (!aResult) {

    return NS_ERROR_NULL_POINTER;

  }

  nsControllerCommandTable* newCommandTable = new nsControllerCommandTable();

  NS_ADDREF(newCommandTable);

  *aResult = newCommandTable;

  return NS_OK;

}