log pls

main.go

@@ -8,6 +8,7 @@ import (
 	"io"
 	"log"
 	"net/http"
+	"os"
 	"os/signal"
 	"syscall"
 	"time"
@@ -19,17 +20,42 @@ func main() {
 
 	jellyFrom, err := NewJelly(ctx, "squeaky2x3", "aFJKcZ4fUuN9FZ", "5c1de748f61145a085f272aea527c759", "213abf9acbe84d9fb9c3b06bbe1eec3b")
 	if err != nil {
-		panic(err)
+		log.Fatal(err)
 	}
 	log.Printf("from %+v", jellyFrom)
 
 	jellyTo, err := NewJelly(ctx, "belly", "qBentOcpHMUjhD", "497b212a22e34b54be091055edbe264d", "b71b931108ba4323b75b675871a7738f")
 	if err != nil {
-		panic(err)
+		log.Fatal(err)
 	}
 	log.Printf("to   %+v", jellyTo)
 
-	log.Fatalf("not impl: %+v => %+v", jellyFrom, jellyTo)
+	folders, err := jellyFrom.VirtualFolders(ctx)
+	if err != nil {
+		log.Fatal(err)
+	}
+	for len(folders) > 0 {
+		folder := folders[0]
+		folders = folders[1:]
+
+		items, err := jellyFrom.ListFolder(ctx, folder)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		for _, item := range items {
+			if item.IsFolder {
+				folders = append([]FolderItem{item}, folders...)
+			} else if userData, err := jellyFrom.UserDataOf(ctx, item); err != nil {
+				log.Fatal(err)
+			} else if userData.PlayCount == 0 && userData.PlaybackPositionTicks == 0 {
+			} else if userDataB, err := jellyTo.UserDataOf(ctx, item); err != nil {
+				log.Fatal(err)
+			} else if err := jellyTo.SetUserData(ctx, userData.Plus(userDataB)); err != nil {
+				log.Fatalf("failed to set user data of %+v: %v", userData, err)
+			}
+		}
+	}
 }
 
 type Jelly struct {
@@ -42,7 +68,7 @@ func NewJelly(ctx context.Context, u, p, apitoken, optAccessToken string) (Jelly
 	jelly := Jelly{u: u, apitoken: apitoken, accesstoken: optAccessToken}
 
 	if optAccessToken != "" {
-	} else if accessToken, err := jelly.login(ctx, p); err != nil {
+	} else if accessToken, err := jelly.Login(ctx, p); err != nil {
 		return Jelly{}, err
 	} else {
 		jelly.accesstoken = accessToken
@@ -51,7 +77,108 @@ func NewJelly(ctx context.Context, u, p, apitoken, optAccessToken string) (Jelly
 	return jelly, nil
 }
 
-func (jelly Jelly) login(ctx context.Context, p string) (string, error) {
+type FolderItem struct {
+	Name     string
+	Id       string
+	IsFolder bool
+	Parent   *FolderItem
+}
+
+func (jelly Jelly) ListFolder(ctx context.Context, folder FolderItem) ([]FolderItem, error) {
+	resp, err := jelly.do(ctx, http.MethodGet, fmt.Sprintf("/Items?parentId=%s", folder.Id), nil)
+	if err != nil {
+		return nil, err
+	}
+	a := resp.(map[string]any)["Items"].([]any)
+	result := make([]FolderItem, len(a))
+	for i := range a {
+		m := a[i].(map[string]any)
+		result[i] = FolderItem{
+			Name:     m["Name"].(string),
+			Id:       m["Id"].(string),
+			IsFolder: m["IsFolder"].(bool),
+			Parent:   &folder,
+		}
+	}
+	return result, nil
+}
+
+// get+set item via /UserItems/{itemId}/UserData
+
+type UserData struct {
+	IsFavorite            bool
+	ItemId                string
+	Key                   string
+	PlayCount             float64
+	PlaybackPositionTicks float64
+	Played                bool
+	FolderItem            FolderItem `json:",omitempty"`
+}
+
+func (this UserData) Plus(that UserData) UserData {
+	this.IsFavorite = this.IsFavorite || that.IsFavorite
+	//this.ItemId
+	//this.Key
+	if this.PlayCount < that.PlayCount {
+		this.PlayCount = that.PlayCount
+	}
+	if this.PlaybackPositionTicks < that.PlaybackPositionTicks {
+		this.PlaybackPositionTicks = that.PlaybackPositionTicks
+	}
+	this.Played = this.Played || that.Played
+	return this
+}
+
+func (jelly Jelly) SetUserData(ctx context.Context, userData UserData) error {
+	fi := userData.FolderItem
+	if os.Getenv("DRY_RUN") != "false" {
+		b, _ := json.Marshal(userData)
+		log.Printf("SET %s", b)
+		return nil
+	}
+	userData.FolderItem = FolderItem{}
+	b, _ := json.Marshal(userData)
+	log.Printf("SET %s", b)
+	_, err := jelly.do(ctx, http.MethodPost, "/UserItems/"+fi.Id+"/UserData", userData)
+	return err
+}
+
+func (jelly Jelly) UserDataOf(ctx context.Context, folderItem FolderItem) (UserData, error) {
+	resp, err := jelly.do(ctx, http.MethodGet, "/UserItems/"+folderItem.Id+"/UserData", nil)
+	if err != nil {
+		return UserData{}, err
+	}
+	return UserData{
+		IsFavorite:            resp.(map[string]any)["IsFavorite"].(bool),
+		ItemId:                resp.(map[string]any)["ItemId"].(string),
+		Key:                   resp.(map[string]any)["Key"].(string),
+		PlayCount:             resp.(map[string]any)["PlayCount"].(float64),
+		PlaybackPositionTicks: resp.(map[string]any)["PlaybackPositionTicks"].(float64),
+		Played:                resp.(map[string]any)["Played"].(bool),
+		FolderItem:            folderItem,
+	}, nil
+}
+
+func (jelly Jelly) VirtualFolders(ctx context.Context) ([]FolderItem, error) {
+	resp, err := jelly.do(ctx, http.MethodGet, "/Library/VirtualFolders", nil)
+	if err != nil {
+		return nil, err
+	}
+
+	a := resp.([]any)
+	result := make([]FolderItem, len(a))
+	for i, v := range a {
+		m := v.(map[string]any)
+		result[i] = FolderItem{
+			Id:       m["ItemId"].(string),
+			Name:     m["Name"].(string),
+			IsFolder: true,
+		}
+	}
+	return result, nil
+}
+
+func (jelly Jelly) Login(ctx context.Context, p string) (string, error) {
 	resp, err := jelly.post(ctx, "/Users/AuthenticateByName", map[string]string{
 		"Username": jelly.u,
 		"Pw":       p,
@@ -60,14 +187,33 @@ func (jelly Jelly) login(ctx context.Context, p string) (string, error) {
 		return "", err
 	}
 
-	var AccessToken struct {
-		AccessToken string
+	m, _ := resp.(map[string]any)
+	if m == nil {
+		return "", fmt.Errorf("no map response")
 	}
-	err = json.NewDecoder(resp.Body).Decode(&AccessToken)
-	return AccessToken.AccessToken, err
+	token, _ := m["AccessToken"]
+	if token == nil {
+		return "", fmt.Errorf("no .AccessToken response")
+	}
+	s, _ := token.(string)
+	return s, nil
 }
 
-func (jelly Jelly) post(ctx context.Context, path string, body any, headers ...string) (*http.Response, error) {
+func (jelly Jelly) post(ctx context.Context, path string, body any, headers ...string) (any, error) {
+	return jelly.do(ctx, http.MethodPost, path, body, headers...)
+}
+
+func (jelly Jelly) do(ctx context.Context, method, path string, body any, headers ...string) (any, error) {
+	resp, err := jelly._do(ctx, method, path, body, headers...)
+	if err != nil {
+		return nil, err
+	}
+	var result any
+	err = json.NewDecoder(resp.Body).Decode(&result)
+	return result, err
+}
+
+func (jelly Jelly) _do(ctx context.Context, method, path string, body any, headers ...string) (*http.Response, error) {
 	t := http.Transport{
 		DisableKeepAlives: true,
 	}
@@ -77,7 +223,7 @@ func (jelly Jelly) post(ctx context.Context, path string, body any, headers ...s
 	}
 
 	b, _ := json.Marshal(body)
-	req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("https://jellyfin.home.blapointe.com%s", path), bytes.NewReader(b))
+	req, err := http.NewRequest(method, fmt.Sprintf("https://jellyfin.home.blapointe.com%s", path), bytes.NewReader(b))
 	if err != nil {
 		return nil, err
 	}