From 110301a975e43739192577166d089e28c22ae266 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Sat, 29 Jul 2023 21:20:12 +0800
Subject: Add API server

---
 recording/runner.go |  7 ++++++-
 recording/task.go   | 16 +++++++++++++---
 2 files changed, 19 insertions(+), 4 deletions(-)

(limited to 'recording')

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,
diff --git a/recording/task.go b/recording/task.go
index 26f3638..43b6678 100644
--- a/recording/task.go
+++ b/recording/task.go
@@ -10,6 +10,7 @@ import (
 	"fmt"
 	"github.com/keuin/slbr/common/retry"
 	"github.com/keuin/slbr/logging"
+	"sync/atomic"
 	"time"
 )
 
@@ -51,7 +52,16 @@ type RunningTask struct {
 	// hookStopped: called asynchronously when the task is stopped. This won't be called when restarting.
 	hookStopped func()
 	// logger: where to print logs
-	logger logging.Logger
+	logger    logging.Logger
+	roomTitle atomic.Pointer[string]
+}
+
+func (t *RunningTask) GetStatus() TaskStatus {
+	return t.status
+}
+
+func (t *RunningTask) GetRoomTitle() *string {
+	return t.roomTitle.Load()
 }
 
 func NewRunningTask(
@@ -60,8 +70,8 @@ func NewRunningTask(
 	hookStarted func(),
 	hookStopped func(),
 	logger logging.Logger,
-) RunningTask {
-	return RunningTask{
+) *RunningTask {
+	return &RunningTask{
 		TaskConfig:  config,
 		ctx:         ctx,
 		status:      StNotStarted,
-- 
cgit v1.2.3