43 lines
1.5 KiB
Ruby
43 lines
1.5 KiB
Ruby
module Carto
|
|
module QueryRewriter
|
|
def rewrite_query(query, old_username, new_user, renamed_tables)
|
|
new_query = query
|
|
new_query = rewrite_query_for_new_user(query, old_username, new_user) if old_username != new_user.database_schema
|
|
new_query = rewrite_query_for_renamed_tables(new_query, renamed_tables) if renamed_tables.present?
|
|
|
|
test_query(new_user, new_query) ? new_query : query
|
|
end
|
|
|
|
def qualify_query(query, table_name, username)
|
|
query.gsub(/(?<!\.)("?#{table_name}"?)/, username + '.\\1')
|
|
end
|
|
|
|
private
|
|
|
|
def test_query(user, query)
|
|
user.in_database.execute("EXPLAIN (#{query})")
|
|
true
|
|
rescue StandardError
|
|
false
|
|
end
|
|
|
|
def rewrite_query_for_new_user(query, old_username, new_user)
|
|
if new_user.username == new_user.database_schema
|
|
new_schema = new_user.sql_safe_database_schema
|
|
query.gsub(" #{old_username}.", " #{new_schema}.").gsub(" \"#{old_username}\".", " #{new_schema}.")
|
|
else
|
|
query.gsub(" #{old_username}.", " ").gsub(" \"#{old_username}\".", " ")
|
|
end
|
|
end
|
|
|
|
PSQL_WORD_CHARS = '[$_[[:alnum:]]]'.freeze
|
|
def rewrite_query_for_renamed_tables(query, renamed_tables)
|
|
renamed_tables.reduce(query) do |sql, (old_name, new_name)|
|
|
# Replaces the table name only if it matches the whole word
|
|
# i.e: previous and next characters are not PSQL_WORD_CHARS (alphanumerics, _ or $)
|
|
sql.gsub(/(?<!#{PSQL_WORD_CHARS})#{Regexp.escape(old_name)}(?!#{PSQL_WORD_CHARS})/, new_name)
|
|
end
|
|
end
|
|
end
|
|
end
|