lib/saml/kit/cli/report.rb in saml-kit-cli-0.3.6 vs lib/saml/kit/cli/report.rb in saml-kit-cli-0.3.7

@@ -1,5 +1,7 @@
+# frozen_string_literal: true
+
 module Saml
   module Kit
     module Cli
       class Report
         attr_reader :document
@@ -7,105 +9,33 @@
         def initialize(document)
           @document = document
         end
 
         def print(shell)
-          shell.say_status :success, "Decoded #{document.send(:name)}"
-          shell.print_table build_table_for(document)
-          shell.say ""
-          if document.signature.present? && document.signature.certificate.present?
-            shell.say(document.signature.certificate.x509.to_text)
-          end
-          shell.say ""
-          shell.say document.to_xml(pretty: true), :green
-          shell.say ""
-          document.errors.full_messages.each do |error|
-            shell.say_status :error, error, :red
-          end
+          shell.say_status status, "Decoded #{document.send(:name)}"
+          shell.print_table document.build_table
+          print_signature(document.signature, shell)
+          print_xml(shell)
+          print_errors(document.errors.full_messages, shell)
         end
 
         private
 
-        def truncate(text, max: 50)
-          if text.length >= max
-            "#{text[0..max]}..."
-          else
-            text
-          end
+        def status
+          document.is_a?(Saml::Kit::InvalidDocument) ? :error : :sucess
         end
 
-        def build_table_for(document)
-          table = [ ]
-          case document
-          when Saml::Kit::Document
-            table.push(['ID', document.id])
-            table.push(['Issuer', document.issuer])
-            table.push(['Version', document.version])
-            table.push(['Issue Instant', document.issue_instant.iso8601])
-            table.push(['Type', document.send(:name)])
-            table.push(['Valid', document.valid?])
-            table.push(['Signed?', !!document.signed?])
-            table.push(['Trusted?', !!document.trusted?])
-          when Saml::Kit::Metadata
-            table.push(['Entity Id', document.entity_id])
-            table.push(['Type', document.send(:name)])
-            table.push(['Valid', document.valid?])
-            table.push(['Name Id Formats', document.name_id_formats.inspect])
-            table.push(['Organization', document.organization_name])
-            table.push(['Url', document.organization_url])
-            table.push(['Contact', document.contact_person_company])
-            [
-              'SingleSignOnService',
-              'SingleLogoutService',
-              'AssertionConsumerService'
-            ].each do |type|
-              document.services(type).each do |service|
-                table.push([type, [service.location, service.binding]])
-              end
-            end
-            document.certificates.each do |certificate|
-              table.push(['', certificate.x509.to_text])
-            end
-          end
-          if document.signature.present?
-            signature = document.signature
-            table.push(['Digest Value', signature.digest_value])
-            table.push(['Expected Digest Value', signature.expected_digest_value])
-            table.push(['Digest Method', signature.digest_method])
-            table.push(['Signature Value', truncate(signature.signature_value)])
-            table.push(['Signature Method', signature.signature_method])
-            table.push(['Canonicalization Method', signature.canonicalization_method])
-            table.push(['', signature.certificate.x509.to_text])
-          end
-          case document
-          when Saml::Kit::AuthenticationRequest
-            table.push(['ACS', document.assertion_consumer_service_url])
-            table.push(['Name Id Format', document.name_id_format])
-          when Saml::Kit::LogoutRequest
-            table.push(['Name Id', document.name_id])
-          when Saml::Kit::Response
-            table.push(['Assertion Present?', document.assertion.present?])
-            table.push(['Issuer', document.assertion.issuer])
-            table.push(['Name Id', document.assertion.name_id])
-            table.push(['Signed?', !!document.assertion.signed?])
-            table.push(['Attributes', document.assertion.attributes.inspect])
-            table.push(['Not Before', document.assertion.started_at])
-            table.push(['Not After', document.assertion.expired_at])
-            table.push(['Audiences', document.assertion.audiences.inspect])
-            table.push(['Encrypted?', document.assertion.encrypted?])
-            table.push(['Decryptable', document.assertion.decryptable?])
-            if document.assertion.present?
-              signature = document.assertion.signature
-              table.push(['Digest Value', signature.digest_value])
-              table.push(['Expected Digest Value', signature.expected_digest_value])
-              table.push(['Digest Method', signature.digest_method])
-              table.push(['Signature Value', truncate(signature.signature_value)])
-              table.push(['Signature Method', signature.signature_method])
-              table.push(['Canonicalization Method', signature.canonicalization_method])
-              table.push(['', signature.certificate.x509.to_text])
-            end
-          end
-          table
+        def print_errors(errors, shell)
+          errors.each { |x| shell.say_status :error, x, :red }
+        end
+
+        def print_signature(signature, shell)
+          return if !signature.present? || !signature.certificate.present?
+          shell.say(signature.certificate.x509.to_text)
+        end
+
+        def print_xml(shell)
+          shell.say document.to_xml(pretty: true), :green
         end
       end
     end
   end
 end