From 283cc9a91828ea1c053ba2ae4d0910a6d12636d0 Mon Sep 17 00:00:00 2001
From: Benjamin Ryan <ben@ryben.dev>
Date: Tue, 24 Jan 2023 06:02:50 -0600
Subject: [PATCH] Add files via upload

---
 OBSReplayFolders.lua | 62 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 OBSReplayFolders.lua

diff --git a/OBSReplayFolders.lua b/OBSReplayFolders.lua
new file mode 100644
index 0000000..c7c4a7e
--- /dev/null
+++ b/OBSReplayFolders.lua
@@ -0,0 +1,62 @@
+obs = obslua
+
+function script_description()
+    return [[Saves replays to sub-folders using the Game Capture window name.
+    
+Author: redraskal]]
+end
+
+function script_load()
+    obs.obs_frontend_add_event_callback(obs_frontend_callback)
+end
+
+function obs_frontend_callback(event)
+    if event == obs.OBS_FRONTEND_EVENT_REPLAY_BUFFER_SAVED then
+        local path = get_replay_buffer_output()
+        local folder = get_game_capture_window_name()
+        if path ~= nil and folder ~= nil then
+            move(path, folder)
+        end
+    end
+end
+
+function get_replay_buffer_output()
+    local replay_buffer = obs.obs_frontend_get_replay_buffer_output()
+    local cd = obs.calldata_create()
+    local ph = obs.obs_output_get_proc_handler(replay_buffer)
+    obs.proc_handler_call(ph, "get_last_replay", cd)
+    local path = obs.calldata_string(cd, "path")
+    obs.calldata_destroy(cd)
+    obs.obs_output_release(replay_buffer)
+    return path
+end
+
+function get_game_capture_window_name()
+    -- TODO: Infer source
+    -- current_scene = obs.obs_frontend_get_current_scene()
+    -- game_capture = obs.obs_scene_find_source(current_scene, "Game Capture")
+    game_capture = obs.obs_get_source_by_name("Game Capture")
+    local properties = obs.obs_source_properties(game_capture)
+    local prop = obs.obs_properties_get(properties, "window")
+    local setting = obs.obs_property_name(prop)
+    local settings = obs.obs_source_get_settings(game_capture)
+    local value = obs.obs_data_get_string(settings, setting)
+    obs.obs_data_release(settings)
+    obs.obs_properties_destroy(properties)
+    obs.obs_source_release(game_capture)
+    -- obs.obs_source_release(current_scene)
+    local i, j = string.find(value, "  :")
+    local title = string.sub(value, 1, j - 3)
+    return title
+end
+
+function move(path, folder)
+    local sep = string.match(path, "^.*()/")
+    local root = string.sub(path, 1, sep) .. folder
+    local file_name = string.sub(path, sep, string.len(path))
+    local adjusted_path = root .. file_name
+    if obs.os_file_exists(root) == false then
+        obs.os_mkdir(root)
+    end
+    obs.os_rename(path, adjusted_path)
+end