Fixed WebP loading and auto-update avatar

pages/upload/avatar.go

@@ -32,5 +32,5 @@ func Avatar(ctx *aero.Context) string {
 	// Save avatar information
 	user.Save()
 
-	return "ok"
+	return user.SmallAvatar()
 }

scripts/Actions/Upload.ts

@@ -48,6 +48,12 @@ function uploadFile(file: File, endpoint: string, arn: AnimeNotifier) {
 			body: reader.result
 		})
 
+		let newURL = await response.text()
+		let sidebar = document.getElementById("sidebar")
+		let userImage = sidebar.getElementsByClassName("user-image")[0] as HTMLImageElement
+		userImage.dataset.src = newURL
+		userImage["became visible"]()
+
 		if(response.ok) {
 			arn.statusMessage.showInfo("Successfully uploaded your new avatar.")
 		} else {

scripts/AnimeNotifier.ts

@@ -510,15 +510,21 @@ export class AnimeNotifier {
 		// Once the image becomes visible, load it
 		element["became visible"] = () => {
 			let dataSrc = element.dataset.src
-			let dot = dataSrc.lastIndexOf(".")
+			let dotPos = dataSrc.lastIndexOf(".")
-			let base = dataSrc.substring(0, dot)
+			let base = dataSrc.substring(0, dotPos)
 			let extension = ""
 
 			// Replace URL with WebP if supported
 			if(this.webpEnabled && element.dataset.webp) {
-				extension = ".webp"
+				let queryPos = dataSrc.lastIndexOf("?")
+
+				if(queryPos !== -1) {
+					extension = ".webp" + dataSrc.substring(queryPos)
 				} else {
-				extension = dataSrc.substring(dot)
+					extension = ".webp"
+				}
+			} else {
+				extension = dataSrc.substring(dotPos)
 			}
 
 			// Anime images on Retina displays