Extract base mappergh-refactor

1 files changed, 37 insertions, 0 deletions

diff --git a/lib/gitlab/import/github/mapper/base.rb b/lib/gitlab/import/github/mapper/base.rb
new file mode 100644
index 00000000000..78e8b97eb03
--- /dev/null
+++ b/lib/gitlab/import/github/mapper/base.rb
@@ -0,0 +1,37 @@
+module Gitlab
+  module Import
+    module Github
+      module Mapper
+        class Base
+          def initialize(project, client)
+            @project = project
+            @client  = client
+          end
+
+          def each
+            return enum_for(:each) unless block_given?
+
+            method = klass.to_s.underscore.pluralize
+
+            client.public_send(method).each do |raw|
+              yield(klass.new(attributes_for(raw)))
+            end
+          end
+
+          private
+
+          attr_reader :project, :client
+
+          def attributes_for
+            {}
+          end
+
+          def klass
+            raise NotImplementedError,
+              "#{self.class} does not implement #{__method__}"
+          end
+        end
+      end
+    end
+  end
+end