import regex as re

from watfag.parsers.generic import GenericParser
from watfag.parsers.generic.parsers import CheckParser
from watfag.parsers.generic.watfag import VideoCodec

checks = {
    re.compile(r"avc|[hx][\. -]?264", re.IGNORECASE): VideoCodec.AVC,
    re.compile(r"hevc|[hx][\. -]?265", re.IGNORECASE): VideoCodec.HEVC,
    re.compile(r"vp9", re.IGNORECASE): VideoCodec.VP9,
    re.compile(r"av1", re.IGNORECASE): VideoCodec.AV1,
    re.compile(r"mpeg-4|mpeg4|mp4v|xvid", re.IGNORECASE): VideoCodec.XVID
}

class VideoCodecParser(CheckParser, GenericParser):
    def __init__(self, release):
        super().__init__(release)
        self.checks = checks
        self.remove_checks = []  # No remove checks for video codec
        self.attribute_name = "video_codec"