golift.io/starr@v1.0.0/lidarr/artist.go

package lidarr

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"net/url"
	"path"
	"time"

	"golift.io/starr"
)

const bpArtist = APIver + "/artist"

// Artist represents the /api/v1/artist endpoint, and it's part of an Album.
type Artist struct {
	ID                int64             `json:"id"`
	Status            string            `json:"status,omitempty"`
	LastInfoSync      time.Time         `json:"lastInfoSync,omitempty"`
	ArtistName        string            `json:"artistName,omitempty"`
	ForeignArtistID   string            `json:"foreignArtistId,omitempty"`
	TadbID            int64             `json:"tadbId,omitempty"`
	DiscogsID         int64             `json:"discogsId,omitempty"`
	QualityProfileID  int64             `json:"qualityProfileId,omitempty"`
	MetadataProfileID int64             `json:"metadataProfileId,omitempty"`
	Overview          string            `json:"overview,omitempty"`
	ArtistType        string            `json:"artistType,omitempty"`
	Disambiguation    string            `json:"disambiguation,omitempty"`
	RootFolderPath    string            `json:"rootFolderPath,omitempty"`
	Path              string            `json:"path,omitempty"`
	CleanName         string            `json:"cleanName,omitempty"`
	SortName          string            `json:"sortName,omitempty"`
	Links             []*starr.Link     `json:"links,omitempty"`
	Images            []*starr.Image    `json:"images,omitempty"`
	Genres            []string          `json:"genres,omitempty"`
	Tags              []int             `json:"tags,omitempty"`
	Added             time.Time         `json:"added,omitempty"`
	Ratings           *starr.Ratings    `json:"ratings,omitempty"`
	Statistics        *Statistics       `json:"statistics,omitempty"`
	LastAlbum         *Album            `json:"lastAlbum,omitempty"`
	NextAlbum         *Album            `json:"nextAlbum,omitempty"`
	AddOptions        *ArtistAddOptions `json:"addOptions,omitempty"`
	AlbumFolder       bool              `json:"albumFolder,omitempty"`
	Monitored         bool              `json:"monitored"`
	Ended             bool              `json:"ended,omitempty"`
}

// Statistics is part of Artist and Album.
type Statistics struct {
	AlbumCount      int     `json:"albumCount,omitempty"`
	TrackFileCount  int     `json:"trackFileCount"`
	TrackCount      int     `json:"trackCount"`
	TotalTrackCount int     `json:"totalTrackCount"`
	SizeOnDisk      int     `json:"sizeOnDisk"`
	PercentOfTracks float64 `json:"percentOfTracks"`
}

// GetArtist returns an artist or all artists.
func (l *Lidarr) GetArtist(mbID string) ([]*Artist, error) {
	return l.GetArtistContext(context.Background(), mbID)
}

// GetArtistContext returns an artist or all artists.
func (l *Lidarr) GetArtistContext(ctx context.Context, mbID string) ([]*Artist, error) {
	req := starr.Request{URI: bpArtist, Query: make(url.Values)}
	if mbID != "" {
		req.Query.Add("mbId", mbID)
	}

	var output []*Artist

	if err := l.GetInto(ctx, req, &output); err != nil {
		return nil, fmt.Errorf("api.Get(%s): %w", &req, err)
	}

	return output, nil
}

// GetArtistByID returns an artist from an ID.
func (l *Lidarr) GetArtistByID(artistID int64) (*Artist, error) {
	return l.GetArtistByIDContext(context.Background(), artistID)
}

// GetArtistByIDContext returns an artist from an ID.
func (l *Lidarr) GetArtistByIDContext(ctx context.Context, artistID int64) (*Artist, error) {
	var output Artist

	req := starr.Request{URI: path.Join(bpArtist, fmt.Sprint(artistID))}
	if err := l.GetInto(ctx, req, &output); err != nil {
		return nil, fmt.Errorf("api.Get(%s): %w", &req, err)
	}

	return &output, nil
}

// AddArtist adds a new artist to Lidarr, and probably does not yet work.
func (l *Lidarr) AddArtist(artist *Artist) (*Artist, error) {
	return l.AddArtistContext(context.Background(), artist)
}

// AddArtistContext adds a new artist to Lidarr, and probably does not yet work.
func (l *Lidarr) AddArtistContext(ctx context.Context, artist *Artist) (*Artist, error) {
	var body bytes.Buffer
	if err := json.NewEncoder(&body).Encode(artist); err != nil {
		return nil, fmt.Errorf("json.Marshal(%s): %w", bpArtist, err)
	}

	req := starr.Request{URI: bpArtist, Query: make(url.Values), Body: &body}

	var output Artist
	if err := l.PostInto(ctx, req, &output); err != nil {
		return nil, fmt.Errorf("api.Post(%s): %w", &req, err)
	}

	return &output, nil
}

// UpdateArtist updates an artist in place.
func (l *Lidarr) UpdateArtist(artist *Artist, moveFiles bool) (*Artist, error) {
	return l.UpdateArtistContext(context.Background(), artist, moveFiles)
}

// UpdateArtistContext updates an artist in place.
func (l *Lidarr) UpdateArtistContext(ctx context.Context, artist *Artist, moveFiles bool) (*Artist, error) {
	var body bytes.Buffer
	if err := json.NewEncoder(&body).Encode(artist); err != nil {
		return nil, fmt.Errorf("json.Marshal(%s): %w", bpArtist, err)
	}

	req := starr.Request{URI: path.Join(bpArtist, fmt.Sprint(artist.ID)), Query: make(url.Values), Body: &body}
	req.Query.Add("moveFiles", fmt.Sprint(moveFiles))

	var output Artist
	if err := l.PutInto(ctx, req, &output); err != nil {
		return nil, fmt.Errorf("api.Put(%s): %w", &req, err)
	}

	return &output, nil
}

// DeleteArtist removes an artist from the database.
// Setting deleteFiles true will delete all content for the artist.
func (l *Lidarr) DeleteArtist(artistID int64, deleteFiles, addImportExclusion bool) error {
	return l.DeleteArtistContext(context.Background(), artistID, deleteFiles, addImportExclusion)
}

// DeleteArtistContext removes an artist from the database.
// Setting deleteFiles true will delete all content for the artist.
func (l *Lidarr) DeleteArtistContext(ctx context.Context, artistID int64, deleteFiles, addImportExclusion bool) error {
	req := starr.Request{URI: path.Join(bpArtist, fmt.Sprint(artistID)), Query: make(url.Values)}
	req.Query.Set("deleteFiles", fmt.Sprint(deleteFiles))
	req.Query.Set("addImportListExclusion", fmt.Sprint(addImportExclusion))

	if err := l.DeleteAny(ctx, req); err != nil {
		return fmt.Errorf("api.Delete(%s): %w", &req, err)
	}

	return nil
}