Initial commit

internal/models/track.go

@@ -0,0 +1,173 @@
+package models
+
+import (
+	"encoding/json"
+)
+
+// Track represents a full track
+type Track struct {
+	BaseModel
+	ID                       string           `json:"id"`
+	Title                    string           `json:"title"`
+	DurationMs               int              `json:"duration_ms"`
+	StorageID                string           `json:"storage_id"`
+	CoverURI                 string           `json:"cover_uri"`
+	Artists                  []Artist         `json:"artists"`
+	Albums                   []Album          `json:"albums"`
+	Genre                    string           `json:"genre"`
+	Version                  string           `json:"version"`
+	Explicit                 bool             `json:"explicit"`
+	Available                bool             `json:"available"`
+	AvailableForPremiumUsers bool             `json:"available_for_premium_users"`
+	R128                     float64          `json:"r128"`
+	TrackSharingFlag         string           `json:"track_sharing_flag"`
+	Major                    *Major           `json:"major"`
+	Lyrics                   *TrackLyricsInfo `json:"lyrics"`
+	SimilarTracks            []Track          `json:"similar_tracks"`
+	DownloadInfo             []DownloadInfo   `json:"download_info"`
+}
+
+// DeJSON deserializes a Track from JSON
+func (t *Track) DeJSON(data []byte, c ClientInterface) (Model, error) {
+	var raw map[string]json.RawMessage
+	if err := json.Unmarshal(data, &raw); err != nil {
+		return nil, err
+	}
+
+	track := &Track{BaseModel: *NewBaseModel(c)}
+
+	// Simple fields
+	if v, ok := raw["id"]; ok {
+		json.Unmarshal(v, &track.ID)
+	}
+	if v, ok := raw["title"]; ok {
+		json.Unmarshal(v, &track.Title)
+	}
+	if v, ok := raw["duration_ms"]; ok {
+		json.Unmarshal(v, &track.DurationMs)
+	}
+	if v, ok := raw["storage_id"]; ok {
+		json.Unmarshal(v, &track.StorageID)
+	}
+	if v, ok := raw["cover_uri"]; ok {
+		json.Unmarshal(v, &track.CoverURI)
+	}
+	if v, ok := raw["genre"]; ok {
+		json.Unmarshal(v, &track.Genre)
+	}
+	if v, ok := raw["version"]; ok {
+		json.Unmarshal(v, &track.Version)
+	}
+	if v, ok := raw["explicit"]; ok {
+		json.Unmarshal(v, &track.Explicit)
+	}
+	if v, ok := raw["available"]; ok {
+		json.Unmarshal(v, &track.Available)
+	}
+	if v, ok := raw["available_for_premium_users"]; ok {
+		json.Unmarshal(v, &track.AvailableForPremiumUsers)
+	}
+	if v, ok := raw["r128"]; ok {
+		json.Unmarshal(v, &track.R128)
+	}
+	if v, ok := raw["track_sharing_flag"]; ok {
+		json.Unmarshal(v, &track.TrackSharingFlag)
+	}
+
+	// Nested objects
+	if v, ok := raw["artists"]; ok {
+		var artists []Artist
+		if err := json.Unmarshal(v, &artists); err == nil {
+			track.Artists = artists
+		}
+	}
+	if v, ok := raw["albums"]; ok {
+		var albums []Album
+		if err := json.Unmarshal(v, &albums); err == nil {
+			track.Albums = albums
+		}
+	}
+	if v, ok := raw["major"]; ok {
+		var major Major
+		if err := json.Unmarshal(v, &major); err == nil {
+			track.Major = &major
+		}
+	}
+	if v, ok := raw["lyrics"]; ok {
+		var lyrics TrackLyricsInfo
+		if err := json.Unmarshal(v, &lyrics); err == nil {
+			track.Lyrics = &lyrics
+		}
+	}
+	if v, ok := raw["similar_tracks"]; ok {
+		var similar []Track
+		if err := json.Unmarshal(v, &similar); err == nil {
+			track.SimilarTracks = similar
+		}
+	}
+	if v, ok := raw["download_info"]; ok {
+		var downloadInfo []DownloadInfo
+		if err := json.Unmarshal(v, &downloadInfo); err == nil {
+			track.DownloadInfo = downloadInfo
+		}
+	}
+
+	return track, nil
+}
+
+// DeList deserializes a list of Tracks
+func (t *Track) DeList(data []byte, c ClientInterface) ([]Model, error) {
+	var rawList []json.RawMessage
+	if err := json.Unmarshal(data, &rawList); err != nil {
+		return nil, err
+	}
+
+	result := make([]Model, 0, len(rawList))
+	for _, item := range rawList {
+		track := &Track{BaseModel: *NewBaseModel(c)}
+		if err := json.Unmarshal(item, track); err != nil {
+			continue
+		}
+		result = append(result, track)
+	}
+	return result, nil
+}
+
+// GetDownloadInfo returns download info for a specific quality
+func (t *Track) GetDownloadInfo(quality string) *DownloadInfo {
+	for _, info := range t.DownloadInfo {
+		if info.Quality == quality {
+			return &info
+		}
+	}
+	return nil
+}
+
+// Note: Artist and Album types are defined in their respective files
+// These are imported automatically since they're in the same package
+
+// Major represents major info
+type Major struct {
+	ID          string `json:"id"`
+	Name        string `json:"name"`
+	Description string `json:"description"`
+}
+
+// TrackLyricsInfo represents minimal lyrics info for a track
+type TrackLyricsInfo struct {
+	ID        string `json:"id"`
+	Text      string `json:"text"`
+	Major     *Major `json:"major"`
+	Available bool   `json:"available"`
+	HasText   bool   `json:"has_text"`
+}
+
+// DownloadInfo represents file download info
+type DownloadInfo struct {
+	Quality    string `json:"quality"`
+	Codec      string `json:"codec"`
+	Bitrate    int    `json:"bitrate"`
+	Size       int64  `json:"size"`
+	URL        string `json:"url"`
+	DirectLink bool   `json:"direct_link"`
+}