Add basic representations for the Github API results

1 files changed, 52 insertions, 0 deletions

diff --git a/lib/github/representation/branch.rb b/lib/github/representation/branch.rb
new file mode 100644
index 00000000000..7c65a948ede
--- /dev/null
+++ b/lib/github/representation/branch.rb
@@ -0,0 +1,52 @@
+module Github
+  module Representation
+    class Branch < Representation::Base
+      attr_reader :repository
+
+      def initialize(repository, raw)
+        @repository = repository
+        @raw = raw
+      end
+
+      def user
+        raw.dig('user', 'login') || 'unknown'
+      end
+
+      def repo
+        return @repo if defined?(@repo)
+
+        @repo = Github::Representation::Repo.new(raw['repo']) if raw['repo'].present?
+      end
+
+      def ref
+        raw['ref']
+      end
+
+      def sha
+        raw['sha']
+      end
+
+      def short_sha
+        Commit.truncate_sha(sha)
+      end
+
+      def exists?
+        branch_exists? && commit_exists?
+      end
+
+      def valid?
+        sha.present? && ref.present?
+      end
+
+      private
+
+      def branch_exists?
+        repository.branch_exists?(ref)
+      end
+
+      def commit_exists?
+        repository.branch_names_contains(sha).include?(ref)
+      end
+    end
+  end
+end