1 files changed, 46 insertions, 0 deletions
diff --git a/lib/gitlab/github_import/representation/protected_branch.rb b/lib/gitlab/github_import/representation/protected_branch.rb
new file mode 100644
index 00000000000..b80b7cf1076
--- /dev/null
+++ b/lib/gitlab/github_import/representation/protected_branch.rb
@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module GithubImport
+    module Representation
+      class ProtectedBranch
+        include ToHash
+        include ExposeAttribute
+
+        attr_reader :attributes
+
+        expose_attribute :id, :allow_force_pushes
+
+        # Builds a Branch Protection info from a GitHub API response.
+        # Resource structure details:
+        # https://docs.github.com/en/rest/branches/branch-protection#get-branch-protection
+        # branch_protection - An instance of `Sawyer::Resource` containing the protection details.
+        def self.from_api_response(branch_protection, _additional_object_data = {})
+          branch_name = branch_protection.url.match(%r{/branches/(\S{1,255})/protection$})[1]
+
+          hash = {
+            id: branch_name,
+            allow_force_pushes: branch_protection.allow_force_pushes.enabled
+          }
+
+          new(hash)
+        end
+
+        # Builds a new Protection using a Hash that was built from a JSON payload.
+        def self.from_json_hash(raw_hash)
+          new(Representation.symbolize_hash(raw_hash))
+        end
+
+        # attributes - A Hash containing the raw Protection details. The keys of this
+        #              Hash (and any nested hashes) must be symbols.
+        def initialize(attributes)
+          @attributes = attributes
+        end
+
+        def github_identifiers
+          { id: id }
+        end
+      end
+    end
+  end
+end