#!/usr/bin/env python3
#
#   Copyright 2020 - The Android Open Source Project
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

from blueberry.tests.gd.cert.capture import Capture
from blueberry.tests.gd.cert.matchers import HciMatchers
from blueberry.tests.gd.cert.matchers import SecurityMatchers
import hci_packets as hci


class HalCaptures(object):

    @staticmethod
    def ReadBdAddrCompleteCapture():
        return Capture(lambda packet: packet.payload[0:5] == b'\x0e\x0a\x01\x09\x10',
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def ConnectionRequestCapture():
        return Capture(lambda packet: packet.payload[0:2] == b'\x04\x0a',
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def ConnectionCompleteCapture():
        return Capture(lambda packet: packet.payload[0:3] == b'\x03\x0b\x00',
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def DisconnectionCompleteCapture():
        return Capture(lambda packet: packet.payload[0:2] == b'\x05\x04',
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def LeConnectionCompleteCapture():
        return Capture(
            lambda packet: packet.payload[0] == 0x3e and (packet.payload[2] == 0x01 or packet.payload[2] == 0x0a),
            lambda packet: hci.Event.parse_all(packet.payload))


class HciCaptures(object):

    @staticmethod
    def ReadLocalOobDataCompleteCapture():
        return Capture(
            HciMatchers.CommandComplete(hci.OpCode.READ_LOCAL_OOB_DATA),
            lambda packet: HciMatchers.ExtractMatchingCommandComplete(packet.payload, hci.OpCode.READ_LOCAL_OOB_DATA))

    @staticmethod
    def ReadLocalOobExtendedDataCompleteCapture():
        return Capture(
            HciMatchers.CommandComplete(hci.OpCode.READ_LOCAL_OOB_EXTENDED_DATA), lambda packet: HciMatchers.
            ExtractMatchingCommandComplete(packet.payload, hci.OpCode.READ_LOCAL_OOB_EXTENDED_DATA))

    @staticmethod
    def ReadBdAddrCompleteCapture():
        return Capture(HciMatchers.CommandComplete(hci.OpCode.READ_BD_ADDR),
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def ConnectionRequestCapture():
        return Capture(HciMatchers.EventWithCode(hci.EventCode.CONNECTION_REQUEST),
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def ConnectionCompleteCapture():
        return Capture(HciMatchers.EventWithCode(hci.EventCode.CONNECTION_COMPLETE),
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def DisconnectionCompleteCapture():
        return Capture(HciMatchers.EventWithCode(hci.EventCode.DISCONNECTION_COMPLETE),
                       lambda packet: hci.Event.parse_all(packet.payload))

    @staticmethod
    def LeConnectionCompleteCapture():
        return Capture(HciMatchers.LeConnectionComplete(),
                       lambda packet: HciMatchers.ExtractLeConnectionComplete(packet.payload))

    @staticmethod
    def SimplePairingCompleteCapture():
        return Capture(HciMatchers.EventWithCode(hci.EventCode.SIMPLE_PAIRING_COMPLETE),
                       lambda packet: hci.Event.parse_all(packet.payload))