138 lines
3.0 KiB
Ruby
Executable File
138 lines
3.0 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
|
|
# A script to automatically generate SQL files from an interface definition.
|
|
# To be called like this: sql-template-renderer interface.yaml sql-template.erb
|
|
|
|
require 'yaml'
|
|
require 'erb'
|
|
|
|
class SqlTemplateRenderer
|
|
|
|
DATASERVICES_CLIENT_SCHEMA = 'cdb_dataservices_client'
|
|
DATASERVICES_SERVER_SCHEMA = 'cdb_dataservices_server'
|
|
|
|
def initialize(template_file, function_signature)
|
|
@function_signature = function_signature
|
|
@template = File.read(template_file)
|
|
end
|
|
|
|
def render
|
|
ERB.new(@template).result(binding)
|
|
end
|
|
|
|
def name
|
|
@function_signature['name']
|
|
end
|
|
|
|
def return_type
|
|
@function_signature['return_type']
|
|
end
|
|
|
|
def multi_field
|
|
@function_signature['multi_field']
|
|
end
|
|
|
|
def multi_row
|
|
@function_signature['multi_row']
|
|
end
|
|
|
|
def user_config_key
|
|
@function_signature['user_config_key']
|
|
end
|
|
|
|
def geocoder_config_key
|
|
@function_signature['geocoder_config_key']
|
|
end
|
|
|
|
def parameters_info(with_user_org)
|
|
parameters = []
|
|
if with_user_org
|
|
parameters << { 'name' => 'username', 'type' => 'text' }
|
|
parameters << { 'name' => 'orgname', 'type' => 'text' }
|
|
end
|
|
parameters + @function_signature['params'].reject(&:empty?)
|
|
end
|
|
|
|
def user_org_declaration()
|
|
"username text;\n orgname text;" unless superuser_function?
|
|
end
|
|
|
|
def params(with_user_org = superuser_function?)
|
|
parameters_info(with_user_org).map { |p| p['name'].to_s }
|
|
end
|
|
|
|
def params_with_type(with_user_org = superuser_function?)
|
|
parameters_info(with_user_org).map { |p| "#{p['name']} #{p['type']}" }
|
|
end
|
|
|
|
def params_with_type_and_default(with_user_org = superuser_function?)
|
|
parameters = parameters_info(with_user_org).map do |p|
|
|
if not p['default'].nil?
|
|
"#{p['name']} #{p['type']} DEFAULT #{p['default']}"
|
|
else
|
|
"#{p['name']} #{p['type']}"
|
|
end
|
|
end
|
|
return parameters
|
|
end
|
|
|
|
def superuser_function?
|
|
!!@function_signature['superuser']
|
|
end
|
|
|
|
def void_return_type?
|
|
return_type.downcase == 'void'
|
|
end
|
|
|
|
def return_declaration
|
|
"ret #{return_type};" unless void_return_type? || multi_row
|
|
end
|
|
|
|
def return_statement(&block)
|
|
if block
|
|
erb_out = block.binding.eval('_erbout')
|
|
|
|
if multi_row
|
|
erb_out << 'RETURN QUERY SELECT * FROM '
|
|
elsif multi_field
|
|
erb_out << 'SELECT * FROM '
|
|
elsif void_return_type?
|
|
erb_out << 'PERFORM '
|
|
else
|
|
erb_out << 'SELECT '
|
|
end
|
|
yield
|
|
if multi_row || void_return_type?
|
|
erb_out << ';'
|
|
else
|
|
erb_out << ' INTO ret;'
|
|
end
|
|
if !multi_row && !void_return_type?
|
|
erb_out << ' RETURN ret;'
|
|
end
|
|
else
|
|
if !multi_row && !void_return_type?
|
|
' RETURN ret;'
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
end
|
|
|
|
|
|
if ARGV.length != 2 then
|
|
puts "Usage: sql-template-renderer <interface.yaml> <template.erb>"
|
|
exit
|
|
end
|
|
|
|
interface_source_file = ARGV[0]
|
|
template_file = ARGV[1]
|
|
|
|
|
|
functions = YAML.load(File.open(interface_source_file))
|
|
|
|
functions.each do |f|
|
|
puts SqlTemplateRenderer.new(template_file, f).render
|
|
end
|