Added kmb schedules

datasources/kmb/query.go

@@ -11,13 +11,13 @@ func Query( lang string, message string ) query.IQueryResult {
     var err error
 
     qr := QueryResult{ Lang: lang }
-    busStops, err := getBusStops()
+    routeStops, err := getRouteStops()
     if err != nil {
         qr.Error = err
         goto qrReturn
     }
 
-    qo, err = query.Parse( strings.ToUpper( message ), busStops )
+    qo, err = query.Parse( strings.ToUpper( message ), routeStops )
     if err != nil {
         qr.Error = err
         goto qrReturn
@@ -25,6 +25,23 @@ func Query( lang string, message string ) query.IQueryResult {
 
     qr.Query = qo
 
+    if 0 < len( *qo.Results ) && 1 < len( *qo.SearchTerms ) {
+
+        rSchedules := map[*RouteStop] *[] *Schedule{}
+        for _, item := range *qo.Results {
+            var r *RouteStop
+            r = any( item ).( *RouteStop )
+            schedules, err := getSchedule( r )
+            if err != nil {
+                qr.Error = err
+                break
+            }
+
+            rSchedules[r] = schedules
+        }
+        qr.Schedules = &rSchedules
+    }
+
     qrReturn:
     var iqr query.IQueryResult
     iqr = &qr