lib/saml/kit/xml_templatable.rb in saml-kit-1.0.6 vs lib/saml/kit/xml_templatable.rb in saml-kit-1.0.7

@@ -2,31 +2,36 @@
   module Kit
     module XmlTemplatable
       include ::Xml::Kit::Templatable
 
       def template_path
-        root_path = File.expand_path(File.dirname(__FILE__))
-        template_name = "#{self.class.name.split("::").last.underscore}.builder"
-        File.join(root_path, "builders/templates/", template_name)
+        root_path = __dir__
+        template_name = "#{self.class.name.split('::').last.underscore}.builder"
+        File.join(root_path, 'builders/templates/', template_name)
       end
 
       # Returns true if an embedded signature is requested and at least one signing certificate is available via the configuration.
       def sign?
         return configuration.sign? if embed_signature.nil?
         (embed_signature && configuration.sign?) ||
-          (embed_signature && @signing_key_pair.present?)
+          (embed_signature && signing_key_pair.present?)
       end
 
+      def encrypt_with(key_pair)
+        self.encrypt = true
+        self.encryption_certificate = key_pair.certificate
+      end
+
       def digest_method
         configuration.digest_method
       end
 
       def signature_method
         configuration.signature_method
       end
 
       def signing_key_pair
-        configuration.key_pairs(use: :signing).last
+        @signing_key_pair || configuration.key_pairs(use: :signing).last
       end
     end
   end
 end