From c6f12ab2a885e89b938b1e3f3ecd78bd919297dd Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@zoriya.dev>
Date: Wed, 20 Mar 2024 19:35:30 +0100
Subject: [PATCH] Add a simkl sync implementation

---
 autosync/autosync/services/simkl.py | 85 +++++++++++++++++++++++++++++
 autosync/requirements.txt           |  1 +
 scanner/pyproject.toml              |  3 +
 shell.nix                           |  1 +
 4 files changed, 90 insertions(+)
 create mode 100644 autosync/autosync/services/simkl.py

diff --git a/autosync/autosync/services/simkl.py b/autosync/autosync/services/simkl.py
new file mode 100644
index 00000000..12c18a21
--- /dev/null
+++ b/autosync/autosync/services/simkl.py
@@ -0,0 +1,85 @@
+import requests
+import logging
+from ..models.user import User
+from ..models.show import Show
+from ..models.movie import Movie
+from ..models.episode import Episode
+from ..models.watch_status import WatchStatus, Status
+
+
+class Simkl:
+	def __init__(self) -> None:
+		self._api_key = ""
+
+	def update(self, user: User, resource: Movie | Show | Episode, status: WatchStatus):
+		if "simkl" not in user.external_id:
+			return
+
+		watch_date = status.played_date or status.added_date
+
+		if resource.kind == "episode":
+			if status.status != Status.COMPLETED:
+				return
+			resp = requests.post(
+				"https://api.simkl.com/sync/history",
+				json={
+					"episodes": {
+						"watched_at": watch_date,
+						"ids": {
+							service: id.data_id
+							for service, id in resource.external_id.items()
+						},
+					}
+				},
+				headers={
+					"Authorization": f"Bearer {user.external_id["simkl"].token.access_token}",
+					"simkl_api_key": self._api_key,
+				},
+			)
+			logging.debug("Simkl response: %s", resp.json())
+			return
+
+		category = "movies" if resource.kind == "movie" else "shows"
+
+		simkl_status = self._to_simkl_status(status.status)
+		if simkl_status is None:
+			return
+
+		resp = requests.post(
+			"https://api.simkl.com/sync/add-to-list",
+			json={
+				category: {
+					"to": simkl_status,
+					"watched_at": watch_date
+					if status.status == Status.COMPLETED
+					else None,
+					"title": resource.name,
+					"year": resource.year,
+					"ids": {
+						service: id.data_id
+						for service, id in resource.external_id.items()
+					},
+				}
+			},
+			headers={
+				"Authorization": f"Bearer {user.external_id["simkl"].token.access_token}",
+				"simkl_api_key": self._api_key,
+			},
+		)
+		logging.debug("Simkl response: %s", resp.json())
+
+	def _to_simkl_status(self, status: Status):
+		match status:
+			case Status.COMPLETED:
+				return "completed"
+			case Status.WATCHING:
+				return "watching"
+			case Status.COMPLETED:
+				return "completed"
+			case Status.PLANNED:
+				return "plantowatch"
+			case Status.DELETED:
+				# do not delete items on simkl, most of deleted status are for a rewatch.
+				return None
+			case _:
+				return None
diff --git a/autosync/requirements.txt b/autosync/requirements.txt
index df7f4230..9f75faf9 100644
--- a/autosync/requirements.txt
+++ b/autosync/requirements.txt
@@ -1 +1,2 @@
 pika
+requets
diff --git a/scanner/pyproject.toml b/scanner/pyproject.toml
index 84e5d38b..ce8becbf 100644
--- a/scanner/pyproject.toml
+++ b/scanner/pyproject.toml
@@ -1,2 +1,5 @@
 [tool.ruff.format]
 indent-style = "tab"
+
+[tool.pyright]
+reportAbstractUsage = false
diff --git a/shell.nix b/shell.nix
index 8f4f17a6..5c7729e3 100644
--- a/shell.nix
+++ b/shell.nix
@@ -6,6 +6,7 @@
       jsons
       watchfiles
       pika
+      requests
     ]);
   dotnet = with pkgs.dotnetCorePackages;
     combinePackages [