class RubyLLM::Generators::UpgradeToV17Generator

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 30
def self.next_migration_number(dirname)
  ::ActiveRecord::Generators::Base.next_migration_number(dirname)
end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 18
def self.source_paths
  [
    File.expand_path('templates', __dir__),
    File.expand_path('../install/templates', __dir__)
  ]
end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 34
def create_migration_file
  @model_table_already_existed = table_exists?(table_name_for(model_model_name))

  # First check if models table exists, if not create it
  unless @model_table_already_existed
    migration_template 'create_models_migration.rb.tt',
                       "db/migrate/create_#{table_name_for(model_model_name)}.rb",
                       migration_version: migration_version,
                       model_model_name: model_model_name
  end

  migration_template 'migration.rb.tt',
                     'db/migrate/migrate_to_ruby_llm_model_references.rb',
                     migration_version: migration_version,
                     chat_model_name: chat_model_name,
                     message_model_name: message_model_name,
                     tool_call_model_name: tool_call_model_name,
                     model_model_name: model_model_name,
                     model_table_already_existed: @model_table_already_existed
end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 55
def create_model_file
  create_namespace_modules

  template 'model_model.rb.tt', "app/models/#{model_model_name.underscore}.rb"
end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 88
      def show_next_steps
        say_status :success, 'Upgrade prepared!', :green
        say <<~INSTRUCTIONS

          Next steps:
          1. Review the generated migrations
          2. Run: bin/rails db:migrate
          3. Update your code to use the new API: #{chat_model_name}.create! now has the same signature as RubyLLM.chat

          ⚠️  If you get "undefined method 'acts_as_model'" during migration:
            Add this to config/application.rb BEFORE your Application class:

            RubyLLM.configure do |config|
              config.use_new_acts_as = true
            end

          📚 See the full migration guide: https://rubyllm.com/upgrading-to-1-7/

        INSTRUCTIONS
      end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 61
def update_existing_models
  update_model_acts_as(chat_model_name, 'acts_as_chat', acts_as_chat_declaration)
  update_model_acts_as(message_model_name, 'acts_as_message', acts_as_message_declaration)
  update_model_acts_as(tool_call_model_name, 'acts_as_tool_call', acts_as_tool_call_declaration)
end

# File lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb, line 67
def update_initializer
  initializer_path = 'config/initializers/ruby_llm.rb'

  unless File.exist?(initializer_path)
    say_status :warning, 'No initializer found. Creating one...', :yellow
    template 'initializer.rb.tt', initializer_path
    return
  end

  initializer_content = File.read(initializer_path)

  return if initializer_content.include?('config.use_new_acts_as')

  inject_into_file initializer_path, before: /^end/ do
    lines = ["\n  # Enable the association-based Rails API", '  config.use_new_acts_as = true']
    lines << "  config.model_registry_class = \"#{model_model_name}\"" if model_model_name != 'Model'
    lines << "\n"
    lines.join("\n")
  end
end