Alexander
24f355c5ae
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/claude-agent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/riverqueue/river"
|
|
"github.com/rs/zerolog/log"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
pb "homelab.lan/music-agregator/gen/music_agregator/v1"
|
|
"homelab.lan/music-agregator/internal/config"
|
|
"homelab.lan/music-agregator/internal/database"
|
|
"homelab.lan/music-agregator/internal/torrent"
|
|
)
|
|
|
|
type MusicAgregatorServer struct {
|
|
service *MusicAgregatorService
|
|
pb.UnimplementedMusicAgregatorServiceServer
|
|
}
|
|
|
|
func NewMusicAgregatorServer(cfg config.Config, riverClient *river.Client[pgx.Tx], torrentClient torrent.TorrentClient, pathMapper *torrent.PathMapper, db *database.DB) (*MusicAgregatorServer, error) {
|
|
service, err := NewMusicAgregatorService(cfg, riverClient, torrentClient, pathMapper, db)
|
|
if err != nil {
|
|
log.Err(err).Msg("failed to create MusicAgregatorService")
|
|
return nil, err
|
|
}
|
|
return &MusicAgregatorServer{
|
|
service: service,
|
|
}, nil
|
|
}
|
|
|
|
func NewMusicAgregatorServerWithService(service *MusicAgregatorService) *MusicAgregatorServer {
|
|
return &MusicAgregatorServer{service: service}
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) GetArtists(ctx context.Context, req *pb.GetArtistsRequest) (*pb.GetArtistsResponse, error) {
|
|
return s.service.GetArtists(ctx, req)
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) GetAlbum(ctx context.Context, req *pb.GetAlbumRequest) (*pb.GetAlbumResponse, error) {
|
|
return s.service.GetAlbum(ctx, req)
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) MonitorAlbum(ctx context.Context, req *pb.MonitorAlbumRequest) (*pb.MonitorAlbumResponse, error) {
|
|
return s.service.MonitorAlbum(ctx, req)
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) MonitorAlbumStream(stream pb.MusicAgregatorService_MonitorAlbumStreamServer) error {
|
|
msg, err := stream.Recv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
startReq := msg.GetStart()
|
|
if startReq == nil {
|
|
return status.Error(codes.InvalidArgument, "first message must be StartMonitorRequest")
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(stream.Context())
|
|
defer cancel()
|
|
|
|
workflow := newMonitorWorkflow(stream, startReq, s.service, cancel)
|
|
|
|
if startReq.Mode == pb.InteractionMode_INTERACTION_MODE_MANUAL {
|
|
go workflow.receiveDecisions(ctx)
|
|
}
|
|
|
|
return workflow.run(ctx)
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) AnalyzeAlbumRelease(ctx context.Context, req *pb.AnalyzeAlbumReleaseRequest) (*pb.AnalyzeAlbumReleaseResponse, error) {
|
|
return s.service.AnalyzeAlbumRelease(ctx, req)
|
|
}
|
|
|
|
func (s *MusicAgregatorServer) Register(server *grpc.Server) {
|
|
pb.RegisterMusicAgregatorServiceServer(server, s)
|
|
}
|