K8s/Playlist: Refactor apis packages so the types and registry are in different packages (#77586)

2026-07-29 15:59:50 -05:00 · 2023-11-03 08:07:55 -07:00
parent 72ed6434ca
commit dd654fdc87
19 changed files with 91 additions and 517 deletions
@@ -0,0 +1,113 @@
+package playlist
+
+import (
+	"encoding/json"
+	"fmt"
+	"strconv"
+	"time"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/types"
+
+	playlist "github.com/grafana/grafana/pkg/apis/playlist/v0alpha1"
+	"github.com/grafana/grafana/pkg/kinds"
+	"github.com/grafana/grafana/pkg/services/grafana-apiserver/endpoints/request"
+	playlistsvc "github.com/grafana/grafana/pkg/services/playlist"
+)
+
+func UnstructuredToLegacyPlaylist(item unstructured.Unstructured) *playlistsvc.Playlist {
+	spec := item.Object["spec"].(map[string]any)
+	return &playlistsvc.Playlist{
+		UID:      item.GetName(),
+		Name:     spec["title"].(string),
+		Interval: spec["interval"].(string),
+		Id:       getLegacyID(&item),
+	}
+}
+
+func UnstructuredToLegacyPlaylistDTO(item unstructured.Unstructured) *playlistsvc.PlaylistDTO {
+	spec := item.Object["spec"].(map[string]any)
+	dto := &playlistsvc.PlaylistDTO{
+		Uid:      item.GetName(),
+		Name:     spec["title"].(string),
+		Interval: spec["interval"].(string),
+		Id:       getLegacyID(&item),
+	}
+	items := spec["items"]
+	if items != nil {
+		b, err := json.Marshal(items)
+		if err == nil {
+			_ = json.Unmarshal(b, &dto.Items)
+		}
+	}
+	return dto
+}
+
+func convertToK8sResource(v *playlistsvc.PlaylistDTO, namespacer request.NamespaceMapper) *playlist.Playlist {
+	spec := playlist.Spec{
+		Title:    v.Name,
+		Interval: v.Interval,
+	}
+	for _, item := range v.Items {
+		spec.Items = append(spec.Items, playlist.Item{
+			Type:  playlist.ItemType(item.Type),
+			Value: item.Value,
+		})
+	}
+
+	meta := kinds.GrafanaResourceMetadata{}
+	meta.SetUpdatedTimestampMillis(v.UpdatedAt)
+	if v.Id > 0 {
+		meta.SetOriginInfo(&kinds.ResourceOriginInfo{
+			Name: "SQL",
+			Key:  fmt.Sprintf("%d", v.Id),
+		})
+	}
+	return &playlist.Playlist{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:              v.Uid,
+			UID:               types.UID(v.Uid),
+			ResourceVersion:   fmt.Sprintf("%d", v.UpdatedAt),
+			CreationTimestamp: metav1.NewTime(time.UnixMilli(v.CreatedAt)),
+			Namespace:         namespacer(v.OrgID),
+			Annotations:       meta.Annotations,
+		},
+		Spec: spec,
+	}
+}
+
+func convertToLegacyUpdateCommand(p *playlist.Playlist, orgId int64) (*playlistsvc.UpdatePlaylistCommand, error) {
+	spec := p.Spec
+	cmd := &playlistsvc.UpdatePlaylistCommand{
+		UID:      p.Name,
+		Name:     spec.Title,
+		Interval: spec.Interval,
+		OrgId:    orgId,
+	}
+	for _, item := range spec.Items {
+		if item.Type == playlist.ItemTypeDashboardById {
+			return nil, fmt.Errorf("unsupported item type: %s", item.Type)
+		}
+		cmd.Items = append(cmd.Items, playlistsvc.PlaylistItem{
+			Type:  string(item.Type),
+			Value: item.Value,
+		})
+	}
+	return cmd, nil
+}
+
+// Read legacy ID from metadata annotations
+func getLegacyID(item *unstructured.Unstructured) int64 {
+	meta := kinds.GrafanaResourceMetadata{
+		Annotations: item.GetAnnotations(),
+	}
+	info := meta.GetOriginInfo()
+	if info != nil && info.Name == "SQL" {
+		i, err := strconv.ParseInt(info.Key, 10, 64)
+		if err == nil {
+			return i
+		}
+	}
+	return 0
+}