#--
# Copyright (c) 2005-2011, John Mettraux, jmettraux@gmail.com
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
# Made in Japan.
#++


module Ruote

  #
  # A container for the pretty printing logic of TestLogger.
  #
  module PrettyLogging

    protected

    #--
    # <ESC>[{attr1};...;{attrn}m
    #
    # 0 Reset all attributes
    # 1 Bright
    # 2 Dim
    # 4 Underscore
    # 5 Blink
    # 7 Reverse
    # 8 Hidden
    #
    # Foreground Colours
    # 30 Black
    # 31 Red
    # 32 Green
    # 33 Yellow
    # 34 Blue
    # 35 Magenta
    # 36 Cyan
    # 37 White
    #
    # Background Colours
    # 40 Black
    # 41 Red
    # 42 Green
    # 43 Yellow
    # 44 Blue
    # 45 Magenta
    # 46 Cyan
    # 47 White
    #++

    def color(mod, s, clear=false)

      return s if Ruote::WIN
      return s unless STDOUT.tty?

      "[#{mod}m#{s}#{clear ? '' : "[#{@color}m"}"
    end

    def pretty_print(msg)

      @count += 1
      @count = 0 if @count > 9

      ei = self.object_id.to_s[-2..-1]

      fei = msg['fei']
      depth = fei ? fei['expid'].split('_').size : 0

      i = fei ?
        [ fei['wfid'], (fei['subid'] || '')[0, 5], fei['expid'] ].join(' ') :
        msg['wfid']
      wfid = fei ? fei['wfid'] : msg['wfid']

      rest = msg.dup
      %w[
        _id put_at _rev
        type action
        fei wfid variables
      ].each { |k| rest.delete(k) }

      if v = rest['parent_id']
        rest['parent_id'] = Ruote.to_storage_id(v)
      end
      if v = rest.delete('workitem')
        rest[:wi] = [
          v['fei'] ? Ruote.to_storage_id(v['fei']) : nil,
          v['fields'].size ]
      end

      #if t = rest.delete('tree')
      #  rest[:t] = color(37, t.inspect, true)
      #end

      { 'tree' => :t, 'parent_id' => :pi }.each do |k0, k1|
        if v = rest.delete(k0)
          rest[k1] = v
        end
      end

      act = msg['action'][0, 2]
      act = case msg['action']
        when 'receive' then 'rc'
        when 'dispatched' then 'dd'
        when 'dispatch_cancel' then 'dc'
        when 'dispatch_pause' then 'dp'
        when 'dispatch_resume' then 'dr'
        when 'pause', 'pause_process' then 'pz'
        when 'resume', 'resume_process' then 'rz'
        else act
      end
      act = case act
        when 'la' then color('4;32', act)
        when 'te' then color('4;31', act)
        when 'ce' then color('31', act)
        when 'ca' then color('31', act)
        when 'rc' then color('4;33', act)
        when 'di' then color('4;33', act)
        when 'dd' then color('4;33', act)
        when 'dc' then color('4;31', act)
        when 'pz' then color('4;31', act)
        when 'rz' then color('4;32', act)
        when 'dp' then color('4;31', act)
        when 'dr' then color('4;32', act)
        else act
      end

      tail = if msg['action'] == 'error_intercepted'

        tail = []
        tail << "  #{wfid} #{rest['error']['class']}"
        tail << "  #{wfid} #{rest['error']['message']}"
        rest['error']['trace'].each do |line|
          tail << "  #{wfid} #{line}"
        end

        color(
          @color,
          "#{@count} #{ei} #{'  ' * depth}#{act} * #{i}",
          true
        ) +
        "\n" +
        color(
          @color,
          tail.join("\n"),
          true)

      else

        pa = %w[ receive dispatch ].include?(msg['action']) ?
          color('34', msg['participant_name']) + ' ' : ''

        color(
          @color,
          "#{@count} #{ei} #{'  ' * depth}#{act} * #{pa}#{i} #{rest.inspect}",
          true)
      end
    end
  end
end