Introduce pipeline build seeds

1 files changed, 28 insertions, 0 deletions

diff --git a/lib/gitlab/ci/pipeline/seed/build.rb b/lib/gitlab/ci/pipeline/seed/build.rb
new file mode 100644
index 00000000000..f5dbb3a1c40
--- /dev/null
+++ b/lib/gitlab/ci/pipeline/seed/build.rb
@@ -0,0 +1,28 @@
+module Gitlab
+  module Ci
+    module Pipeline
+      module Seed
+        class Build < Seed::Base
+          def initialize(pipeline, attributes)
+            @pipeline = pipeline
+            @attributes = attributes
+          end
+
+          # TODO find a different solution
+          #
+          def user=(current_user)
+            @attributes.merge!(user: current_user)
+          end
+
+          def attributes
+            @attributes.merge(project: @pipeline.project,
+                              ref: @pipeline.ref,
+                              tag: @pipeline.tag,
+                              trigger_request: @pipeline.legacy_trigger,
+                              protected: @pipeline.protected_ref?)
+          end
+        end
+      end
+    end
+  end
+end