routedetail.goも殆ど従う

src/routedetail.go

@@ -6,19 +6,47 @@ import (
   "github.com/gocolly/colly"
 )
 
+func getSummary(i int, e *colly.HTMLElement) Route {
+  r := Route{}
+  if e.Attr("class") == "icnPriTime"  { r.Badges = append(r.Badges, 1) }
+  if e.Attr("class") == "icnPriFare"  { r.Badges = append(r.Badges, 2) }
+  if e.Attr("class") == "icnPriTrans" { r.Badges = append(r.Badges, 3) }
+
+  return r
+}
+
+func handleFare(el *colly.HTMLElement, f Fare, s Stop) Fare {
+  f.Train = strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(el.ChildText("li.transport div"), "[train]", "【電車】"), "[bus]", "【バス】"), "[air]", "【空路】"), "当駅始発", "【当駅始発】") + "　"
+
+  f.Platform = el.ChildText("li.platform")
+  f.Color = strings.ReplaceAll(el.ChildAttr("span", "style"), "border-color:#", "")
+  el.ForEach("li.stop ul", func (js int, els *colly.HTMLElement) {
+    s.Time = els.ChildText("li dl dt")
+    s.Name = strings.ReplaceAll(els.ChildText("li dl dd"), "○", "")
+    f.Stops = append(f.Stops, s)
+  })
+
+  return f
+}
+
+func handleWalk(el *colly.HTMLElement, f Fare) Fare {
+  f.Train = strings.ReplaceAll(
+    el.ChildText("li.transport"), "[line][walk]", "",
+  )
+  f.Platform = ""
+  f.Color = "a8a8a8"
+
+  return f
+}
+
 func getRouteDetail(e *colly.HTMLElement) Route {
   r := Route{}
-  e.ForEach("div.routeSummary div ul.priority li span", func (j int, el *colly.HTMLElement) {
-    if el.Attr("class") == "icnPriTime" {
-      r.Badges = append(r.Badges, 1)
-    }
-    if el.Attr("class") == "icnPriFare" {
-      r.Badges = append(r.Badges, 2)
-    }
-    if el.Attr("class") == "icnPriTrans" {
-      r.Badges = append(r.Badges, 3)
-    }
+  onDivs := "div.routeSummary div ul.priority li span"
+  e.ForEach(onDivs, func(i int, el *colly.HTMLElement) {
+    summary := getSummary(i, el)
+    r.Badges = append(r.Badges, summary.Badges...)
   })
+
   base := e.ChildText("ul.summary li.time span")
   time := strings.ReplaceAll(base, e.ChildText("ul.summary li.time span.small"), "")
   time2 := strings.Split(time, "着")
@@ -28,37 +56,38 @@ func getRouteDetail(e *colly.HTMLElement) Route {
   duraei := strings.Index(durabase[durasi:], "分") + len("分") + durasi
 
   r.Duration = durabase[durasi:duraei]
-  r.TransitCunt = strings.ReplaceAll(e.ChildText("ul.summary li.transfer"), "乗換：", "")
-  r.Fare = strings.ReplaceAll(e.ChildText("ul.summary li.fare"), "[priic]IC優先：", "")
+  r.TransitCunt = strings.ReplaceAll(
+    e.ChildText("ul.summary li.transfer"), "乗換：", "",
+  )
+  r.Fare = strings.ReplaceAll(
+    e.ChildText("ul.summary li.fare"), "[priic]IC優先：", "",
+  )
+
   Stations := Station{}
   Fares := Fare{}
   Stops := Stop{}
-  e.ForEach("div.routeDetail div.station", func (j int, el *colly.HTMLElement) {
+
+  onDivs = "div.routeDetail div.station"
+  e.ForEach(onDivs, func (j int, el *colly.HTMLElement) {
     Stations.Time = el.ChildText("ul.time li")
     if el.ChildText("p.icon span") == "[dep]" { Stations.Time += "発" }
     if el.ChildText("p.icon span") == "[arr]" { Stations.Time += "着" }
     Stations.Name = el.ChildText("dl dt a")
-    e.ForEach("div.routeDetail div.fareSection div.access", func (jf int, elf *colly.HTMLElement) {
+
+    onDivs = "div.routeDetail div.fareSection div.access"
+    e.ForEach(onDivs, func (jf int, elf *colly.HTMLElement) {
       Fares.Stops = nil
       if jf == j {
-        Fares.Train = strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(elf.ChildText("li.transport div"), "[train]", "【電車】"), "[bus]", "【バス】"), "[air]", "【空路】"), "当駅始発", "【当駅始発】") + "　"
-
-        Fares.Platform = elf.ChildText("li.platform")
-        Fares.Color = strings.ReplaceAll(elf.ChildAttr("span", "style"), "border-color:#", "")
-        elf.ForEach("li.stop ul", func (js int, els *colly.HTMLElement) {
-          Stops.Time = els.ChildText("li dl dt")
-          Stops.Name = strings.ReplaceAll(els.ChildText("li dl dd"), "○", "")
-          Fares.Stops = append(Fares.Stops, Stops)
-        })
-        Stations.Fares = append(Stations.Fares, Fares)
+        f := handleFare(elf, Fares, Stops)
+        Stations.Fares = append(Stations.Fares, f)
       }
     })
-    e.ForEach("div.routeDetail div.walk ul.info", func (jw int, elw *colly.HTMLElement) {
+
+    onDivs = "div.routeDetail div.walk ul.info"
+    e.ForEach(onDivs, func (jw int, elw *colly.HTMLElement) {
       if jw == j {
-        Fares.Train = strings.ReplaceAll(elw.ChildText("li.transport"), "[line][walk]", "")
-        Fares.Platform = ""
-        Fares.Color = "a8a8a8"
-        Stations.Fares = append(Stations.Fares, Fares)
+        f := handleWalk(elw, Fares)
+        Stations.Fares = append(Stations.Fares, f)
       }
     })
     r.Stations = append(r.Stations, Stations)