|
|
|
@ -14,40 +14,28 @@ import ( |
|
|
|
|
var ( |
|
|
|
|
canteen *openmensa.Canteen |
|
|
|
|
|
|
|
|
|
emojis = map[string][]string{ |
|
|
|
|
":pizza:": {"pizza"}, |
|
|
|
|
":hotdog:": {"wurst", "würstchen", "hotdog"}, |
|
|
|
|
":fries:": {"pommes"}, |
|
|
|
|
":hamburger:": {"burger"}, |
|
|
|
|
":fish:": {"lachs", "filet", "fisch", "kabeljau", "hering", "makrele", "sardine", "schellfisch", "seezunge"}, |
|
|
|
|
":apple:": {"apfel", "äpfel"}, |
|
|
|
|
":poultry_leg:": {"hähnchen"}, |
|
|
|
|
":meat_on_bone:": {"schnitzel", "steak", "schwein", "huhn", "geflügel"}, |
|
|
|
|
":spaghetti:": {"nudel", "spaghetti", "pasta"}, |
|
|
|
|
":bread:": {"brot", "brötchen"}, |
|
|
|
|
":rice:": {"reis"}, |
|
|
|
|
emojis = map[string][]string { |
|
|
|
|
":pizza:": {"pizza"}, |
|
|
|
|
":hotdog:": {"wurst"}, |
|
|
|
|
":fries:": {"pommes"}, |
|
|
|
|
":hamburger:": {"burger"}, |
|
|
|
|
":fish:": {"lachs", "filet", "fisch", "kabeljau"}, |
|
|
|
|
":apple:": {"apfel", "äpfel"}, |
|
|
|
|
":poultry_leg:": {"hänchen"}, |
|
|
|
|
":meat_on_bone:": {"schnitzel"}, |
|
|
|
|
":spaghetti:": {"nudel", "spaghetti", "pasta"}, |
|
|
|
|
} |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
func main() { |
|
|
|
|
token := os.Getenv("MENSABOT_TOKEN") |
|
|
|
|
if (len(token) == 0 || token == "token") && len(os.Args) <= 0 { |
|
|
|
|
fmt.Println("Please add token as environment variable (MENSABOT_TOKEN) or cli argument.") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if len(os.Args) > 1 { |
|
|
|
|
token = os.Args[1] |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
dg, err := discordgo.New(fmt.Sprintf("Bot %s", token)) |
|
|
|
|
if err != nil { |
|
|
|
|
fmt.Println("error creating discord bot: ", err) |
|
|
|
|
if len(os.Args) <= 0 { |
|
|
|
|
println("Please add token as cli argument.") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
dg, err := discordgo.New(fmt.Sprintf("Bot %s", os.Args[1])) |
|
|
|
|
canteens, err := openmensa.GetCanteens(175) |
|
|
|
|
if err != nil { |
|
|
|
|
fmt.Println("error getting mensa: ", err) |
|
|
|
|
fmt.Println("error creating discord bot: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
dg.AddHandler(messageCreate) |
|
|
|
@ -78,16 +66,7 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if m.Content == "!mensa" { |
|
|
|
|
date := time.Now() |
|
|
|
|
switch date.Weekday() { |
|
|
|
|
case time.Saturday: |
|
|
|
|
date = date.AddDate(0, 0, -1) |
|
|
|
|
break |
|
|
|
|
case time.Sunday: |
|
|
|
|
date = date.AddDate(0, 0, -2) |
|
|
|
|
break |
|
|
|
|
} |
|
|
|
|
sendMealsForDate(s, date, m.ChannelID) |
|
|
|
|
sendMealsForDate(s, time.Now(), m.ChannelID) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -95,7 +74,6 @@ func sendMealsForDate(s *discordgo.Session, t time.Time, channel string) { |
|
|
|
|
meals, err := canteen.GetMeals(t) |
|
|
|
|
if err != nil { |
|
|
|
|
s.ChannelMessageSend(channel, ":angry: an error occured") |
|
|
|
|
fmt.Println("an error occured while getting the meals: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
var messages = make([]*discordgo.MessageEmbedField, len(meals)) |
|
|
|
@ -103,12 +81,11 @@ func sendMealsForDate(s *discordgo.Session, t time.Time, channel string) { |
|
|
|
|
for i, meal := range meals { |
|
|
|
|
prefix := "" |
|
|
|
|
|
|
|
|
|
outer: |
|
|
|
|
for emoji, keywords := range emojis { |
|
|
|
|
outer: for emoji, keywords := range emojis { |
|
|
|
|
for _, keyword := range keywords { |
|
|
|
|
if strings.Contains(strings.ToLower(meal.Name), keyword) { |
|
|
|
|
prefix += emoji |
|
|
|
|
continue outer |
|
|
|
|
continue outer; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
@ -118,24 +95,21 @@ func sendMealsForDate(s *discordgo.Session, t time.Time, channel string) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
messages[i] = &discordgo.MessageEmbedField{ |
|
|
|
|
Name: prefix + meal.Name, |
|
|
|
|
Name: prefix + meal.Name, |
|
|
|
|
Value: fmt.Sprintf("%.2f€", *meal.Prices.Students), |
|
|
|
|
} |
|
|
|
|
if len(meal.Notes) > 0 { |
|
|
|
|
footer += fmt.Sprintf("[%d] %s\n", i+1, strings.Join(meal.Notes, ", ")) |
|
|
|
|
footer += fmt.Sprintf("[%d] %s\n", i + 1, strings.Join(meal.Notes, ", ")) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
_, err = s.ChannelMessageSendComplex(channel, &discordgo.MessageSend{ |
|
|
|
|
s.ChannelMessageSendComplex(channel, &discordgo.MessageSend{ |
|
|
|
|
Embed: &discordgo.MessageEmbed{ |
|
|
|
|
Title: fmt.Sprintf("**Menü vom %s**", t.Format("02.01.2006")), |
|
|
|
|
Title: fmt.Sprintf("** :spaghetti: Menü vom %s**", t.Format("02.01.2006")), |
|
|
|
|
Fields: messages, |
|
|
|
|
Footer: &discordgo.MessageEmbedFooter{ |
|
|
|
|
IconURL: "https://pbs.twimg.com/profile_images/643755515118505984/xzZMK7fU_400x400.png", |
|
|
|
|
Text: footer, |
|
|
|
|
Text: footer, |
|
|
|
|
}, |
|
|
|
|
}, |
|
|
|
|
}) |
|
|
|
|
if err != nil { |
|
|
|
|
fmt.Println("an error occured while sending the meals: ", err) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |