135 lines
3.2 KiB
Go
135 lines
3.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
oapi "nyanimedb/api"
|
|
sqlc "nyanimedb/sql"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func Word2Sqlc(s *string) *string {
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
if *s == "" {
|
|
return nil
|
|
}
|
|
return s
|
|
}
|
|
|
|
func TitleStatus2Sqlc(s *oapi.TitleStatus) (*sqlc.TitleStatusT, error) {
|
|
if s == nil {
|
|
return nil, nil
|
|
}
|
|
var t sqlc.TitleStatusT
|
|
if *s == "finished" {
|
|
t = "finished"
|
|
} else if *s == "ongoing" {
|
|
t = "ongoing"
|
|
} else if *s == "planned" {
|
|
t = "planned"
|
|
} else {
|
|
return nil, fmt.Errorf("unexpected tittle status: %s", *s)
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
func ReleaseSeason2sqlc(s *oapi.ReleaseSeason) (*sqlc.ReleaseSeasonT, error) {
|
|
if s == nil {
|
|
return nil, nil
|
|
}
|
|
//TODO
|
|
var t sqlc.ReleaseSeasonT
|
|
if *s == oapi.Winter {
|
|
t = sqlc.ReleaseSeasonTWinter
|
|
} else if *s == "spring" {
|
|
t = "spring"
|
|
} else if *s == "summer" {
|
|
t = "summer"
|
|
} else if *s == "fall" {
|
|
t = "fall"
|
|
} else {
|
|
return nil, fmt.Errorf("unexpected release season: %s", *s)
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
// unmarshall jsonb to map[string][]string
|
|
func jsonb2map4names(b []byte) (*map[string][]string, error) {
|
|
var t map[string][]string
|
|
if err := json.Unmarshal(b, &t); err != nil {
|
|
return nil, fmt.Errorf("invalid title_names JSON for title: %w", err)
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
func jsonb2map4len(b []byte) (*map[string]float64, error) {
|
|
var t map[string]float64
|
|
if err := json.Unmarshal(b, &t); err != nil {
|
|
return nil, fmt.Errorf("invalid episodes_len JSON for title: %w", err)
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
func (s Server) GetTitle(ctx context.Context, request oapi.GetTitleRequestObject) (oapi.GetTitleResponseObject, error) {
|
|
var result []oapi.Title
|
|
|
|
word := Word2Sqlc(request.Params.Word)
|
|
status, err := TitleStatus2Sqlc(request.Params.Status)
|
|
if err != nil {
|
|
log.Errorf("%v", err)
|
|
return oapi.GetTitle400Response{}, err
|
|
}
|
|
season, err := ReleaseSeason2sqlc(request.Params.ReleaseSeason)
|
|
if err != nil {
|
|
log.Errorf("%v", err)
|
|
return oapi.GetTitle400Response{}, err
|
|
}
|
|
// param = nil means it will not be used
|
|
titles, err := s.db.SearchTitles(ctx, sqlc.SearchTitlesParams{
|
|
Word: word,
|
|
Status: status,
|
|
Rating: request.Params.Rating,
|
|
ReleaseYear: request.Params.ReleaseYear,
|
|
ReleaseSeason: season,
|
|
})
|
|
if err != nil {
|
|
return oapi.GetTitle500Response{}, nil
|
|
}
|
|
if len(titles) == 0 {
|
|
return oapi.GetTitle204Response{}, nil
|
|
}
|
|
|
|
for _, title := range titles {
|
|
title_names, err := jsonb2map4names(title.TitleNames)
|
|
if err != nil {
|
|
log.Errorf("%v", err)
|
|
return oapi.GetTitle500Response{}, err
|
|
}
|
|
episodes_lens, err := jsonb2map4len(title.EpisodesLen)
|
|
if err != nil {
|
|
log.Errorf("%v", err)
|
|
return oapi.GetTitle500Response{}, err
|
|
}
|
|
t := oapi.Title{
|
|
Id: &title.ID,
|
|
PosterId: title.PosterID,
|
|
Rating: title.Rating,
|
|
RatingCount: title.RatingCount,
|
|
ReleaseSeason: (*oapi.ReleaseSeason)(title.ReleaseSeason),
|
|
ReleaseYear: title.ReleaseYear,
|
|
StudioId: &title.StudioID,
|
|
TitleNames: title_names,
|
|
TitleStatus: (*oapi.TitleStatus)(&title.TitleStatus),
|
|
EpisodesAired: title.EpisodesAired,
|
|
EpisodesAll: title.EpisodesAll,
|
|
EpisodesLen: episodes_lens,
|
|
}
|
|
result = append(result, t)
|
|
}
|
|
|
|
return oapi.GetTitle200JSONResponse(result), nil
|
|
}
|