default.rb 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. #
  2. # Author:: Noah Kantrowitz <noah@opscode.com>
  3. # Cookbook Name:: application
  4. # Resource:: default
  5. #
  6. # Copyright:: 2011, Opscode, Inc <legal@opscode.com>
  7. #
  8. # Licensed under the Apache License, Version 2.0 (the "License");
  9. # you may not use this file except in compliance with the License.
  10. # You may obtain a copy of the License at
  11. #
  12. # http://www.apache.org/licenses/LICENSE-2.0
  13. #
  14. # Unless required by applicable law or agreed to in writing, software
  15. # distributed under the License is distributed on an "AS IS" BASIS,
  16. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17. # See the License for the specific language governing permissions and
  18. # limitations under the License.
  19. #
  20. require 'weakref'
  21. include Chef::Mixin::RecipeDefinitionDSLCore
  22. def initialize(*args)
  23. super
  24. @action = :deploy
  25. @sub_resources = []
  26. end
  27. actions :deploy, :remove
  28. attribute :name, :kind_of => String, :name_attribute => true
  29. attribute :environment_name, :kind_of => String, :default => (node.chef_environment =~ /_default/ ? "production" : node.chef_environment)
  30. attribute :path, :kind_of => String
  31. attribute :owner, :kind_of => String
  32. attribute :group, :kind_of => String
  33. attribute :strategy, :kind_of => [String, Symbol], :default => :deploy_revision
  34. attribute :revision, :kind_of => String
  35. attribute :repository, :kind_of => String
  36. attribute :environment, :kind_of => Hash, :default => {}
  37. attribute :deploy_key, :kind_of => [String, NilClass], :default => nil
  38. attribute :force, :kind_of => [TrueClass, FalseClass], :default => false
  39. attribute :purge_before_symlink, :kind_of => Array, :default => []
  40. attribute :create_dirs_before_symlink, :kind_of => Array, :default => []
  41. attribute :symlinks, :kind_of => Hash, :default => {}
  42. attribute :symlink_before_migrate, :kind_of => Hash, :default => {}
  43. attribute :migrate, :kind_of => [TrueClass, FalseClass], :default => false
  44. attribute :migration_command, :kind_of => [String, NilClass], :default => nil
  45. attribute :restart_command, :kind_of => [String, NilClass], :default => nil
  46. attribute :packages, :kind_of => [Array, Hash], :default => []
  47. attr_reader :sub_resources
  48. # Callback fires before deploy is started.
  49. def before_deploy(arg=nil, &block)
  50. arg ||= block
  51. set_or_return(:before_deploy, arg, :kind_of => [Proc, String])
  52. end
  53. # Callback fires before migration is run.
  54. def before_migrate(arg=nil, &block)
  55. arg ||= block
  56. set_or_return(:before_migrate, arg, :kind_of => [Proc, String])
  57. end
  58. # Callback fires before symlinking
  59. def before_symlink(arg=nil, &block)
  60. arg ||= block
  61. set_or_return(:before_symlink, arg, :kind_of => [Proc, String])
  62. end
  63. # Callback fires before restart
  64. def before_restart(arg=nil, &block)
  65. arg ||= block
  66. set_or_return(:before_restart, arg, :kind_of => [Proc, String])
  67. end
  68. # Callback fires after restart
  69. def after_restart(arg=nil, &block)
  70. arg ||= block
  71. set_or_return(:after_restart, arg, :kind_of => [Proc, String])
  72. end
  73. def method_missing(name, &block)
  74. # Build the set of names to check for a valid resource
  75. lookup_path = ["application_#{name}"]
  76. run_context.cookbook_collection.each do |cookbook_name, cookbook_ver|
  77. if cookbook_name.start_with?("application_")
  78. lookup_path << "#{cookbook_name}_#{name}"
  79. end
  80. end
  81. lookup_path << name
  82. resource = nil
  83. # Try to find our resource
  84. lookup_path.each do |resource_name|
  85. begin
  86. Chef::Log.debug "Trying to load application resource #{resource_name} for #{name}"
  87. resource = super(resource_name.to_sym, name, &block)
  88. break
  89. rescue NameError
  90. next
  91. end
  92. end
  93. raise NameError, "No resource found for #{name}. Tried #{lookup_path.join(', ')}" unless resource
  94. # Enforce action :nothing in case people forget
  95. resource.action :nothing
  96. # Make this a weakref to prevent a cycle between the application resource and the sub resources
  97. resource.application WeakRef.new(self)
  98. resource.type name
  99. @sub_resources << resource
  100. resource
  101. end