managers/rcm_file_manager.rb (d16ba15)

rcm_file_manager.rb

require_relative '../objects/rcm_file'
require_relative '../rcm_utils'
require 'fileutils'

module RCM
  class FileManager
    def get_id_to_name_mapping(file)
      id_to_name = {}
      ::File.open(file, 'r').each do |l|
        matches = /(?<name>\w+):.+:(?<id>\d+):.*\s/.match(l)
        id_to_name[matches[:id]] = matches[:name]
      end
      id_to_name
    end

def apply_attributes(file)
      raise 'apply_attributes only works with ::RCM::File' unless file.is_a?(::RCM::File)

# Will throw exception if something poopy happens
      ::FileUtils.chown(file.owner, file.group, file.path)
      ::FileUtils.chmod(file.mode, file.path)

file.changed = true
      file
    end

def copy(file)
      dir = ::File.dirname(file.path)
      ::FileUtils.mkdir_p(dir)
      ::FileUtils.copy(file.src_path, file.path, remove_destination: true)

file.changed = true
      file
    end

def remove(file)
      file_existed = ::File.file?(file.path)
      ::FileUtils.rm(file.path)
      file.changed = true if file_existed
      file
    end

def initialize(logger)
      @logger = logger
    end

def get_current_state(wanted_files)
      got = {}
      uids = get_id_to_name_mapping('/etc/passwd')
      gids = get_id_to_name_mapping('/etc/group')
      wanted_files.each do |path, p|
        f = ::RCM::File.new('', '', '', '', '')
        # Put in an empty object if file does not exist
        unless ::File.exist?(path)
          got[path] = f
          next
        end

f_stat = ::File.stat(path)
        f.path = path
        f.owner = uids[f_stat.uid.to_s]
        f.group = gids[f_stat.gid.to_s]
        f.mode = f_stat.mode.to_s(8)[-4, 4]
        got[path] = f
      end

got
    end
  end
end

rcm

Now detecting difference in current and wanted state.