fog/fog-aws

module Fog
  module AWS
    class RDS
      class Real
        require 'fog/aws/parsers/rds/describe_db_snapshots'

        # Describe all or specified db snapshots
        # http://docs.amazonwebservices.com/AmazonRDS/latest/APIReference/API_DescribeDBSnapshots.html
        # ==== Parameters
        # * DBInstanceIdentifier <~String> - ID of instance to retrieve information for. if absent information for all instances is returned
        # * DBSnapshotIdentifier <~String> - ID of snapshot to retrieve information for. if absent information for all snapshots is returned
        # * SnapshotType       <~String> - type of snapshot to retrieve (automated|manual)
        # * Marker               <~String> - An optional marker provided in the previous DescribeDBInstances request
        # * MaxRecords           <~Integer> - Max number of records to return (between 20 and 100)
        # Only one of DBInstanceIdentifier or DBSnapshotIdentifier can be specified
        # ==== Returns
        # * response<~Excon::Response>:
        #   * body<~Hash>:
        def describe_db_snapshots(opts={})
          params = {}
          params['SnapshotType'] = opts[:type] if opts[:type]
          params['DBInstanceIdentifier'] = opts[:identifier] if opts[:identifier]
          params['DBSnapshotIdentifier'] = opts[:snapshot_id] if opts[:snapshot_id]
          params['Marker'] = opts[:marker] if opts[:marker]
          params['MaxRecords'] = opts[:max_records] if opts[:max_records]
          request({
            'Action'  => 'DescribeDBSnapshots',
            :parser   => Fog::Parsers::AWS::RDS::DescribeDBSnapshots.new
          }.merge(params))
        end
      end

      class Mock
        def describe_db_snapshots(opts={})
          response = Excon::Response.new
          snapshots = self.data[:snapshots].values
          if opts[:identifier]
            snapshots = snapshots.select { |snapshot| snapshot['DBInstanceIdentifier'] == opts[:identifier] }

          end

          if opts[:snapshot_id]
            snapshots = snapshots.select { |snapshot| snapshot['DBSnapshotIdentifier'] == opts[:snapshot_id] }
            raise Fog::AWS::RDS::NotFound.new("DBSnapshot #{opts[:snapshot_id]} not found") if snapshots.empty?
          end

          snapshots.each do |snapshot|
            case snapshot['Status']
            when 'creating'
              if Time.now - snapshot['SnapshotCreateTime'] > Fog::Mock.delay
                snapshot['Status'] = 'available'
              end
            end
          end

          # Build response
          response.status = 200
          response.body = {
            "ResponseMetadata"=>{ "RequestId"=> Fog::AWS::Mock.request_id },
            "DescribeDBSnapshotsResult" => { "DBSnapshots" => snapshots }
          }
          response
        end
      end
    end
  end
end