50 lines
878 B
Go

package main
import (
"flag"
"github.com/akyoto/color"
"github.com/animenotifier/notify.moe/arn"
)
// Shell parameters
var nick string
var service string
var serviceID string
// Shell flags
func init() {
flag.StringVar(&nick, "nick", "", "Nick of the user")
flag.StringVar(&service, "service", "", "Service name (Google or Facebook)")
flag.StringVar(&serviceID, "serviceID", "", "ID of the user on the given service")
flag.Parse()
}
func main() {
if nick == "" || service == "" || serviceID == "" {
flag.Usage()
return
}
color.Yellow("Updating user service ID")
defer color.Green("Finished.")
defer arn.Node.Close()
user, err := arn.GetUserByNick(nick)
arn.PanicOnError(err)
switch service {
case "Google":
user.ConnectGoogle(serviceID)
case "Facebook":
user.ConnectFacebook(serviceID)
default:
panic("Unknown service")
}
user.Save()
}