zypper-upgraderepo 1.0.2 → 1.5.0

data/lib/zypper/upgraderepo/request.rb

@@ -0,0 +1,42 @@
+require 'delegate'
+require_relative 'traversable.rb'
+require_relative 'requests/local.rb'
+require_relative 'requests/http.rb'
+
+module Zypper
+  module Upgraderepo
+
+
+    class Request
+
+      def self.build(repo, timeout)
+        @@registry ||= self.load_requests
+
+        raise InvalidProtocol, repo unless @@registry.include? repo.protocol
+
+        Object.const_get(@@registry[repo.protocol]).new(repo, timeout)
+      end
+
+      def self.protocols
+        self.load_requests.keys
+      end
+
+      private
+
+      def self.load_requests
+        res = {}
+        Requests.constants.each do |klass|
+          Object.const_get("Zypper::Upgraderepo::Requests::#{klass}").register_protocol.each do |protocol|
+            res[protocol] = "Zypper::Upgraderepo::Requests::#{klass}"
+          end
+        end
+
+        res
+      end
+
+    end
+
+
+
+  end
+end

data/lib/zypper/upgraderepo/requests/http.rb

@@ -0,0 +1,131 @@
+require 'delegate'
+require 'net/http'
+
+module Zypper
+  module Upgraderepo
+
+    class PageRequest < SimpleDelegator
+
+      attr_reader :page
+
+      USER_AGENT = 'Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:59.0) Gecko/20100101 Firefox/59.0'
+
+      def initialize(obj, timeout = 60)
+        super obj
+        @timeout = timeout
+      end
+
+      def available?
+        ping.is_a?(Net::HTTPSuccess)
+      end
+
+      def redirected?
+        ping.is_a?(Net::HTTPRedirection)
+      end
+
+      def redirected_to
+        ping['location']
+      end
+
+      def not_found?
+        ping.is_a?(Net::HTTPNotFound)
+      end
+
+      def forbidden?
+        ping.is_a?(Net::HTTPForbidden)
+      end
+
+      def timeout?
+        ping.is_a?(Net::HTTPRequestTimeOut)
+      end
+
+      def status
+        ping.class.to_s
+      end
+
+      def cache!
+        @page = nil
+      end
+
+
+      private
+
+      def get_request(uri, head)
+
+        if head
+          request = Net::HTTP::Head.new(uri.request_uri)
+        else
+          request = Net::HTTP::Get.new(uri.request_uri)
+        end
+
+        request['User-Agent'] = USER_AGENT
+
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = (uri.scheme == 'https')
+        http.open_timeout = @timeout
+
+        http.request(request)
+      end
+
+      def ping(uri = nil, head = true)
+        begin
+          if @page.nil? || uri
+            @page = get_request(uri, head)
+          end
+        rescue SocketError
+          raise NoConnection
+        rescue Net::OpenTimeout
+          @page = Net::HTTPRequestTimeOut.new('1.1', '', '')
+        end
+        @page
+      end
+
+    end
+
+
+    module Requests
+
+      class HttpRequest < PageRequest
+
+        include Traversable
+
+        def max_drop_back; 0; end
+
+        def self.register_protocol; ['https', 'http'] end
+
+        def evaluate_alternative(version)
+          if not_found?
+            return traverse_url(URI(url), version)
+          elsif redirected?
+            return { url: redirected_to, message: 'Redirected to:' }
+          end
+        end
+
+
+        private
+
+        def get_request(uri, head)
+          #super uri || URI(url), head
+          super uri || repodata_uri, head
+        end
+
+        def has_repodata?(uri)
+          ping(repodata_uri(uri))
+          available?
+        end
+
+        def subfolders
+          res = ping.body.to_s.scan(Regexp.new('href=[\'\"][^\/\"]+\/[\'\"]')).delete_if do |x|
+            x =~ /^\// || x =~ /^\.\./ || x =~ /\:\/\// || x =~ /href=[\"\'](media\.1|boot|EFI)\/[\"\']/
+          end.uniq.map do |d|
+            d.scan(/href=[\"\']([^"]+)[\'\"]/).pop.pop
+          end
+
+          res
+        end
+      end
+
+    end
+
+  end
+end

data/lib/zypper/upgraderepo/requests/local.rb

@@ -0,0 +1,94 @@
+require 'delegate'
+
+module Zypper
+  module Upgraderepo
+
+
+    class DirRequest < SimpleDelegator
+
+      attr_reader :dir_path
+
+      def initialize(obj, timeout)
+        super obj
+      end
+
+      def available?
+        Dir.exist? ping
+      end
+
+      def redirected?
+        File.symlink? ping
+      end
+
+      def redirected_to
+        File.realpath ping
+      end
+
+      def not_found?
+        !available?
+      end
+
+      def forbidden?
+        File.readable? ping
+      end
+
+      def timeout?
+        false
+      end
+
+      def status
+        File.stat ping
+      end
+
+      def cache!
+        @dir_path = nil
+      end
+
+
+      private
+
+      def ping(uri = nil, head = true)
+        @dir_path ||= URI(url).path
+
+        @dir_path = uri.to_s =~ /^\// ? uri.to_s : URI(uri.to_s).path if uri
+
+        URI.unescape(@dir_path)
+      end
+
+    end
+
+
+    module Requests
+
+      class LocalRequest < DirRequest
+
+        include Traversable
+
+        def max_drop_back; 1 end
+
+        def self.register_protocol; ['dir'] end
+
+        def evaluate_alternative(version)
+          if not_found?
+            return traverse_url(URI(url), version)
+          elsif redirected?
+            return { url: redirected_to, message: 'Linked to' }
+          end
+        end
+
+
+        private
+
+        def has_repodata?(uri)
+          File.exist? URI.unescape(repodata_uri(uri).path)
+        end
+
+        def subfolders
+          Dir.glob(ping.gsub(/\/$/, '') + '/*/').map { |x| URI.escape(x.gsub(/\/$/, '').gsub(ping, '').gsub(/^\//, '')) }
+        end
+      end
+
+    end
+
+  end
+end

data/lib/zypper/upgraderepo/traversable.rb

@@ -0,0 +1,93 @@
+module Zypper
+  module Upgraderepo
+
+    module Traversable
+
+      def traverse_url(uri, version)
+        ping(uri)
+
+        if forbidden?
+          res =  { url: url, message: 'Can\'t navigate through the repository!' }
+        elsif available? && uri.to_s =~ /#{version}/
+          res = traverse_url_forward(uri, version)
+        else
+          res = traverse_url_backward(uri, version)
+        end
+
+        res || { url: '', message: 'Can\'t find a valid alternative, try manually!' }
+      end
+
+
+      private
+
+      def traverse_url_backward(uri, version)
+        uri.path = File.dirname(uri.path)
+
+        return nil if uri.path == '/' || uri.path == '.' || (versioned? && (drop_back_level(uri) > max_drop_back))
+
+        uri.path += '/' if uri.path[-1] != '/'
+        ping(uri, false)
+
+        if not_found?
+          return traverse_url_backward(uri, version)
+        elsif available?
+          if res = traverse_url_forward(uri, version)
+            return res
+          else
+            return traverse_url_backward(uri, version)
+          end
+        elsif forbidden?
+          return { url: uri.to_s, message: 'Try to replace with this one' } if has_repodata?(uri)
+
+          return traverse_url_backward(uri, version)
+        end
+
+        nil
+      end
+
+      def traverse_url_forward(uri, version)
+        uri.path += '/' if uri.path[-1] != '/'
+        ping(uri, false)
+
+        subfolders.each do |dir|
+          u = URI(uri.to_s)
+          u.path += dir
+
+          if has_repodata?(u)
+            if (versioned?) && (u.to_s =~ /#{version}/)
+              return { url: u.to_s, message: 'Override with this one' }
+            end
+          else
+            res = traverse_url_forward(u, version)
+            return res if res.class == Hash
+          end
+        end
+
+        nil
+      end
+
+      def repodata_uri(uri = nil)
+        if uri
+          uri = URI(uri.to_s)
+        else
+          uri = URI(url)
+        end
+
+        uri.path = uri.path.gsub(/\/$/, '') + '/repodata/repomd.xml'
+
+        uri
+      end
+      
+      def drop_back_level(uri)
+        URI(url).path.split('/').index { |x| x =~ /\d\d.\d/ } - uri.path.split('/').count
+      end
+      
+      # to implement on each repository type class
+      #
+      # def has_repodata?(uri)
+      #
+      # def subfolders
+
+    end
+  end
+end

data/lib/zypper/upgraderepo/utils.rb

@@ -25,12 +25,19 @@ module Zypper
       def new_line;       "\n#{self}" end
     end
 
+    class ::StandardError
+      def error_code
+        1
+      end
+    end
 
     class Messages
 
       def self.error(e)
         if e.class == String
           puts ' [E] '.bold.red + e
+        elsif e.class == Interruption
+          STDERR.puts e.message =~ /\(/ ? e.message.gsub(/.*\((.*)\).*/, '\1').green : e.message.green
         else
           STDERR.puts 'Error! '.bold.red + e.message
         end
@@ -44,59 +51,89 @@ module Zypper
         puts ' [W] '.bold.yellow + m
       end
 
-      def self.available(num, name, url, max_col)
-        Messages.ok("| #{num.to_s.rjust(2)} | #{name.ljust(max_col, ' ')} |")
+    end
+
+    class FileNotFound < StandardError
+      def initialize(filename)
+        super "The File #{filename} doesn't exist."
+      end
+    end
+
+    class ReleaseFileNotFound < StandardError
+      def initialize
+        super 'The release file is not found.'
       end
+    end
 
-      def self.redirected(num, name, url, max_col, redirected)
-        Messages.warning("| #{num.to_s.rjust(2)} | #{name.ljust(max_col, ' ')} | Redirection of #{url} ")
-        puts " #{' ' * 3} | #{' ' * 2} | #{ ' ' * max_col} | #{'To:'.bold.yellow} #{redirected}"
+    class InvalidProtocol < StandardError
+      def initialize(repo)
+        super "The repository #{repo.name} has an unknown protocol: #{repo.protocol}; disable it to continue."
       end
+    end
 
-      def self.not_found(num, name, url, max_col)
-        Messages.error("| #{num.to_s.rjust(2)} | #{name.ljust(max_col, ' ')}")
+    class InvalidVersion < StandardError
+      def initialize(version)
+        super "The version #{version} is not valid"
       end
+    end
 
-      def self.alternative(num, name, url, max_col, res)
-        Messages.error("| #{num.to_s.rjust(2)} | #{name.ljust(max_col, ' ')} | #{res[:message].bold.yellow}")
-        puts " #{' ' * 3} | #{' ' * 2} | #{' ' * max_col} | #{res[:url]}" unless res[:url].to_s.empty?
+    class InvalidWritePermissions < StandardError
+      def initialize(filename)
+        super "Don't have the right permission to write #{filename}"
       end
 
-      def self.separator
-        puts '-' * 90
+      def error_code
+        4
       end
+    end
 
-      def self.header(max_col)
-        puts " St. |  # | #{'Name'.ljust(max_col, ' ')} | Hint"
+    class SystemUpdateRunning < StandardError
+      def initialize(args)
+        super "The application #{args[:process].bold} with pid #{args[:pid].bold} is running a system update!"
       end
 
-      def self.footer
-        self.separator
+      def error_code
+        5
       end
     end
 
+    class UnableToUpgrade < StandardError
+      def initialize(args)
+        super "The repository n.#{args[:num].to_s.bold.red} named #{args[:repo].name.bold.red} can't be upgraded, a manual check is required!"
+      end
 
-    class ReleaseFileNotFound < StandardError
-      def initialize
-        super 'The release file is not found.'
+      def error_code
+        7
       end
     end
-    
-    class InvalidVersion < StandardError
-      def initialize(version)
-        super "The version #{version} is not valid"
+
+    class MissingOverride < StandardError
+      def initialize(args)
+        super "The repository n.#{args[:num].to_s.bold.red} named #{args[:ini]['name'].bold.red} doesn't contain the URL key!"
+      end
+
+      def error_code
+        8
       end
     end
 
-    class InvalidPermissions < StandardError
-      def initialize(filename)
-        super "Don't have the right permission to write #{filename}"
+    class UnmatchingOverrides < StandardError
+      def initialize(args)
+        super "The repository n.#{args[:num]} named #{args[:repo].name.bold.red} doesn't match with the repository named #{args[:ini]['name'].bold.red} in the ini file"
+      end
+
+      def error_code
+        9
       end
     end
 
     class AlreadyUpgraded < StandardError
       def initialize(version)
-        super "The system is already upgraded to #{version}"
+        super "The system is already upgraded to the #{version} version"
+      end
+
+      def error_code
+        2
       end
     end
 
@@ -104,6 +141,20 @@ module Zypper
       def initialize
         super 'Internet connection has some trouble'
       end
+
+      def error_code
+        6
+      end
     end
+
+    class Interruption < StandardError
+      def initialize
+        super 'Ok ok... Exiting!'
+      end
+    end
+
+    Signal.trap('INT') { raise Interruption }
+
+    Signal.trap('TERM') { raise Interruption }
   end
 end