feat: implement updating profile pictures

Author: Preloading

bluesky/blueskyapi.go

@@ -278,7 +278,7 @@ type RecordValue struct { // TODO: Figure out how to make it get different types
 	CreatedAt   time.Time     `json:"createdAt,omitempty"`
 	Description string        `json:"description,omitempty"`
 	DisplayName string        `json:"displayName,omitempty"`
-	Avatar      interface{}   `json:"avatar,omitempty"` // TODO: implement avatar
+	Avatar      Blob          `json:"avatar,omitempty"`
 }
 
 type Relationships struct {
@@ -349,6 +349,25 @@ func SendRequest(token *string, method string, url string, body io.Reader) (*htt
 	return resp, nil
 }
 
+func SendRequestWithContentType(token *string, method string, url string, body io.Reader, content_type string) (*http.Response, error) {
+	client := &http.Client{}
+	req, err := http.NewRequest(method, url, body)
+	if err != nil {
+		return nil, err
+	}
+	if token != nil {
+		req.Header.Set("Authorization", "Bearer "+*token)
+	}
+	req.Header.Set("Content-Type", content_type) // 99% sure all bluesky requests are json.
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp, nil
+}
+
 func GetUserInfo(pds string, token string, screen_name string, nocache bool) (*bridge.TwitterUser, error) {
 	if !nocache {
 		if user, found := userCache.Get(screen_name); found {
@@ -1613,6 +1632,33 @@ func GetNotifications(pds string, token string, limit int, context string) (*Not
 	return &notifications, nil
 }
 
+func UploadBlob(pds string, token string, data []byte, content_type string) (*Blob, error) {
+	url := pds + "/xrpc/com.atproto.repo.uploadBlob"
+
+	resp, err := SendRequestWithContentType(&token, http.MethodPost, url, bytes.NewReader(data), content_type)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		bodyBytes, _ := io.ReadAll(resp.Body)
+		bodyString := string(bodyBytes)
+		fmt.Println("Response Status:", resp.StatusCode)
+		fmt.Println("Response Body:", bodyString)
+		return nil, errors.New("failed to upload blob")
+	}
+
+	blob := struct {
+		Blob Blob `json:"blob"`
+	}{}
+	if err := json.NewDecoder(resp.Body).Decode(&blob); err != nil {
+		return nil, err
+	}
+
+	return &blob.Blob, nil
+}
+
 // Gets the URI components
 //
 // @param uri: The URI to ge split

twitterv1/account.go

@@ -2,13 +2,28 @@ package twitterv1
 
 import (
 	"fmt"
+	"io"
 	"strings"
+	"sync"
 
 	blueskyapi "github.com/Preloading/MastodonTwitterAPI/bluesky"
 	"github.com/Preloading/MastodonTwitterAPI/bridge"
 	"github.com/gofiber/fiber/v2"
 )
 
+// Mutex map to store mutexes for each user
+var userMutexes = make(map[string]*sync.Mutex)
+var mutexMapLock sync.Mutex
+
+func getUserMutex(userID string) *sync.Mutex {
+	mutexMapLock.Lock()
+	defer mutexMapLock.Unlock()
+	if _, exists := userMutexes[userID]; !exists {
+		userMutexes[userID] = &sync.Mutex{}
+	}
+	return userMutexes[userID]
+}
+
 func PushDestinations(c *fiber.Ctx) error {
 	// TODO: figure out what the hell this is supposed to do to make notifications not crash.
 	old_udid := c.Query("old_udid")
@@ -87,6 +102,17 @@ func GetSettings(c *fiber.Ctx) error {
 }
 
 func UpdateProfile(c *fiber.Ctx) error {
+	// auth
+	my_did, pds, _, oauthToken, err := GetAuthFromReq(c)
+	if err != nil {
+		return c.Status(fiber.StatusUnauthorized).SendString("OAuth token not found in Authorization header")
+	}
+
+	// Lock the mutex for this user
+	userMutex := getUserMutex(*my_did)
+	userMutex.Lock()
+	defer userMutex.Unlock()
+
 	description := c.FormValue("description")
 	name := c.FormValue("name")
 	// These don't exist in bluesky.
@@ -100,20 +126,87 @@ func UpdateProfile(c *fiber.Ctx) error {
 	// some quality of life features
 	description = strings.ReplaceAll(description, "\\n", "\n")
 
+	oldProfile, err := blueskyapi.GetRecord(*pds, "app.bsky.actor.profile", *my_did, "self")
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to get profile")
+	}
+
+	oldProfile.Value.DisplayName = name
+	oldProfile.Value.Description = description
+
+	if err := blueskyapi.UpdateRecord(*pds, *oauthToken, "app.bsky.actor.profile", *my_did, "self", oldProfile.CID, oldProfile.Value); err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to update profile")
+	}
+
+	user, err := blueskyapi.GetUserInfo(*pds, *oauthToken, *my_did, true)
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to fetch user info")
+	}
+
+	user.Description = description
+	user.Name = name
+
+	xml, err := bridge.XMLEncoder(user, "TwitterUser", "user")
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to encode user info")
+	}
+
+	return c.SendString(*xml)
+}
+
+func UpdateProfilePicture(c *fiber.Ctx) error {
 	// auth
 	my_did, pds, _, oauthToken, err := GetAuthFromReq(c)
 	if err != nil {
 		return c.Status(fiber.StatusUnauthorized).SendString("OAuth token not found in Authorization header")
 	}
 
+	// Lock the mutex for this user
+	userMutex := getUserMutex(*my_did)
+	userMutex.Lock()
+	defer userMutex.Unlock()
+
+	// get the old profile
 	oldProfile, err := blueskyapi.GetRecord(*pds, "app.bsky.actor.profile", *my_did, "self")
 	if err != nil {
 		fmt.Println("Error:", err)
 		return c.Status(fiber.StatusInternalServerError).SendString("Failed to get profile")
 	}
 
-	oldProfile.Value.DisplayName = name
-	oldProfile.Value.Description = description
+	// get our new image
+	image, err := c.FormFile("image")
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusBadRequest).SendString("Please upload an image")
+	}
+
+	// read the image file content
+	file, err := image.Open()
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to open image file")
+	}
+	defer file.Close()
+
+	imageData, err := io.ReadAll(file)
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to read image file")
+	}
+
+	// upload our new profile picture
+	profilePictureBlob, err := blueskyapi.UploadBlob(*pds, *oauthToken, imageData, c.Get("Content-Type"))
+	if err != nil {
+		fmt.Println("Error:", err)
+		return c.Status(fiber.StatusInternalServerError).SendString("Failed to upload profile picture")
+	}
+
+	// change our thing
+	oldProfile.Value.Avatar = *profilePictureBlob
 
 	if err := blueskyapi.UpdateRecord(*pds, *oauthToken, "app.bsky.actor.profile", *my_did, "self", oldProfile.CID, oldProfile.Value); err != nil {
 		fmt.Println("Error:", err)
@@ -126,8 +219,7 @@ func UpdateProfile(c *fiber.Ctx) error {
 		return c.Status(fiber.StatusInternalServerError).SendString("Failed to fetch user info")
 	}
 
-	user.Description = description
-	user.Name = name
+	// ...
 
 	xml, err := bridge.XMLEncoder(user, "TwitterUser", "user")
 	if err != nil {

twitterv1/twitterv1.go

@@ -85,6 +85,7 @@ func InitServer(config *config.Config) {
 
 	// Account / Settings
 	app.Post("/1/account/update_profile.xml", UpdateProfile)
+	app.Post("/1/account/update_profile_image.xml", UpdateProfilePicture)
 	app.Get("/1/account/settings.xml", GetSettings)
 	app.Get("/1/account/push_destinations/device.xml", PushDestinations)