require 'socket'
require 'timeout'
require 'fastdfs-client/cmd'


module Fastdfs
  module Client

    class Socket
      attr_accessor :socket, :host, :port

      def initialize(host, port, options = {})
        @host, @port = host, port
        @header_len = ProtoCommon::HEAD_LEN
        @options = options || {}
        @connection_timeout = @options[:connection_timeout] || 3
        @recv_timeout = @options[:recv_timeout] || 20
      end

      def write(*args)
        @cmd = args.shift
        pkg = args.shift

        pkg = pkg.pack("C*") if pkg.is_a?(Array)
        @socket.write pkg
      end

      def close 
        @socket.close if connected
      end

      def connection(&block)
        if @socket.nil? || !connected
          @socket = Timeout.timeout(@connection_timeout) do
            TCPSocket.new(@host, @port)  
          end
        end
        yield if block_given?
      end

      def connected
        !@socket.closed?
      end

      def receive(&block)
        @header = timeout_recv do 
          @socket.recv(@header_len).unpack("C*")
        end
        res_header = parseHeader
        if res_header[:status]
          recv_body if is_recv?
          res = yield(@content) if block_given?
          res_header[:result] = res unless res.nil?
        end
        res_header
      end

      private
      def parseHeader
        err_msg = nil
        err_msg = "recv package size #{@header} is not equal #{@header_len}, cmd: #{CMD::MAPPING_NAME[@cmd]}" unless @header.length == @header_len || err_msg
        err_msg = "recv cmd: #{@header[8]} is not correct, expect recv code: #{CMD::RESP_CODE}, cmd: #{CMD::MAPPING_NAME[@cmd]}" unless @header[8] == CMD::RESP_CODE || err_msg
        err_msg = "recv erron #{@header[9]}, 0 is correct cmd: #{CMD::MAPPING_NAME[@cmd]}" unless @header[9] == 0 || err_msg
        {status: err_msg.nil?, err_msg: err_msg}
      end

      def timeout_recv
        Timeout.timeout(@recv_timeout) do 
          yield if block_given?
        end
      end

      def is_recv?
        recv_body_len > 0
      end

      def recv_body_len
        @header[0...8].to_pack_long
      end

      def recv_body
        @content = ""
        max_len, body_len = ProtoCommon::RECV_MAX_LEN, recv_body_len

        while body_len > 0
          timeout_recv do 
            len = [body_len, max_len].min
            @content << @socket.recv(len) 
            body_len -= len
          end
        end
        @content = nil if @content.blank?
      end

    end
  end
end