Class: Middleman::Spreadsheet::Extension

Inherits:
Extension
  • Object
show all
Defined in:
lib/middleman/spreadsheet/extension.rb

Instance Method Summary collapse

Constructor Details

#initialize(app, options_hash = {}, &block) ⇒ Extension

Returns a new instance of Extension.

Raises:



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/middleman/spreadsheet/extension.rb', line 16

def initialize(app, options_hash={}, &block)
  super

  require "google_drive"
  require "sheet_wrap"

  session = drive_session(options)

  if options.spreadsheet_key
    spreadsheet = session.spreadsheet_by_key(options.spreadsheet_key)
  elsif options.spreadsheet_url
    spreadsheet = session.spreadsheet_by_url(options.spreadsheet_url)
  elsif options.spreadsheet_title
    spreadsheet = session.spreadsheet_by_title(options.spreadsheet_title)
  end

  raise SpreadsheetNotFound, "No spreadsheet specified" unless spreadsheet

  data = spreadsheet.worksheets.each_with_object({}) do |worksheet, hash|
    sheet_wrap = SheetWrap.wrap(worksheet)
    hash[sheet_wrap.title.to_sym] = sheet_wrap.rows.map(&:to_h)
  end
  @app.data.store(options.data_name.to_sym, data)
end