mirror of
https://github.com/discourse/discourse.git
synced 2025-06-04 22:14:44 +08:00
17ba19c7ae
* Splits the existing script into multiple classes * Adds command for generating IntermediateDB schema (`migrations/bin/cli schema generate`) * Changes the syntax of the IntermediateDB schema config * Adds validation for the schema config * It uses YAML schema aka JSON schema to validate the config file * It generates the SQL schema file and Ruby classes for storing data in the IntermediateDB
91 lines
2.6 KiB
Ruby
91 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Migrations::CLI
|
|
class SchemaSubCommand < Thor
|
|
Schema = ::Migrations::Database::Schema
|
|
|
|
desc "generate", "Generates the database schema"
|
|
method_option :db, type: :string, default: "intermediate_db", desc: "Name of the database"
|
|
def generate
|
|
db = options[:db]
|
|
config = load_config_file(db)
|
|
|
|
puts "Generating schema for #{db.bold}..."
|
|
::Migrations.load_rails_environment(quiet: true)
|
|
|
|
validate_config(config)
|
|
|
|
loader = Schema::Loader.new(config[:schema])
|
|
schema = loader.load_schema
|
|
header = file_header(db)
|
|
|
|
write_db_schema(config, header, schema)
|
|
write_db_models(config, header, schema)
|
|
|
|
puts "Done"
|
|
end
|
|
|
|
private
|
|
|
|
def validate_config(config)
|
|
validator = Schema::ConfigValidator.new
|
|
validator.validate(config)
|
|
|
|
if validator.has_errors?
|
|
validator.errors.each { |error| print_error(error) }
|
|
exit(2)
|
|
end
|
|
end
|
|
|
|
def write_db_schema(config, header, schema)
|
|
schema_file_path = File.expand_path(config.dig(:output, :schema_file), ::Migrations.root_path)
|
|
|
|
File.open(schema_file_path, "w") do |schema_file|
|
|
writer = Schema::TableWriter.new(schema_file)
|
|
writer.output_file_header(header)
|
|
|
|
schema.each { |table| writer.output_table(table) }
|
|
end
|
|
end
|
|
|
|
def write_db_models(config, header, schema)
|
|
writer = Schema::ModelWriter.new(config.dig(:output, :models_namespace), header)
|
|
models_path = File.expand_path(config.dig(:output, :models_directory), ::Migrations.root_path)
|
|
|
|
schema.each do |table|
|
|
model_file_path = File.join(models_path, Schema::ModelWriter.filename_for(table))
|
|
File.open(model_file_path, "w") { |model_file| writer.output_table(table, model_file) }
|
|
end
|
|
|
|
Schema::ModelWriter.format_files(models_path)
|
|
end
|
|
|
|
def relative_config_path(db)
|
|
File.join("config", "#{db}.yml")
|
|
end
|
|
|
|
def file_header(db)
|
|
<<~HEADER
|
|
This file is auto-generated from the IntermediateDB schema. To make changes,
|
|
update the "#{relative_config_path(db)}" configuration file and then run
|
|
`bin/cli schema generate` to regenerate this file.
|
|
HEADER
|
|
end
|
|
|
|
def load_config_file(db)
|
|
config_path = File.join(::Migrations.root_path, relative_config_path(db))
|
|
|
|
if !File.exist?(config_path)
|
|
print_error("Configuration file for #{db} wasn't found at '#{config_path}'")
|
|
exit 1
|
|
end
|
|
|
|
YAML.load_file(config_path, symbolize_names: true)
|
|
end
|
|
|
|
def print_error(message)
|
|
$stderr.puts "ERROR: ".red + message
|
|
end
|
|
end
|
|
end
|