tecnh-chef 0.9.9.vpc

Sign up to get free protection for your applications and to get access to all the features.
Files changed (337) hide show
  1. data/LICENSE +201 -0
  2. data/README.rdoc +172 -0
  3. data/bin/chef-client +26 -0
  4. data/bin/chef-solo +25 -0
  5. data/bin/knife +26 -0
  6. data/bin/shef +34 -0
  7. data/distro/README +2 -0
  8. data/distro/common/man/man1/chef-indexer.1 +42 -0
  9. data/distro/common/man/man1/chef-server-webui.1 +106 -0
  10. data/distro/common/man/man1/chef-server.1 +107 -0
  11. data/distro/common/man/man1/chef-solr-indexer.1 +55 -0
  12. data/distro/common/man/man1/chef-solr.1 +55 -0
  13. data/distro/common/man/man8/chef-client.8 +63 -0
  14. data/distro/common/man/man8/chef-solo.8 +57 -0
  15. data/distro/common/man/man8/chef-solr-rebuild.8 +37 -0
  16. data/distro/common/man/man8/knife.8 +1339 -0
  17. data/distro/common/man/man8/shef.8 +45 -0
  18. data/distro/common/markdown/README +3 -0
  19. data/distro/common/markdown/knife.mkd +825 -0
  20. data/distro/debian/etc/default/chef-client +4 -0
  21. data/distro/debian/etc/default/chef-server +6 -0
  22. data/distro/debian/etc/default/chef-server-webui +6 -0
  23. data/distro/debian/etc/default/chef-solr +4 -0
  24. data/distro/debian/etc/default/chef-solr-indexer +4 -0
  25. data/distro/debian/etc/init.d/chef-client +175 -0
  26. data/distro/debian/etc/init.d/chef-server +120 -0
  27. data/distro/debian/etc/init.d/chef-server-webui +121 -0
  28. data/distro/debian/etc/init.d/chef-solr +177 -0
  29. data/distro/debian/etc/init.d/chef-solr-indexer +176 -0
  30. data/distro/debian/etc/init/chef-client.conf +17 -0
  31. data/distro/debian/etc/init/chef-server-webui.conf +17 -0
  32. data/distro/debian/etc/init/chef-server.conf +17 -0
  33. data/distro/debian/etc/init/chef-solr-indexer.conf +17 -0
  34. data/distro/debian/etc/init/chef-solr.conf +17 -0
  35. data/distro/redhat/etc/init.d/chef-client +106 -0
  36. data/distro/redhat/etc/init.d/chef-server +112 -0
  37. data/distro/redhat/etc/init.d/chef-server-webui +112 -0
  38. data/distro/redhat/etc/init.d/chef-solr +104 -0
  39. data/distro/redhat/etc/init.d/chef-solr-indexer +104 -0
  40. data/distro/redhat/etc/logrotate.d/chef-client +8 -0
  41. data/distro/redhat/etc/logrotate.d/chef-server +8 -0
  42. data/distro/redhat/etc/logrotate.d/chef-server-webui +8 -0
  43. data/distro/redhat/etc/logrotate.d/chef-solr +8 -0
  44. data/distro/redhat/etc/logrotate.d/chef-solr-indexer +8 -0
  45. data/distro/redhat/etc/sysconfig/chef-client +15 -0
  46. data/distro/redhat/etc/sysconfig/chef-server +14 -0
  47. data/distro/redhat/etc/sysconfig/chef-server-webui +14 -0
  48. data/distro/redhat/etc/sysconfig/chef-solr +8 -0
  49. data/distro/redhat/etc/sysconfig/chef-solr-indexer +7 -0
  50. data/lib/chef.rb +39 -0
  51. data/lib/chef/api_client.rb +264 -0
  52. data/lib/chef/application.rb +125 -0
  53. data/lib/chef/application/agent.rb +18 -0
  54. data/lib/chef/application/client.rb +237 -0
  55. data/lib/chef/application/knife.rb +160 -0
  56. data/lib/chef/application/solo.rb +216 -0
  57. data/lib/chef/applications.rb +4 -0
  58. data/lib/chef/cache.rb +61 -0
  59. data/lib/chef/cache/checksum.rb +91 -0
  60. data/lib/chef/certificate.rb +194 -0
  61. data/lib/chef/checksum.rb +118 -0
  62. data/lib/chef/client.rb +340 -0
  63. data/lib/chef/config.rb +240 -0
  64. data/lib/chef/cookbook/cookbook_collection.rb +44 -0
  65. data/lib/chef/cookbook/file_system_file_vendor.rb +54 -0
  66. data/lib/chef/cookbook/file_vendor.rb +48 -0
  67. data/lib/chef/cookbook/metadata.rb +500 -0
  68. data/lib/chef/cookbook/metadata/version.rb +87 -0
  69. data/lib/chef/cookbook/remote_file_vendor.rb +80 -0
  70. data/lib/chef/cookbook/syntax_check.rb +136 -0
  71. data/lib/chef/cookbook_loader.rb +225 -0
  72. data/lib/chef/cookbook_site_streaming_uploader.rb +215 -0
  73. data/lib/chef/cookbook_uploader.rb +102 -0
  74. data/lib/chef/cookbook_version.rb +877 -0
  75. data/lib/chef/couchdb.rb +239 -0
  76. data/lib/chef/daemon.rb +172 -0
  77. data/lib/chef/data_bag.rb +215 -0
  78. data/lib/chef/data_bag_item.rb +228 -0
  79. data/lib/chef/exceptions.rb +61 -0
  80. data/lib/chef/file_access_control.rb +140 -0
  81. data/lib/chef/file_cache.rb +218 -0
  82. data/lib/chef/handler.rb +163 -0
  83. data/lib/chef/handler/json_file.rb +58 -0
  84. data/lib/chef/index_queue.rb +29 -0
  85. data/lib/chef/index_queue/amqp_client.rb +113 -0
  86. data/lib/chef/index_queue/consumer.rb +76 -0
  87. data/lib/chef/index_queue/indexable.rb +76 -0
  88. data/lib/chef/knife.rb +496 -0
  89. data/lib/chef/knife/bootstrap.rb +167 -0
  90. data/lib/chef/knife/bootstrap/centos5-gems.erb +50 -0
  91. data/lib/chef/knife/bootstrap/fedora13-gems.erb +47 -0
  92. data/lib/chef/knife/bootstrap/ubuntu10.04-apt.erb +41 -0
  93. data/lib/chef/knife/bootstrap/ubuntu10.04-gems.erb +53 -0
  94. data/lib/chef/knife/client_bulk_delete.rb +40 -0
  95. data/lib/chef/knife/client_create.rb +70 -0
  96. data/lib/chef/knife/client_delete.rb +45 -0
  97. data/lib/chef/knife/client_edit.rb +45 -0
  98. data/lib/chef/knife/client_list.rb +40 -0
  99. data/lib/chef/knife/client_reregister.rb +56 -0
  100. data/lib/chef/knife/client_show.rb +50 -0
  101. data/lib/chef/knife/configure.rb +140 -0
  102. data/lib/chef/knife/configure_client.rb +50 -0
  103. data/lib/chef/knife/cookbook_bulk_delete.rb +58 -0
  104. data/lib/chef/knife/cookbook_create.rb +209 -0
  105. data/lib/chef/knife/cookbook_delete.rb +143 -0
  106. data/lib/chef/knife/cookbook_download.rb +130 -0
  107. data/lib/chef/knife/cookbook_list.rb +41 -0
  108. data/lib/chef/knife/cookbook_metadata.rb +82 -0
  109. data/lib/chef/knife/cookbook_metadata_from_file.rb +40 -0
  110. data/lib/chef/knife/cookbook_show.rb +98 -0
  111. data/lib/chef/knife/cookbook_site_download.rb +58 -0
  112. data/lib/chef/knife/cookbook_site_list.rb +56 -0
  113. data/lib/chef/knife/cookbook_site_search.rb +51 -0
  114. data/lib/chef/knife/cookbook_site_share.rb +108 -0
  115. data/lib/chef/knife/cookbook_site_show.rb +57 -0
  116. data/lib/chef/knife/cookbook_site_unshare.rb +52 -0
  117. data/lib/chef/knife/cookbook_site_vendor.rb +127 -0
  118. data/lib/chef/knife/cookbook_test.rb +82 -0
  119. data/lib/chef/knife/cookbook_upload.rb +95 -0
  120. data/lib/chef/knife/data_bag_create.rb +59 -0
  121. data/lib/chef/knife/data_bag_delete.rb +48 -0
  122. data/lib/chef/knife/data_bag_edit.rb +50 -0
  123. data/lib/chef/knife/data_bag_from_file.rb +44 -0
  124. data/lib/chef/knife/data_bag_list.rb +43 -0
  125. data/lib/chef/knife/data_bag_show.rb +41 -0
  126. data/lib/chef/knife/ec2_instance_data.rb +46 -0
  127. data/lib/chef/knife/ec2_server_create.rb +211 -0
  128. data/lib/chef/knife/ec2_server_delete.rb +90 -0
  129. data/lib/chef/knife/ec2_server_list.rb +88 -0
  130. data/lib/chef/knife/index_rebuild.rb +51 -0
  131. data/lib/chef/knife/node_bulk_delete.rb +43 -0
  132. data/lib/chef/knife/node_create.rb +47 -0
  133. data/lib/chef/knife/node_delete.rb +44 -0
  134. data/lib/chef/knife/node_edit.rb +44 -0
  135. data/lib/chef/knife/node_from_file.rb +42 -0
  136. data/lib/chef/knife/node_list.rb +41 -0
  137. data/lib/chef/knife/node_run_list_add.rb +64 -0
  138. data/lib/chef/knife/node_run_list_remove.rb +45 -0
  139. data/lib/chef/knife/node_show.rb +54 -0
  140. data/lib/chef/knife/rackspace_server_create.rb +156 -0
  141. data/lib/chef/knife/rackspace_server_delete.rb +57 -0
  142. data/lib/chef/knife/rackspace_server_list.rb +59 -0
  143. data/lib/chef/knife/recipe_list.rb +32 -0
  144. data/lib/chef/knife/role_bulk_delete.rb +44 -0
  145. data/lib/chef/knife/role_create.rb +52 -0
  146. data/lib/chef/knife/role_delete.rb +44 -0
  147. data/lib/chef/knife/role_edit.rb +45 -0
  148. data/lib/chef/knife/role_from_file.rb +46 -0
  149. data/lib/chef/knife/role_list.rb +40 -0
  150. data/lib/chef/knife/role_show.rb +51 -0
  151. data/lib/chef/knife/search.rb +94 -0
  152. data/lib/chef/knife/slicehost_images_list.rb +53 -0
  153. data/lib/chef/knife/slicehost_server_create.rb +103 -0
  154. data/lib/chef/knife/slicehost_server_delete.rb +61 -0
  155. data/lib/chef/knife/slicehost_server_list.rb +64 -0
  156. data/lib/chef/knife/ssh.rb +311 -0
  157. data/lib/chef/knife/status.rb +70 -0
  158. data/lib/chef/knife/terremark_server_create.rb +152 -0
  159. data/lib/chef/knife/terremark_server_delete.rb +87 -0
  160. data/lib/chef/knife/terremark_server_list.rb +77 -0
  161. data/lib/chef/log.rb +61 -0
  162. data/lib/chef/mixin/check_helper.rb +31 -0
  163. data/lib/chef/mixin/checksum.rb +32 -0
  164. data/lib/chef/mixin/command.rb +220 -0
  165. data/lib/chef/mixin/command/unix.rb +215 -0
  166. data/lib/chef/mixin/command/windows.rb +72 -0
  167. data/lib/chef/mixin/convert_to_class_name.rb +63 -0
  168. data/lib/chef/mixin/create_path.rb +56 -0
  169. data/lib/chef/mixin/deep_merge.rb +225 -0
  170. data/lib/chef/mixin/deprecation.rb +65 -0
  171. data/lib/chef/mixin/find_preferred_file.rb +92 -0
  172. data/lib/chef/mixin/from_file.rb +50 -0
  173. data/lib/chef/mixin/language.rb +156 -0
  174. data/lib/chef/mixin/language_include_attribute.rb +61 -0
  175. data/lib/chef/mixin/language_include_recipe.rb +52 -0
  176. data/lib/chef/mixin/params_validate.rb +225 -0
  177. data/lib/chef/mixin/recipe_definition_dsl_core.rb +81 -0
  178. data/lib/chef/mixin/shell_out.rb +38 -0
  179. data/lib/chef/mixin/template.rb +95 -0
  180. data/lib/chef/mixin/xml_escape.rb +140 -0
  181. data/lib/chef/mixins.rb +16 -0
  182. data/lib/chef/monkey_patches/dir.rb +36 -0
  183. data/lib/chef/monkey_patches/string.rb +28 -0
  184. data/lib/chef/monkey_patches/tempfile.rb +64 -0
  185. data/lib/chef/node.rb +605 -0
  186. data/lib/chef/node/attribute.rb +450 -0
  187. data/lib/chef/openid_registration.rb +187 -0
  188. data/lib/chef/platform.rb +354 -0
  189. data/lib/chef/provider.rb +124 -0
  190. data/lib/chef/provider/breakpoint.rb +36 -0
  191. data/lib/chef/provider/cookbook_file.rb +96 -0
  192. data/lib/chef/provider/cron.rb +186 -0
  193. data/lib/chef/provider/deploy.rb +319 -0
  194. data/lib/chef/provider/deploy/revision.rb +73 -0
  195. data/lib/chef/provider/deploy/timestamped.rb +33 -0
  196. data/lib/chef/provider/directory.rb +72 -0
  197. data/lib/chef/provider/env.rb +152 -0
  198. data/lib/chef/provider/env/windows.rb +75 -0
  199. data/lib/chef/provider/erl_call.rb +72 -0
  200. data/lib/chef/provider/execute.rb +58 -0
  201. data/lib/chef/provider/file.rb +212 -0
  202. data/lib/chef/provider/git.rb +203 -0
  203. data/lib/chef/provider/group.rb +133 -0
  204. data/lib/chef/provider/group/dscl.rb +121 -0
  205. data/lib/chef/provider/group/gpasswd.rb +53 -0
  206. data/lib/chef/provider/group/groupadd.rb +78 -0
  207. data/lib/chef/provider/group/pw.rb +84 -0
  208. data/lib/chef/provider/group/usermod.rb +57 -0
  209. data/lib/chef/provider/group/windows.rb +79 -0
  210. data/lib/chef/provider/http_request.rb +106 -0
  211. data/lib/chef/provider/ifconfig.rb +132 -0
  212. data/lib/chef/provider/link.rb +156 -0
  213. data/lib/chef/provider/log.rb +53 -0
  214. data/lib/chef/provider/mdadm.rb +91 -0
  215. data/lib/chef/provider/mount.rb +117 -0
  216. data/lib/chef/provider/mount/mount.rb +227 -0
  217. data/lib/chef/provider/mount/windows.rb +80 -0
  218. data/lib/chef/provider/package.rb +160 -0
  219. data/lib/chef/provider/package/apt.rb +110 -0
  220. data/lib/chef/provider/package/dpkg.rb +112 -0
  221. data/lib/chef/provider/package/easy_install.rb +114 -0
  222. data/lib/chef/provider/package/freebsd.rb +123 -0
  223. data/lib/chef/provider/package/macports.rb +105 -0
  224. data/lib/chef/provider/package/pacman.rb +101 -0
  225. data/lib/chef/provider/package/portage.rb +124 -0
  226. data/lib/chef/provider/package/rpm.rb +101 -0
  227. data/lib/chef/provider/package/rubygems.rb +462 -0
  228. data/lib/chef/provider/package/solaris.rb +127 -0
  229. data/lib/chef/provider/package/yum-dump.py +125 -0
  230. data/lib/chef/provider/package/yum.rb +222 -0
  231. data/lib/chef/provider/package/zypper.rb +133 -0
  232. data/lib/chef/provider/remote_directory.rb +134 -0
  233. data/lib/chef/provider/remote_file.rb +120 -0
  234. data/lib/chef/provider/route.rb +195 -0
  235. data/lib/chef/provider/ruby_block.rb +33 -0
  236. data/lib/chef/provider/script.rb +55 -0
  237. data/lib/chef/provider/service.rb +128 -0
  238. data/lib/chef/provider/service/arch.rb +109 -0
  239. data/lib/chef/provider/service/debian.rb +105 -0
  240. data/lib/chef/provider/service/freebsd.rb +156 -0
  241. data/lib/chef/provider/service/gentoo.rb +54 -0
  242. data/lib/chef/provider/service/init.rb +71 -0
  243. data/lib/chef/provider/service/redhat.rb +60 -0
  244. data/lib/chef/provider/service/simple.rb +118 -0
  245. data/lib/chef/provider/service/solaris.rb +85 -0
  246. data/lib/chef/provider/service/upstart.rb +192 -0
  247. data/lib/chef/provider/service/windows.rb +129 -0
  248. data/lib/chef/provider/subversion.rb +156 -0
  249. data/lib/chef/provider/template.rb +105 -0
  250. data/lib/chef/provider/user.rb +187 -0
  251. data/lib/chef/provider/user/dscl.rb +280 -0
  252. data/lib/chef/provider/user/pw.rb +113 -0
  253. data/lib/chef/provider/user/useradd.rb +109 -0
  254. data/lib/chef/provider/user/windows.rb +124 -0
  255. data/lib/chef/providers.rb +89 -0
  256. data/lib/chef/recipe.rb +130 -0
  257. data/lib/chef/resource.rb +417 -0
  258. data/lib/chef/resource/apt_package.rb +34 -0
  259. data/lib/chef/resource/bash.rb +33 -0
  260. data/lib/chef/resource/breakpoint.rb +35 -0
  261. data/lib/chef/resource/cookbook_file.rb +45 -0
  262. data/lib/chef/resource/cron.rb +188 -0
  263. data/lib/chef/resource/csh.rb +33 -0
  264. data/lib/chef/resource/deploy.rb +371 -0
  265. data/lib/chef/resource/deploy_revision.rb +35 -0
  266. data/lib/chef/resource/directory.rb +76 -0
  267. data/lib/chef/resource/dpkg_package.rb +34 -0
  268. data/lib/chef/resource/easy_install_package.rb +41 -0
  269. data/lib/chef/resource/env.rb +58 -0
  270. data/lib/chef/resource/erl_call.rb +83 -0
  271. data/lib/chef/resource/execute.rb +127 -0
  272. data/lib/chef/resource/file.rb +92 -0
  273. data/lib/chef/resource/freebsd_package.rb +35 -0
  274. data/lib/chef/resource/gem_package.rb +49 -0
  275. data/lib/chef/resource/git.rb +36 -0
  276. data/lib/chef/resource/group.rb +70 -0
  277. data/lib/chef/resource/http_request.rb +52 -0
  278. data/lib/chef/resource/ifconfig.rb +134 -0
  279. data/lib/chef/resource/link.rb +78 -0
  280. data/lib/chef/resource/log.rb +62 -0
  281. data/lib/chef/resource/macports_package.rb +29 -0
  282. data/lib/chef/resource/mdadm.rb +82 -0
  283. data/lib/chef/resource/mount.rb +135 -0
  284. data/lib/chef/resource/package.rb +80 -0
  285. data/lib/chef/resource/pacman_package.rb +33 -0
  286. data/lib/chef/resource/perl.rb +33 -0
  287. data/lib/chef/resource/portage_package.rb +33 -0
  288. data/lib/chef/resource/python.rb +33 -0
  289. data/lib/chef/resource/remote_directory.rb +100 -0
  290. data/lib/chef/resource/remote_file.rb +83 -0
  291. data/lib/chef/resource/route.rb +135 -0
  292. data/lib/chef/resource/ruby.rb +33 -0
  293. data/lib/chef/resource/ruby_block.rb +40 -0
  294. data/lib/chef/resource/scm.rb +146 -0
  295. data/lib/chef/resource/script.rb +51 -0
  296. data/lib/chef/resource/service.rb +160 -0
  297. data/lib/chef/resource/solaris_package.rb +36 -0
  298. data/lib/chef/resource/subversion.rb +36 -0
  299. data/lib/chef/resource/template.rb +69 -0
  300. data/lib/chef/resource/timestamped_deploy.rb +31 -0
  301. data/lib/chef/resource/user.rb +130 -0
  302. data/lib/chef/resource/yum_package.rb +43 -0
  303. data/lib/chef/resource_collection.rb +216 -0
  304. data/lib/chef/resource_collection/stepable_iterator.rb +124 -0
  305. data/lib/chef/resource_definition.rb +67 -0
  306. data/lib/chef/resource_definition_list.rb +38 -0
  307. data/lib/chef/resources.rb +62 -0
  308. data/lib/chef/rest.rb +388 -0
  309. data/lib/chef/rest/auth_credentials.rb +78 -0
  310. data/lib/chef/rest/cookie_jar.rb +31 -0
  311. data/lib/chef/rest/rest_request.rb +186 -0
  312. data/lib/chef/role.rb +287 -0
  313. data/lib/chef/run_context.rb +110 -0
  314. data/lib/chef/run_list.rb +150 -0
  315. data/lib/chef/run_list/run_list_expansion.rb +172 -0
  316. data/lib/chef/run_list/run_list_item.rb +78 -0
  317. data/lib/chef/run_status.rb +121 -0
  318. data/lib/chef/runner.rb +131 -0
  319. data/lib/chef/sandbox.rb +153 -0
  320. data/lib/chef/search/query.rb +60 -0
  321. data/lib/chef/shef.rb +325 -0
  322. data/lib/chef/shef/ext.rb +565 -0
  323. data/lib/chef/shef/model_wrapper.rb +120 -0
  324. data/lib/chef/shef/shef_rest.rb +28 -0
  325. data/lib/chef/shef/shef_session.rb +271 -0
  326. data/lib/chef/shell_out.rb +413 -0
  327. data/lib/chef/streaming_cookbook_uploader.rb +201 -0
  328. data/lib/chef/tasks/chef_repo.rake +256 -0
  329. data/lib/chef/util/file_edit.rb +122 -0
  330. data/lib/chef/util/windows.rb +56 -0
  331. data/lib/chef/util/windows/net_group.rb +101 -0
  332. data/lib/chef/util/windows/net_use.rb +121 -0
  333. data/lib/chef/util/windows/net_user.rb +198 -0
  334. data/lib/chef/util/windows/volume.rb +59 -0
  335. data/lib/chef/version.rb +21 -0
  336. data/lib/chef/webui_user.rb +232 -0
  337. metadata +614 -0
@@ -0,0 +1,120 @@
1
+ #--
2
+ # Author:: Daniel DeLeo (<dan@opscode.com>)
3
+ # Copyright:: Copyright (c) 2010 Opscode, Inc.
4
+ # License:: Apache License, Version 2.0
5
+ #
6
+ # Licensed under the Apache License, Version 2.0 (the "License");
7
+ # you may not use this file except in compliance with the License.
8
+ # You may obtain a copy of the License at
9
+ #
10
+ # http://www.apache.org/licenses/LICENSE-2.0
11
+ #
12
+ # Unless required by applicable law or agreed to in writing, software
13
+ # distributed under the License is distributed on an "AS IS" BASIS,
14
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
+ # See the License for the specific language governing permissions and
16
+ # limitations under the License.
17
+ #
18
+
19
+ require 'chef/mixin/convert_to_class_name'
20
+ require 'chef/mixin/language'
21
+
22
+ module Shef
23
+ class ModelWrapper
24
+
25
+ include Chef::Mixin::ConvertToClassName
26
+
27
+ attr_reader :model_symbol
28
+
29
+ def initialize(model_class, symbol=nil)
30
+ @model_class = model_class
31
+ @model_symbol = symbol || convert_to_snake_case(model_class.name, "Chef").to_sym
32
+ end
33
+
34
+ def search(query)
35
+ return all if query.to_s == "all"
36
+ results = []
37
+ Chef::Search::Query.new.search(@model_symbol, format_query(query)) do |obj|
38
+ if block_given?
39
+ results << yield(obj)
40
+ else
41
+ results << obj
42
+ end
43
+ end
44
+ results
45
+ end
46
+
47
+ alias :find :search
48
+
49
+ def all(&block)
50
+ all_objects = list_objects
51
+ block_given? ? all_objects.map(&block) : all_objects
52
+ end
53
+
54
+ alias :list :all
55
+
56
+ def show(obj_id)
57
+ @model_class.load(obj_id)
58
+ end
59
+
60
+ alias :load :show
61
+
62
+ def transform(what_to_transform, &block)
63
+ if what_to_transform == :all
64
+ objects_to_transform = list_objects
65
+ else
66
+ objects_to_transform = search(what_to_transform)
67
+ end
68
+ objects_to_transform.each do |obj|
69
+ if result = yield(obj)
70
+ obj.save
71
+ end
72
+ end
73
+ end
74
+
75
+ alias :bulk_edit :transform
76
+
77
+ private
78
+
79
+ # paper over inconsistencies in the model classes APIs, and return the objects
80
+ # the user wanted instead of the URI=>object stuff
81
+ def list_objects
82
+ objects = @model_class.method(:list).arity == 0? @model_class.list : @model_class.list(true)
83
+ objects.map { |obj| Array(obj).find {|o| o.kind_of?(@model_class)} }
84
+ end
85
+
86
+ def format_query(query)
87
+ if query.respond_to?(:keys)
88
+ query.map { |key, value| "#{key}:#{value}" }.join(" AND ")
89
+ else
90
+ query
91
+ end
92
+ end
93
+ end
94
+
95
+ class NamedDataBagWrapper < ModelWrapper
96
+
97
+ def initialize(databag_name)
98
+ @model_symbol = @databag_name = databag_name
99
+ end
100
+
101
+
102
+ alias :list :all
103
+
104
+ def show(item)
105
+ Chef::DataBagItem.load(@databag_name, item)
106
+ end
107
+
108
+ private
109
+
110
+ def list_objects
111
+ all_items = []
112
+ Chef::Search::Query.new.search(@databag_name) do |item|
113
+ all_items << item
114
+ end
115
+ all_items
116
+ end
117
+
118
+ end
119
+
120
+ end
@@ -0,0 +1,28 @@
1
+ #--
2
+ # Author:: Daniel DeLeo (<dan@opscode.com>)
3
+ # Copyright:: Copyright (c) 2010 Opscode, Inc.
4
+ # License:: Apache License, Version 2.0
5
+ #
6
+ # Licensed under the Apache License, Version 2.0 (the "License");
7
+ # you may not use this file except in compliance with the License.
8
+ # You may obtain a copy of the License at
9
+ #
10
+ # http://www.apache.org/licenses/LICENSE-2.0
11
+ #
12
+ # Unless required by applicable law or agreed to in writing, software
13
+ # distributed under the License is distributed on an "AS IS" BASIS,
14
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
+ # See the License for the specific language governing permissions and
16
+ # limitations under the License.
17
+ #
18
+
19
+ module Shef
20
+ class ShefREST < Chef::REST
21
+
22
+ alias :get :get_rest
23
+ alias :put :put_rest
24
+ alias :post :post_rest
25
+ alias :delete :delete_rest
26
+
27
+ end
28
+ end
@@ -0,0 +1,271 @@
1
+ #--
2
+ # Author:: Daniel DeLeo (<dan@kallistec.com>)
3
+ # Copyright:: Copyright (c) 2009 Daniel DeLeo
4
+ # License:: Apache License, Version 2.0
5
+ #
6
+ # Licensed under the Apache License, Version 2.0 (the "License");
7
+ # you may not use this file except in compliance with the License.
8
+ # You may obtain a copy of the License at
9
+ #
10
+ # http://www.apache.org/licenses/LICENSE-2.0
11
+ #
12
+ # Unless required by applicable law or agreed to in writing, software
13
+ # distributed under the License is distributed on an "AS IS" BASIS,
14
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
+ # See the License for the specific language governing permissions and
16
+ # limitations under the License.
17
+ #
18
+
19
+ require 'chef/recipe'
20
+ require 'chef/run_context'
21
+ require 'chef/config'
22
+ require 'chef/client'
23
+ require 'chef/cookbook/cookbook_collection'
24
+ require 'chef/cookbook_loader'
25
+
26
+ module Shef
27
+ class ShefSession
28
+ include Singleton
29
+
30
+ def self.session_type(type=nil)
31
+ @session_type = type if type
32
+ @session_type
33
+ end
34
+
35
+ attr_accessor :node, :compile, :recipe, :run_context
36
+ attr_reader :node_attributes, :client
37
+ def initialize
38
+ @node_built = false
39
+ end
40
+
41
+ def node_built?
42
+ !!@node_built
43
+ end
44
+
45
+ def reset!
46
+ loading do
47
+ rebuild_node
48
+ @node = client.node
49
+ shorten_node_inspect
50
+ Shef::Extensions.extend_context_node(@node)
51
+ rebuild_context
52
+ node.consume_attributes(node_attributes) if node_attributes
53
+ @recipe = Chef::Recipe.new(nil, nil, run_context)
54
+ Shef::Extensions.extend_context_recipe(@recipe)
55
+ @node_built = true
56
+ end
57
+ end
58
+
59
+ def node_attributes=(attrs)
60
+ @node_attributes = attrs
61
+ @node.consume_attributes(@node_attributes)
62
+ end
63
+
64
+ def resource_collection
65
+ run_context.resource_collection
66
+ end
67
+
68
+ def run_context
69
+ @run_context || rebuild_context
70
+ end
71
+
72
+ def definitions
73
+ nil
74
+ end
75
+
76
+ def cookbook_loader
77
+ nil
78
+ end
79
+
80
+ def save_node
81
+ raise "Not Supported! #{self.class.name} doesn't support #save_node, maybe you need to run shef in client mode?"
82
+ end
83
+
84
+ def rebuild_context
85
+ raise "Not Implemented! :rebuild_collection should be implemented by subclasses"
86
+ end
87
+
88
+ private
89
+
90
+ def loading
91
+ show_loading_progress
92
+ begin
93
+ yield
94
+ rescue => e
95
+ loading_complete(false)
96
+ raise e
97
+ else
98
+ loading_complete(true)
99
+ end
100
+ end
101
+
102
+ def show_loading_progress
103
+ print "Loading"
104
+ @loading = true
105
+ @dot_printer = Thread.new do
106
+ while @loading
107
+ print "."
108
+ sleep 0.5
109
+ end
110
+ end
111
+ end
112
+
113
+ def loading_complete(success)
114
+ @loading = false
115
+ @dot_printer.join
116
+ msg = success ? "done.\n\n" : "epic fail!\n\n"
117
+ print msg
118
+ end
119
+
120
+ def shorten_node_inspect
121
+ def @node.inspect
122
+ "<Chef::Node:0x#{self.object_id.to_s(16)} @name=\"#{self.name}\">"
123
+ end
124
+ end
125
+
126
+ def rebuild_node
127
+ raise "Not Implemented! :rebuild_node should be implemented by subclasses"
128
+ end
129
+
130
+ end
131
+
132
+ class StandAloneSession < ShefSession
133
+
134
+ session_type :standalone
135
+
136
+ def rebuild_context
137
+ @run_context = Chef::RunContext.new(@node, {}) # no recipes
138
+ end
139
+
140
+ private
141
+
142
+ def rebuild_node
143
+ Chef::Config[:solo] = true
144
+ @client = Chef::Client.new
145
+ @client.run_ohai
146
+ @client.build_node
147
+ end
148
+
149
+ end
150
+
151
+ class SoloSession < ShefSession
152
+
153
+ session_type :solo
154
+
155
+ def definitions
156
+ @run_context.definitions
157
+ end
158
+
159
+ def rebuild_context
160
+ @run_context = Chef::RunContext.new(@node, Chef::CookbookCollection.new(Chef::CookbookLoader.new))
161
+ end
162
+
163
+ private
164
+
165
+ def rebuild_node
166
+ # Tell the client we're chef solo so it won't try to contact the server
167
+ Chef::Config[:solo] = true
168
+ @client = Chef::Client.new
169
+ @client.run_ohai
170
+ @client.build_node
171
+ end
172
+
173
+ end
174
+
175
+ class ClientSession < SoloSession
176
+
177
+ session_type :solo
178
+
179
+ def save_node
180
+ @client.save_node
181
+ end
182
+
183
+ private
184
+
185
+ def rebuild_node
186
+ # Make sure the client knows this is not chef solo
187
+ Chef::Config[:solo] = false
188
+ @client = Chef::Client.new
189
+ @client.run_ohai
190
+ @client.register
191
+ @client.build_node
192
+
193
+ @client.sync_cookbooks({})
194
+ end
195
+
196
+ end
197
+
198
+ class DoppelGangerClient < Chef::Client
199
+
200
+ attr_reader :node_name
201
+
202
+ def initialize(node_name)
203
+ @node_name = node_name
204
+ @ohai = Ohai::System.new
205
+ end
206
+
207
+ # Run the very smallest amount of ohai we can get away with and still
208
+ # hope to have things work. Otherwise we're not very good doppelgangers
209
+ def run_ohai
210
+ @ohai.require_plugin('os')
211
+ end
212
+
213
+ # DoppelGanger implementation of build_node. preserves as many of the node's
214
+ # attributes, and does not save updates to the server
215
+ def build_node
216
+ Chef::Log.debug("Building node object for #{@node_name}")
217
+
218
+ @node = Chef::Node.find_or_create(node_name)
219
+
220
+ ohai_data = @ohai.data.merge(@node.automatic_attrs)
221
+
222
+ @node.process_external_attrs(ohai_data,nil)
223
+ @node.reset_defaults_and_overrides
224
+
225
+ @node
226
+ end
227
+
228
+ def register
229
+ @rest = Chef::REST.new(Chef::Config[:chef_server_url], Chef::Config[:node_name], Chef::Config[:client_key])
230
+ end
231
+
232
+ end
233
+
234
+ class DoppelGangerSession < ClientSession
235
+
236
+ session_type "doppelganger client"
237
+
238
+ def save_node
239
+ puts "A doppelganger should think twice before saving the node"
240
+ end
241
+
242
+ def assume_identity(node_name)
243
+ Chef::Config[:doppelganger] = @node_name = node_name
244
+ reset!
245
+ rescue Exception => e
246
+ puts "#{e.class.name}: #{e.message}"
247
+ puts Array(e.backtrace).join("\n")
248
+ puts
249
+ puts "* " * 40
250
+ puts "failed to assume the identity of node '#{node_name}', resetting"
251
+ puts "* " * 40
252
+ puts
253
+ Chef::Config[:doppelganger] = false
254
+ @node_built = false
255
+ Shef.session
256
+ end
257
+
258
+ def rebuild_node
259
+ # Make sure the client knows this is not chef solo
260
+ Chef::Config[:solo] = false
261
+ @client = DoppelGangerClient.new(@node_name)
262
+ @client.run_ohai
263
+ @client.register
264
+ @client.build_node
265
+
266
+ @client.sync_cookbooks({})
267
+ end
268
+
269
+ end
270
+
271
+ end
@@ -0,0 +1,413 @@
1
+ #--
2
+ # Author:: Daniel DeLeo (<dan@opscode.com>)
3
+ # Copyright:: Copyright (c) 2010 Opscode, Inc.
4
+ # License:: Apache License, Version 2.0
5
+ #
6
+ # Licensed under the Apache License, Version 2.0 (the "License");
7
+ # you may not use this file except in compliance with the License.
8
+ # You may obtain a copy of the License at
9
+ #
10
+ # http://www.apache.org/licenses/LICENSE-2.0
11
+ #
12
+ # Unless required by applicable law or agreed to in writing, software
13
+ # distributed under the License is distributed on an "AS IS" BASIS,
14
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
+ # See the License for the specific language governing permissions and
16
+ # limitations under the License.
17
+ #
18
+
19
+ require 'etc'
20
+ require 'tmpdir'
21
+ require 'chef/log'
22
+ require 'fcntl'
23
+ require 'chef/exceptions'
24
+
25
+ class Chef
26
+
27
+ # == Chef::ShellOut
28
+ # Provides a simplified interface to shelling out yet still collecting both
29
+ # standard out and standard error and providing full control over environment,
30
+ # working directory, uid, gid, etc.
31
+ #
32
+ # No means for passing input to the subprocess is provided, nor is there any
33
+ # way to inspect the output of the command as it is being read. If you need
34
+ # to do that, you have to use popen4 (in Chef::Mixin::Command)
35
+ #
36
+ # === Platform Support
37
+ # Chef::ShellOut uses Kernel.fork() and is therefore unsuitable for Windows
38
+ # or jruby.
39
+ class ShellOut
40
+ READ_WAIT_TIME = 0.01
41
+ READ_SIZE = 4096
42
+ DEFAULT_READ_TIMEOUT = 60
43
+ DEFAULT_ENVIRONMENT = {'LC_ALL' => 'C'}
44
+
45
+ attr_accessor :user, :group, :cwd, :valid_exit_codes
46
+ attr_reader :command, :umask, :environment
47
+ attr_writer :timeout
48
+ attr_reader :execution_time
49
+
50
+ attr_reader :stdout, :stderr, :status
51
+
52
+ attr_reader :stdin_pipe, :stdout_pipe, :stderr_pipe, :process_status_pipe
53
+
54
+ # === Arguments:
55
+ # Takes a single command, or a list of command fragments. These are used
56
+ # as arguments to Kernel.exec. See the Kernel.exec documentation for more
57
+ # explanation of how arguments are evaluated. The last argument can be an
58
+ # options Hash.
59
+ # === Options:
60
+ # If the last argument is a Hash, it is removed from the list of args passed
61
+ # to exec and used as an options hash. The following options are available:
62
+ # * +user+: the user the commmand should run as. if an integer is given, it is
63
+ # used as a uid. A string is treated as a username and resolved to a uid
64
+ # with Etc.getpwnam
65
+ # * +group+: the group the command should run as. works similarly to +user+
66
+ # * +cwd+: the directory to chdir to before running the command
67
+ # * +umask+: a umask to set before running the command. If given as an Integer,
68
+ # be sure to use two leading zeros so it's parsed as Octal. A string will
69
+ # be treated as an octal integer
70
+ # * +returns+: one or more Integer values to use as valid exit codes for the
71
+ # subprocess. This only has an effect if you call +error!+ after
72
+ # +run_command+.
73
+ # * +environment+: a Hash of environment variables to set before the command
74
+ # is run. By default, the environment will *always* be set to 'LC_ALL' => 'C'
75
+ # to prevent issues with multibyte characters in Ruby 1.8. To avoid this,
76
+ # use :environment => nil for *no* extra environment settings, or
77
+ # :environment => {'LC_ALL'=>nil, ...} to set other environment settings
78
+ # without changing the locale.
79
+ # * +timeout+: a Numeric value for the number of seconds to wait on the
80
+ # child process before raising an Exception. This is calculated as the
81
+ # total amount of time that ShellOut waited on the child process without
82
+ # receiving any output (i.e., IO.select returned nil). Default is 60
83
+ # seconds. Note: the stdlib Timeout library is not used.
84
+ # === Examples:
85
+ # Invoke find(1) to search for .rb files:
86
+ # find = Chef::ShellOut.new("find . -name '*.rb'")
87
+ # find.run_command
88
+ # # If all went well, the results are on +stdout+
89
+ # puts find.stdout
90
+ # # find(1) prints diagnostic info to STDERR:
91
+ # puts "error messages" + find.stderr
92
+ # # Raise an exception if it didn't exit with 0
93
+ # find.error!
94
+ # Run a command as the +www+ user with no extra ENV settings from +/tmp+
95
+ # cmd = Chef::ShellOut.new("apachectl", "start", :user => 'www', :env => nil, :cwd => '/tmp')
96
+ # cmd.run_command # etc.
97
+ def initialize(*command_args)
98
+ @stdout, @stderr = '', ''
99
+ @environment = DEFAULT_ENVIRONMENT
100
+ @cwd = Dir.tmpdir
101
+ @valid_exit_codes = [0]
102
+
103
+ if command_args.last.is_a?(Hash)
104
+ parse_options(command_args.pop)
105
+ end
106
+
107
+ @command = command_args.size == 1 ? command_args.first : command_args
108
+ end
109
+
110
+ def umask=(new_umask)
111
+ @umask = (new_umask.respond_to?(:oct) ? new_umask.oct : new_umask.to_i) & 007777
112
+ end
113
+
114
+ def uid
115
+ return nil unless user
116
+ user.kind_of?(Integer) ? user : Etc.getpwnam(user.to_s).uid
117
+ end
118
+
119
+ def gid
120
+ return nil unless group
121
+ group.kind_of?(Integer) ? group : Etc.getgrnam(group.to_s).gid
122
+ end
123
+
124
+ def timeout
125
+ @timeout || DEFAULT_READ_TIMEOUT
126
+ end
127
+
128
+ # Creates a String showing the output of the command, including a banner
129
+ # showing the exact command executed. Used by +invalid!+ to show command
130
+ # results when the command exited with an unexpected status.
131
+ def format_for_exception
132
+ msg = ""
133
+ msg << "---- Begin output of #{command} ----\n"
134
+ msg << "STDOUT: #{stdout.strip}\n"
135
+ msg << "STDERR: #{stderr.strip}\n"
136
+ msg << "---- End output of #{command} ----\n"
137
+ msg << "Ran #{command} returned #{status.exitstatus}" if status
138
+ msg
139
+ end
140
+
141
+ def exitstatus
142
+ @status && @status.exitstatus
143
+ end
144
+
145
+ # Run the command, writing the command's standard out and standard error
146
+ # to +stdout+ and +stderr+, and saving its exit status object to +status+
147
+ # === Returns
148
+ # returns +self+; +stdout+, +stderr+, +status+, and +exitstatus+ will be
149
+ # populated with results of the command
150
+ # === Raises
151
+ # * Errno::EACCES when you are not privileged to execute the command
152
+ # * Errno::ENOENT when the command is not available on the system (or not
153
+ # in the current $PATH)
154
+ # * Chef::Exceptions::CommandTimeout when the command does not complete
155
+ # within +timeout+ seconds (default: 60s)
156
+ def run_command
157
+ Chef::Log.debug("sh(#{@command})")
158
+ @child_pid = fork_subprocess
159
+
160
+ configure_parent_process_file_descriptors
161
+ propagate_pre_exec_failure
162
+
163
+ @result = nil
164
+ @execution_time = 0
165
+
166
+ # Ruby 1.8.7 and 1.8.6 from mid 2009 try to allocate objects during GC
167
+ # when calling IO.select and IO#read. Some OS Vendors are not interested
168
+ # in updating their ruby packages (Apple, *cough*) and we *have to*
169
+ # make it work. So I give you this epic hack:
170
+ GC.disable
171
+ until @status
172
+ ready = IO.select(open_pipes, nil, nil, READ_WAIT_TIME)
173
+ unless ready
174
+ @execution_time += READ_WAIT_TIME
175
+ if @execution_time >= timeout && !@result
176
+ raise Chef::Exceptions::CommandTimeout, "command timed out:\n#{format_for_exception}"
177
+ end
178
+ end
179
+
180
+ if ready && ready.first.include?(child_stdout)
181
+ read_stdout_to_buffer
182
+ end
183
+ if ready && ready.first.include?(child_stderr)
184
+ read_stderr_to_buffer
185
+ end
186
+
187
+ unless @status
188
+ # make one more pass to get the last of the output after the
189
+ # child process dies
190
+ if results = Process.waitpid2(@child_pid, Process::WNOHANG)
191
+ @status = results.last
192
+ redo
193
+ end
194
+ end
195
+ end
196
+ self
197
+ rescue Exception
198
+ # do our best to kill zombies
199
+ Process.waitpid2(@child_pid, Process::WNOHANG) rescue nil
200
+ raise
201
+ ensure
202
+ # no matter what happens, turn the GC back on, and hope whatever busted
203
+ # version of ruby we're on doesn't allocate some objects during the next
204
+ # GC run.
205
+ GC.enable
206
+ close_all_pipes
207
+ end
208
+
209
+ # Checks the +exitstatus+ against the set of +valid_exit_codes+. If
210
+ # +exitstatus+ is not in the list of +valid_exit_codes+, calls +invalid!+,
211
+ # which raises an Exception.
212
+ # === Returns
213
+ # nil::: always returns nil when it does not raise
214
+ # === Raises
215
+ # Chef::Exceptions::ShellCommandFailed::: via +invalid!+
216
+ def error!
217
+ unless Array(valid_exit_codes).include?(exitstatus)
218
+ invalid!("Expected process to exit 0, but it exited with #{exitstatus}")
219
+ end
220
+ end
221
+
222
+ # Raises a Chef::Exceptions::ShellCommandFailed exception, appending the
223
+ # command's stdout, stderr, and exitstatus to the exception message.
224
+ # === Arguments
225
+ # +msg+: A String to use as the basis of the exception message. The
226
+ # default explanation is very generic, providing a more informative message
227
+ # is highly encouraged.
228
+ # === Raises
229
+ # Chef::Exceptions::ShellCommandFailed always
230
+ def invalid!(msg=nil)
231
+ msg ||= "Command produced unexpected results"
232
+ raise Chef::Exceptions::ShellCommandFailed, msg + "\n" + format_for_exception
233
+ end
234
+
235
+ def inspect
236
+ "<#{self.class.name}##{object_id}: command: '#@command' process_status: #{@status.inspect} " +
237
+ "stdout: '#{stdout.strip}' stderr: '#{stderr.strip}' child_pid: #{@child_pid.inspect} " +
238
+ "environment: #{@environment.inspect} timeout: #{timeout} user: #@user group: #@group working_dir: #@cwd >"
239
+ end
240
+
241
+ private
242
+
243
+ def parse_options(opts)
244
+ opts.each do |option, setting|
245
+ case option.to_s
246
+ when 'cwd'
247
+ self.cwd = setting
248
+ when 'user'
249
+ self.user = setting
250
+ when 'group'
251
+ self.group = setting
252
+ when 'umask'
253
+ self.umask = setting
254
+ when 'timeout'
255
+ self.timeout = setting
256
+ when 'returns'
257
+ self.valid_exit_codes = Array(setting)
258
+ when 'environment', 'env'
259
+ # passing :environment => nil means don't set any new ENV vars
260
+ setting.nil? ? @environment = {} : @environment.merge!(setting)
261
+ else
262
+ raise Chef::Exceptions::InvalidCommandOption, "option '#{option.inspect}' is not a valid option for #{self.class.name}"
263
+ end
264
+ end
265
+ end
266
+
267
+ def set_user
268
+ if user
269
+ Process.euid = uid
270
+ Process.uid = uid
271
+ end
272
+ end
273
+
274
+ def set_group
275
+ if group
276
+ Process.egid = gid
277
+ Process.gid = gid
278
+ end
279
+ end
280
+
281
+ def set_environment
282
+ environment.each do |env_var,value|
283
+ ENV[env_var] = value
284
+ end
285
+ end
286
+
287
+ def set_umask
288
+ File.umask(umask) if umask
289
+ end
290
+
291
+ def initialize_ipc
292
+ @stdout_pipe, @stderr_pipe, @process_status_pipe = IO.pipe, IO.pipe, IO.pipe
293
+ @process_status_pipe.last.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC)
294
+ end
295
+
296
+ def child_stdout
297
+ @stdout_pipe[0]
298
+ end
299
+
300
+ def child_stderr
301
+ @stderr_pipe[0]
302
+ end
303
+
304
+ def child_process_status
305
+ @process_status_pipe[0]
306
+ end
307
+
308
+ def close_all_pipes
309
+ child_stdout.close unless child_stdout.closed?
310
+ child_stderr.close unless child_stderr.closed?
311
+ child_process_status.close unless child_process_status.closed?
312
+ end
313
+
314
+ # replace stdout, and stderr with pipes to the parent, and close the
315
+ # reader side of the error marshaling side channel. Close STDIN so when we
316
+ # exec, the new program will know it's never getting input ever.
317
+ def configure_subprocess_file_descriptors
318
+ process_status_pipe.first.close
319
+
320
+ # HACK: for some reason, just STDIN.close isn't good enough when running
321
+ # under ruby 1.9.2, so make it good enough:
322
+ stdin_reader, stdin_writer = IO.pipe
323
+ stdin_writer.close
324
+ STDIN.reopen stdin_reader
325
+ stdin_reader.close
326
+
327
+ stdout_pipe.first.close
328
+ STDOUT.reopen stdout_pipe.last
329
+ stdout_pipe.last.close
330
+
331
+ stderr_pipe.first.close
332
+ STDERR.reopen stderr_pipe.last
333
+ stderr_pipe.last.close
334
+
335
+ STDOUT.sync = STDERR.sync = true
336
+ end
337
+
338
+ def configure_parent_process_file_descriptors
339
+ # Close the sides of the pipes we don't care about
340
+ stdout_pipe.last.close
341
+ stderr_pipe.last.close
342
+ process_status_pipe.last.close
343
+ # Get output as it happens rather than buffered
344
+ child_stdout.sync = true
345
+ child_stderr.sync = true
346
+
347
+ true
348
+ end
349
+
350
+ # Some patch levels of ruby in wide use (in particular the ruby 1.8.6 on OSX)
351
+ # segfault when you IO.select a pipe that's reached eof. Weak sauce.
352
+ def open_pipes
353
+ @open_pipes ||= [child_stdout, child_stderr]
354
+ end
355
+
356
+ def read_stdout_to_buffer
357
+ while chunk = child_stdout.read_nonblock(READ_SIZE)
358
+ @stdout << chunk
359
+ end
360
+ rescue Errno::EAGAIN
361
+ rescue EOFError
362
+ open_pipes.delete_at(0)
363
+ end
364
+
365
+ def read_stderr_to_buffer
366
+ while chunk = child_stderr.read_nonblock(READ_SIZE)
367
+ @stderr << chunk
368
+ end
369
+ rescue Errno::EAGAIN
370
+ rescue EOFError
371
+ open_pipes.delete_at(1)
372
+ end
373
+
374
+ def fork_subprocess
375
+ initialize_ipc
376
+
377
+ fork do
378
+ configure_subprocess_file_descriptors
379
+
380
+ set_user
381
+ set_group
382
+ set_environment
383
+ set_umask
384
+
385
+ begin
386
+ command.kind_of?(Array) ? exec(*command) : exec(command)
387
+
388
+ raise 'forty-two' # Should never get here
389
+ rescue Exception => e
390
+ Marshal.dump(e, process_status_pipe.last)
391
+ process_status_pipe.last.flush
392
+ end
393
+ process_status_pipe.last.close unless (process_status_pipe.last.closed?)
394
+ exit!
395
+ end
396
+ end
397
+
398
+ # Attempt to get a Marshaled error from the side-channel.
399
+ # If it's there, un-marshal it and raise. If it's not there,
400
+ # assume everything went well.
401
+ def propagate_pre_exec_failure
402
+ begin
403
+ e = Marshal.load child_process_status
404
+ raise(Exception === e ? e : "unknown failure: #{e.inspect}")
405
+ rescue EOFError # If we get an EOF error, then the exec was successful
406
+ true
407
+ ensure
408
+ child_process_status.close
409
+ end
410
+ end
411
+
412
+ end
413
+ end