mslouka/lunch-go
1
0
Fork 0
forked from zv0n/lunch-go
Code Pull Requests Activity

Fresh's triumphant return

Browse Source
This commit is contained in:
zv0n 2024-04-10 20:01:35 +02:00
parent 1aab0d7a4f
commit 4a9f5296a5
2 changed files with 30 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/lunch-go/main.go
View File

@ -54,7 +54,7 @@ func refresh(c *gin.Context) {
}
func main() {
//restaurants = append(restaurants, r.NewFreshRestaurant("http://www.fresh-menu.cz/", "Fresh", 1))
restaurants = append(restaurants, r.NewFreshRestaurant("http://www.fresh-menu.cz/", "Fresh", 1))
restaurants = append(restaurants, r.NewMenickaRestaurant("https://www.menicka.cz/2752-u-drevaka-beergrill.html", "U Dřeváka", 2))
restaurants = append(restaurants, r.NewMenickaRestaurant("https://www.menicka.cz/4116-padagali.html", "Padagali", 3))
restaurants = append(restaurants, r.NewMenickaRestaurant("https://www.menicka.cz/5448-light-of-india.html", "Light of India", 4))

34
pkg/restaurants/fresh.go
View File

@ -24,7 +24,31 @@ func NewFreshRestaurant(url string, name string, id int) *FreshRestaurant {
return restaurant
}
func getIndexFromDay(line string) int {
if len(line) < 3 || line[:3] == "Pol" {
return -1
}
ascii := line[:2]
unicode := line[:3]
if ascii == "Po" {
return 0
} else if unicode == "Út" {
return 1
} else if ascii == "St" {
return 2
} else if unicode == "Čt" {
return 3
} else if unicode == "Pá" {
return 4
}
return -1
}
func (restaurant *FreshRestaurant) Parse() {
defer func() {
recover()
}()
restaurant.clearMenus()
resp, err := http.Get(restaurant.url)
if err != nil {
@ -62,12 +86,16 @@ func (restaurant *FreshRestaurant) Parse() {
meals := [5][]string{}
prices := [5][]int{}
scanner := bufio.NewScanner(strings.NewReader(pdftxt.Body))
curIndex := -1
curIndex := 0
pricesIndex := -1
pricesSection := false
emptyLine := false
for scanner.Scan() {
line := scanner.Text()
dayIndex := getIndexFromDay(line)
if dayIndex != -1 {
curIndex = dayIndex
}
if len(line) == 0 {
emptyLine = true
pricesSection = false
@ -80,13 +108,9 @@ func (restaurant *FreshRestaurant) Parse() {
emptyLine = false
// 10 because soup name starts after 10 and vacations have no soup provided
if !pricesSection && len(line) > 10 && line[:3] == "Pol" {
curIndex++
meals[curIndex] = append(meals[curIndex], line[10:])
prices[curIndex] = append(prices[curIndex], -1)
}
if curIndex < 0 {
curIndex++
}
if !pricesSection && line[1] == '.' && len(line) > 2 {
meals[curIndex] = append(meals[curIndex], line[3:])
}