Scanner: Get + POST Chaptet thumbnail

Arthi-chaud · Apr 6, 2024 · 45987b0 · 45987b0
1 parent 10114e3
commit 45987b0
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 7 deletions.
diff --git a/scanner/pkg/actions/register.go b/scanner/pkg/actions/register.go
@@ -39,6 +39,15 @@ func RegisterFile(path string, c *config.Config) error {
 			glg.Failf(err.Error())
 			return err
 		}
+		go func() {
+			for i, chapterId := range res.ChaptersId {
+				thumbnailBytes, err := pkg.GetFrame(path, int64(dto.Chapters[i].StartTimestamp)+int64(dto.Chapters[i].EndTimestamp-dto.Chapters[i].StartTimestamp)/2)
+				api.SaveChapterThumbnail(chapterId, thumbnailBytes, *c)
+				if err != nil {
+					glg.Fail(err)
+				}
+			}
+		}()
 	} else if parsedPath.Extra != nil {
 		dto, err := buildExtraDto(path, parsedPath.Extra, mediainfo)
 
@@ -53,14 +62,14 @@ func RegisterFile(path string, c *config.Config) error {
 			return err
 		}
 	}
-	thumbnailBytes, err := pkg.GetFrame(path, int64(mediainfo.Duration) / 2)
-	if (err != nil) {
+	thumbnailBytes, err := pkg.GetFrame(path, int64(mediainfo.Duration)/2)
+	if err != nil {
 		glg.Failf(err.Error())
 		return err
 	}
-	if (parsedPath.Extra != nil) {
+	if parsedPath.Extra != nil {
 		err = api.SaveExtraThumbnail(resourceUuid, thumbnailBytes, *c)
-	} else if (parsedPath.Movie != nil) {
+	} else if parsedPath.Movie != nil {
 		err = api.SaveMovieThumbnail(resourceUuid, thumbnailBytes, *c)
 	}
 	return err

diff --git a/scanner/pkg/api/models.go b/scanner/pkg/api/models.go
@@ -18,6 +18,11 @@ type NewExtraResponse struct {
 	ExtraId string `validate:"required" json:"extra_id"`
 }
 
+type NewChapterResponse struct {
+	ExtraId string `validate:"required" json:"extra_id"`
+}
+
 type NewMovieResponse struct {
-	MovieId string `validate:"required" json:"movie_id"`
-}
+	MovieId    string   `validate:"required" json:"movie_id"`
+	ChaptersId []string `validate:"required,dive,required" json:"chapters_id"`
+}
diff --git a/scanner/pkg/thumbnail.go b/scanner/pkg/thumbnail.go
@@ -21,4 +21,3 @@ func GetFrame(filepath string, timestamp int64) (io.Reader, error) {
 		WithOutput(buf, os.Stdout).Run()
 	return buf, err
 }
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -21,4 +21,3 @@ func GetFrame(filepath string, timestamp int64) (io.Reader, error) {
		WithOutput(buf, os.Stdout).Run()
		return buf, err
		}