#!/usr/bin/ruby
#
# This file is part of CPEE.
#
# CPEE is free software: you can redistribute it and/or modify it under the terms
# of the GNU General Public License as published by the Free Software Foundation,
# either version 3 of the License, or (at your option) any later version.
#
# CPEE is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
# PARTICULAR PURPOSE.  See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# CPEE (file COPYING in the main directory).  If not, see
# <http://www.gnu.org/licenses/>.

require 'json'
require 'redis'
require 'daemonite'
require_relative '../../lib/cpee/value_helper'

EVENTS = %w{
  event:state/change
  event:handlerwrapper/change
  event:description/change
  event:dataelements/change
  event:endpoints/change
  event:attributes/change
  event:transformation/change
  event:status/change
  event:position/change
  event:handler/change
  callback:activity/content
}

Daemonite.new do |opts|
  redis = Redis.new(path: "/tmp/redis.sock", db: 3)
  pubsubredis = Redis.new(path: "/tmp/redis.sock", db: 3)

  run do
    pubsubredis.subscribe(EVENTS) do |on|
      on.message do |what, message|
        mess = JSON.parse(message[message.index(' ')+1..-1])
        instance = mess.dig('instance')
        case what
          when 'callback:activity/content'
            key = mess.dig('content','key')
            redis.multi do |multi|
              multi.sadd("instance:#{instance}/callbacks",key)
              multi.set("instance:#{instance}/callback/#{key}/uuid",mess.dig('content','activity_uuid'))
              multi.set("instance:#{instance}/callback/#{key}/label",mess.dig('content','label'))
              multi.set("instance:#{instance}/callback/#{key}/position",mess.dig('content','activity'))
              multi.set("instance:#{instance}/callback/#{key}/type",'callback')
            end
          when 'event:state/change'
            redis.multi do |multi|
              multi.set("instance:#{instance}/state",mess.dig('content','state'))
              multi.set("instance:#{instance}/state/@changed",mess.dig('content','timestamp'))
            end
          when 'event:handlerwrapper/change'
            redis.set("instance:#{instance}/handlerwrapper",mess.dig('content','handlerwrapper'))
          when 'event:description/change'
            redis.multi do |multi|
              multi.set("instance:#{instance}/description",mess.dig('content','description'))
              multi.set("instance:#{instance}/dslx",mess.dig('content','dslx'))
              multi.set("instance:#{instance}/dsl",mess.dig('content','dsl'))
            end
          when 'event:dataelements/change', 'event:endpoints/change', 'event:attributes/change'
            topic = mess.dig('topic')
            redis.multi do |multi|
              mess.dig('content','changed')&.each_with_index do |c,i|
                unless what == 'event:attributes/change' && c == 'uuid'
                  multi.zadd("instance:#{instance}/#{topic}",i,c)
                  if what == 'event:dataelements/change'
                    multi.set("instance:#{instance}/#{topic}/#{c}",CPEE::ValueHelper::generate(mess.dig('content','values',c)))
                  else
                    multi.set("instance:#{instance}/#{topic}/#{c}",mess.dig('content','values',c))
                  end
                end
              end
              mess.dig('content','deleted')&.to_a&.each do |c|
                unless what == 'event:attributes/change' && c == 'uuid'
                  multi.zrem("instance:#{instance}/#{topic}",c)
                  multi.del("instance:#{instance}/#{topic}/#{c}")
                end
              end
            end
          when 'event:transformation/change'
            redis.multi do |multi|
              multi.set("instance:#{instance}/transformation/description/",mess.dig('content','description'))
              multi.set("instance:#{instance}/transformation/description/@type",mess.dig('content','description_type'))
              multi.set("instance:#{instance}/transformation/dataelements/",mess.dig('content','dataelements'))
              multi.set("instance:#{instance}/transformation/dataelements/@type",mess.dig('content','dataelements_type'))
              multi.set("instance:#{instance}/transformation/endpoints/",mess.dig('content','endpoints'))
              multi.set("instance:#{instance}/transformation/endpoints/@type",mess.dig('content','endpoints_type'))
            end
          when 'event:status/change'
            redis.multi do |multi|
              multi.set("instance:#{instance}/status/id",mess.dig('content','id'))
              multi.set("instance:#{instance}/status/message",mess.dig('content','message'))
            end
          when 'event:position/change'
            redis.multi do |multi|
              c = mess.dig('content')
              c.dig('unmark')&.each do |ele|
                multi.srem("instance:#{instance}/positions",ele['position'])
                multi.del("instance:#{instance}/positions/#{ele['position']}")
                multi.del("instance:#{instance}/positions/#{ele['position']}/@passthrough")
              end
              c.dig('before')&.each do |ele|
                multi.sadd("instance:#{instance}/positions",ele['position'])
                multi.set("instance:#{instance}/positions/#{ele['position']}",'before')
              end
              c.dig('at')&.each do |ele|
                multi.sadd("instance:#{instance}/positions",ele['position'])
                multi.set("instance:#{instance}/positions/#{ele['position']}",'at')
                multi.set("instance:#{instance}/positions/#{ele['position']}/@passthrough",ele['passthrough']) if ele['passthrough']
              end
              c.dig('wait')&.each do |ele|
                multi.sadd("instance:#{instance}/positions",ele['position'])
                multi.set("instance:#{instance}/positions/#{ele['position']}",'at')
                multi.set("instance:#{instance}/positions/#{ele['position']}/@passthrough",ele['passthrough']) if ele['passthrough']
              end
              c.dig('after')&.each do |ele|
                multi.sadd("instance:#{instance}/positions",ele['position'])
                multi.set("instance:#{instance}/positions/#{ele['position']}",'after')
              end
            end
          when 'event:handler/change'
            redis.multi do |multi|
              mess.dig('content','changed').each do |c|
                multi.sadd("instance:#{instance}/handlers",mess.dig('content','key'))
                multi.sadd("instance:#{instance}/handlers/#{mess.dig('content','key')}",c)
                multi.set("instance:#{instance}/handlers/#{mess.dig('content','key')}/url",mess.dig('content','url'))
                multi.sadd("instance:#{instance}/handlers/#{c}",mess.dig('content','key'))
              end
              mess.dig('content','deleted').to_a.each do |c|
                multi.srem("instance:#{instance}/handlers/#{mess.dig('content','key')}",c)
                multi.srem("instance:#{instance}/handlers/#{c}",mess.dig('content','key'))
              end
            end
            if redis.scard("instance:#{instance}/handlers/#{mess.dig('content','key')}") < 1
              redis.multi do |multi|
                multi.del("instance:#{instance}/handlers/#{mess.dig('content','key')}/url")
                multi.srem("instance:#{instance}/handlers",mess.dig('content','key'))
              end
            end
        end
      rescue => e
        puts e.message
        puts e.backtrace
      end
    end
  end
end.go!