Class: SitemapGenerator::FileAdapter

Inherits:
Object
  • Object
show all
Defined in:
lib/sitemap_generator/adapters/file_adapter.rb

Overview

Class for writing out data to a file.

Instance Method Summary collapse

Instance Method Details

#gzip(stream, data) ⇒ Object

Write `data` to a stream, passing the data through a GzipWriter to compress it.


30
31
32
33
34
# File 'lib/sitemap_generator/adapters/file_adapter.rb', line 30

def gzip(stream, data)
  gz = Zlib::GzipWriter.new(stream)
  gz.write data
  gz.close
end

#plain(stream, data) ⇒ Object

Write `data` to a stream as is.


37
38
39
40
# File 'lib/sitemap_generator/adapters/file_adapter.rb', line 37

def plain(stream, data)
  stream.write data
  stream.close
end

#write(location, raw_data) ⇒ Object

Write data to a file.


11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# File 'lib/sitemap_generator/adapters/file_adapter.rb', line 11

def write(location, raw_data)
  # Ensure that the directory exists
  dir = location.directory
  if !File.exist?(dir)
    FileUtils.mkdir_p(dir)
  elsif !File.directory?(dir)
    raise SitemapError.new("#{dir} should be a directory!")
  end

  stream = open(location.path, 'wb')
  if location.path.to_s =~ /.gz$/
    gzip(stream, raw_data)
  else
    plain(stream, raw_data)
  end
end