From 2d3dcee01f9d6963afbf79eb7654914fb0e5a9e0 Mon Sep 17 00:00:00 2001
From: Emmie Maeda <emmie.maeda@gmail.com>
Date: Sun, 28 Jan 2024 19:04:23 -0500
Subject: [PATCH] Add handling for job types.

---
 yellowstone/job/__init__.py               | 18 +++++++++++++++++-
 yellowstone/job/index_forum_categories.py |  2 +-
 yellowstone/job/index_forum_threads.py    |  2 +-
 3 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/yellowstone/job/__init__.py b/yellowstone/job/__init__.py
index 0dbfa84..21b57f2 100644
--- a/yellowstone/job/__init__.py
+++ b/yellowstone/job/__init__.py
@@ -14,7 +14,13 @@
 
 from ..exception import UnknownJobError
 from ..types import Json
-from . import get_user, get_user_avatar, index_site_members
+from . import (
+    get_user,
+    get_user_avatar,
+    index_forum_categories,
+    index_forum_threads,
+    index_site_members,
+)
 from .get_user import GetUserJob
 from .get_user_avatar import GetUserAvatarJob
 from .index_forum_threads import ForumThreadsJob
@@ -116,6 +122,16 @@ def process(self, core: "BackupDispatcher", job: JobDict) -> None:
                         core,
                         cast(index_site_members.SiteMemberJob, data),
                     )
+                case JobType.INDEX_FORUM_CATEGORIES:
+                    index_forum_categories.run(
+                        core,
+                        cast(index_forum_categories.ForumCategoriesJob, data),
+                    )
+                case JobType.INDEX_FORUM_THREADS:
+                    index_forum_threads.run(
+                        core,
+                        cast(index_forum_threads.ForumThreadsJob, data),
+                    )
                 case JobType.FETCH_USER:
                     get_user.run(
                         core,
diff --git a/yellowstone/job/index_forum_categories.py b/yellowstone/job/index_forum_categories.py
index 511122a..1944807 100644
--- a/yellowstone/job/index_forum_categories.py
+++ b/yellowstone/job/index_forum_categories.py
@@ -26,7 +26,7 @@ class ForumCategoryProgressRow(TypedDict):
     last_post_id: Optional[int]
 
 
-def run(core: "BackupDispatcher", *, data: ForumCategoriesJob) -> None:
+def run(core: "BackupDispatcher", data: ForumCategoriesJob) -> None:
     site_slug = data["site_slug"]
 
     # Clear out previous forum groups
diff --git a/yellowstone/job/index_forum_threads.py b/yellowstone/job/index_forum_threads.py
index fd1f82b..2a945be 100644
--- a/yellowstone/job/index_forum_threads.py
+++ b/yellowstone/job/index_forum_threads.py
@@ -19,6 +19,6 @@ class ForumThreadsJob(TypedDict):
     category_id: int
 
 
-def run(core: "BackupDispatcher", *, data: ForumThreadsJob) -> None:
+def run(core: "BackupDispatcher", data: ForumThreadsJob) -> None:
     # TODO
     pass