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)   @got = {

18)     ::RCM::ConfigManager::PACKAGES => {},
19)     ::RCM::ConfigManager::FILES => {},
20)     ::RCM::ConfigManager::SERVICES => {}

21)   }

22)  
23)   @logger = @config_mgr.logger

24)   @pkg_mgr = ::RCM::Apt.new(@logger)
25)   @file_mgr = ::RCM::FileManager.new(@logger)
26)   @svc_mgr = ::RCM::ServiceManager.new(@logger)

27) 

28)   def self.converge_packages(current_state_of_packages)

29)     # We'll install them in one go (minor optimization)
30)     missing_packages = []
31) 

32)     @pkg_mgr.update unless @env == 'dev'
33) 
34)     @config_mgr.packages.each do |pkg_name, pkg_obj|
35)       current_pkg = current_state_of_packages[pkg_name]

36) 

37)       if current_pkg == pkg_obj

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

39)         next
40)       end

41) 

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

46)         @pkg_mgr.remove(pkg_obj)

47)         missing_packages.push(pkg_obj)

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

49)         missing_packages.push(pkg_obj)

50)       end

51)     end

52) 

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

54)   end
55) 

56)   def self.converge_files(current_state_of_files)
57)     @config_mgr.files.each do |path, file_obj|
58)       file_on_fs = current_state_of_files[path]

59) 

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

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

63)         next
64)       end
65) 

66)       if file_on_fs == file_obj

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

68)         next

69)       end

70) 

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

72)       status = @file_mgr.copy(file_obj)
73)       @file_mgr.apply_attributes(file_obj) if status == 0

74)     end

75)   end
76) 

77)   def self.restart_services_if_needed

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

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

80)     end

81)   end

82) 
83)   def main

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

86)     current_state_of_packages = @pkg_mgr.get_current_state(@config_mgr.packages)
87)     converge_packages(current_state_of_packages)
88)     current_state_of_files = @file_mgr.get_current_state(@config_mgr.files)
89)     converge_files(current_state_of_files)

90)     restart_services_if_needed

91)   end

92) 

93)   module_function :main

94) end
95) 

96) RCM.main