3 files changed, 15 insertions, 0 deletions
diff --git a/packages/shared-react/hooks/users.ts b/packages/shared-react/hooks/users.ts
index eecde3f1..b1909761 100644
--- a/packages/shared-react/hooks/users.ts
+++ b/packages/shared-react/hooks/users.ts
@@ -13,6 +13,19 @@ export function useUpdateUserSettings(
   });
 }
 
+export function useUpdateUserAvatar(
+  ...opts: Parameters<typeof api.users.updateAvatar.useMutation>
+) {
+  const apiUtils = api.useUtils();
+  return api.users.updateAvatar.useMutation({
+    ...opts[0],
+    onSuccess: (res, req, meta, context) => {
+      apiUtils.users.whoami.invalidate();
+      return opts[0]?.onSuccess?.(res, req, meta, context);
+    },
+  });
+}
+
 export function useDeleteAccount(
   ...opts: Parameters<typeof api.users.deleteAccount.useMutation>
 ) {
diff --git a/packages/shared/types/bookmarks.ts b/packages/shared/types/bookmarks.ts
index cbaa4574..8a294422 100644
--- a/packages/shared/types/bookmarks.ts
+++ b/packages/shared/types/bookmarks.ts
@@ -25,6 +25,7 @@ export const zAssetTypesSchema = z.enum([
   "bookmarkAsset",
   "precrawledArchive",
   "userUploaded",
+  "avatar",
   "unknown",
 ]);
 export type ZAssetType = z.infer<typeof zAssetTypesSchema>;
diff --git a/packages/shared/types/users.ts b/packages/shared/types/users.ts
index d4fff9a1..7338ee15 100644
--- a/packages/shared/types/users.ts
+++ b/packages/shared/types/users.ts
@@ -38,6 +38,7 @@ export const zWhoAmIResponseSchema = z.object({
   id: z.string(),
   name: z.string().nullish(),
   email: z.string().nullish(),
+  image: z.string().nullish(),
   localUser: z.boolean(),
 });