module Flydata
  module Command
    class Sender < Base
      FLYDATA_HOME=Flydata::HOME_DIR
      def start
        # Check if process exist
        if process_exist?
          say("Process exist. Please stop process first.")
          return
        end

        wait_until_server_ready

        # Start sender(fluentd) process
        say('Starting sender process.')
        Dir.chdir(FLYDATA_HOME){
          system("fluentd -d #{FLYDATA_HOME}/flydata.pid -l #{FLYDATA_HOME}/flydata.log -c #{FLYDATA_HOME}/flydata.conf")
        }
        sleep 5

        wait_until_client_ready
        wait_until_logs_uploaded
        say("Go to your Dashboard! #{Flydata::FLYDATA_API_HOST}")
        say <<EOF
Please Note: Records and Total Size are updated every 20 minutes. You can download logs one hour after they are generated as they are sent to s3 once per hour.
EOF
      end
      def stop
        unless process_exist?
          say("Process doesn't exist.")
          return true
        end

        say('Stopping sender process.')
        if system("kill `cat #{FLYDATA_HOME}/flydata.pid`")
          say('Done.')
          return true
        end
        raise 'Something wrong..'
      end
      def restart
        if process_exist?
          say('Restarting sender process.')
          if system("kill -HUP `cat #{FLYDATA_HOME}/flydata.pid`")
            say('Done.')
            return true
          else
            raise 'Something wrong..'
          end
        else
          say("Process doesn't exist.")
          start
        end
      end

      private
      def wait_until_server_ready
        retry_count = 10
        1.upto(retry_count) do |i|
          return true if server_ready?
          say("Waiting server side active... (#{i}/#{retry_count})")
          sleep 30
        end
        false
      end
      def wait_until_client_ready
        retry_count = 10
        1.upto(retry_count) do |i|
          if client_ready?
            say("Done! Client is ready now.")
            return true
          end
          say("Waiting client side active... (#{i}/#{retry_count})")
          sleep 30
        end
        raise "Somthing wrong... Please try setup command again."
      end
      def wait_until_logs_uploaded
        say('Starting to check uploading from your server.')
        data_port = flydata.data_port.get
        data_port_id = data_port['id']

        retry_count = 10
        1.upto(retry_count) do |i|
          if uploaded_successfully?(data_port_id)
            say("Uploading your logs correctly.")
            return true
          end
          say("Waiting logs uploading... (#{i}/#{retry_count})")
          sleep 30
        end
        raise 'Cannot confirm your logs on flydata server. Something wrong..'
      end

      def server_ready?
        data_port = flydata.data_port.get
        data_port['server_status'] == 'active'
      end
      def client_ready?
        process_exist?
      end
      def process_exist?
        process_count = `ps aux|grep -v grep|grep "\\.flydata/flydata\\.conf"|grep fluentd|wc -l`.to_i
        process_count > 0
      end
      def uploaded_successfully?(data_port_id)
        res = flydata.get("/data_ports/#{data_port_id}/tail.json")
        res and res['logs'] and res['logs'].size > 0
      end
    end
  end
end