started query builder

2023-11-01 19:55:09 +01:00
parent fbdf6116ff
commit 9e2d1358bb
16 changed files with 866 additions and 34 deletions
@@ -0,0 +1,152 @@
+#ifndef QUERY_DIALECT_HPP
+#define QUERY_DIALECT_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;
+
+  /**
+ * 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::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, ""}
+  };
+};
+
+}
+
+#endif //QUERY_DIALECT_HPP