forked from Zach-Houston/Taco-Bell-Alert
-
Notifications
You must be signed in to change notification settings - Fork 0
/
taco_bell.go
97 lines (87 loc) · 2.31 KB
/
taco_bell.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package main
import (
"fmt"
"strings"
"time"
"github.com/gocolly/colly"
)
// sendAlert will be responsible for all alerts that are sent after detecting a change
func sendAlert(additions []string, removals []string) {
if len(additions) != 0 {
fmt.Println("NEW ITEMS!")
for _, item := range additions {
fmt.Println(item)
}
}
if len(removals) != 0 {
fmt.Println("REMOVED ITEMS!")
for _, item := range removals {
fmt.Println(item)
}
}
}
// source:
// https://stackoverflow.com/questions/15323767/does-go-have-if-x-in-construct-similar-to-python
// Checks if a string exists within a slice
func stringInSlice(a string, list []string) bool {
for _, b := range list {
if b == a {
return true
}
}
return false
}
// checkSite is responsible for querying the Taco Bell "new item" page and returning
// an alert if it detects any additions to the lineup.
func checkSite(items *[]string, firstRun bool, debug bool) {
// Check all product-name divs
c := colly.NewCollector()
newItems := make([]string, 0)
c.OnHTML("div.product-name", func(e *colly.HTMLElement) {
if firstRun {
*items = append(*items, strings.Trim(e.Text, " \t\n"))
}
{
newItems = append(newItems, strings.Trim(e.Text, " \t\n"))
}
})
c.Visit("https://www.tacobell.com/food/new")
// Debug stuff
if debug && !firstRun {
newItems = newItems[:len(newItems)-1]
newItems = append(newItems, "Buffalo Chicken Taco")
}
// Check for changes in the page
additions := make([]string, 0)
removals := make([]string, 0)
if !firstRun {
fmt.Println("Checking for new items...")
for _, item := range newItems {
if !stringInSlice(item, *items) {
additions = append(additions, item)
}
}
for _, item := range *items {
if !stringInSlice(item, newItems) {
removals = append(removals, item)
}
}
// Changes were made, send alert
if len(additions) != 0 || len(removals) != 0 {
sendAlert(additions, removals)
*items = newItems
}
}
}
func main() {
items := make([]string, 0)
checkSite(&items, true, false)
for {
checkSite(&items, false, false)
fmt.Println("Waiting half an hour.")
time.Sleep(time.Hour / 2)
}
//checkSite(&items, false, false)
//checkSite(&items, false, true)
//checkSite(&items, false, true)
}