module BulletTrain module SuperScaffolding module Scaffolders class JoinModelScaffolder < Scaffolder def run unless argv.count >= 3 puts "" puts "🚅 usage: bin/super-scaffold join-model " puts "" puts "E.g. Add project-specific tags to a project:" puts "" puts " Given the following example models:".blue puts "" puts " rails g model Project team:references name:string description:text" puts " bin/super-scaffold crud Project Team name:text_field description:trix_editor" puts "" puts " rails g model Projects::Tag team:references name:string" puts " bin/super-scaffold crud Projects::Tag Team name:text_field" puts "" puts " 1️⃣ Use the standard Rails model generator to generate the join model:".blue puts "" puts " rails g model Projects::AppliedTag project:references tag:references" puts "" puts " 👋 Don't run migrations yet! Sometimes Super Scaffolding updates them for you.".yellow puts "" puts " 2️⃣ Use `join-model` scaffolding to prepare the join model for use in `crud-field` scaffolding:".blue puts "" puts " bin/super-scaffold join-model Projects::AppliedTag project_id{class_name=Project} tag_id{class_name=Projects::Tag}" puts "" puts " 3️⃣ Now you can use `crud-field` scaffolding to actually add the field to the form of the parent model:".blue puts "" puts " bin/super-scaffold crud-field Project tag_ids:super_select{class_name=Projects::Tag}" puts "" puts " 👋 Heads up! There will be one follow-up step output by this command that you need to take action on." puts "" puts " 4️⃣ Now you can run your migrations.".blue exit end child = argv[0] primary_parent = argv[1].split("class_name=").last.split(",").first.split("]").first secondary_parent = argv[2].split("class_name=").last.split(",").first.split("]").first # There should only be two attributes. attributes = [argv[1], argv[2]] # Pretend we're doing a `super_select` scaffolding because it will do the correct thing. attributes = attributes.map { |attribute| attribute.gsub("{", ":super_select{") } attributes = attributes.map { |attribute| attribute.gsub("}", ",required}") } transformer = Scaffolding::Transformer.new(child, [primary_parent], @options) # We need this transformer to reflect on the class names _just_ between e.g. `Project` and `Projects::Tag`, without the join model. has_many_through_transformer = Scaffolding::Transformer.new(secondary_parent, [primary_parent], @options) # We need this transformer to reflect on the association between `Projects::Tag` and `Projects::AppliedTag` backwards. inverse_transformer = Scaffolding::Transformer.new(child, [secondary_parent], @options) # We need this transformer to reflect on the class names _just_ between e.g. `Projects::Tag` and `Project`, without the join model. inverse_has_many_through_transformer = Scaffolding::Transformer.new(primary_parent, [secondary_parent], @options) # However, for the first attribute, we actually don't need the scope validator (and can't really implement it). attributes[0] = attributes[0].gsub("}", ",unscoped}") has_many_through_association = has_many_through_transformer.transform_string("completely_concrete_tangible_things") source = transformer.transform_string("absolutely_abstract_creative_concept.valid_$HAS_MANY_THROUGH_ASSOCIATION") source.gsub!("$HAS_MANY_THROUGH_ASSOCIATION", has_many_through_association) # For the second one, we don't want users to have to define the list of valid options in the join model, so we do this: attributes[1] = attributes[1].gsub("}", ",source=#{source}}") # This model hasn't been crud scaffolded, so a bunch of views are skipped here, but that's OK! # It does what we need on the files that exist. transformer.add_scaffolding_hooks_to_model transformer.suppress_could_not_find = true transformer.add_attributes_to_various_views(attributes, type: :crud_field) transformer.suppress_could_not_find = false # Add the `has_many ... through:` association in both directions. transformer.add_has_many_through_associations(has_many_through_transformer) inverse_transformer.add_has_many_through_associations(inverse_has_many_through_transformer) additional_steps = (transformer.additional_steps + has_many_through_transformer.additional_steps + inverse_transformer.additional_steps + inverse_has_many_through_transformer.additional_steps).uniq additional_steps.each_with_index do |additional_step, index| color, message = additional_step puts "" puts "#{index + 1}. #{message}".send(color) end puts "" end end end end end