From 715ade831ca0387e5d45c447201e86bab5389552 Mon Sep 17 00:00:00 2001
From: bytedream <bytedream@protonmail.com>
Date: Mon, 20 Jun 2022 10:22:17 +0200
Subject: [PATCH] Add more account and profile endpoints

---
 account.go     | 116 ++++++++++++++++++++++++++++++++++++++++++++++++-
 crunchyroll.go |   4 +-
 2 files changed, 117 insertions(+), 3 deletions(-)

diff --git a/account.go b/account.go
index d22eba5..87a6ef8 100644
--- a/account.go
+++ b/account.go
@@ -1,9 +1,16 @@
 package crunchyroll
 
-import "time"
+import (
+	"bytes"
+	"encoding/json"
+	"net/http"
+	"time"
+)
 
 // Account contains information about a crunchyroll account.
 type Account struct {
+	crunchy *Crunchyroll
+
 	AccountID     string    `json:"account_id"`
 	ExternalID    string    `json:"external_id"`
 	EmailVerified bool      `json:"email_verified"`
@@ -25,5 +32,110 @@ type Account struct {
 	PreferredCommunicationLanguage   LOCALE `json:"preferred_communication_language"`
 	PreferredContentSubtitleLanguage LOCALE `json:"preferred_content_subtitle_language"`
 	QaUser                           bool   `json:"qa_user"`
-	Username                         string `json:"username"`
+
+	Username  string     `json:"username"`
+	Wallpaper *Wallpaper `json:"wallpaper"`
+}
+
+// UpdateEmailLanguage sets in which language emails should be received.
+func (a *Account) UpdateEmailLanguage(language LOCALE) error {
+	err := a.updatePreferences("preferred_communication_language", string(language))
+	if err == nil {
+		a.PreferredCommunicationLanguage = language
+	}
+	return err
+}
+
+// UpdateVideoSubtitleLanguage sets in which language default subtitles should be shown
+func (a *Account) UpdateVideoSubtitleLanguage(language LOCALE) error {
+	err := a.updatePreferences("preferred_content_subtitle_language", string(language))
+	if err == nil {
+		a.PreferredContentSubtitleLanguage = language
+	}
+	return err
+}
+
+// UpdateMatureVideoContent sets if mature video content / 18+ content should be shown
+func (a *Account) UpdateMatureVideoContent(enabled bool) error {
+	if enabled {
+		return a.updatePreferences("maturity_rating", "M3")
+	} else {
+		return a.updatePreferences("maturity_rating", "M2")
+	}
+}
+
+// UpdateMatureMangaContent sets if mature manga content / 18+ content should be shown
+func (a *Account) UpdateMatureMangaContent(enabled bool) error {
+	if enabled {
+		return a.updatePreferences("mature_content_flag_manga", "1")
+	} else {
+		return a.updatePreferences("mature_content_flag_manga", "0")
+	}
+}
+
+func (a *Account) updatePreferences(name, value string) error {
+	endpoint := "https://beta.crunchyroll.com/accounts/v1/me/profile"
+	body, _ := json.Marshal(map[string]string{name: value})
+	req, err := http.NewRequest(http.MethodPatch, endpoint, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Add("Content-Type", "application/json")
+	_, err = a.crunchy.requestFull(req)
+	return err
+}
+
+// ChangePassword changes the password for the current account.
+func (a *Account) ChangePassword(currentPassword, newPassword string) error {
+	endpoint := "https://beta.crunchyroll.com/accounts/v1/me/credentials"
+	body, _ := json.Marshal(map[string]string{"accountId": a.AccountID, "current_password": currentPassword, "new_password": newPassword})
+	req, err := http.NewRequest(http.MethodPatch, endpoint, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Add("Content-Type", "application/json")
+	_, err = a.crunchy.requestFull(req)
+	return err
+}
+
+// ChangeEmail changes the email address for the current account.
+func (a *Account) ChangeEmail(currentPassword, newEmail string) error {
+	endpoint := "https://beta.crunchyroll.com/accounts/v1/me/credentials"
+	body, _ := json.Marshal(map[string]string{"current_password": currentPassword, "email": newEmail})
+	req, err := http.NewRequest(http.MethodPatch, endpoint, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Add("Content-Type", "application/json")
+	_, err = a.crunchy.requestFull(req)
+	return err
+}
+
+// AvailableWallpapers returns all available wallpapers which can be set as profile wallpaper.
+func (a *Account) AvailableWallpapers() (w []*Wallpaper, err error) {
+	endpoint := "https://beta.crunchyroll.com/assets/v1/wallpaper"
+	resp, err := a.crunchy.request(endpoint, http.MethodGet)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var jsonBody map[string]any
+	json.NewDecoder(resp.Body).Decode(&jsonBody)
+
+	err = decodeMapToStruct(jsonBody["items"].([]any), &w)
+	return
+}
+
+// ChangeWallpaper changes the profile wallpaper of the current user. Use AvailableWallpapers
+// to get all available ones.
+func (a *Account) ChangeWallpaper(wallpaper *Wallpaper) error {
+	endpoint := "https://beta.crunchyroll.com/accounts/v1/me/profile"
+	body, _ := json.Marshal(map[string]string{"wallpaper": string(*wallpaper)})
+	req, err := http.NewRequest(http.MethodPatch, endpoint, bytes.NewBuffer(body))
+	if err != nil {
+		return err
+	}
+	_, err = a.crunchy.requestFull(req)
+	return err
 }
diff --git a/crunchyroll.go b/crunchyroll.go
index 8b4fad8..704003d 100644
--- a/crunchyroll.go
+++ b/crunchyroll.go
@@ -931,7 +931,9 @@ func (c *Crunchyroll) Account() (*Account, error) {
 	}
 	defer resp.Body.Close()
 
-	account := &Account{}
+	account := &Account{
+		crunchy: c,
+	}
 
 	if err = json.NewDecoder(resp.Body).Decode(&account); err != nil {
 		return nil, fmt.Errorf("failed to parse 'me' response: %w", err)