summaryrefslogtreecommitdiff
path: root/lib/gitlab/graphql/representation/tree_entry.rb
blob: 7ea83db5876b53b9df555db87e28541800538447 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# frozen_string_literal: true

module Gitlab
  module Graphql
    module Representation
      class TreeEntry < SimpleDelegator
        class << self
          def decorate(entries, repository)
            return if entries.nil?

            entries.map do |entry|
              if entry.is_a?(TreeEntry)
                entry
              else
                self.new(entry, repository)
              end
            end
          end
        end

        attr_accessor :repository

        def initialize(raw_entry, repository)
          @repository = repository

          super(raw_entry)
        end
      end
    end
  end
end