# frozen_string_literal: true

require 'rails/generators'
require 'rails/generators/named_base'
require_relative 'core'

module Ibrain
  module Graphql
    # TODO: What other options should be supported?
    #
    # @example Generate a `GraphQL::Schema::Resolver` by name
    #     rails g graphql:resolver PostsResolver
    class ResolversGenerator < Rails::Generators::Base
      include Core

      desc "Create a resolver by name"
      source_root File.expand_path('templates', __dir__)

      argument :name, type: :string
      class_option :model, type: :string, default: nil
      class_option :prefix, type: :string, default: nil

      def initialize(args, *options) # :nodoc:
        # Unfreeze name in case it's given as a frozen string
        args[0] = args[0].dup if args[0].is_a?(String) && args[0].frozen?
        super

        assign_names!(name)
      end

      attr_reader :file_name, :resolver_name, :field_name, :model_name

      def create_resolver_file
        if @behavior == :revoke
          log :gsub, "#{options[:directory]}/types/query_type.rb"
        else
          create_resolver_root_type
        end

        if options[:model].present?
          system("bundle exec rails generate ibrain:graphql:object #{options[:model].downcase}")
        end

        template "resolvers.erb", "#{options[:directory]}/resolvers/#{file_name}.rb"
        template "aggregate.erb", "#{options[:directory]}/resolvers/#{file_name}_aggregate.rb"

        in_root do
          gsub_file "#{options[:directory]}/types/query_type.rb", /  \# TODO: Add Resolvers as fields\s*\n/m, ""
          inject_into_file "#{options[:directory]}/types/query_type.rb", "\n    field :#{field_name}, resolver: Resolvers::#{resolver_name}\n    field :#{field_name}_aggregate, resolver: Resolvers::#{resolver_name}Aggregate \n  ", before: "end\nend", verbose: true, force: true
        end
      end

      private

      def assign_names!(name)
        underscore_name = name&.camelize&.underscore
        prefix = options[:prefix].try(:underscore)

        @model_name = options[:model].blank? ? 'Post' : options[:model].try(:capitalize)

        if prefix.blank?
          @resolver_name = name.camelize(:upper)
          @file_name = underscore_name
          @field_name = underscore_name

          return
        end

        @resolver_name = "#{prefix.try(:capitalize)}::#{name.camelize(:upper)}"
        @file_name = "#{prefix}/#{underscore_name}"
        @field_name = "#{prefix}_#{underscore_name}"
      end
    end
  end
end