cartodb-4.42/lib/carto/table_utils.rb

35 lines
1.0 KiB
Ruby
Raw Normal View History

2024-04-06 13:25:13 +08:00
module Carto
module TableUtils
# Returns a (double) quoted table name if needed (if it contains a dash, for example).
# Coupled to lib/assets/javascripts/builder/helpers/utils.js#safeTableNameQuoting
def safe_table_name_quoting(table_name)
can_be_quoted?(table_name) ? dashes_quoting(table_name) : table_name
end
def safe_schema_name_quoting(schema_name)
can_be_quoted?(schema_name) ? dashes_quoting(schema_name) : schema_name
end
def safe_schema_and_table_quoting(schema_name, table_name)
return nil if table_name.nil?
safe_schema = safe_schema_name_quoting(schema_name)
safe_table_name = safe_table_name_quoting(table_name)
"#{safe_schema}.#{safe_table_name}"
end
private
ALREADY_QUOTED = /\A".*"\Z/
VALID_IDENTIFIER = /^[a-zA-Z_][a-zA-Z0-9_$]*$/
def dashes_quoting(name)
name && !name.match(ALREADY_QUOTED) && !name.match(VALID_IDENTIFIER) ? "\"#{name}\"" : name
end
def can_be_quoted?(name)
name && !name.include?('.')
end
end
end