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
|
#
# Author:: John Keiser (<jkeiser@getchef.com>)
# Copyright:: Copyright (c) 2013 Opscode, 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 'chef/config'
if Chef::Platform.windows?
if Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.1')
require 'chef/monkey_patches/webrick-utils'
end
end
class Chef
module LocalMode
# Create a chef local server (if the configuration requires one) for the
# duration of the given block.
#
# # This ...
# with_server_connectivity { stuff }
#
# # Is exactly equivalent to this ...
# Chef::LocalMode.setup_server_connectivity
# begin
# stuff
# ensure
# Chef::LocalMode.destroy_server_connectivity
# end
#
def self.with_server_connectivity
setup_server_connectivity
begin
yield
ensure
destroy_server_connectivity
end
end
# If Chef::Config.chef_zero.enabled is true, sets up a chef-zero server
# according to the Chef::Config.chef_zero and path options, and sets
# chef_server_url to point at it.
def self.setup_server_connectivity
if Chef::Config.chef_zero.enabled
destroy_server_connectivity
require 'chef_zero/server'
require 'chef/chef_fs/chef_fs_data_store'
require 'chef/chef_fs/config'
@chef_fs = Chef::ChefFS::Config.new.local_fs
@chef_fs.write_pretty_json = true
data_store = Chef::ChefFS::ChefFSDataStore.new(@chef_fs)
data_store = ChefZero::DataStore::V1ToV2Adapter.new(data_store, 'chef')
server_options = {}
server_options[:data_store] = data_store
server_options[:log_level] = Chef::Log.level
server_options[:host] = Chef::Config.chef_zero.host
server_options[:port] = parse_port(Chef::Config.chef_zero.port)
@chef_zero_server = ChefZero::Server.new(server_options)
if Chef::Config[:listen]
@chef_zero_server.start_background
else
@chef_zero_server.start_socketless
end
local_mode_url = @chef_zero_server.local_mode_url
Chef::Log.info("Started chef-zero at #{local_mode_url} with #{@chef_fs.fs_description}")
Chef::Config.chef_server_url = local_mode_url
end
end
# Return the current chef-zero server set up by setup_server_connectivity.
def self.chef_zero_server
@chef_zero_server
end
# Return the chef_fs object for the current chef-zero server.
def self.chef_fs
@chef_fs
end
# If chef_zero_server is non-nil, stop it and remove references to it.
def self.destroy_server_connectivity
if @chef_zero_server
@chef_zero_server.stop
@chef_zero_server = nil
end
end
def self.parse_port(port)
if port.is_a?(String)
parts = port.split(',')
if parts.size == 1
a,b = parts[0].split('-',2)
if b
a.to_i.upto(b.to_i)
else
[ a.to_i ]
end
else
array = []
parts.each do |part|
array += parse_port(part).to_a
end
array
end
else
port
end
end
end
end
|
