// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_DEVELOPER_SHELL_INTERPRETER_SRC_NODES_H_
#define SRC_DEVELOPER_SHELL_INTERPRETER_SRC_NODES_H_

#include <cstdint>
#include <memory>
#include <ostream>
#include <string>

#include "src/developer/shell/interpreter/src/value.h"

namespace shell {
namespace interpreter {

namespace code {
class Code;
}

class Addition;
class ExecutionContext;
class ExecutionScope;
class Expression;
class ExpressionVariable;
class Instruction;
class IntegerLiteral;
class Interpreter;
class ObjectDeclaration;
class ObjectSchema;
class ObjectFieldSchema;
class Scope;
class StringLiteral;
class TypeObject;
class Variable;
class VariableDefinition;

struct NodeId {
  NodeId(uint64_t file_id, uint64_t node_id) : file_id(file_id), node_id(node_id) {}

  // The id of the file which defines the node.
  uint64_t file_id;
  // The node id.
  uint64_t node_id;

  bool operator==(const NodeId& ref) const {
    return (node_id == ref.node_id) && (file_id == ref.file_id);
  }

  bool operator<(const NodeId& ref) const {
    return (node_id < ref.node_id) || (file_id < ref.file_id);
  }

  // Returns a text representation.
  std::string StringId() const { return std::to_string(file_id) + ":" + std::to_string(node_id); }
};

// Base class for a type.
class Type {
 public:
  enum class TypeKind {
    kUndefined,
    kBool,
    kChar,
    kString,
    kInt8,
    kUint8,
    kInt16,
    kUint16,
    kInt32,
    kUint32,
    kInt64,
    kUint64,
    kInteger,
    kFloat32,
    kFloat64,
    kObject
  };

  Type() = default;
  virtual ~Type() = default;

  // The size for the type in bytes.
  virtual size_t Size() const = 0;

  // The alignment for instances of the type.
  virtual size_t Alignment() const = 0;

  // Returns the type kind.
  virtual TypeKind Kind() const = 0;

  // Returns true if the type is the undefined type.
  bool IsUndefined() const { return Kind() == TypeKind::kUndefined; }

  // Returns true if the type is the string type.
  bool IsString() const { return Kind() == TypeKind::kString; }

  // Returns true if the type is the object type.
  bool IsObject() const { return Kind() == TypeKind::kObject; }

  // Returns a reference to this if the object is of type ObjectType.
  virtual TypeObject* AsTypeObject() { return nullptr; }

  // Creates an exact copy of the type.
  virtual std::unique_ptr<Type> Duplicate() const = 0;

  // Prints the type.
  virtual void Dump(std::ostream& os) const = 0;

  // Creates a variable of this type in the scope.
  virtual Variable* CreateVariable(ExecutionContext* context, Scope* scope, NodeId id,
                                   const std::string& name, bool is_mutable) const;

  // Generates a default value for this type. When the generated code is executed, it pushes the
  // value to the thread's stack values.
  virtual void GenerateDefaultValue(ExecutionContext* context, code::Code* code) const;

  // Generates an integer literal for this type. When the generated code is executed, it pushes the
  // value to the thread's stack value. The generation can generate an error if the literal is not
  // compatible with the type.
  virtual bool GenerateIntegerLiteral(ExecutionContext* context, code::Code* code,
                                      const IntegerLiteral* literal) const;

  // Generates a string literal for this type. When the generated code is executed, it pushes the
  // value to the thread's stack value. The generation can generate an error if the literal is not
  // compatible with the type.
  virtual bool GenerateStringLiteral(ExecutionContext* context, code::Code* code,
                                     const StringLiteral* literal) const;

  // Generates a variable load. It pushes the variable value to the stack.
  virtual bool GenerateVariable(ExecutionContext* context, code::Code* code, const NodeId& id,
                                const Variable* variable) const;

  // Generates a variable store. It pops a value from the stack and assigns the variable with it.
  virtual void GenerateAssignVariable(ExecutionContext* context, code::Code* code, const NodeId& id,
                                      const Variable* variable) const;

  // Generates an addition. it pops two values, do an addition and pushes the result. It generates
  // an error if the type doesn't support the addition or if the operand types are not supported.
  virtual bool GenerateAddition(ExecutionContext* context, code::Code* code,
                                const Addition* addition) const;

  // Loads the current value of the variable stored at |index| in |scope| into |value|.
  virtual void LoadVariable(const ExecutionScope* scope, size_t index, Value* value) const;

  // Clears the current value of the variable stored at |index| in |scope|. This also deallocate the
  // data for reference counted values.
  virtual void ClearVariable(ExecutionScope* scope, size_t index) const;

  // Sets a new value for an object field or a global variable. If free_old_value is false, this is
  // an initialization. If free_old_value is true, this is an assignment (this case can also be used
  // to deallocate a field by assigning 0).
  virtual void SetData(uint8_t* data, uint64_t value, bool free_old_value) const;

  // Interprets the value using the type, sends it back to the client and releases the value.
  virtual void EmitResult(ExecutionContext* context, uint64_t value) const;
};

inline std::ostream& operator<<(std::ostream& os, const Type& type) {
  type.Dump(os);
  return os;
}

// Base class for all the AST nodes.
class Node {
 public:
  Node(Interpreter* interpreter, uint64_t file_id, uint64_t node_id);
  virtual ~Node();

  Interpreter* interpreter() const { return interpreter_; }
  const NodeId& id() const { return id_; }
  uint64_t file_id() const { return id_.file_id; }
  uint64_t node_id() const { return id_.node_id; }

  // Returns a text representation of the node id.
  std::string StringId() const { return id_.StringId(); }

  // Downcast to a VariableDefinition.
  virtual const VariableDefinition* AsVariableDefinition() const { return nullptr; }

 private:
  // The interpreter which owns the node.
  Interpreter* interpreter_;
  // The node id.
  NodeId id_;
};

// Base class for all the expressions. Expressions generate a result which can be used by another
// expression or by an instruction.
class Expression : public Node {
 public:
  Expression(Interpreter* interpreter, uint64_t file_id, uint64_t node_id)
      : Node(interpreter, file_id, node_id) {}

  // Prints the expression.
  virtual void Dump(std::ostream& os) const = 0;

  // Returns true is the node is constant. That includes, for example, nodes which compute a value
  // only using constants.
  virtual bool IsConstant() const { return false; }

  // Infer the type of the expression. This is used when we don't know the type of the destination
  // for the computed value.
  virtual std::unique_ptr<Type> InferType(ExecutionContext* context) const = 0;

  // Compiles the expression (perform the semantic checks and generates code).
  virtual bool Compile(ExecutionContext* context, code::Code* code, const Type* for_type) const = 0;

  // Used by the string concatenation. It generates the string terms for the expression. It usually
  // generates one string (which is pushed to the stack). For Addition, it generates the strings for
  // both terms. This way, we can optimize the string concatenation.
  // Returns the number of strings generated (pushed to the stack).
  virtual size_t GenerateStringTerms(ExecutionContext* context, code::Code* code,
                                     const Type* for_type) const;

  // Generates code which will assign this expression with the last value pushed to the stack.
  virtual void Assign(ExecutionContext* context, code::Code* code) const;
};

inline std::ostream& operator<<(std::ostream& os, const Expression& expression) {
  expression.Dump(os);
  return os;
}

// Base class for all the instructions.
class Instruction : public Node {
 public:
  Instruction(Interpreter* interpreter, uint64_t file_id, uint64_t node_id)
      : Node(interpreter, file_id, node_id) {}

  // Prints the instruction.
  virtual void Dump(std::ostream& os) const = 0;

  // Compiles the instruction (performs the semantic checks and generates code).
  virtual void Compile(ExecutionContext* context, code::Code* code) = 0;
};

inline std::ostream& operator<<(std::ostream& os, const Instruction& instruction) {
  instruction.Dump(os);
  return os;
}

}  // namespace interpreter
}  // namespace shell

#endif  // SRC_DEVELOPER_SHELL_INTERPRETER_SRC_NODES_H_