summaryrefslogtreecommitdiff
path: root/lib/gitlab/template/base_template.rb
blob: 760ff3e614a699b6055fb7573add67df79fca1e5 (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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
module Gitlab
  module Template
    class BaseTemplate
      def initialize(path)
        @path = path
      end

      def name
        File.basename(@path, self.class.extension)
      end

      def content
        File.read(@path)
      end

      class << self
        def all
          self.categories.keys.flat_map { |cat| by_category(cat) }
        end

        def find(key)
          file_name = "#{key}#{self.extension}"

          directory = select_directory(file_name)
          directory ? new(File.join(category_directory(directory), file_name)) : nil
        end

        def categories
          raise NotImplementedError
        end

        def extension
          raise NotImplementedError
        end

        def base_dir
          raise NotImplementedError
        end

        def by_category(category)
          templates_for_directory(category_directory(category))
        end

        def category_directory(category)
          File.join(base_dir, categories[category])
        end

        private

        def select_directory(file_name)
          categories.keys.find do |category|
            File.exist?(File.join(category_directory(category), file_name))
          end
        end

        def templates_for_directory(dir)
          dir << '/' unless dir.end_with?('/')
          Dir.glob(File.join(dir, "*#{self.extension}")).select { |f| f =~ filter_regex }.map { |f| new(f) }
        end

        def filter_regex
          @filter_reges ||= /#{Regexp.escape(extension)}\z/
        end
      end
    end
  end
end