summaryrefslogtreecommitdiff
path: root/lib/chef/chef_fs/file_system/repository/file_system_entry.rb
blob: 85f62d57ae6b21e421486e1b9470d89ba0b8fc59 (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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#
# Author:: John Keiser (<jkeiser@chef.io>)
# Copyright:: Copyright (c) Chef Software Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require_relative "../base_fs_dir"
require_relative "../chef_server/rest_list_dir"
require_relative "../exceptions"
require_relative "../../path_utils"
require "fileutils" unless defined?(FileUtils)

class Chef
  module ChefFS
    module FileSystem
      module Repository
        class FileSystemEntry < BaseFSDir
          def initialize(name, parent, file_path = nil, data_handler = nil)
            super(name, parent)
            @file_path = file_path || "#{parent.file_path}/#{name}"
            @data_handler = data_handler
          end

          attr_reader :file_path

          def write_pretty_json=(value)
            @write_pretty_json = value
          end

          def write_pretty_json
            @write_pretty_json.nil? ? root.write_pretty_json : @write_pretty_json
          end

          def data_handler
            @data_handler || parent.data_handler
          end

          def path_for_printing
            file_path
          end

          def chef_object
            data_handler.chef_object(Chef::JSONCompat.parse(read))
          rescue
            Chef::Log.error("Could not read #{path_for_printing} into a Chef object: #{$!}")
            nil
          end

          def can_have_child?(name, is_dir)
            !is_dir && File.extname(name) == ".json"
          end

          def name_valid?
            !name.start_with?(".")
          end

          # basic implementation to support Repository::Directory API
          def fs_entry_valid?
            name_valid? && File.exist?(file_path)
          end

          def minimize(file_contents, entry)
            object = Chef::JSONCompat.parse(file_contents)
            object = data_handler.normalize(object, entry)
            object = data_handler.minimize(object, entry)
            Chef::JSONCompat.to_json_pretty(object)
          end

          def children
            # Except cookbooks and data bag dirs, all things must be json files
            Dir.entries(file_path).sort
              .map { |child_name| make_child_entry(child_name) }
              .select { |new_child| new_child.fs_entry_valid? && can_have_child?(new_child.name, new_child.dir?) }
          rescue Errno::ENOENT
            raise Chef::ChefFS::FileSystem::NotFoundError.new(self, $!)
          end

          def create_child(child_name, file_contents = nil)
            child = make_child_entry(child_name)
            if child.exists?
              raise Chef::ChefFS::FileSystem::AlreadyExistsError.new(:create_child, child)
            end

            if file_contents
              child.write(file_contents)
            else
              Dir.mkdir(child.file_path)
            end
            child
          rescue Errno::EEXIST
            raise Chef::ChefFS::FileSystem::AlreadyExistsError.new(:create_child, child)
          end

          def dir?
            File.directory?(file_path)
          end

          def delete(recurse)
            if dir?
              unless recurse
                raise MustDeleteRecursivelyError.new(self, $!)
              end

              FileUtils.rm_r(file_path)
            else
              File.delete(file_path)
            end
          rescue Errno::ENOENT
            raise Chef::ChefFS::FileSystem::NotFoundError.new(self, $!)
          end

          def exists?
            File.exists?(file_path) && (parent.nil? || parent.can_have_child?(name, dir?))
          end

          def read
            File.open(file_path, "rb", &:read)
          rescue Errno::ENOENT
            raise Chef::ChefFS::FileSystem::NotFoundError.new(self, $!)
          end

          def write(file_contents)
            if file_contents && write_pretty_json && File.extname(name) == ".json"
              file_contents = minimize(file_contents, self)
            end
            File.open(file_path, "wb") do |file|
              file.write(file_contents)
            end
          end
          alias :create :write

          protected

          def make_child_entry(child_name)
            FileSystemEntry.new(child_name, self)
          end
        end
      end
    end
  end
end