feat: v0.0.10 (#23)

* feat: single SOT for themes + basic custom support * fix: adjust colors for yuu theme * feat: Allow loading of environment variables from file (#20) * feat: allow loading of environment variables from file * Panic if a file for an environment variable cannot be read * Use log.Fatalf + os.Exit instead of panic * fix: remove supurfluous call to os.Exit() --------- Co-authored-by: adaexec <nixos-git.s1pht@simplelogin.com> Co-authored-by: Gabe Farrell <90876006+gabehf@users.noreply.github.com> * chore: add pr test workflow * chore: changelog * feat: make all activity grids configurable * fix: adjust activity grid style * fix: make background gradient consistent size * revert: remove year from activity grid opts * style: adjust top item list min size to 200px * feat: add support for custom themes * fix: stabilized the order of top items * chore: update changelog * feat: native import & export * fix: use correct request body for alias requests * fix: clear input when closing edit modal * chore: changelog * docs: make endpoint clearer for some apps * feat: add ui and handler for export * fix: fix pr test workflow --------- Co-authored-by: adaexec <78047743+adaexec@users.noreply.github.com> Co-authored-by: adaexec <nixos-git.s1pht@simplelogin.com>
2026-03-08 23:18:15 -07:00 · 2025-06-18 08:48:19 -04:00 · 2025-06-18 08:48:19 -04:00 · c16b557c21
commit c16b557c21
parent 486f5d0269
51 changed files with 1754 additions and 866 deletions
--- a/internal/export/export.go
+++ b/internal/export/export.go
@ -0,0 +1,145 @@
+package export
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"time"
+
+	"github.com/gabehf/koito/internal/db"
+	"github.com/gabehf/koito/internal/logger"
+	"github.com/gabehf/koito/internal/models"
+	"github.com/google/uuid"
+)
+
+type KoitoExport struct {
+	Version    string        `json:"version"`
+	ExportedAt time.Time     `json:"exported_at"` // RFC3339
+	User       string        `json:"user"`        // username
+	Listens    []KoitoListen `json:"listens"`
+}
+type KoitoListen struct {
+	ListenedAt time.Time     `json:"listened_at"`
+	Track      KoitoTrack    `json:"track"`
+	Album      KoitoAlbum    `json:"album"`
+	Artists    []KoitoArtist `json:"artists"`
+}
+type KoitoTrack struct {
+	MBID     *uuid.UUID     `json:"mbid"`
+	Duration int            `json:"duration"`
+	Aliases  []models.Alias `json:"aliases"`
+}
+type KoitoAlbum struct {
+	ImageUrl       string         `json:"image_url"`
+	MBID           *uuid.UUID     `json:"mbid"`
+	Aliases        []models.Alias `json:"aliases"`
+	VariousArtists bool           `json:"various_artists"`
+}
+type KoitoArtist struct {
+	ImageUrl  string         `json:"image_url"`
+	MBID      *uuid.UUID     `json:"mbid"`
+	IsPrimary bool           `json:"is_primary"`
+	Aliases   []models.Alias `json:"aliases"`
+}
+
+func ExportData(ctx context.Context, user *models.User, store db.DB, out io.Writer) error {
+	lastTime := time.Unix(0, 0)
+	lastTrackId := int32(0)
+	pageSize := int32(1000)
+
+	l := logger.FromContext(ctx)
+	l.Info().Msg("ExportData: Generating Koito export file...")
+
+	exportedAt := time.Now()
+	// exportFile := path.Join(cfg.ConfigDir(), fmt.Sprintf("koito_export_%d.json", exportedAt.Unix()))
+	// f, err := os.Create(exportFile)
+	// if err != nil {
+	// 	return fmt.Errorf("ExportData: %w", err)
+	// }
+	// defer f.Close()
+
+	// Write the opening of the JSON manually
+	_, err := fmt.Fprintf(out, "{\n  \"version\": \"1\",\n  \"exported_at\": \"%s\",\n  \"user\": \"%s\",\n  \"listens\": [\n", exportedAt.UTC().Format(time.RFC3339), user.Username)
+	if err != nil {
+		return fmt.Errorf("ExportData: %w", err)
+	}
+
+	first := true
+	for {
+		rows, err := store.GetExportPage(ctx, db.GetExportPageOpts{
+			UserID:     user.ID,
+			ListenedAt: lastTime,
+			TrackID:    lastTrackId,
+			Limit:      pageSize,
+		})
+		if err != nil {
+			return fmt.Errorf("ExportData: %w", err)
+		}
+		if len(rows) == 0 {
+			break
+		}
+
+		for _, r := range rows {
+			// Adds a comma after each listen item
+			if !first {
+				_, _ = out.Write([]byte(",\n"))
+			}
+			first = false
+
+			exported := convertToExportFormat(r)
+
+			raw, err := json.MarshalIndent(exported, "    ", "  ")
+
+			// needed to make the listen item start at the right indent level
+			out.Write([]byte("    "))
+
+			if err != nil {
+				return fmt.Errorf("ExportData: marshal: %w", err)
+			}
+			_, _ = out.Write(raw)
+
+			if r.TrackID > lastTrackId {
+				lastTrackId = r.TrackID
+			}
+			if r.ListenedAt.After(lastTime) {
+				lastTime = r.ListenedAt
+			}
+		}
+	}
+
+	// Write closing of the JSON array and object
+	_, err = out.Write([]byte("\n  ]\n}\n"))
+	if err != nil {
+		return fmt.Errorf("ExportData: f.Write: %w", err)
+	}
+
+	l.Info().Msgf("Export successfully created")
+	return nil
+}
+
+func convertToExportFormat(item *db.ExportItem) *KoitoListen {
+	ret := &KoitoListen{
+		ListenedAt: item.ListenedAt.UTC(),
+		Track: KoitoTrack{
+			MBID:     item.TrackMbid,
+			Duration: int(item.TrackDuration),
+			Aliases:  item.TrackAliases,
+		},
+		Album: KoitoAlbum{
+			MBID:           item.ReleaseMbid,
+			ImageUrl:       item.ReleaseImageSource,
+			VariousArtists: item.VariousArtists,
+			Aliases:        item.ReleaseAliases,
+		},
+	}
+	for i := range item.Artists {
+		ret.Artists = append(ret.Artists, KoitoArtist{
+			IsPrimary: item.Artists[i].IsPrimary,
+			MBID:      item.Artists[i].MbzID,
+			Aliases:   item.Artists[i].Aliases,
+			ImageUrl:  item.Artists[i].ImageSource,
+		})
+	}
+	return ret
+}