# frozen_string_literal: true

module BulkImports
  module Pipeline
    module Attributes
      extend ActiveSupport::Concern
      include Gitlab::ClassAttributes

      class_methods do
        def extractor(klass, options = nil)
          add_attribute(:extractors, klass, options)
        end

        def transformer(klass, options = nil)
          add_attribute(:transformers, klass, options)
        end

        def loader(klass, options = nil)
          add_attribute(:loaders, klass, options)
        end

        def add_attribute(sym, klass, options)
          class_attributes[sym] ||= []
          class_attributes[sym] << { klass: klass, options: options }
        end

        def extractors
          class_attributes[:extractors]
        end

        def transformers
          class_attributes[:transformers]
        end

        def loaders
          class_attributes[:loaders]
        end
      end
    end
  end
end