| 1 | #include "duckdb/parser/statement/pragma_statement.hpp" |
|---|---|
| 2 | #include "duckdb/parser/transformer.hpp" |
| 3 | #include "duckdb/parser/statement/show_statement.hpp" |
| 4 | #include "duckdb/parser/query_node/select_node.hpp" |
| 5 | #include "duckdb/parser/expression/star_expression.hpp" |
| 6 | #include "duckdb/parser/tableref/basetableref.hpp" |
| 7 | |
| 8 | namespace duckdb { |
| 9 | |
| 10 | static void TransformShowName(unique_ptr<PragmaStatement> &result, const string &name) { |
| 11 | auto &info = *result->info; |
| 12 | auto lname = StringUtil::Lower(str: name); |
| 13 | |
| 14 | if (lname == "\"databases\"") { |
| 15 | info.name = "show_databases"; |
| 16 | } else if (lname == "\"tables\"") { |
| 17 | // show all tables |
| 18 | info.name = "show_tables"; |
| 19 | } else if (lname == "__show_tables_expanded") { |
| 20 | info.name = "show_tables_expanded"; |
| 21 | } else { |
| 22 | // show one specific table |
| 23 | info.name = "show"; |
| 24 | info.parameters.emplace_back(args: name); |
| 25 | } |
| 26 | } |
| 27 | |
| 28 | unique_ptr<SQLStatement> Transformer::TransformShow(duckdb_libpgquery::PGVariableShowStmt &stmt) { |
| 29 | // we transform SHOW x into PRAGMA SHOW('x') |
| 30 | if (stmt.is_summary) { |
| 31 | auto result = make_uniq<ShowStatement>(); |
| 32 | auto &info = *result->info; |
| 33 | info.is_summary = stmt.is_summary; |
| 34 | |
| 35 | auto select = make_uniq<SelectNode>(); |
| 36 | select->select_list.push_back(x: make_uniq<StarExpression>()); |
| 37 | auto basetable = make_uniq<BaseTableRef>(); |
| 38 | auto qualified_name = QualifiedName::Parse(input: stmt.name); |
| 39 | basetable->schema_name = qualified_name.schema; |
| 40 | basetable->table_name = qualified_name.name; |
| 41 | select->from_table = std::move(basetable); |
| 42 | |
| 43 | info.query = std::move(select); |
| 44 | return std::move(result); |
| 45 | } |
| 46 | |
| 47 | auto result = make_uniq<PragmaStatement>(); |
| 48 | |
| 49 | auto show_name = stmt.name; |
| 50 | TransformShowName(result, name: show_name); |
| 51 | return std::move(result); |
| 52 | } |
| 53 | |
| 54 | } // namespace duckdb |
| 55 |