rcm

1) #!/usr/bin/ruby
2) 

3) require_relative 'objects/rcm_file'
4) require_relative 'objects/rcm_package'
5) require_relative 'objects/rcm_service'

6) require_relative 'managers/rcm_config_manager'

7) require_relative 'managers/rcm_package_manager'

8) require_relative 'managers/rcm_file_manager'
9) require_relative 'managers/rcm_service_manager'
10) require_relative 'rcm_utils'

11) require 'yaml'

12) require 'logger'

13) 
14) module RCM

15) 

16)   @config_mgr = ::RCM::ConfigManager.new()
17)   @logger = @config_mgr.logger

18)   @pkg_mgr = ::RCM::Apt.new(@logger)
19)   @file_mgr = ::RCM::FileManager.new(@logger)
20)   @svc_mgr = ::RCM::ServiceManager.new(@logger)

21) 

22)   def self.converge_packages(current_state_of_packages)

23)     # We'll install them in one go (minor optimization)
24)     missing_packages = []
25) 

26)     @config_mgr.packages.each do |pkg_name, pkg_obj|
27)       current_pkg = current_state_of_packages[pkg_name]

28) 

29)       if current_pkg == pkg_obj

30)         @logger.info("#{pkg_name} is in expected state.")

31)         next
32)       end

33) 

34)       if current_pkg.version != pkg_obj.version && current_pkg.state == ::RCM::Package::INSTALLED
35)         # Wrong version is installed. Remove current version and install the correct one.
36)         # Not the best way to go about it... but we are not implementing a legit solution.
37)         @logger.debug("#{current_pkg.name}=#{current_pkg.version} is installed. Uninstalling first.")

38)         @pkg_mgr.remove(pkg_obj)

39)         missing_packages.push(pkg_obj)

40)       elsif current_pkg.state == ::RCM::Package::REMOVED

41)         missing_packages.push(pkg_obj)

42)       end

43)     end

44) 

45)     @pkg_mgr.update unless @env == 'dev' || missing_packages.empty?

46)     @pkg_mgr.install(missing_packages) unless missing_packages.empty?

47)   end
48) 

49)   def self.converge_files(current_state_of_files)
50)     @config_mgr.files.each do |path, file_obj|
51)       file_on_fs = current_state_of_files[path]

52) 

53)       if file_on_fs.state == ::RCM::File::PRESENT && file_obj.state == ::RCM::File::ABSENT

54)         @logger.info("#{path} is present on disk when it should not be. Removing...")
55)         @file_mgr.remove(file_obj)

56)         next
57)       end
58) 

59)       if file_on_fs == file_obj

60)         @logger.info("#{path} is in expected state (#{file_on_fs.state}).")

61)         next

62)       end

63) 

64)       @logger.info("#{path} is not in expected state on disk.")

65)       status = @file_mgr.copy(file_obj)
66)       @file_mgr.apply_attributes(file_obj) if status == 0

67)     end

68)   end
69) 

70)   def self.restart_services_if_needed

71)     @config_mgr.services.values.each do |svc_obj|

72)       @svc_mgr.restart(svc_obj) if @svc_mgr.dependencies_changed?(svc_obj)

73)     end

74)   end

75) 
76)   def main

77)     # Installing packages creates file that we want to delete.
78)     # So finish managing packages and then deal with files.

79)     current_state_of_packages = @pkg_mgr.get_current_state(@config_mgr.packages)
80)     converge_packages(current_state_of_packages)
81)     current_state_of_files = @file_mgr.get_current_state(@config_mgr.files)
82)     converge_files(current_state_of_files)

83)     restart_services_if_needed

84)   end

85) 

86)   module_function :main

87) end
88) 

89) RCM.main