184 lines
7 KiB
Lua
Executable file
184 lines
7 KiB
Lua
Executable file
local output_path = obslua.obs_frontend_get_current_record_output_path()
|
|
local output_path_prop_key = "output_path"
|
|
local remux_enabled_prop_key = "remux_enabled"
|
|
local remux_format_prop_key = "remux_format"
|
|
local input_format_prop_key = "input_format"
|
|
local delete_original_prop_key = "delete_original"
|
|
local ffmpeg_path_prop_key = "ffmpeg_path"
|
|
|
|
local REPLAY_EVENT = 0
|
|
local RECORDING_EVENT = 1
|
|
local SCREENSHOT_EVENT = 2
|
|
local remux_enabled = false
|
|
local remux_format = "mp4"
|
|
local input_format = "mkv"
|
|
local delete_original = false
|
|
local ffmpeg_path = "ffmpeg"
|
|
|
|
function script_description()
|
|
return [[
|
|
<h2>OBSReplay</h2>
|
|
<p>Moves recordings, replays, and screenshots to a folder named after the current scene and optionally remuxes them to a different format.</p>
|
|
]]
|
|
end
|
|
|
|
function script_load()
|
|
obslua.obs_frontend_add_event_callback(obs_frontend_callback)
|
|
end
|
|
|
|
function obs_frontend_callback(callback, _private_data)
|
|
if callback == obslua.OBS_FRONTEND_EVENT_REPLAY_BUFFER_SAVED then
|
|
save(REPLAY_EVENT)
|
|
elseif callback == obslua.OBS_FRONTEND_EVENT_RECORDING_STOPPED then
|
|
save(RECORDING_EVENT)
|
|
elseif callback == obslua.OBS_FRONTEND_EVENT_SCREENSHOT_TAKEN then
|
|
save(SCREENSHOT_EVENT)
|
|
end
|
|
end
|
|
|
|
function script_properties()
|
|
local prop = obslua.obs_properties_create()
|
|
obslua.obs_properties_add_path(prop, output_path_prop_key, "Output path:", obslua.OBS_PATH_DIRECTORY, nil, nil)
|
|
obslua.obs_properties_add_path(prop, ffmpeg_path_prop_key, "FFmpeg Path:", obslua.OBS_PATH_FILE, nil, nil)
|
|
obslua.obs_properties_add_bool(prop, remux_enabled_prop_key, "Enable Remuxing")
|
|
obslua.obs_properties_add_bool(prop, delete_original_prop_key, "Delete Original After Remux")
|
|
|
|
local input_format_list = obslua.obs_properties_add_list(
|
|
prop, input_format_prop_key, "Input Format", obslua.OBS_COMBO_TYPE_EDITABLE, obslua.OBS_COMBO_FORMAT_STRING
|
|
)
|
|
obslua.obs_property_list_add_string(input_format_list, "MKV", "mkv")
|
|
obslua.obs_property_list_add_string(input_format_list, "MOV", "mov")
|
|
obslua.obs_property_list_add_string(input_format_list, "FLV", "flv")
|
|
obslua.obs_property_list_add_string(input_format_list, "TS", "ts")
|
|
obslua.obs_property_list_add_string(input_format_list, "MP4", "mp4")
|
|
|
|
local remux_format_list = obslua.obs_properties_add_list(
|
|
prop, remux_format_prop_key, "Remux To", obslua.OBS_COMBO_TYPE_EDITABLE, obslua.OBS_COMBO_FORMAT_STRING
|
|
)
|
|
obslua.obs_property_list_add_string(remux_format_list, "MP4", "mp4")
|
|
obslua.obs_property_list_add_string(remux_format_list, "MKV", "mkv")
|
|
obslua.obs_property_list_add_string(remux_format_list, "MOV", "mov")
|
|
obslua.obs_property_list_add_string(remux_format_list, "FLV", "flv")
|
|
obslua.obs_property_list_add_string(remux_format_list, "TS", "ts")
|
|
|
|
return prop
|
|
end
|
|
|
|
function script_update(settings)
|
|
output_path = obslua.obs_data_get_string(settings, output_path_prop_key)
|
|
remux_enabled = obslua.obs_data_get_bool(settings, remux_enabled_prop_key)
|
|
delete_original = obslua.obs_data_get_bool(settings, delete_original_prop_key)
|
|
ffmpeg_path = obslua.obs_data_get_string(settings, ffmpeg_path_prop_key)
|
|
input_format = obslua.obs_data_get_string(settings, input_format_prop_key)
|
|
remux_format = obslua.obs_data_get_string(settings, remux_format_prop_key)
|
|
|
|
if ffmpeg_path == "" then ffmpeg_path = "ffmpeg" end
|
|
print("Updated output path:", output_path)
|
|
print("Remux Enabled:", remux_enabled)
|
|
print("Input Format:", input_format)
|
|
print("Remux To:", remux_format)
|
|
print("Delete Original After Remux:", delete_original)
|
|
print("FFmpeg Path:", ffmpeg_path)
|
|
end
|
|
|
|
function script_defaults(settings)
|
|
obslua.obs_data_set_default_string(settings, output_path_prop_key, output_path)
|
|
obslua.obs_data_set_default_bool(settings, remux_enabled_prop_key, false)
|
|
obslua.obs_data_set_default_string(settings, input_format_prop_key, "mkv")
|
|
obslua.obs_data_set_default_string(settings, remux_format_prop_key, "mp4")
|
|
obslua.obs_data_set_default_bool(settings, delete_original_prop_key, false)
|
|
obslua.obs_data_set_default_string(settings, ffmpeg_path_prop_key, "ffmpeg")
|
|
end
|
|
|
|
function save(event)
|
|
local latest_file_path = (event == REPLAY_EVENT and obslua.obs_frontend_get_last_replay())
|
|
or (event == RECORDING_EVENT and obslua.obs_frontend_get_last_recording())
|
|
or (event == SCREENSHOT_EVENT and obslua.obs_frontend_get_last_screenshot())
|
|
|
|
if latest_file_path and latest_file_path ~= "" then
|
|
local current_scene_name = get_current_scene_name()
|
|
local scene_folder_path = output_path .. "/" .. current_scene_name
|
|
|
|
if package.config:sub(1, 1) == "\\" then
|
|
os.execute(('powershell -Command "if (!(Test-Path \\"%s\\")) { New-Item -ItemType Directory -Path \\"%s\\" }"')
|
|
:format(scene_folder_path, scene_folder_path))
|
|
else
|
|
os.execute(('mkdir -p "%s"'):format(scene_folder_path))
|
|
end
|
|
|
|
local file_name = latest_file_path:match("([^/\\]+)$")
|
|
local moved_file_path = scene_folder_path .. "/" .. file_name
|
|
|
|
local move_command
|
|
if package.config:sub(1, 1) == "\\" then
|
|
move_command = ('powershell -Command "Move-Item -Path \\"%s\\" -Destination \\"%s\\" -Force"')
|
|
:format(latest_file_path, moved_file_path)
|
|
else
|
|
move_command = ('mv "%s" "%s"'):format(latest_file_path, moved_file_path)
|
|
end
|
|
print("Moving file:", move_command)
|
|
os.execute(move_command)
|
|
|
|
local file_exists = io.open(moved_file_path, "r")
|
|
if not file_exists then
|
|
print("Error: Moved file does not exist, remuxing skipped.")
|
|
return
|
|
else
|
|
file_exists:close()
|
|
end
|
|
|
|
local base_name, ext = file_name:match("(.+)%.([^.]+)$")
|
|
if not base_name or not ext then
|
|
print("Error: Could not extract base name and extension from '" .. file_name .. "'")
|
|
return
|
|
end
|
|
|
|
ext = ext:lower()
|
|
local normalized_input_format = input_format:lower()
|
|
|
|
if remux_enabled and ext == normalized_input_format then
|
|
local new_file_path = scene_folder_path .. "/" .. base_name .. "." .. remux_format:lower()
|
|
|
|
local remux_command = ('%s -y -i "%s" -c:v copy -c:a copy "%s" 2>&1')
|
|
:format(ffmpeg_path, moved_file_path, new_file_path)
|
|
|
|
print("Executing remux command:", remux_command)
|
|
|
|
local handle = io.popen(remux_command, "r")
|
|
local result = handle:read("*a")
|
|
handle:close()
|
|
|
|
print("FFmpeg output:", result)
|
|
|
|
local remuxed_file_exists = io.open(new_file_path, "r")
|
|
if not remuxed_file_exists then
|
|
print("Error: Remuxing failed, original file will not be deleted.")
|
|
return
|
|
else
|
|
remuxed_file_exists:close()
|
|
end
|
|
|
|
if delete_original then
|
|
local delete_command
|
|
if package.config:sub(1, 1) == "\\" then
|
|
delete_command = ('powershell -Command "Remove-Item -Path \\"%s\\" -Force"'):format(moved_file_path)
|
|
else
|
|
delete_command = ('rm "%s"'):format(moved_file_path)
|
|
end
|
|
print("Deleting original:", delete_command)
|
|
os.execute(delete_command)
|
|
end
|
|
else
|
|
print("Remuxing skipped: File format '" .. ext .. "' does not match selected input format '" .. normalized_input_format .. "'")
|
|
end
|
|
else
|
|
print("No file found to move.")
|
|
end
|
|
end
|
|
|
|
function get_current_scene_name()
|
|
local current_scene = obslua.obs_frontend_get_current_scene()
|
|
local name = obslua.obs_source_get_name(current_scene)
|
|
obslua.obs_source_release(current_scene)
|
|
return name
|
|
end
|