require 'json'
require 'logger'
require 'trace-component'
require 'pathname'
require 'helper-run-class'
require 'prepare-environment'

class PlanStep
    def initialize ()
        @basePath = Pathname.new(Canzea::config[:catalog_location]).realpath
        @log = Logger.new(Canzea::config[:logging_root] + '/plans.log')
    end

    def runPhaseInstall (role, solution, test, task)

        plan = JSON.parse("{ \"plan\": [ { \"role\": \"#{role}\", \"solution\": \"#{solution}\" } ] }")

        n = Worker.new
        n.test ( test )

        start = Integer(task)
        lines = 1

        begin

            plan['plan'].each do |item|
                @log.info(item['solution'])

                root = "#{@basePath}/roles/#{item['role']}/#{item['solution']}"
                if File.exist?(root) == false
                    puts "-- ERROR #{root} does not exist!"
                    raise "#{root} does not exist!"
                end

                cmd = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/install.sh"
                if File.exist?(cmd)
                    lines = n.run cmd, start, lines - 1

                    @log.info "#{ lines } lines read"
                end
            end
        rescue => exception
            @log.error(cmd)
            @log.error(exception.to_s)
            @log.error(exception.backtrace)
            abort()
        end
    end

    def runPhaseConfigure (role, solution, test, task)

        plan = JSON.parse("{ \"plan\": [ { \"role\": \"#{role}\", \"solution\": \"#{solution}\" } ] }")

        n = Worker.new
        n.test ( test )

        start = Integer(task)
        lines = 1

        cmd = "undefined"

        begin

            plan['plan'].each do |item|
                @log.info(item['solution'])

                root = "#{@basePath}/roles/#{item['role']}/#{item['solution']}"
                if File.exist?(root) == false
                    puts "-- ERROR #{root} does not exist!"
                    raise "#{root} does not exist!"
                end

                # Register the service with Consul, if consul is ready
                # If metadata.json exists, then use the information to register
                cmd = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/metadata.json"
                if File.exist?(cmd)
                    md = File.read(cmd)
                    md = JSON.parse(md)
                    if (md['services'].size() > 0)
                        svc = md['services'][0]
                        adef = {"listener"=>svc['listener'], "serviceName"=>svc['name'], "tags"=>{}, "port"=>svc['port']}
                        puts "-- Registering"
                        if (test == false)
                            h = HelperRun.new
                            h.run "consul", "add_service", JSON.generate(adef)
                        end
                    end
                end

                envScript = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/environment.json"
                if File.exist?(envScript)
                    envPush = PrepareEnvironment.new
                    envPush.addToEnv "#{envScript}"
                end

                cmd = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/configure.sh"
                if File.exist?(cmd)
                  lines = n.run cmd, start, lines - 1

                  @log.info("#{ lines } lines read")
                end

                cmd = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/enable.sh"
                if File.exist?(cmd)
                  lines = n.run cmd, start, lines - 1

                  @log.info("#{ lines } lines read")
                end

                cmd = "#{@basePath}/roles/#{item['role']}/#{item['solution']}/status.sh"
                if File.exist?(cmd)
                  lines = n.run cmd, start, lines - 1, true

                  @log.info("#{ lines } lines read")
                end

            end
        rescue => exception
            @log.error(cmd)
            @log.error(exception.to_s)
            @log.error(exception.backtrace)
            abort()
        end
    end
end