bob

unifi/cams/reolink.py

@@ -0,0 +1,148 @@
+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.username}:{self.args.password}@{self.args.ip}:1935"
+            f"//h264Preview_{int(self.args.channel) + 1:02}_{stream}"
+        )