Add API server

1 files changed, 6 insertions, 1 deletions

diff --git a/recording/runner.go b/recording/runner.go
index 9be78c9..7bf3790 100644
--- a/recording/runner.go
+++ b/recording/runner.go
@@ -175,7 +175,9 @@ func tryRunTask(t *RunningTask) error {
 			var err error
 			run := true
 			for run {
-				err = record(t.ctx, bi, &t.TaskConfig, t.logger)
+				err = record(t.ctx, bi, &t.TaskConfig, t.logger, func(resp types.RoomProfileResponse) {
+					t.roomTitle.Store(&resp.Data.Title)
+				})
 				if err == nil {
 					// live is ended
 					t.logger.Info("The live is ended. Restarting current task...")
@@ -247,6 +249,7 @@ func record(
 	bi *bilibili.Bilibili,
 	task *TaskConfig,
 	logger logging.Logger,
+	profileConsumer func(types.RoomProfileResponse),
 ) error {
 	logger.Info("Getting room profile...")
 
@@ -265,6 +268,8 @@ func record(
 		return errs.NewError(errs.GetRoomInfo, err)
 	}
 
+	profileConsumer(profile)
+
 	logger.Info("Getting stream url...")
 	urlInfo, err := AutoRetryWithConfig(
 		ctx,