grafana/public/app/plugins/datasource/grafana-postgresql-datasource/postgresMetaQuery.ts
Gábor Farkas 279aa4863b
Postgres: Handle single quotes in table names in the query editor (#80951)
postgres: handle single quotes in table names
2024-01-22 15:36:45 +01:00

47 lines
1.7 KiB
TypeScript

export function getVersion() {
return "SELECT current_setting('server_version_num')::int/100 as version";
}
export function getTimescaleDBVersion() {
return "SELECT extversion FROM pg_extension WHERE extname = 'timescaledb'";
}
export function showTables() {
return `select quote_ident(table_name) as "table" from information_schema.tables
where quote_ident(table_schema) not in ('information_schema',
'pg_catalog',
'_timescaledb_cache',
'_timescaledb_catalog',
'_timescaledb_internal',
'_timescaledb_config',
'timescaledb_information',
'timescaledb_experimental')
and ${buildSchemaConstraint()}`;
}
export function getSchema(table: string) {
// we will put table-name between single-quotes, so we need to escape single-quotes
// in the table-name
const tableNamePart = "'" + table.replace(/'/g, "''") + "'";
return `select quote_ident(column_name) as "column", data_type as "type"
from information_schema.columns
where quote_ident(table_name) = ${tableNamePart};
`;
}
function buildSchemaConstraint() {
// quote_ident protects hyphenated schemes
return `
quote_ident(table_schema) IN (
SELECT
CASE WHEN trim(s[i]) = '"$user"' THEN user ELSE trim(s[i]) END
FROM
generate_series(
array_lower(string_to_array(current_setting('search_path'),','),1),
array_upper(string_to_array(current_setting('search_path'),','),1)
) as i,
string_to_array(current_setting('search_path'),',') s
)`;
}