Added kmb schedules

datasources/kmb/QueryResult.go

@@ -2,18 +2,52 @@ package kmb
 
 import (
     "fmt"
+    "sort"
     "strings"
+    "time"
 
     query "github.com/tgckpg/golifehk/query"
     utils "github.com/tgckpg/golifehk/utils"
 )
 
 type QueryResult struct {
+    Schedules *map[*RouteStop] *[] *Schedule
     Lang string
     Error error
     Query *query.QueryObject
 }
 
+func writeRouteHead( sb *strings.Builder, r *RouteStop ) {
+    utils.WriteMDv2Text( sb, r.RouteId )
+    if r.Direction == "O" {
+        sb.WriteString( "↑" )
+    } else if r.Direction == "I" {
+        sb.WriteString( "↓" )
+    }
+    if r.ServiceType != "1" {
+        utils.WriteMDv2Text( sb, utils.ToPower( r.ServiceType ) )
+    }
+}
+
+func writeShortRoute( lang *string, sb *strings.Builder, r *RouteStop ) {
+
+    if r.PrevStop() != nil {
+        utils.WriteMDv2Text( sb, (*(r.PrevStop().BusStop).Name)[ *lang ] )
+        sb.WriteString( " \\> " )
+    }
+
+    sb.WriteString( "*" )
+    utils.WriteMDv2Text( sb, (*(r.BusStop).Name)[ *lang ] )
+    sb.WriteString( "*" )
+
+    if r.NextStop() != nil {
+        sb.WriteString( " \\> " )
+        utils.WriteMDv2Text( sb, (*(r.NextStop().BusStop).Name)[ *lang ] )
+    }
+
+    sb.WriteString( "\n" )
+}
+
 func ( this *QueryResult ) Message() ( string, error ) {
 
     if this.Error != nil {
@@ -23,35 +57,119 @@ func ( this *QueryResult ) Message() ( string, error ) {
     sb := strings.Builder{}
 
     if 0 < len( *this.Query.Results ) {
-        // Print Stop Name, the print the list of routes
-        if this.Query.Key == "" {
-            for _, item := range *this.Query.Results {
-                var b *BusStop
-                b = any( item ).( *BusStop )
 
+        // Print Stop Names, then print the list of routes
+        if this.Query.Key == "" {
+            busStops := map[string] *BusStop{}
+            for _, item := range *this.Query.Results {
+                var r *RouteStop
+                r = any( item ).( *RouteStop )
+
+                b := r.BusStop
                 if b.Routes == nil {
                     continue
                 }
 
+                busStops[ b.BusStopId ] = b
+            }
+
+            for _, b := range busStops {
                 utils.WriteMDv2Text( &sb, (*b.Name)[ this.Lang ] )
                 sb.WriteString( "\n  " )
                 for _, route := range *b.Routes {
-                    utils.WriteMDv2Text( &sb, route.RouteId )
-                    if route.Direction == "O" {
-                        sb.WriteString( "↑" )
-                    } else if route.Direction == "I" {
-                        sb.WriteString( "↓" )
-                    }
-                    if route.ServiceType != "1" {
-                        utils.WriteMDv2Text( &sb, utils.ToPower( route.ServiceType ) )
-                    }
+                    writeRouteHead( &sb, route )
                     sb.WriteString( " " )
                 }
                 sb.WriteString( "\n" )
             }
+
+        // We got a route key
+        } else {
+
+            // We also got other search keys with 1 < Results
+            // Get the ETA for this stop
+            if 1 < len( *this.Query.SearchTerms )  {
+
+                now := time.Now()
+                for _, item := range *this.Query.Results {
+                    var r *RouteStop
+                    r = any( item ).( *RouteStop )
+                    writeRouteHead( &sb, r )
+                    sb.WriteString( "\n" )
+                    writeShortRoute( &this.Lang, &sb, r )
+
+                    for _, schedule := range *(*this.Schedules)[ r ] {
+
+                        if !schedule.ETA.IsZero() {
+
+                            _m := schedule.ETA.Sub( now ).Minutes()
+
+                            sb.WriteString( "  \\* " )
+                            txt := "%.0f min(s)"
+
+                            if this.Lang == "zh-Hant" {
+                                txt = "%.0f 分鐘"
+                            }
+
+                            utils.WriteMDv2Text( &sb, fmt.Sprintf( txt, _m ) )
+
+                            if _m < 0 {
+                                sb.WriteString( " 走左了？" )
+                            }
+                        }
+
+                        if schedule.Remarks_en != "" {
+                            sb.WriteString( "  \\*\\* " )
+                            switch this.Lang {
+                            case "en":
+                                utils.WriteMDv2Text( &sb, schedule.Remarks_en )
+                            case "zh-Hant":
+                                utils.WriteMDv2Text( &sb, schedule.Remarks_tc )
+                            }
+                        }
+
+                        sb.WriteString( "\n" )
+                    }
+
+                    sb.WriteString( "\n" )
+                }
+
+            // We got only the route key, proceed to list the route stops
+            } else {
+                // Result contains all route stops, we only need the starting one
+                routes := [] *RouteStop{}
+
+                for _, item := range *this.Query.Results {
+                    var r *RouteStop
+                    r = any( item ).( *RouteStop )
+                    if r.PrevStop() == nil {
+                        routes = append( routes, r )
+                    }
+                }
+
+                sort.Sort( ByRoute( routes ) )
+
+                for _, r := range routes {
+                    writeRouteHead( &sb, r )
+                    sb.WriteString( "\n" )
+                    for {
+                        b := *r.BusStop
+                        utils.WriteMDv2Text( &sb, (*b.Name)[ this.Lang ] )
+                        r = r.NextStop()
+                        if r == nil {
+                            break
+                        }
+
+                        sb.WriteString( " \\> " )
+                    }
+                    sb.WriteString( "\n" )
+                }
+
+
+            }
         }
-        fmt.Print( this.Query.Key )
     } else {
+        return "", fmt.Errorf( "No Results" )
     }
 
     return sb.String(), nil