query/include/matador/sql/dialect.hpp

#ifndef QUERY_DIALECT_HPP
#define QUERY_DIALECT_HPP

#include "matador/sql/types.hpp"

#include <cstdint>
#include <string>
#include <unordered_map>

namespace matador::sql {

class dialect
{
public:
  enum class token_t : uint8_t
  {
    CREATE = 0,
    DROP,
    REMOVE,
    INSERT,
    UPDATE,
    SELECT,
    TABLE,
    VALUES,
    VALUE,
    COLUMNS,
    COLUMN,
    FROM,
    INTO,
    WHERE,
    AND,
    OR,
    LIKE,
    ORDER_BY,
    GROUP_BY,
    ASC,
    DESC,
    TOP,
    AS,
    OFFSET,
    DISTINCT,
    CONDITION,
    SET,
    NOT_NULL,
    PRIMARY_KEY,
    BEGIN,
    COMMIT,
    ROLLBACK,
    START_QUOTE,
    END_QUOTE,
    STRING_QUOTE,
    NONE
  };

  /**
   * Holding enums concerning escaping identifiers
   */
  enum class escape_identifier_t : uint8_t {
    ESCAPE_BOTH_SAME,       /**< The escape quotes are the same  */
    ESCAPE_CLOSING_BRACKET  /**< The escape quotes differ; escape the closing one */
  };

public:
  const std::string& token_at(token_t token) const;
  const std::string& data_type_at(data_type_t type) const;

  /**
 * Prepare sql dialect identifier for execution
 * and escape quotes and quote the identifier
 * string
 *
 * @param str The identifier string to be prepared
 * @return The prepared string
 */
  std::string prepare_identifier(const std::string &str) const;

  /**
   * Prepare string literal
   *
   * @param str String literal to be prepared
   */
  std::string prepare_literal(const std::string &str) const;

  /**
   * Wrap identifier quotes around a sql identifier keyword
   *
   * @param str Identifier to put quotes around
   */
  void quote_identifier(std::string &str) const;

  /**
   * Escape identifier quotes inside identifiers.
   *
   * @param str Identifier to be escaped
   */
  void escape_quotes_in_identifier(std::string &str) const;

  /**
   * Escape quotes in string literals
   *
   * @param str String literal to be escaped
   */
  void escape_quotes_in_literals(std::string &str) const;

  /**
   * Returns how the identifier quotes should be
   * escaped.
   *
   * @return How the identifier quotes should be escaped
   */
  virtual escape_identifier_t identifier_escape_type() const;

private:
  using token_to_string_map = std::unordered_map<token_t, std::string>;

  token_to_string_map tokens_ {
    {token_t::CREATE, "CREATE"},
    {token_t::DROP, "DROP"},
    {token_t::REMOVE, "DELETE"},
    {token_t::INSERT, "INSERT"},
    {token_t::TABLE, "TABLE"},
    {token_t::INTO, "INTO"},
    {token_t::VALUES, "VALUES"},
    {token_t::UPDATE, "UPDATE"},
    {token_t::SELECT, "SELECT"},
    {token_t::COLUMNS, "COLUMNS"},
    {token_t::COLUMN, "COLUMN"},
    {token_t::FROM, "FROM"},
    {token_t::WHERE, "WHERE"},
    {token_t::AND, "AND"},
    {token_t::OR, "OR"},
    {token_t::LIKE, "LIKE"},
    {token_t::ORDER_BY, "ORDER BY"},
    {token_t::GROUP_BY, "GROUP BY"},
    {token_t::ASC, "ASC"},
    {token_t::DESC, "DESC"},
    {token_t::TOP, "LIMIT"},
    {token_t::AS, "AS"},
    {token_t::OFFSET, "OFFSET"},
    {token_t::DISTINCT, "DISTINCT"},
    {token_t::SET, "SET"},
    {token_t::NOT_NULL, "NOT NULL"},
    {token_t::PRIMARY_KEY, "PRIMARY KEY"},
    {token_t::BEGIN, "BEGIN TRANSACTION"},
    {token_t::COMMIT, "COMMIT TRANSACTION"},
    {token_t::ROLLBACK, "ROLLBACK TRANSACTION"},
    {token_t::START_QUOTE, "\""},
    {token_t::END_QUOTE, "\""},
    {token_t::STRING_QUOTE, "'"},
    {token_t::NONE, ""}
  };

  using data_type_to_string_map = std::unordered_map<data_type_t, std::string>;
  data_type_to_string_map data_types_ {
    {data_type_t::type_char, "TINYINT"},
    {data_type_t::type_short, "SMALLINT"},
    {data_type_t::type_int, "INTEGER"},
    {data_type_t::type_long, "BIGINT"},
    {data_type_t::type_long_long, "BIGINT"},
    {data_type_t::type_unsigned_char, "TINYINT"},
    {data_type_t::type_unsigned_short, "INTEGER"},
    {data_type_t::type_unsigned_int, "BIGINT"},
    {data_type_t::type_unsigned_long, "BIGINT"},
    {data_type_t::type_unsigned_long_long, "BIGINT"},
    {data_type_t::type_float, "FLOAT"},
    {data_type_t::type_double, "DOUBLE"},
    {data_type_t::type_bool, "BOOLEAN"},
    {data_type_t::type_char_pointer, "VARCHAR"},
    {data_type_t::type_varchar, "VARCHAR"},
    {data_type_t::type_text, "TEXT"},
    {data_type_t::type_date, "DATE"},
    {data_type_t::type_time, "DATETIME"},
    {data_type_t::type_blob, "BLOB"},
    {data_type_t::type_null, "NULL"},
    {data_type_t::type_unknown, "UNKNOWN"}
  };
};

}

#endif //QUERY_DIALECT_HPP