/*
 * Copyright (c) 2017, 2018, Oracle and/or its affiliates. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2.0,
 * as published by the Free Software Foundation.
 *
 * This program is also distributed with certain software (including
 * but not limited to OpenSSL) that is licensed under separate terms,
 * as designated in a particular file or component or in included license
 * documentation.  The authors of MySQL hereby grant you an additional
 * permission to link the program and your derivative works with the
 * separately licensed software that they have included with MySQL.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License, version 2.0, for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA
 */

#ifndef PLUGIN_X_TESTS_DRIVER_PROCESSOR_COMMANDS_MACRO_H_
#define PLUGIN_X_TESTS_DRIVER_PROCESSOR_COMMANDS_MACRO_H_

#include <list>
#include <memory>
#include <string>
#include <vector>

#include "plugin/x/tests/driver/formatters/console.h"
#include "plugin/x/tests/driver/processor/block_processor.h"
#include "plugin/x/tests/driver/processor/script_stack.h"

class Execution_context;

class Macro {
 public:
  using Strings = std::list<std::string>;

 public:
  Macro(const std::string &name, const Strings &argnames,
        const bool accepts_variadic_arguments)
      : m_name(name),
        m_accepts_args(argnames),
        m_accepts_variadic_arguments(accepts_variadic_arguments) {}

  const std::string &name() const { return m_name; }

  void set_macro_body(const std::string &body) { m_body = body; }

  std::string get_expanded_macro_body(const Strings &args,
                                      const Script_stack *stack,
                                      const Console &console) const;

 private:
  std::string m_name;
  Strings m_accepts_args;
  std::string m_body;
  bool m_accepts_variadic_arguments;
};

class Macro_container {
 public:
  using Strings = std::list<std::string>;

 public:
  void add_macro(std::shared_ptr<Macro> macro);
  void set_compress_option(const bool compress);

  bool call(Execution_context *context, const std::string &cmd);

 private:
  std::string get_expanded_macro(Execution_context *context,
                                 const std::string &cmd, std::string *r_name,
                                 const Script_stack *stack,
                                 const Console &console);

  std::list<std::shared_ptr<Macro>> m_macros;
  bool m_compress{true};
};

#endif  // PLUGIN_X_TESTS_DRIVER_PROCESSOR_COMMANDS_MACRO_H_