# frozen_string_literal: true

require 'rails/generators/rails/scaffold/scaffold_generator'
require 'generators/aws_record/resource/resource_generator'

module AwsRecord
  module Generators
    class ScaffoldGenerator < ResourceGenerator
      source_root File.expand_path('../model/templates', __dir__)

      remove_class_option :orm
      remove_class_option :actions

      class_option :api, type: :boolean
      class_option :stylesheets, type: :boolean, desc: 'Generate Stylesheets'
      class_option :stylesheet_engine, desc: 'Engine for Stylesheets'
      class_option :assets, type: :boolean
      class_option :resource_route, type: :boolean
      class_option :scaffold_stylesheet, type: :boolean

      def handle_skip
        @options = @options.merge(stylesheets: false) unless options[:assets]
        return if options[:stylesheets] && options[:scaffold_stylesheet]

        @options = @options.merge(stylesheet_engine: false)
      end

      hook_for :scaffold_controller, in: :aws_record, required: true

      hook_for :assets, in: :rails do |assets|
        invoke assets, [controller_name]
      end

      hook_for :stylesheet_engine, in: :rails do |stylesheet_engine|
        invoke stylesheet_engine, [controller_name] if behavior == :invoke
      end

      private

      def initialize(args, *options)
        options[0] << '--scaffold'
        super
      end
    end
  end
end