module Flydata
  module Command
    class Sender < Base
      FLYDATA_HOME=Flydata::HOME_DIR
      def start
        # Check if process exist
        if process_exist?
          say("Process is still running. 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

        data_port = flydata.data_port.get
        say("Go to your Dashboard! #{flydata.flydata_api_host}/data_ports/#{data_port['id']}")
        say <<EOF
Please Note: Records and Total Size are updated every 10 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`") and wait_until_client_stop
          say('Done.')
          return true
        end
        raise 'Something has gone 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 has gone 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 for the server side to become 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 for the client side to become active... (#{i}/#{retry_count})")
          sleep 30
        end
        raise "Somthing has gone wrong... Please try setup command again."
      end
      def wait_until_client_stop
        retry_count = 5
        1.upto(retry_count) do |i|
          return true unless process_exist?
          say("Waiting for the client stopping... (#{i}/#{retry_count})")
          sleep 3
        end
        false
      end
      def wait_until_logs_uploaded
        say('Starting to check the upload 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 that your logs exist on the FlyData server. Something has gone 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 auxww|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