Avatar improvements

jobs/avatars/download.go

@@ -0,0 +1,79 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+
+	"github.com/animenotifier/arn"
+	"github.com/fatih/color"
+	"github.com/parnurzeal/gorequest"
+	gravatar "github.com/ungerik/go-gravatar"
+)
+
+func findOriginalAvatar(user *arn.User) string {
+	return arn.FindFileWithExtension(
+		user.ID,
+		avatarsDirectoryOriginal,
+		[]string{
+			".jpg",
+			".png",
+			".gif",
+		},
+	)
+}
+
+func downloadAvatar(user *arn.User) bool {
+	if user.Email == "" {
+		return false
+	}
+
+	directory := avatarsDirectoryOriginal
+	fileName := directory + user.ID
+
+	// Build URL
+	url := gravatar.Url(user.Email) + "?s=" + fmt.Sprint(arn.AvatarMaxSize) + "&d=404&r=pg"
+
+	// Skip existing avatars
+	if findOriginalAvatar(user) != "" {
+		color.Yellow(user.Nick)
+		return true
+	}
+
+	// Download
+	response, data, err := gorequest.New().Get(url).EndBytes()
+
+	if err != nil {
+		color.Red(user.Nick)
+		return false
+	}
+
+	contentType := response.Header.Get("content-type")
+
+	if response.StatusCode != 200 {
+		color.Red(user.Nick)
+		return false
+	}
+
+	// Determine file extension
+	extension := ""
+
+	switch contentType {
+	case "image/jpeg":
+		extension = ".jpg"
+	case "image/png":
+		extension = ".png"
+	case "image/gif":
+		extension = ".gif"
+	case "image/webp":
+		extension = ".webp"
+	}
+
+	fileName += extension
+
+	// Write to disk
+	ioutil.WriteFile(fileName, data, 0644)
+
+	color.Green(user.Nick)
+
+	return true
+}