1 files changed, 33 insertions, 0 deletions
diff --git a/lib/bulk_imports/pipeline/context.rb b/lib/bulk_imports/pipeline/context.rb
new file mode 100644
index 00000000000..ad19f5cad7d
--- /dev/null
+++ b/lib/bulk_imports/pipeline/context.rb
@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module BulkImports
+  module Pipeline
+    class Context
+      include Gitlab::Utils::LazyAttributes
+
+      Attribute = Struct.new(:name, :type)
+
+      PIPELINE_ATTRIBUTES = [
+        Attribute.new(:current_user, User),
+        Attribute.new(:entity, ::BulkImports::Entity),
+        Attribute.new(:configuration, ::BulkImports::Configuration)
+      ].freeze
+
+      def initialize(args)
+        assign_attributes(args)
+      end
+
+      private
+
+      PIPELINE_ATTRIBUTES.each do |attr|
+        lazy_attr_reader attr.name, type: attr.type
+      end
+
+      def assign_attributes(values)
+        values.slice(*PIPELINE_ATTRIBUTES.map(&:name)).each do |name, value|
+          instance_variable_set("@#{name}", value)
+        end
+      end
+    end
+  end
+end