# frozen_string_literal: true

module Lokalise
  module Utils
    module AttributeHelpers
      using Lokalise::Utils::StringUtils

      UNIFIED_RESOURCES = %w[comment language].freeze

      # Returns the name of the API resource for the given class.
      # Most class names correspond to resource names (eg, `Project`, `Team`)
      # but some may differ (`ProjectComment` corresponds to `Comment` resource).
      # The resource name is in lowercase, with underscores as separators.
      # Some resources also have different pluralization rules. For example,
      # "CustomTranslationStatus" is "CustomTranslationStatuses" (-es postfix).
      # To address that, we try to fetch `DATA_KEY_PLURAL` set for the individual class.
      #
      # @return [String]
      # @param model_class [String]
      # @param plural [Boolean] Should the returned value be pluralized?
      def data_key_for(model_class:, plural: false, collection: false)
        data_key_plural = get_key(
          name: 'DATA_KEY_PLURAL',
          model_class: model_class,
          collection: true,
          strict: true
        )

        return data_key_plural if collection && data_key_plural

        data_key = get_key name: 'DATA_KEY', model_class: model_class, collection: collection

        return data_key unless plural

        "#{data_key}s"
      end

      # Returns key used to determine resource id (for example `user_id` or `project_id`).
      # Most ids corresponds to resources' class names, but some may differ (for instance,
      # `Contributor` has  `user_id` attribute).
      #
      # @return [String]
      # @param model_class [String]
      def id_key_for(model_class)
        get_key name: 'ID_KEY', model_class: model_class
      end

      # Loads attributes for the given resource based on its name
      #
      # @return [Array<String>]
      def attributes_for(klass)
        @attributes ||= YAML.load_file(File.expand_path('../data/attributes.json', __dir__)).freeze

        name = unify klass.name.snakecase
        @attributes[name]
      end

      private

      def get_key(name:, model_class:, collection: false, strict: false)
        key = if collection && Module.const_defined?("Lokalise::Collections::#{model_class}::#{name}")
                Module.const_get "Lokalise::Collections::#{model_class}::#{name}"
              elsif Module.const_defined? "Lokalise::Resources::#{model_class}::#{name}"
                Module.const_get "Lokalise::Resources::#{model_class}::#{name}"
              else
                strict ? nil : model_class
              end

        # Sometimes key is nil
        key ? key.snakecase : key
      end

      # Unify some resources' names (eg, `ProjectComment` and `KeyComment` have the same attributes which are stored under `comment`)
      #
      # @return [String]
      def unify(name)
        UNIFIED_RESOURCES.each do |u_a|
          return u_a if name.match?(/#{u_a}/)
        end

        name
      end
    end
  end
end