74 lines
2.0 KiB
Go
74 lines
2.0 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"video-factory/shortdrama/model/dto"
|
|
"video-factory/shortdrama/service"
|
|
)
|
|
|
|
type scene struct{}
|
|
|
|
var Scene = new(scene)
|
|
|
|
func (c *scene) ListScene(ctx context.Context, req *dto.ListSceneReq) (res *dto.ListSceneRes, err error) {
|
|
list, err := service.SceneService.ListScene(ctx, req.DramaId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.ListSceneRes{List: list}, nil
|
|
}
|
|
|
|
func (c *scene) AddScene(ctx context.Context, req *dto.AddSceneReq) (res *dto.GetDramaRes, err error) {
|
|
var imagePath string
|
|
if req.ImageFile != nil {
|
|
f, err := req.ImageFile.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer func() { _ = f.Close() }()
|
|
if err := service.ValidateUploadFile(ctx, req.ImageFile.Size, req.ImageFile.Filename, "image"); err != nil {
|
|
return nil, err
|
|
}
|
|
imagePath, err = service.SaveScenePropFile(ctx, req.DramaId, f, req.ImageFile.Filename, "场景", req.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
_, err = service.SceneService.AddScene(ctx, req.DramaId, req.Name, req.Description, imagePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *scene) UpdateScene(ctx context.Context, req *dto.UpdateSceneReq) (res *dto.GetDramaRes, err error) {
|
|
var imagePath string
|
|
if req.ImageFile != nil {
|
|
f, err := req.ImageFile.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer func() { _ = f.Close() }()
|
|
if err := service.ValidateUploadFile(ctx, req.ImageFile.Size, req.ImageFile.Filename, "image"); err != nil {
|
|
return nil, err
|
|
}
|
|
imagePath, err = service.SaveScenePropFile(ctx, req.DramaId, f, req.ImageFile.Filename, "场景", req.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
err = service.SceneService.UpdateScene(ctx, req.Id, req.DramaId, req.Name, req.Description, imagePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *scene) DeleteScene(ctx context.Context, req *dto.DeleteSceneReq) (res *dto.GetDramaRes, err error) {
|
|
err = service.SceneService.DeleteScene(ctx, req.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nil, nil
|
|
}
|