import argparse import json import logging import tempfile from pathlib import Path import aiohttp import reolinkapi from yarl import URL from unifi.cams.base import UnifiCamBase class Reolink(UnifiCamBase): def __init__(self, args: argparse.Namespace, logger: logging.Logger) -> None: super().__init__(args, logger) self.snapshot_dir: str = tempfile.mkdtemp() self.motion_in_progress: bool = False self.substream = args.substream self.cam = reolinkapi.Camera( ip=args.ip, username=args.username, password=args.password, ) self.stream_fps = self.get_stream_info(self.cam) @classmethod def add_parser(cls, parser: argparse.ArgumentParser) -> None: super().add_parser(parser) parser.add_argument("--username", "-u", required=True, help="Camera username") parser.add_argument("--password", "-p", required=True, help="Camera password") parser.add_argument( "--channel", "-c", default=0, help="Camera channel (not needed, leaving for possible future)", ) parser.add_argument( "--stream", "-m", default="main", type=str, choices=["main", "sub"], help="Stream profile to use for the higher quality stream", ) parser.add_argument( "--substream", "-s", default="sub", type=str, choices=["main", "sub"], help="Stream profile to use for the lower quality stream", ) def get_stream_info(self, camera) -> tuple[int, int]: info = camera.get_recording_encoding() return ( info[0]["value"]["Enc"]["mainStream"]["frameRate"], info[0]["value"]["Enc"]["subStream"]["frameRate"], ) async def get_snapshot(self) -> Path: img_file = Path(self.snapshot_dir, "screen.jpg") url = ( f"http://{self.args.ip}" f"/cgi-bin/api.cgi?cmd=Snap&channel={self.args.channel}" f"&rs=6PHVjvf0UntSLbyT&user={self.args.username}" f"&password={self.args.password}" ) self.logger.info(f"Grabbing snapshot: {url}") await self.fetch_to_file(url, img_file) return img_file async def run(self) -> None: url = ( f"http://{self.args.ip}" f"/api.cgi?cmd=GetMdState&user={self.args.username}" f"&password={self.args.password}" ) encoded_url = URL(url, encoded=True) body = ( f'[{{ "cmd":"GetMdState", "param":{{ "channel":{self.args.channel} }} }}]' ) while True: self.logger.info(f"Connecting to motion events API: {url}") try: async with aiohttp.ClientSession( timeout=aiohttp.ClientTimeout(None) ) as session: while True: async with session.post(encoded_url, data=body) as resp: data = await resp.read() try: json_body = json.loads(data) if "value" in json_body[0]: if json_body[0]["value"]["state"] == 1: if not self.motion_in_progress: self.motion_in_progress = True self.logger.info("Trigger motion start") await self.trigger_motion_start() elif json_body[0]["value"]["state"] == 0: if self.motion_in_progress: self.motion_in_progress = False self.logger.info("Trigger motion end") await self.trigger_motion_stop() else: self.logger.error( "Motion API request responded with " "unexpected JSON, retrying. " f"JSON: {data}" ) except json.JSONDecodeError as err: self.logger.error( "Motion API request returned invalid " "JSON, retrying. " f"Error: {err}, " f"Response: {data}" ) except aiohttp.ClientError as err: self.logger.error(f"Motion API request failed, retrying. Error: {err}") def get_extra_ffmpeg_args(self, stream_index: str) -> str: if stream_index == "video1": fps = self.stream_fps[0] else: fps = self.stream_fps[1] return ( "-ar 32000 -ac 1 -codec:a aac -b:a 32k -c:v copy -vbsf" f' "h264_metadata=tick_rate={fps*2}"' ) async def get_stream_source(self, stream_index: str) -> str: if stream_index == "video1": stream = self.args.stream else: stream = self.args.substream return ( f"rtmp://{self.args.ip}/bcs/channel{self.args.channel}_{stream}.bcs" f"?channel={self.args.channel}&stream=0&user={self.args.username}" f"&password={self.args.password}" )