lib/xml/kit/encrypted_data.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

# frozen_string_literal: true

module Xml
  module Kit
    # An implementation of the EncryptedKey element.
    # https://www.w3.org/TR/xmlenc-core1/#sec-EncryptedData
    #
    # @since 0.3.0
    class EncryptedData
      attr_reader :id
      attr_reader :key_info
      attr_reader :symmetric_cipher
      attr_reader :symmetric_cipher_value

      def initialize(raw_xml,
                     id: Id.generate,
                     symmetric_cipher: nil,
                     asymmetric_cipher: nil,
                     key_info: nil)
        @id = id
        @symmetric_cipher = symmetric_cipher ||
          key_info.try(:symmetric_cipher) ||
          Xml::Kit::Crypto::SymmetricCipher.new
        @symmetric_cipher_value = Base64.strict_encode64(
          @symmetric_cipher.encrypt(raw_xml)
        )
        @key_info = key_info ||
          create_key_info_for(@symmetric_cipher, asymmetric_cipher)
      end

      def to_xml(xml: ::Builder::XmlMarkup.new)
        ::Xml::Kit::Template.new(self).to_xml(xml: xml)
      end

      def render(model, options)
        ::Xml::Kit::Template.new(model).to_xml(options)
      end

      private

      def create_key_info_for(symmetric_cipher, asymmetric_cipher)
        KeyInfo.new do |x|
          x.encrypted_key = EncryptedKey.new(
            asymmetric_cipher: asymmetric_cipher,
            symmetric_cipher: symmetric_cipher
          )
        end
      end
    end
  end
end