v-yarotsky/medo

require 'rainbow'

module Medo
  class TextTaskWriter
    module Decorators
      module ColorsDecorator
        extend Support::Decorator

        private

        def present_tasks(tasks)
          super.each { |t| TaskColors.decorate(t) }
        end

        module TaskColors
          extend Support::Decorator

          def number
            str = super.to_s
            preserve_size(str, str.color(:red))
          end

          def tag
            str = super
            unless String(str).empty?
              preserve_size(str, str.bright)
            end
          end

          private

          def preserve_size(str, colored_str)
            colored_str.instance_eval <<-RUBY
              def size; #{str.size}; end
            RUBY
            colored_str
          end
        end
      end
    end
  end
end