Support route listing & search

datasources/mtr/bus/QueryResult.go

@@ -2,6 +2,7 @@ package bus
 
 import (
     "fmt"
+    "sort"
     "strings"
 )
 
@@ -9,8 +10,33 @@ type QueryResult struct {
     Schedules *map[BusStop] *BusStopBuses
     Lang string
     Error error
+    Query *QueryObject
 }
 
+func writeShortRoute( lang *string, sb *strings.Builder, b *BusStop ) {
+    if b.PrevStop() != nil {
+        sb.WriteString( (*b.PrevStop().Name)[ *lang ] )
+        sb.WriteString( " > " )
+    }
+
+    sb.WriteString( "**" )
+    sb.WriteString( (*b.Name)[ *lang ] )
+    sb.WriteString( "**" )
+
+    if b.NextStop() != nil {
+        sb.WriteString( " > " )
+        sb.WriteString( (*b.NextStop().Name)[ *lang ] )
+    }
+
+    sb.WriteString( "\n" )
+}
+
+type ByRouteId []BusStop
+
+func (a ByRouteId) Len() int           { return len(a) }
+func (a ByRouteId) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
+func (a ByRouteId) Less(i, j int) bool { return a[i].RouteId < a[j].RouteId }
+
 func ( result QueryResult ) Message() string {
 
     if result.Error != nil {
@@ -19,33 +45,51 @@ func ( result QueryResult ) Message() string {
 
     sb := strings.Builder{}
 
-    for busStop, buses := range *result.Schedules {
+    if result.Schedules == nil {
 
-        if busStop.PrevStop() != nil {
-            sb.WriteString( (*busStop.PrevStop().Name)[ result.Lang ] )
-            sb.WriteString( " > *" )
+        query := *result.Query
+        if query.Route == "" {
+            sort.Sort( ByRouteId( *query.BusStops ) )
+            for _, busStop := range *query.BusStops {
+                sb.WriteString( busStop.RouteId )
+                sb.WriteString( " " )
+                writeShortRoute( &result.Lang, &sb, &busStop )
+            }
+        } else if query.BusStops == nil && query.RouteStarts != nil {
+            for d, b := range *query.RouteStarts {
+                sb.WriteString( query.Route )
+                sb.WriteString( "-" )
+                sb.WriteString( d )
+                sb.WriteString( "\n " )
+
+                for {
+                    sb.WriteString( (*b.Name)[ result.Lang ] )
+                    b = b.NextStop()
+                    if b == nil {
+                        break
+                    }
+
+                    sb.WriteString( " > " )
+                }
+                sb.WriteString( "\n" )
+            }
+        } else {
+            sb.WriteString( "Unreachable condition occured!?" )
         }
-
-        sb.WriteString( (*busStop.Name)[ result.Lang ] )
-        sb.WriteString( "*" )
-
-        if busStop.NextStop() != nil {
-            sb.WriteString( " > " )
-            sb.WriteString( (*busStop.NextStop().Name)[ result.Lang ] )
-        }
-
-        sb.WriteString( "\n" )
-        for _, bus := range buses.Buses {
-            sb.WriteString( "  * " )
-            if bus.ETAText == "" {
-                sb.WriteString( bus.ETDText )
-            } else {
-                sb.WriteString( bus.ETAText )
+    } else {
+        for busStop, buses := range *result.Schedules {
+            writeShortRoute( &result.Lang, &sb, &busStop )
+            for _, bus := range buses.Buses {
+                sb.WriteString( "  * " )
+                if bus.ETAText == "" {
+                    sb.WriteString( bus.ETDText )
+                } else {
+                    sb.WriteString( bus.ETAText )
+                }
+                sb.WriteString( "\n" )
             }
             sb.WriteString( "\n" )
         }
-
-        sb.WriteString( "\n" )
     }
 
     return sb.String()