package host
import (
"context"
"encoding/json"
extism "github.com/extism/go-sdk"
)
type ArtworkGetArtistUrlRequest struct {
Id string `json:"id"`
Size int32 `json:"size"`
}
type ArtworkGetArtistUrlResponse struct {
Url string `json:"url,omitempty"`
Error string `json:"error,omitempty"`
}
type ArtworkGetAlbumUrlRequest struct {
Id string `json:"id"`
Size int32 `json:"size"`
}
type ArtworkGetAlbumUrlResponse struct {
Url string `json:"url,omitempty"`
Error string `json:"error,omitempty"`
}
type ArtworkGetTrackUrlRequest struct {
Id string `json:"id"`
Size int32 `json:"size"`
}
type ArtworkGetTrackUrlResponse struct {
Url string `json:"url,omitempty"`
Error string `json:"error,omitempty"`
}
type ArtworkGetPlaylistUrlRequest struct {
Id string `json:"id"`
Size int32 `json:"size"`
}
type ArtworkGetPlaylistUrlResponse struct {
Url string `json:"url,omitempty"`
Error string `json:"error,omitempty"`
}
func RegisterArtworkHostFunctions(service ArtworkService) []extism.HostFunction {
return []extism.HostFunction{
newArtworkGetArtistUrlHostFunction(service),
newArtworkGetAlbumUrlHostFunction(service),
newArtworkGetTrackUrlHostFunction(service),
newArtworkGetPlaylistUrlHostFunction(service),
}
}
func newArtworkGetArtistUrlHostFunction(service ArtworkService) extism.HostFunction {
return extism.NewHostFunctionWithStack(
"artwork_getartisturl",
func(ctx context.Context, p *extism.CurrentPlugin, stack []uint64) {
reqBytes, err := p.ReadBytes(stack[0])
if err != nil {
artworkWriteError(p, stack, err)
return
}
var req ArtworkGetArtistUrlRequest
if err := json.Unmarshal(reqBytes, &req); err != nil {
artworkWriteError(p, stack, err)
return
}
url, svcErr := service.GetArtistUrl(ctx, req.Id, req.Size)
if svcErr != nil {
artworkWriteError(p, stack, svcErr)
return
}
resp := ArtworkGetArtistUrlResponse{
Url: url,
}
artworkWriteResponse(p, stack, resp)
},
[]extism.ValueType{extism.ValueTypePTR},
[]extism.ValueType{extism.ValueTypePTR},
)
}
func newArtworkGetAlbumUrlHostFunction(service ArtworkService) extism.HostFunction {
return extism.NewHostFunctionWithStack(
"artwork_getalbumurl",
func(ctx context.Context, p *extism.CurrentPlugin, stack []uint64) {
reqBytes, err := p.ReadBytes(stack[0])
if err != nil {
artworkWriteError(p, stack, err)
return
}
var req ArtworkGetAlbumUrlRequest
if err := json.Unmarshal(reqBytes, &req); err != nil {
artworkWriteError(p, stack, err)
return
}
url, svcErr := service.GetAlbumUrl(ctx, req.Id, req.Size)
if svcErr != nil {
artworkWriteError(p, stack, svcErr)
return
}
resp := ArtworkGetAlbumUrlResponse{
Url: url,
}
artworkWriteResponse(p, stack, resp)
},
[]extism.ValueType{extism.ValueTypePTR},
[]extism.ValueType{extism.ValueTypePTR},
)
}
func newArtworkGetTrackUrlHostFunction(service ArtworkService) extism.HostFunction {
return extism.NewHostFunctionWithStack(
"artwork_gettrackurl",
func(ctx context.Context, p *extism.CurrentPlugin, stack []uint64) {
reqBytes, err := p.ReadBytes(stack[0])
if err != nil {
artworkWriteError(p, stack, err)
return
}
var req ArtworkGetTrackUrlRequest
if err := json.Unmarshal(reqBytes, &req); err != nil {
artworkWriteError(p, stack, err)
return
}
url, svcErr := service.GetTrackUrl(ctx, req.Id, req.Size)
if svcErr != nil {
artworkWriteError(p, stack, svcErr)
return
}
resp := ArtworkGetTrackUrlResponse{
Url: url,
}
artworkWriteResponse(p, stack, resp)
},
[]extism.ValueType{extism.ValueTypePTR},
[]extism.ValueType{extism.ValueTypePTR},
)
}
func newArtworkGetPlaylistUrlHostFunction(service ArtworkService) extism.HostFunction {
return extism.NewHostFunctionWithStack(
"artwork_getplaylisturl",
func(ctx context.Context, p *extism.CurrentPlugin, stack []uint64) {
reqBytes, err := p.ReadBytes(stack[0])
if err != nil {
artworkWriteError(p, stack, err)
return
}
var req ArtworkGetPlaylistUrlRequest
if err := json.Unmarshal(reqBytes, &req); err != nil {
artworkWriteError(p, stack, err)
return
}
url, svcErr := service.GetPlaylistUrl(ctx, req.Id, req.Size)
if svcErr != nil {
artworkWriteError(p, stack, svcErr)
return
}
resp := ArtworkGetPlaylistUrlResponse{
Url: url,
}
artworkWriteResponse(p, stack, resp)
},
[]extism.ValueType{extism.ValueTypePTR},
[]extism.ValueType{extism.ValueTypePTR},
)
}
func artworkWriteResponse(p *extism.CurrentPlugin, stack []uint64, resp any) {
respBytes, err := json.Marshal(resp)
if err != nil {
artworkWriteError(p, stack, err)
return
}
respPtr, err := p.WriteBytes(respBytes)
if err != nil {
stack[0] = 0
return
}
stack[0] = respPtr
}
func artworkWriteError(p *extism.CurrentPlugin, stack []uint64, err error) {
errResp := struct {
Error string `json:"error"`
}{Error: err.Error()}
respBytes, _ := json.Marshal(errResp)
respPtr, _ := p.WriteBytes(respBytes)
stack[0] = respPtr
}