From 360190586fa17630db2d56dc427e6ffc5b8fa38d Mon Sep 17 00:00:00 2001
From: Eduard Urbach <e.urbach@gmail.com>
Date: Thu, 22 Jun 2017 09:29:31 +0200
Subject: [PATCH] Added ability to refresh specific avatars on jobs/avatars

---
 jobs/avatars/main.go  |  4 ++++
 jobs/avatars/shell.go | 45 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 jobs/avatars/shell.go

diff --git a/jobs/avatars/main.go b/jobs/avatars/main.go
index e2bdebc3..064fcf0f 100644
--- a/jobs/avatars/main.go
+++ b/jobs/avatars/main.go
@@ -75,6 +75,10 @@ func main() {
 		},
 	}
 
+	if InvokeShellArgs() {
+		return
+	}
+
 	// Stream of all users
 	users, _ := arn.FilterUsers(func(user *arn.User) bool {
 		return true
diff --git a/jobs/avatars/shell.go b/jobs/avatars/shell.go
new file mode 100644
index 00000000..ab957ac2
--- /dev/null
+++ b/jobs/avatars/shell.go
@@ -0,0 +1,45 @@
+package main
+
+import (
+	"flag"
+
+	"github.com/animenotifier/arn"
+)
+
+// Shell parameters
+var userID string
+var userNick string
+
+// Shell flags
+func init() {
+	flag.StringVar(&userID, "id", "", "ID of the user whose avatar you want to refresh")
+	flag.StringVar(&userNick, "nick", "", "Nickname of the user whose avatar you want to refresh")
+	flag.Parse()
+}
+
+// InvokeShellArgs ...
+func InvokeShellArgs() bool {
+	if userID != "" {
+		user, err := arn.GetUser(userID)
+
+		if err != nil {
+			panic(err)
+		}
+
+		Work(user)
+		return true
+	}
+
+	if userNick != "" {
+		user, err := arn.GetUserByNick(userNick)
+
+		if err != nil {
+			panic(err)
+		}
+
+		Work(user)
+		return true
+	}
+
+	return false
+}