/*
 * GDevelop Core
 * Copyright 2008-2016 Florian Rival (Florian.Rival@gmail.com). All rights
 * reserved. This project is released under the MIT License.
 */
#include "GDCore/IDE/Events/EventsPropertyReplacer.h"

#include <map>
#include <memory>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#include "GDCore/Events/Event.h"
#include "GDCore/Events/EventsList.h"
#include "GDCore/Events/Parsers/ExpressionParser2NodePrinter.h"
#include "GDCore/Events/Parsers/ExpressionParser2NodeWorker.h"
#include "GDCore/Extensions/Metadata/MetadataProvider.h"
#include "GDCore/Extensions/Metadata/ParameterMetadata.h"
#include "GDCore/Extensions/Metadata/ParameterMetadataTools.h"
#include "GDCore/IDE/Events/ExpressionValidator.h"
#include "GDCore/Project/Layout.h"
#include "GDCore/Project/Project.h"
#include "GDCore/Project/ProjectScopedContainers.h"
#include "GDCore/Project/PropertiesContainer.h"
#include "GDCore/String.h"
#include "GDCore/Tools/Log.h"

namespace gd {

/**
 * \brief Go through the nodes and rename properties,
 * or signal if the instruction must be renamed if a removed property is used.
 *
 * \see gd::ExpressionParser2
 */
class GD_CORE_API ExpressionPropertyReplacer
    : public ExpressionParser2NodeWorker {
 public:
  ExpressionPropertyReplacer(
      const gd::Platform& platform_,
      const gd::ProjectScopedContainers& projectScopedContainers_,
      const gd::PropertiesContainer& targetPropertiesContainer_,
      bool isParentTypeAVariable_,
      const std::unordered_map<gd::String, gd::String>& oldToNewPropertyNames_,
      const std::unordered_set<gd::String>& removedPropertyNames_)
      : hasDoneRenaming(false),
        removedPropertyUsed(false),
        platform(platform_),
        projectScopedContainers(projectScopedContainers_),
        targetPropertiesContainer(targetPropertiesContainer_),
        isParentTypeAVariable(isParentTypeAVariable_),
        oldToNewPropertyNames(oldToNewPropertyNames_),
        removedPropertyNames(removedPropertyNames_){};
  virtual ~ExpressionPropertyReplacer(){};

  bool HasDoneRenaming() const { return hasDoneRenaming; }
  bool IsRemovedPropertyUsed() const { return removedPropertyUsed; }

 protected:
  void OnVisitSubExpressionNode(SubExpressionNode& node) override {
    node.expression->Visit(*this);
  }
  void OnVisitOperatorNode(OperatorNode& node) override {
    node.leftHandSide->Visit(*this);
    node.rightHandSide->Visit(*this);
  }
  void OnVisitUnaryOperatorNode(UnaryOperatorNode& node) override {
    node.factor->Visit(*this);
  }
  void OnVisitNumberNode(NumberNode& node) override {}
  void OnVisitTextNode(TextNode& node) override {}
  void OnVisitVariableNode(VariableNode& node) override {
    if (isParentTypeAVariable) {
        // Do nothing, it's a variable.
        if (node.child) node.child->Visit(*this);
        return;
    }

    auto& propertiesContainersList =
        projectScopedContainers.GetPropertiesContainersList();

    // The node represents a variable or an object name on which a variable
    // will be accessed, or a property with a child.

    projectScopedContainers.MatchIdentifierWithName<void>(
      // The property name is changed after the refactor operation.
      node.name,
      [&]() {
        // Do nothing, it's an object variable.
        if (node.child) node.child->Visit(*this);
      }, [&]() {
        // Do nothing, it's a variable.
        if (node.child) node.child->Visit(*this);
      }, [&]() {
        // This is a property, check if it's coming from the target container with
        // properties to replace.
        if (propertiesContainersList.HasPropertiesContainer(
                targetPropertiesContainer)) {
          // The node represents a property, that can come from the target
          // (because the target is in the scope), replace or remove it:
          RenameOrRemovePropertyOfTargetPropertyContainer(node.name);
        }

        if (node.child) node.child->Visit(*this);
      }, [&]() {
        // Do nothing, it's a parameter.
        if (node.child) node.child->Visit(*this);
      }, [&]() {
        // Do nothing, it's something else.
        if (node.child) node.child->Visit(*this);
      });
  }
  void OnVisitVariableAccessorNode(VariableAccessorNode& node) override {
    if (node.child) node.child->Visit(*this);
  }
  void OnVisitVariableBracketAccessorNode(
      VariableBracketAccessorNode& node) override {
    bool isGrandParentTypeAVariable = isParentTypeAVariable;
    isParentTypeAVariable = false;
    node.expression->Visit(*this);
    isParentTypeAVariable = isGrandParentTypeAVariable;
    if (node.child) node.child->Visit(*this);
  }
  void OnVisitIdentifierNode(IdentifierNode& node) override {
    if (isParentTypeAVariable) {
        // Do nothing, it's a variable.
        return;
    }

    auto& propertiesContainersList =
        projectScopedContainers.GetPropertiesContainersList();

    projectScopedContainers.MatchIdentifierWithName<void>(
      // The property name is changed after the refactor operation
      node.identifierName,
      [&]() {
        // Do nothing, it's an object variable.
      }, [&]() {
        // Do nothing, it's a variable.
      }, [&]() {
        // This is a property, check if it's coming from the target container with
        // properties to replace.
        if (propertiesContainersList.HasPropertiesContainer(
                targetPropertiesContainer)) {
          // The node represents a property, that can come from the target
          // (because the target is in the scope), replace or remove it:
          RenameOrRemovePropertyOfTargetPropertyContainer(node.identifierName);
        }
      }, [&]() {
        // Do nothing, it's a parameter.
      }, [&]() {
        // Do nothing, it's something else.
      });
  }
  void OnVisitObjectFunctionNameNode(ObjectFunctionNameNode& node) override {}
  void OnVisitFunctionCallNode(FunctionCallNode &node) override {
    bool isGrandParentTypeAVariable = isParentTypeAVariable;
    for (auto &parameter : node.parameters) {
      const auto &parameterMetadata =
          gd::MetadataProvider::GetFunctionCallParameterMetadata(
              platform, projectScopedContainers.GetObjectsContainersList(),
              node, *parameter);
      if (!parameterMetadata) {
        continue;
      }
      const auto &parameterTypeMetadata =
          parameterMetadata->GetValueTypeMetadata();
      if (gd::EventsPropertyReplacer::CanContainProperty(
              parameterTypeMetadata)) {
        isParentTypeAVariable = parameterTypeMetadata.IsVariableOnly();
        parameter->Visit(*this);
      }
    }
    isParentTypeAVariable = isGrandParentTypeAVariable;
  }
  void OnVisitEmptyNode(EmptyNode& node) override {}

 private:
  bool hasDoneRenaming;
  bool removedPropertyUsed;

  bool RenameOrRemovePropertyOfTargetPropertyContainer(
      gd::String& propertyName) {
    if (oldToNewPropertyNames.count(propertyName) >= 1) {
      propertyName = oldToNewPropertyNames.find(propertyName)->second;
      hasDoneRenaming = true;
      return true;
    } else if (removedPropertyNames.count(propertyName) >= 1) {
      removedPropertyUsed = true;
      return true;
    }

    return false;  // Nothing was changed or done.
  }

  // Scope:
  const gd::Platform& platform;
  const gd::ProjectScopedContainers& projectScopedContainers;

  // Renaming or removing to do:
  const gd::PropertiesContainer& targetPropertiesContainer;
  const std::unordered_map<gd::String, gd::String>& oldToNewPropertyNames;
  const std::unordered_set<gd::String>& removedPropertyNames;

  gd::String objectNameToUseForVariableAccessor;
  bool isParentTypeAVariable;
};

bool EventsPropertyReplacer::DoVisitInstruction(gd::Instruction& instruction,
                                                bool isCondition) {
  const auto& metadata = isCondition
                             ? gd::MetadataProvider::GetConditionMetadata(
                                   platform, instruction.GetType())
                             : gd::MetadataProvider::GetActionMetadata(
                                   platform, instruction.GetType());
  bool shouldDeleteInstruction = false;

  gd::ParameterMetadataTools::IterateOverParametersWithIndex(
      instruction.GetParameters(), metadata.GetParameters(),
      [&](const gd::ParameterMetadata &parameterMetadata,
          const gd::Expression &parameterValue, size_t parameterIndex,
          const gd::String &lastObjectName, size_t lastObjectIndex) {
        if (!gd::EventsPropertyReplacer::CanContainProperty(
          parameterMetadata.GetValueTypeMetadata())) {
          return;
        }
        auto node = parameterValue.GetRootNode();
        if (node) {
          ExpressionPropertyReplacer renamer(
              platform, GetProjectScopedContainers(), targetPropertiesContainer,
              parameterMetadata.GetValueTypeMetadata().IsVariableOnly(),
              oldToNewPropertyNames, removedPropertyNames);
          node->Visit(renamer);

          if (renamer.IsRemovedPropertyUsed()) {
            shouldDeleteInstruction = true;
          } else if (renamer.HasDoneRenaming()) {
            instruction.SetParameter(
                parameterIndex, ExpressionParser2NodePrinter::PrintNode(*node));
          }
        }
      });

  return shouldDeleteInstruction;
}

bool EventsPropertyReplacer::DoVisitEventExpression(
    gd::Expression& expression, const gd::ParameterMetadata& metadata) {
  if (!gd::EventsPropertyReplacer::CanContainProperty(
          metadata.GetValueTypeMetadata())) {
    return false;
  }
  auto node = expression.GetRootNode();
  if (node) {
    ExpressionPropertyReplacer renamer(
        platform, GetProjectScopedContainers(), targetPropertiesContainer,
        metadata.GetValueTypeMetadata().IsVariableOnly(), oldToNewPropertyNames,
        removedPropertyNames);
    node->Visit(renamer);

    if (renamer.IsRemovedPropertyUsed()) {
      return true;
    } else if (renamer.HasDoneRenaming()) {
      expression = ExpressionParser2NodePrinter::PrintNode(*node);
    }
  }

  return false;
}

bool EventsPropertyReplacer::CanContainProperty(
    const gd::ValueTypeMetadata &valueTypeMetadata) {
  return valueTypeMetadata.IsVariable() || valueTypeMetadata.IsNumber() ||
         valueTypeMetadata.IsString() || valueTypeMetadata.IsResource();
}

EventsPropertyReplacer::~EventsPropertyReplacer() {}

}  // namespace gd