evesync 1.0.2

data/lib/evesync/log.rb

@@ -0,0 +1,66 @@
+require 'logger'
+require 'evesync/config'
+require 'evesync/constants'
+
+# This module is responsible for logging
+module Evesync
+  module Log
+    # Supported levels for logging
+    LEVELS = %i[debug info warn error fatal].freeze
+
+    class << self
+      def method_missing(m, *args)
+        # Unlisted methods are not allowed
+        raise NoMethodError unless LEVELS.include?(m)
+
+        check_logger
+        @logger.send(m, to_string(*args))
+        nil
+      end
+
+      def check_logger
+        init_logger unless @logger
+      end
+
+      def level=(lvl)
+        check_logger
+        if lvl.is_a?(Symbol) or lvl.is_a?(String)
+          @logger.level =
+            begin
+              Logger.const_get(lvl.to_s.upcase)
+            rescue NameError
+              Logger::DEBUG
+            end
+        end
+      end
+
+      def level
+        check_logger
+        @logger.level
+      end
+
+      def simple=(bool)
+        init_logger unless @logger
+        if bool
+          @logger.formatter = proc do |_sev, _dt, _prog, msg|
+            "#{msg}\n"
+          end
+        end
+      end
+
+      def init_logger
+        @logger = Logger.new(STDERR)
+        @logger.formatter = proc do |sev, dtime, _prog, msg|
+          time = dtime.strftime('%Y-%m-%d %H:%M:%S')
+          prog = File.basename($PROGRAM_NAME)
+          "[#{time}] #{prog.ljust(8)} #{sev.ljust(5)}: #{msg}\n"
+        end
+      end
+
+      def to_string(*args)
+        to_s_with_space = ->(s) { "#{s} " }
+        args.map(&to_s_with_space).reduce(&:+).strip
+      end
+    end
+  end
+end

data/lib/evesync/ntp.rb

@@ -0,0 +1,16 @@
+require 'net/ntp'
+require 'evesync/config'
+
+module Evesync
+  module NTP
+
+    def self.time
+      Net::NTP.get(Config['ntp']).time
+    end
+
+    def self.timestamp
+      time.to_f.to_s
+    end
+
+  end
+end

data/lib/evesync/os/linux/arch/package_manager.rb

@@ -0,0 +1,29 @@
+module Evesync
+  module OS
+    # FIXME: pacman downgrading requires full url path
+    module PackageManager
+      class << self
+        def install(*args)
+          pacman('-Sy', *args)
+        end
+
+        def remove(*args)
+          pacman('-R', *args)
+        end
+
+        # FIXME: update and downgrade specific version is not fine
+        def update(*args)
+          pacman('-U', *args)
+        end
+
+        def downgrade(*args)
+          pacman('-U', *args)
+        end
+
+        def pacman(cmd, name, _version)
+          `pacman #{cmd} #{name}` # FIXME: do smth with version
+        end
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/arch/package_watcher.rb

@@ -0,0 +1,59 @@
+require 'file-tail'
+require 'evesync/log'
+require 'evesync/config'
+require 'evesync/ipc/data/package'
+require 'evesync/watcher/interface'
+
+module Evesync
+  module OS
+
+    # Watcher for package changes for Arch Linux
+    #
+    # = Example
+    #  Thread.new { IPC::Data::PackageWatcher.new(queue).run }
+    class PackageWatcher < Watcher::Interface
+      ARCH_LOG_FILE = '/var/log/pacman.log'.freeze
+      PKG_REGEXP =
+        /(?<command>reinstalled|installed|removed)
+    \s*
+    (?<package>\w+)
+    \s*
+    \(   (?<version>[\w\d.-]+)   \)
+    /x.freeze
+
+      private_constant :ARCH_LOG_FILE, :PKG_REGEXP
+
+      def initialize(queue)
+        @queue = queue
+        Log.debug('Arch Package watcher initialized')
+      end
+
+      def start
+        Log.debug('Arch Package watcher started')
+        @thr = Thread.new do
+          File.open(ARCH_LOG_FILE) do |log|
+            log.extend(File::Tail)
+            log.interval = Config[:evemond]['watch-interval']
+            log.backward(1)
+            log.tail do |line|
+              m = line.match(PKG_REGEXP)
+              next unless m
+
+              pkg = IPC::Data::Package.new(
+                name: m[:package],
+                version: m[:version],
+                command: m[:command]
+              )
+              @queue << pkg
+              Log.debug 'Arch package watcher enqued:', pkg
+            end
+          end
+        end
+      end
+
+      def stop
+        @thr.exit
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/arch.rb

@@ -0,0 +1,2 @@
+require_relative './arch/package_watcher.rb'
+require_relative './arch/package_manager.rb'

data/lib/evesync/os/linux/base_package_manager.rb

@@ -0,0 +1,80 @@
+require 'hashdiff'
+require 'evesync/ipc/data/package'
+
+module Evesync
+  module OS
+    module BasePackageManager
+      def initialize
+        @packages = make_pkg_snapshot
+      end
+
+      def changes
+        # make hash of all packages
+        snapshot = make_pkg_snapshot
+        # diff with last version
+        diff = HashDiff.diff(@packages, snapshot)
+        # parse changes into IPC::Data::Package array
+        packages = parse_pkg_diff(diff)
+        # use new hash as default
+        # FIXME: wait if the changes were saved
+        #        and properly handled
+        @packages = snapshot
+        packages
+      end
+
+      private
+
+      # Parses changes, given by 'hashdiff' gem into
+      # IPC::Data::Package array
+      def parse_pkg_diff(diffs)
+        packages = []
+        diffs.each do |diff|
+          package = case diff[0]
+                    when '-' then removed_package(diff)
+                    when '+' then installed_package(diff)
+                    when '~' then updated_package(diff)
+                    end
+          packages.push(package)
+        end
+
+        packages
+      end
+
+      def removed_package(diff)
+        IPC::Data::Package.new(
+          name: diff[1],
+          version: diff[2],
+          command: IPC::Data::Package::Command::REMOVE
+        )
+      end
+
+      def installed_package(diff)
+        IPC::Data::Package.new(
+          name: diff[1],
+          version: diff[2],
+          command: IPC::Data::Package::Command::INSTALL
+        )
+      end
+
+      def updated_package(diff)
+        command = if pkg_version_less(diff[2], diff[3])
+                    IPC::Data::Package::Command::UPDATE
+                  else
+                    IPC::Data::Package::Command::DOWNGRADE
+                  end
+
+        IPC::Data::Package.new(
+          name: diff[1],
+          version: diff[3],
+          command: command
+        )
+      end
+
+      # FIXME:
+      #   - research about restrictions and make sure this work
+      def pkg_version_less(a, b)
+        a < b
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/deb/dpkg.rb

@@ -0,0 +1,30 @@
+require 'hashdiff'
+require 'evesync/log'
+require 'evesync/os/linux/base_package_manager'
+
+module Evesync
+  module OS
+    class Dpkg
+      include BasePackageManager
+
+      PKG_QUERY = 'dpkg-query -l'.freeze
+
+      private_constant :PKG_QUERY
+
+      # Snapshot is a hash where key is a package name
+      # and value - it's version.
+      # This function returns a snapshot of a system
+      # package status for a moment
+      def make_pkg_snapshot
+        snapshot = {}
+        query_output = `#{PKG_QUERY}`
+        query_output.lines.each do |line|
+          info = line.split
+          snapshot[info[1]] = info[2]
+        end
+
+        snapshot
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/deb/package_manager.rb

@@ -0,0 +1,38 @@
+module Evesync
+  module OS
+    module PackageManager
+      class << self
+        def install(*args)
+          apt_get('install', *args)
+          exist?(*args)
+        end
+
+        def remove(*args)
+          apt_get('remove', *args)
+          !exist?(*args)
+        end
+
+        def update(*args)
+          apt_get('upgrade', *args)
+          exist?(*args)
+        end
+
+        def downgrade(*args)
+          apt_get('install', *args)
+          exist?(*args)
+        end
+
+        def apt_get(cmd, name, version)
+          Log.debug("Apt-get command: '#{cmd}' on #{name}=#{version}")
+          system("apt-get --assume-yes #{cmd} #{name}=#{version}")
+        end
+
+        def exist?(name, version)
+          Log.debug("Dpkg checking if exists: #{name}-#{version}")
+          `dpkg-query -l #{name}`
+          $CHILD_STATUS == 0
+        end
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/deb/package_watcher.rb

@@ -0,0 +1,32 @@
+require 'evesync/log'
+require 'evesync/watcher/interface'
+require_relative './dpkg'
+
+module Evesync
+  module OS
+    class PackageWatcher < Watcher::Interface
+      def initialize(queue)
+        @queue = queue
+        @dpkg = Dpkg.new
+        Log.debug('Debian Package watcher initialized')
+      end
+
+      def start
+        Log.debug('Debian Package watcher started')
+        @thr = Thread.new do
+          loop do
+            sleep 10
+            @dpkg.changes.each do |pkg|
+              @queue << pkg
+              Log.debug 'Debian Package watcher enqued:', pkg
+            end
+          end
+        end
+      end
+
+      def stop
+        @thr.exit
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/deb.rb

@@ -0,0 +1,2 @@
+require_relative './deb/package_watcher.rb'
+require_relative './deb/package_manager.rb'

data/lib/evesync/os/linux/rhel/package_manager.rb

@@ -0,0 +1,42 @@
+require 'evesync/log'
+
+# Manages package manager things
+# TODO: trigger package_watcher event to update database
+module Evesync
+  module OS
+    module PackageManager
+      class << self
+        def install(*args)
+          return false if exist?(*args)
+          yum('install', *args)
+          exist?(*args)
+        end
+
+        def remove(*args)
+          yum('remove', *args)
+          !exist?(*args)
+        end
+
+        def update(*args)
+          yum('update', *args)
+          exist?(*args)
+        end
+
+        def downgrade(*args)
+          yum('downgrade', *args)
+          exist?(*args)
+        end
+
+        def yum(cmd, name, version)
+          Log.debug("Yum command: '#{cmd}' on #{name}-#{version}")
+          system("yum --assumeyes #{cmd} #{name}-#{version}")
+        end
+
+        def exist?(name, version)
+          Log.debug("Yum checking if exists: #{name}-#{version}")
+          system("rpm -q #{name}-#{version} >/dev/null 2>&1")
+        end
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/rhel/package_watcher.rb

@@ -0,0 +1,32 @@
+require_relative './rpm'
+require 'evesync/log'
+require 'evesync/watcher/interface'
+
+module Evesync
+  module OS
+    class PackageWatcher < Watcher::Interface
+      def initialize(queue)
+        @queue = queue
+        @rpm_packages = Rpm.new
+        Log.debug('Rhel Package watcher initialized')
+      end
+
+      def start
+        Log.debug('Rhel Package watcher started')
+        @thr = Thread.new do
+          loop do
+            sleep 10 # FIXME: don't use magic numbers
+            @rpm_packages.changes.each do |pkg|
+              @queue << pkg
+              Log.debug 'Rhel Package watcher enqued:', pkg
+            end
+          end
+        end
+      end
+
+      def stop
+        @thr.exit
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/rhel/rpm.rb

@@ -0,0 +1,41 @@
+require 'evesync/log'
+require 'evesync/os/linux/base_package_manager'
+
+module Evesync
+  module OS
+
+    # Rpm packages changes watcher. Yum history makes it
+    # difficult to handler package removals. So, rpm is
+    # the only tool that show all packages in the system.
+    # This class handles packages changes.
+    #
+    # = Example:
+    #   rpm = Evesync::Rpm.new
+    #   sleep 1000
+    #   rpm.changed.each do |package|
+    #     p package # print packages that changed
+    #   end
+    #
+    # = TODO:
+    #  * add reinstall handling also
+    #
+    class Rpm
+      include BasePackageManager
+      # Query for rpm list
+      PKG_QUERY = 'rpm -qa --queryformat "%{NAME} %{VERSION}-%{RELEASE}.%{ARCH}\n"'.freeze
+
+      private_constant :PKG_QUERY
+
+      def make_pkg_snapshot
+        snapshot = {}
+        query_output = `#{PKG_QUERY}`
+        query_output.lines.each do |line|
+          info = line.split
+          snapshot[info[0]] = info[1]
+        end
+
+        snapshot
+      end
+    end
+  end
+end

data/lib/evesync/os/linux/rhel.rb

@@ -0,0 +1,3 @@
+# TODO: add check if rpm exists
+require_relative './rhel/package_watcher.rb'
+require_relative './rhel/package_manager.rb'

data/lib/evesync/os/linux.rb

@@ -0,0 +1,9 @@
+text = File.new('/etc/os-release').read
+
+if text =~ /^ID.*(rhel|centos|fedora)/
+  require 'evesync/os/linux/rhel'
+elsif text =~ /ID.*arch/
+  require 'evesync/os/linux/arch'
+elsif text =~ /ID.*debian/
+  require 'evesync/os/linux/deb'
+end

data/lib/evesync/os.rb

@@ -0,0 +1,2 @@
+# TODO: Fix to catch different OSes
+require 'evesync/os/linux'