feat: add GreenStar column to indicate that restaurant has a MICHELIN Green Star award

Author: ngshiheng

pkg/crawler/crawler.go

@@ -148,10 +148,12 @@ func (a *App) Crawl() {
 		description := e.ChildText(restaurantDescriptionXPath)
 
 		distinctions := e.ChildTexts(restaurantDistinctionXPath)
-		distinction := parser.ParseDistinction(distinctions[0]) // NOTE: [Three Stars: Exceptional cuisine MICHELIN Green Star]
-		if distinction == "" {
-			log.WithFields(log.Fields{"url": url, "distinctions": distinctions}).Warn("invalid distinctions")
+		distinction := parser.ParseDistinction(distinctions[0])
+		greenStar := false
+		if len(distinctions) > 1 {
+			greenStar = parser.ParseDistinction(distinctions[len(distinctions)-1]) == michelin.GreenStar
 		}
+
 		priceAndCuisine := e.ChildText(restaurantPriceAndCuisineXPath)
 		price, cuisine := parser.SplitUnpack(priceAndCuisine, "·")
 
@@ -174,6 +176,7 @@ func (a *App) Crawl() {
 			Description:           parser.TrimWhiteSpaces(description),
 			Distinction:           distinction,
 			FacilitiesAndServices: strings.Join(facilitiesAndServices, ","),
+			GreenStar:             greenStar,
 			Latitude:              e.Request.Ctx.Get("latitude"),
 			Location:              e.Request.Ctx.Get("location"),
 			Longitude:             e.Request.Ctx.Get("longitude"),

pkg/michelin/restaurant.go

@@ -9,6 +9,7 @@ type Restaurant struct {
 	Description           string
 	Distinction           string // Comma-separated string
 	FacilitiesAndServices string // Comma-separated string
+	GreenStar             bool
 	Latitude              string
 	Location              string
 	Longitude             string

pkg/parser/parser.go

@@ -49,6 +49,7 @@ func ParseDistinction(distinction string) string {
 	case "michelin green star":
 		return michelin.GreenStar
 	default:
+		log.WithFields(log.Fields{"distinctions": distinction}).Warn("invalid distinctions")
 		return ""
 	}
 }