module Fog
  module AWS
    class Compute
      class Real

        require 'fog/aws/parsers/compute/describe_snapshots'

        # Describe all or specified snapshots
        #
        # ==== Parameters
        # * snapshot_id<~Array> - List of snapshots to describe, defaults to all
        # * options<~Array>:
        #   * 'Owner'<~String> - Owner of snapshot in ['self', 'amazon', account_id]
        #   * 'RestorableBy'<~String> - Account id of user who can create volumes from this snapshot
        #
        # ==== Returns
        # * response<~Excon::Response>:
        #   * body<~Hash>:
        #     * 'requestId'<~String> - Id of request
        #     * 'snapshotSet'<~Array>:
        #       * 'progress'<~String>: The percentage progress of the snapshot
        #       * 'snapshotId'<~String>: Id of the snapshot
        #       * 'startTime'<~Time>: Timestamp of when snapshot was initiated
        #       * 'status'<~String>: Snapshot state, in ['pending', 'completed']
        #       * 'volumeId'<~String>: Id of volume that snapshot contains
        def describe_snapshots(snapshot_id = [], options = {})
          options['Owner'] ||= 'self'
          options.merge!(AWS.indexed_param('SnapshotId', snapshot_id))
          request({
            'Action'    => 'DescribeSnapshots',
            :idempotent => true,
            :parser     => Fog::Parsers::AWS::Compute::DescribeSnapshots.new
          }.merge!(options))
        end

      end

      class Mock

        def describe_snapshots(snapshot_id = [])
          response = Excon::Response.new
          snapshot_id = [*snapshot_id]
          if snapshot_id != []
            snapshot_set = @data[:snapshots].reject {|key,value| !snapshot_id.include?(key)}.values
          else
            snapshot_set = @data[:snapshots].values
          end

          if snapshot_id.length == 0 || snapshot_id.length == snapshot_set.length
            snapshot_set.each do |snapshot|
              case snapshot['status']
              when 'in progress', 'pending'
                if Time.now - snapshot['startTime'] > Fog::Mock.delay * 2
                  snapshot['progress']  = '100%'
                  snapshot['status']    = 'completed'
                elsif Time.now - snapshot['startTime'] > Fog::Mock.delay
                  snapshot['progress']  = '50%'
                  snapshot['status']    = 'in progress'
                else
                  snapshot['progress']  = '0%'
                  snapshot['status']    = 'in progress'
                end
              end
            end

            response.status = 200
            response.body = {
              'requestId' => Fog::AWS::Mock.request_id,
              'snapshotSet' => snapshot_set
            }
            response
          else
            raise Fog::AWS::Compute::NotFound.new("The snapshot #{snapshot_id.inspect} does not exist.")
          end
        end

      end
    end
  end
end