From bea9ce1a2e0af97be5481289c2a79d7ebf39c35e Mon Sep 17 00:00:00 2001
From: Tweeticoats <60335703+Tweeticoats@users.noreply.github.com>
Date: Thu, 26 Sep 2024 13:34:55 +0930
Subject: [PATCH] Adding a plugin to get markers from funscripts. (#439)

Co-authored-by: Tweeticoats <Tweeticoats@github.com>
---
 plugins/funscriptMarkers/funscriptMarkers.py  | 95 +++++++++++++++++++
 plugins/funscriptMarkers/funscriptMarkers.yml | 13 +++
 2 files changed, 108 insertions(+)
 create mode 100644 plugins/funscriptMarkers/funscriptMarkers.py
 create mode 100644 plugins/funscriptMarkers/funscriptMarkers.yml

diff --git a/plugins/funscriptMarkers/funscriptMarkers.py b/plugins/funscriptMarkers/funscriptMarkers.py
new file mode 100644
index 00000000..0e11c23c
--- /dev/null
+++ b/plugins/funscriptMarkers/funscriptMarkers.py
@@ -0,0 +1,95 @@
+import stashapi.log as log
+from stashapi.stashapp import StashInterface
+import stashapi.marker_parse as mp
+import sys
+from pathlib import Path
+import json
+import re
+import time
+
+
+def processScene(scene):
+    log.debug(scene["scene_markers"])
+    if len(scene["scene_markers"]) == 0:
+        for f in scene["files"]:
+            scriptfile = Path(f["path"]).parent / (Path(f["path"]).stem + ".funscript")
+            log.debug(scriptfile)
+            if scriptfile.exists():
+                with open(scriptfile) as f2:
+                    script_json = json.load(f2)
+                    if "metadata" in script_json:
+                        markers = []
+                        if "chapters" in script_json["metadata"]:
+                            for c in script_json["metadata"]["chapters"]:
+                                name = c["name"]
+                                if len(name) == 0:
+                                    name = "#"
+                                marker = {
+                                    "seconds": sum(
+                                        x * float(t)
+                                        for x, t in zip(
+                                            [3600, 60, 1],
+                                            re.split(r"[:]", c["startTime"]),
+                                        )
+                                    ),
+                                    "primary_tag": name,
+                                    "tags": [],
+                                    "title": name,
+                                }
+                                log.debug(marker)
+                                markers.append(marker)
+                            if len(markers) > 0:
+                                mp.import_scene_markers(stash, markers, scene["id"], 15)
+
+
+def processAll():
+    query = {
+        "has_markers": "false",
+        "interactive": True,
+    }
+    per_page = 100
+    log.info("Getting scene count")
+    count = stash.find_scenes(
+        f=query,
+        filter={"per_page": 1},
+        get_count=True,
+    )[0]
+    log.info(str(count) + " scenes to process.")
+    #    i = 0
+    # 98
+    for r in range(1, int(count / per_page) + 2):
+        i = (r - 1) * per_page
+        log.info(
+            "fetching data: %s - %s %0.1f%%"
+            % (
+                (r - 1) * per_page,
+                r * per_page,
+                (i / count) * 100,
+            )
+        )
+        scenes = stash.find_scenes(
+            f=query,
+            filter={"page": r, "per_page": per_page},
+        )
+        for s in scenes:
+            processScene(s)
+            i = i + 1
+            log.progress((i / count))
+            time.sleep(2)
+
+
+json_input = json.loads(sys.stdin.read())
+
+FRAGMENT_SERVER = json_input["server_connection"]
+stash = StashInterface(FRAGMENT_SERVER)
+
+if "mode" in json_input["args"]:
+    PLUGIN_ARGS = json_input["args"]["mode"]
+    if "processAll" == PLUGIN_ARGS:
+        processAll()
+elif "hookContext" in json_input["args"]:
+    _id = json_input["args"]["hookContext"]["id"]
+    _type = json_input["args"]["hookContext"]["type"]
+    if _type == "Scene.Update.Post":
+        scene = stash.find_scene(_id)
+        processScene(scene)
diff --git a/plugins/funscriptMarkers/funscriptMarkers.yml b/plugins/funscriptMarkers/funscriptMarkers.yml
new file mode 100644
index 00000000..d7053b65
--- /dev/null
+++ b/plugins/funscriptMarkers/funscriptMarkers.yml
@@ -0,0 +1,13 @@
+name: Funscript Markers
+description: Create markers if there is a funscript with "chapters" included in the metadata of the script
+version: 0.1
+url: https://github.com/stashapp/CommunityScripts/
+exec:
+  - python
+  - "{pluginDir}/funscriptMarkers.py"
+interface: raw
+tasks:
+  - name: "Re-process All"
+    description: Look at funscript files and see if there are chapters in the metadata
+    defaultArgs:
+      mode: processAll