Add record split hotkey and websocket request

Author: exeldro

source-record.c

@@ -38,6 +38,7 @@ struct source_record_filter_context {
 	bool replayBuffer;
 	obs_hotkey_pair_id enableHotkey;
 	obs_hotkey_pair_id pauseHotkeys;
+	obs_hotkey_id splitHotkey;
 	int audio_track;
 	obs_weak_source_t *audio_source;
 	bool closing;
@@ -263,7 +264,8 @@ static void start_stream_output_task(void *data)
 	}
 	context->starting_stream_output = false;
 }
-static void release_encoders(void* param) {
+static void release_encoders(void *param)
+{
 	struct source_record_filter_context *context = param;
 	if (obs_source_enabled(context->source) && (context->replayBuffer || context->record || context->stream))
 		return;
@@ -974,6 +976,7 @@ static void *source_record_filter_create(obs_data_t *settings, obs_source_t *sou
 	context->last_frontend_event = -1;
 	context->enableHotkey = OBS_INVALID_HOTKEY_PAIR_ID;
 	context->pauseHotkeys = OBS_INVALID_HOTKEY_PAIR_ID;
+	context->splitHotkey = OBS_INVALID_HOTKEY_ID;
 	source_record_filter_update(context, settings);
 	obs_frontend_add_event_callback(frontend_event, context);
 	return context;
@@ -1054,6 +1057,9 @@ static void source_record_filter_destroy(void *data)
 	if (context->pauseHotkeys != OBS_INVALID_HOTKEY_PAIR_ID)
 		obs_hotkey_pair_unregister(context->pauseHotkeys);
 
+	if (context->splitHotkey != OBS_INVALID_HOTKEY_ID)
+		obs_hotkey_unregister(context->splitHotkey);
+
 	source_record_delayed_destroy(context);
 }
 
@@ -1114,6 +1120,21 @@ static bool source_record_unpause_hotkey(void *data, obs_hotkey_pair_id id, obs_
 	return true;
 }
 
+static void source_record_split_hotkey(void *data, obs_hotkey_id id, obs_hotkey_t *hotkey, bool pressed) {
+	UNUSED_PARAMETER(id);
+	UNUSED_PARAMETER(hotkey);
+	if (!pressed)
+		return;
+	struct source_record_filter_context *context = data;
+	if (!context->fileOutput)
+		return;
+	proc_handler_t *ph = obs_output_get_proc_handler(context->fileOutput);
+	struct calldata cd;
+	calldata_init(&cd);
+	proc_handler_call(ph, "split_file", &cd);
+	calldata_free(&cd);
+}
+
 static void source_record_filter_tick(void *data, float seconds)
 {
 	UNUSED_PARAMETER(seconds);
@@ -1137,6 +1158,11 @@ static void source_record_filter_tick(void *data, float seconds)
 			"source_record.UnpauseRecording", obs_frontend_get_locale_string("Basic.Main.UnpauseRecording"),
 			source_record_pause_hotkey, source_record_unpause_hotkey, context, context);
 
+	if (context->splitHotkey == OBS_INVALID_HOTKEY_ID)
+		context->splitHotkey = obs_hotkey_register_source(parent, "source_record.SplitRecording",
+								  obs_frontend_get_locale_string("Basic.Main.SplitFile"),
+								  source_record_split_hotkey, context);
+
 	uint32_t width = obs_source_get_width(parent);
 	width += (width & 1);
 	uint32_t height = obs_source_get_height(parent);
@@ -1689,10 +1715,10 @@ static bool pause_record_source(obs_source_t *source, obs_data_t *request_data,
 		return false;
 
 	struct source_record_filter_context *context = obs_obj_get_data(filter);
+	obs_source_release(filter);
 	if (!context->fileOutput)
 		return false;
 	obs_output_pause(context->fileOutput, true);
-	obs_source_release(filter);
 	return true;
 }
 
@@ -1703,10 +1729,31 @@ static bool unpause_record_source(obs_source_t *source, obs_data_t *request_data
 		return false;
 
 	struct source_record_filter_context *context = obs_obj_get_data(filter);
+	obs_source_release(filter);
 	if (!context->fileOutput)
 		return false;
 	obs_output_pause(context->fileOutput, false);
+	return true;
+}
+
+static bool split_record_source(obs_source_t *source, obs_data_t *request_data, obs_data_t *response_data)
+{
+	obs_source_t *filter = get_source_record_filter(source, request_data, response_data, false);
+	if (!filter)
+		return false;
+
+	struct source_record_filter_context *context = obs_obj_get_data(filter);
 	obs_source_release(filter);
+	if (!context->fileOutput)
+		return false;
+	proc_handler_t *ph = obs_output_get_proc_handler(context->fileOutput);
+	struct calldata cd;
+	calldata_init(&cd);
+	if (!proc_handler_call(ph, "split_file", &cd)) {
+		calldata_free(&cd);
+		return false;
+	}
+	calldata_free(&cd);
 	return true;
 }
 
@@ -1819,6 +1866,37 @@ static void websocket_unpause_record(obs_data_t *request_data, obs_data_t *respo
 	obs_data_set_bool(response_data, "success", success);
 }
 
+static void websocket_split_record(obs_data_t *request_data, obs_data_t *response_data, void *param)
+{
+	UNUSED_PARAMETER(param);
+	const char *source_name = obs_data_get_string(request_data, "source");
+	bool success = true;
+	if (strlen(source_name)) {
+		obs_source_t *source = obs_get_source_by_name(source_name);
+		if (!source) {
+			obs_data_set_string(response_data, "error", "source not found");
+			obs_data_set_bool(response_data, "success", false);
+			return;
+		}
+		success = split_record_source(source, request_data, response_data);
+		obs_source_release(source);
+	} else {
+		DARRAY(obs_source_t *) sources = {0};
+		obs_enum_sources(find_source, &sources);
+		obs_enum_scenes(find_source, &sources);
+		if (!sources.num) {
+			obs_data_set_string(response_data, "error", "no source found");
+			obs_data_set_bool(response_data, "success", false);
+			return;
+		}
+		for (size_t i = 0; i < sources.num; i++) {
+			success = split_record_source(sources.array[i], request_data, response_data) && success;
+		}
+		da_free(sources);
+	}
+	obs_data_set_bool(response_data, "success", success);
+}
+
 static void websocket_stop_record(obs_data_t *request_data, obs_data_t *response_data, void *param)
 {
 	UNUSED_PARAMETER(param);
@@ -2119,6 +2197,7 @@ bool obs_module_load(void)
 	obs_websocket_vendor_register_request(vendor, "record_start", websocket_start_record, NULL);
 	obs_websocket_vendor_register_request(vendor, "record_pause", websocket_pause_record, NULL);
 	obs_websocket_vendor_register_request(vendor, "record_unpause", websocket_unpause_record, NULL);
+	obs_websocket_vendor_register_request(vendor, "record_split", websocket_split_record, NULL);
 	obs_websocket_vendor_register_request(vendor, "record_stop", websocket_stop_record, NULL);
 	obs_websocket_vendor_register_request(vendor, "replay_buffer_start", websocket_start_replay_buffer, NULL);
 	obs_websocket_vendor_register_request(vendor, "replay_buffer_stop", websocket_stop_replay_buffer, NULL);