aboutsummaryrefslogtreecommitdiffstats
path: root/lib/action_text/trix_attachment.rb
blob: 717d130d12e30d3c16b973d88fa22d12fc6f7b6d (plain) (blame)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
module ActionText
  class TrixAttachment
    TAG_NAME = "figure"
    SELECTOR = "[data-trix-attachment]"

    ATTRIBUTES = %w( sgid contentType url href filename filesize width height previewable content caption )
    ATTRIBUTE_TYPES = {
      "previewable" => ->(value) { value.to_s == "true" },
      "filesize"    => ->(value) { Integer(value.to_s) rescue value },
      "width"       => ->(value) { Integer(value.to_s) rescue nil },
      "height"      => ->(value) { Integer(value.to_s) rescue nil },
      :default      => ->(value) { value.to_s }
    }

    class << self
      def from_attributes(attributes)
        attributes = process_attributes(attributes)

        trix_attachment_attributes = attributes.except("caption")
        trix_attributes = attributes.slice("caption")

        node = ActionText::HtmlConversion.create_element(TAG_NAME)
        node["data-trix-attachment"] = JSON.generate(trix_attachment_attributes)
        node["data-trix-attributes"] = JSON.generate(trix_attributes) if trix_attributes.any?

        new(node)
      end

      private
        def process_attributes(attributes)
          typecast_attribute_values(transform_attribute_keys(attributes))
        end

        def transform_attribute_keys(attributes)
          attributes.transform_keys { |key| key.to_s.underscore.camelize(:lower) }
        end

        def typecast_attribute_values(attributes)
          attributes.map do |key, value|
            typecast = ATTRIBUTE_TYPES[key] || ATTRIBUTE_TYPES[:default]
            [key, typecast.call(value)]
          end.to_h
        end
    end

    attr_reader :node

    def initialize(node)
      @node = node
    end

    def attributes
      @attributes ||= attachment_attributes.merge(composed_attributes).slice(*ATTRIBUTES)
    end

    def to_html
      ActionText::HtmlConversion.node_to_html(node)
    end

    def to_s
      to_html
    end

    private
      def attachment_attributes
        read_json_object_attribute("data-trix-attachment")
      end

      def composed_attributes
        read_json_object_attribute("data-trix-attributes")
      end

      def read_json_object_attribute(name)
        read_json_attribute(name) || {}
      end

      def read_json_attribute(name)
        if value = node[name]
          begin
            JSON.parse(value)
          rescue => e
            Rails.logger.error "[#{self.class.name}] Couldn't parse JSON #{value} from NODE #{node.inspect}"
            Rails.logger.error "[#{self.class.name}] Failed with #{e.class}: #{e.backtrace}"
            nil
          end
        end
      end
  end
end