Refactoring codes for more tg message types
This commit is contained in:
@@ -1,176 +1,181 @@
|
||||
package kmb
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
"fmt"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
query "github.com/tgckpg/golifehk/query"
|
||||
utils "github.com/tgckpg/golifehk/utils"
|
||||
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
|
||||
Schedules *map[*RouteStop]*[]*Schedule
|
||||
Lang string
|
||||
Error error
|
||||
Query *query.QueryObject
|
||||
|
||||
isConsumed bool
|
||||
}
|
||||
|
||||
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 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 ) {
|
||||
func writeShortRoute(lang *string, sb *strings.Builder, r *RouteStop) {
|
||||
|
||||
if r.PrevStop() != nil {
|
||||
utils.WriteMDv2Text( sb, (*(r.PrevStop().BusStop).Name)[ *lang ] )
|
||||
sb.WriteString( " \\> " )
|
||||
}
|
||||
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( "*" )
|
||||
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 ] )
|
||||
}
|
||||
if r.NextStop() != nil {
|
||||
sb.WriteString(" \\> ")
|
||||
utils.WriteMDv2Text(sb, (*(r.NextStop().BusStop).Name)[*lang])
|
||||
}
|
||||
|
||||
sb.WriteString( "\n" )
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
|
||||
func ( this *QueryResult ) Message() ( string, error ) {
|
||||
func (this QueryResult) DataType() string { return "MarkdownV2" }
|
||||
func (this QueryResult) Consumed() bool { return this.isConsumed }
|
||||
func (this QueryResult) GetTableData() [][]map[string]string { return nil }
|
||||
|
||||
if this.Error != nil {
|
||||
return "", this.Error
|
||||
}
|
||||
func (this *QueryResult) Message() (string, error) {
|
||||
|
||||
sb := strings.Builder{}
|
||||
if this.Error != nil {
|
||||
return "", this.Error
|
||||
}
|
||||
|
||||
if 0 < len( *this.Query.Results ) {
|
||||
sb := strings.Builder{}
|
||||
|
||||
// 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 )
|
||||
if 0 < len(*this.Query.Results) {
|
||||
|
||||
b := r.BusStop
|
||||
if b.Routes == nil {
|
||||
continue
|
||||
}
|
||||
// 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)
|
||||
|
||||
busStops[ b.BusStopId ] = b
|
||||
}
|
||||
b := r.BusStop
|
||||
if b.Routes == nil {
|
||||
continue
|
||||
}
|
||||
|
||||
for _, b := range busStops {
|
||||
utils.WriteMDv2Text( &sb, (*b.Name)[ this.Lang ] )
|
||||
sb.WriteString( "\n " )
|
||||
for _, route := range *b.Routes {
|
||||
writeRouteHead( &sb, route )
|
||||
sb.WriteString( " " )
|
||||
}
|
||||
sb.WriteString( "\n" )
|
||||
}
|
||||
busStops[b.BusStopId] = b
|
||||
}
|
||||
|
||||
// We got a route key
|
||||
} else {
|
||||
for _, b := range busStops {
|
||||
utils.WriteMDv2Text(&sb, (*b.Name)[this.Lang])
|
||||
sb.WriteString("\n ")
|
||||
for _, route := range *b.Routes {
|
||||
writeRouteHead(&sb, route)
|
||||
sb.WriteString(" ")
|
||||
}
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
|
||||
// We also got other search keys with 1 < Results
|
||||
// Get the ETA for this stop
|
||||
if 1 < len( *this.Query.SearchTerms ) {
|
||||
// We got a route key
|
||||
} else {
|
||||
|
||||
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 )
|
||||
// We also got other search keys with 1 < Results
|
||||
// Get the ETA for this stop
|
||||
if 1 < len(*this.Query.SearchTerms) {
|
||||
|
||||
for _, schedule := range *(*this.Schedules)[ r ] {
|
||||
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)
|
||||
|
||||
if !schedule.ETA.IsZero() {
|
||||
for _, schedule := range *(*this.Schedules)[r] {
|
||||
|
||||
_m := schedule.ETA.Sub( now ).Minutes()
|
||||
if !schedule.ETA.IsZero() {
|
||||
|
||||
sb.WriteString( " \\* " )
|
||||
txt := "%.0f min(s)"
|
||||
_m := schedule.ETA.Sub(now).Minutes()
|
||||
|
||||
if this.Lang == "zh-Hant" {
|
||||
txt = "%.0f 分鐘"
|
||||
}
|
||||
sb.WriteString(" \\* ")
|
||||
txt := "%.0f min(s)"
|
||||
|
||||
utils.WriteMDv2Text( &sb, fmt.Sprintf( txt, _m ) )
|
||||
if this.Lang == "zh-Hant" {
|
||||
txt = "%.0f 分鐘"
|
||||
}
|
||||
|
||||
if _m < 0 {
|
||||
sb.WriteString( " 走左了？" )
|
||||
}
|
||||
}
|
||||
utils.WriteMDv2Text(&sb, fmt.Sprintf(txt, _m))
|
||||
|
||||
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 )
|
||||
}
|
||||
}
|
||||
if _m < 0 {
|
||||
sb.WriteString(" 走左了？")
|
||||
}
|
||||
}
|
||||
|
||||
sb.WriteString( "\n" )
|
||||
}
|
||||
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{}
|
||||
sb.WriteString("\n")
|
||||
}
|
||||
|
||||
for _, item := range *this.Query.Results {
|
||||
var r *RouteStop
|
||||
r = any( item ).( *RouteStop )
|
||||
if r.PrevStop() == nil {
|
||||
routes = append( routes, r )
|
||||
}
|
||||
}
|
||||
// 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{}
|
||||
|
||||
sort.Sort( ByRoute( routes ) )
|
||||
for _, item := range *this.Query.Results {
|
||||
var r *RouteStop
|
||||
r = any(item).(*RouteStop)
|
||||
if r.PrevStop() == nil {
|
||||
routes = append(routes, r)
|
||||
}
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
sort.Sort(ByRoute(routes))
|
||||
|
||||
sb.WriteString( " \\> " )
|
||||
}
|
||||
sb.WriteString( "\n" )
|
||||
}
|
||||
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")
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return "", fmt.Errorf( "No Results" )
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return "", fmt.Errorf("No Results")
|
||||
}
|
||||
|
||||
return sb.String(), nil
|
||||
return sb.String(), nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user