about summary refs log tree commit diff
path: root/lib/paperclip/gif_transcoder.rb
blob: 9f3c8e8be339fc0784b6688069cd8fbb4e41241c (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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# frozen_string_literal: true

class GifReader
  attr_reader :animated

  EXTENSION_LABELS = [0xf9, 0x01, 0xff].freeze
  GIF_HEADERS      = %w(GIF87a GIF89a).freeze

  class GifReaderException < StandardError; end

  class UnknownImageType < GifReaderException; end

  class CannotParseImage < GifReaderException; end

  def self.animated?(path)
    new(path).animated
  rescue GifReaderException
    false
  end

  def initialize(path, max_frames = 2)
    @path      = path
    @nb_frames = 0

    File.open(path, 'rb') do |s|
      raise UnknownImageType unless GIF_HEADERS.include?(s.read(6))

      # Skip to "packed byte"
      s.seek(4, IO::SEEK_CUR)

      # "Packed byte" gives us the size of the GIF color table
      packed_byte, = s.read(1).unpack('C')

      # Skip background color and aspect ratio
      s.seek(2, IO::SEEK_CUR)

      if packed_byte & 0x80 != 0
        # GIF uses a global color table, skip it
        s.seek(3 * (1 << ((packed_byte & 0x07) + 1)), IO::SEEK_CUR)
      end

      # Now read data
      while @nb_frames < max_frames
        separator = s.read(1)

        case separator
        when ',' # Image block
          @nb_frames += 1

          # Skip to "packed byte"
          s.seek(8, IO::SEEK_CUR)
          packed_byte, = s.read(1).unpack('C')

          if packed_byte & 0x80 != 0
            # Image uses a local color table, skip it
            s.seek(3 * (1 << ((packed_byte & 0x07) + 1)), IO::SEEK_CUR)
          end

          # Skip lzw min code size
          raise InvalidValue unless s.read(1).unpack('C')[0] >= 2

          # Skip image data sub-blocks
          skip_sub_blocks!(s)
        when '!' # Extension block
          skip_extension_block!(s)
        when ';' # Trailer
          break
        else
          raise CannotParseImage
        end
      end
    end

    @animated = @nb_frames > 1
  end

  private

  def skip_extension_block!(file)
    if EXTENSION_LABELS.include?(file.read(1).unpack('C')[0])
      block_size, = file.read(1).unpack('C')
      file.seek(block_size, IO::SEEK_CUR)
    end

    # Read until extension block end marker
    skip_sub_blocks!(file)
  end

  # Skip sub-blocks up until block end marker
  def skip_sub_blocks!(file)
    loop do
      size, = file.read(1).unpack('C')

      break if size.zero?

      file.seek(size, IO::SEEK_CUR)
    end
  end
end

module Paperclip
  # This transcoder is only to be used for the MediaAttachment model
  # to convert animated gifs to webm
  class GifTranscoder < Paperclip::Processor
    def make
      return File.open(@file.path) unless needs_convert?

      final_file = Paperclip::Transcoder.make(file, options, attachment)

      attachment.instance.file_file_name    = File.basename(attachment.instance.file_file_name, '.*') + '.mp4'
      attachment.instance.file_content_type = 'video/mp4'
      attachment.instance.type              = MediaAttachment.types[:gifv]

      final_file
    end

    private

    def needs_convert?
      options[:style] == :original && GifReader.animated?(file.path)
    end
  end
end