module RubyXL::WorkbookConvenienceMethods

def [](ind)
  case ind
  when Integer then worksheets[ind]
  when String  then worksheets.find { |ws| ws.sheet_name == ind }
  end
end

def add_worksheet(name = nil)
  if name.nil? then
    n = 0
    begin
      name = SHEET_NAME_TEMPLATE % (n += 1)
    end until self[name].nil?
  end
  new_worksheet = Worksheet.new(:workbook => self, :sheet_name => name)
  worksheets << new_worksheet
  new_worksheet
end

def application
  root.document_properties.application && root.document_properties.application.value
end

def application=(v)
  root.document_properties.application ||= StringNode.new
  root.document_properties.application.value = v
end

def appversion
  root.document_properties.app_version && root.document_properties.app_version.value
end

def appversion=(v)
  root.document_properties.app_version ||= StringNode.new
  root.document_properties.app_version.value = v
end

def borders # Stylesheet should be pre-filled with defaults on initialize()
  stylesheet.borders
end

def cell_xfs # Stylesheet should be pre-filled with defaults on initialize()
  stylesheet.cell_xfs
end

def company
  root.document_properties.company && root.document_properties.company.value
end

def company=(v)
  root.document_properties.company ||= StringNode.new
  root.document_properties.company.value = v
end

def created_at
  root.core_properties.created_at
end

def created_at=(v)
  root.core_properties.created_at = v
end

def creator
  root.core_properties.creator
end

def creator=(v)
  root.core_properties.creator = v
end

def date1904
  workbook_properties && workbook_properties.date1904
end

def date1904=(v)
  self.workbook_properties ||= RubyXL::WorkbookProperties.new
  workbook_properties.date1904 = v
end

def each
  worksheets.each{ |i| yield i }
end

def fills # Stylesheet should be pre-filled with defaults on initialize()
  stylesheet.fills
end

def fonts # Stylesheet should be pre-filled with defaults on initialize()
  stylesheet.fonts
end

def get_fill_color(xf)
  fill = fills[xf.fill_id]
  pattern = fill && fill.pattern_fill
  color = pattern && pattern.fg_color
  color && color.rgb || 'ffffff'
end

def modified_at
  root.core_properties.modified_at
end

def modified_at=(v)
  root.core_properties.modified_at = v
end

def modifier
  root.core_properties.modifier
end

def modifier=(v)
  root.core_properties.modifier = v
end

def modify_alignment(style_index, &block)
  xf = cell_xfs[style_index].dup
  xf.alignment ||= RubyXL::Alignment.new
  xf.apply_alignment = true
  yield(xf.alignment)
  register_new_xf(xf, style_index)
end

def modify_border(style_index, direction, weight)
  old_xf = cell_xfs[style_index].dup
  new_border = borders[old_xf.border_id].dup
  new_border.set_edge_style(direction, weight)
  new_xf = old_xf.dup
  new_xf.apply_border = true
  new_xf.border_id = borders.find_index { |x| x == new_border } # Use existing border, if it exists
  new_xf.border_id ||= borders.size # If this border has never existed before, add it to collection.
  borders[new_xf.border_id] = new_border
  register_new_xf(new_xf, style_index)
end

def modify_fill(style_index, rgb)
  xf = cell_xfs[style_index].dup
  new_fill = RubyXL::Fill.new(:pattern_fill =>
               RubyXL::PatternFill.new(:pattern_type => 'solid',
                                       :fg_color => RubyXL::Color.new(:rgb => rgb)))
  new_xf = register_new_fill(new_fill, xf)
  register_new_xf(new_xf, style_index)
end

def register_new_fill(new_fill, old_xf)
  new_xf = old_xf.dup
  new_xf.apply_fill = true
  new_xf.fill_id = fills.find_index { |x| x == new_fill } # Use existing fill, if it exists
  new_xf.fill_id ||= fills.size # If this fill has never existed before, add it to collection.
  fills[new_xf.fill_id] = new_fill
  new_xf
end

def register_new_font(new_font, old_xf)
  new_xf = old_xf.dup
  new_xf.apply_font = true
  new_xf.font_id = fonts.find_index { |x| x == new_font } # Use existing font, if it exists
  new_xf.font_id ||= fonts.size # If this font has never existed before, add it to collection.
  fonts[new_xf.font_id] = new_font
  new_xf
end

def register_new_xf(new_xf, old_style_index)
  new_xf_id = cell_xfs.find_index { |xf| xf == new_xf } # Use existing XF, if it exists
  new_xf_id ||= cell_xfs.size # If this XF has never existed before, add it to collection.
  cell_xfs[new_xf_id] = new_xf
  new_xf_id
end