module Timetrap module Formatters class Text LONGEST_NOTE_LENGTH = 50 attr_accessor :output include Timetrap::Helpers def initialize entries @entries = entries self.output = '' sheets = entries.inject({}) do |h, e| h[e.sheet] ||= [] h[e.sheet] << e h end (sheet_names = sheets.keys.sort).each do |sheet| self.output << "Timesheet: #{sheet}\n" id_heading = Timetrap::CLI.args['-v'] ? "Id#{' ' * (max_id_length - 3)}" : " " self.output << "#{id_heading} Day Start End Duration Notes\n" last_start = nil from_current_day = [] sheets[sheet].each_with_index do |e, i| from_current_day << e self.output << "%-#{max_id_length + 1}s%16s%11s -%9s%10s %s\n" % [ (Timetrap::CLI.args['-v'] ? e.id : ''), format_date_if_new(e.start, last_start), format_time(e.start), format_time(e.end), format_duration(e.duration), format_note(e.note) ] nxt = sheets[sheet].to_a[i+1] if nxt == nil or !same_day?(e.start, nxt.start) self.output << "%#{49 + max_id_length}s\n" % format_total(from_current_day) from_current_day = [] else end last_start = e.start end self.output << "#{' ' * (max_id_length + 1)}%s\n" % ('-'*(52+longest_note)) self.output << "#{' ' * (max_id_length + 1)}Total%43s\n" % format_total(sheets[sheet]) self.output << "\n" unless sheet == sheet_names.last end if sheets.size > 1 self.output << "%s\n" % ('-'*(4+52+longest_note)) self.output << "Grand Total%41s\n" % format_total(sheets.values.flatten) end end private attr_reader :entries def longest_note @longest_note ||= begin entries.inject('Notes'.length) {|l, e| [e.note.to_s.rstrip.length, LONGEST_NOTE_LENGTH].min} end end def max_id_length @max_id_length ||= begin if Timetrap::CLI.args['-v'] entries.inject(3) {|l, e| [e.id.to_s.length, l].max} else 3 end end end def format_note(note) return "" unless note lines = [] line_number = 0 note.lines.each do |line| while index = line.index(/\s/, LONGEST_NOTE_LENGTH) do shorter_line = line.slice!(0..(index - 1)) lines << padded_line(shorter_line.strip, line_number) line_number += 1 end lines << padded_line(line.strip, line_number) line_number += 1 end lines.join("\n") end def padded_line(content, line_number) return content if line_number == 0 "#{" " * (56 + max_id_length - 3) }#{content}" end end end end