command_line_utils.go

package main

import (
	"flag"
	"fmt"
)

type FlightCommand struct {
	UsageLine string
	Run       func(args []string) int
	Flag      flag.FlagSet
}

func (c *FlightCommand) Name() string {
	return c.UsageLine
}

var (
	flightTableCommand      = &FlightCommand{UsageLine: "schedule"}
	flightDepartureCityName string
	flightArrivalCityName   string
	flightDate              string
	flightTripType          string
)

var (
	flightOverSeaTableCommand      = &FlightCommand{UsageLine: "oversea"}
	flightOverSeaDepartureCityName string
	flightOverSeaArrivalCityName   string
	flightOverSeaDate              string
	flightOverSeaCabinType         string
)

var (
	flightNumberInfoCommand = &FlightCommand{UsageLine: "code"}
	flightNumber            string
	flightNumberCheckDate   string
)

var (
	airportInfoCommand = &FlightCommand{UsageLine: "airport"}
	airportName        string
	airportDepOrArr    string
)

var flightCommands = []*FlightCommand{
	flightTableCommand,
	flightNumberInfoCommand,
	airportInfoCommand,
	flightOverSeaTableCommand,
}

// 查询机场信息
func executeAirportInfoTableFunc(args []string) int {
	airportInfoTable := NewVariFlightCrawler()
	airportInfoTable.runAirportInfo(args[0], args[1])
	return 1
}

// 查询航班号信息
func executeFlightNumberInfoTableFunc(args []string) int {
	flightNumberTable := NewVariFlightCrawler()
	flightNumberTable.runFlightInfo(args[0], args[1])
	return 1
}

// 查询国际航班信息
func executeOverSeaFlightTableFunc(args []string) int {
	flightTable := NewCtripCrawler()
	if len(args) < 4 {
		args = append(args, "")
	}
	flightTable.runOverSeaFlightTableCrawler(args[0], args[1], args[2], args[3])
	return 1
}

// 查询国内航班信息
func executeFlightTableFunc(args []string) int {
	flightTable := NewCtripCrawler()
	flightTable.runMainLandFlightTableCrawler(args[0], args[1], args[2], "Oneway", true)
	return 1
}

// 命令行初始化
func commandLineInit() {
	// 国内航班信息
	flightTableCommand.Run = executeFlightTableFunc
	flightTableCommand.Flag.StringVar(&flightDepartureCityName, "dep", "", "需要查询的始发地")
	flightTableCommand.Flag.StringVar(&flightArrivalCityName, "arr", "", "需要查询的目的地")
	flightTableCommand.Flag.StringVar(&flightDate, "date", "", "需要搜索的日期（格式: YYYY-MM-DD 例如: 2019-10-17）")

	// 国际航班信息
	flightOverSeaTableCommand.Run = executeOverSeaFlightTableFunc
	flightOverSeaTableCommand.Flag.StringVar(&flightOverSeaDepartureCityName, "dep", "", "需要查询的始发地")
	flightOverSeaTableCommand.Flag.StringVar(&flightOverSeaArrivalCityName, "arr", "", "需要查询的目的地")
	flightOverSeaTableCommand.Flag.StringVar(&flightOverSeaDate, "date", "", "需要搜索的日期（格式: YYYY-MM-DD 例如: 2019-10-17）")
	flightOverSeaTableCommand.Flag.StringVar(&flightOverSeaCabinType, "cabin", "", "舱位等级（经济舱，超级经济舱，商务/头等舱，商务舱，公务舱，头等舱）")

	// 航班号信息
	flightNumberInfoCommand.Run = executeFlightNumberInfoTableFunc
	flightNumberInfoCommand.Flag.StringVar(&flightNumber, "flightNumber", "", "需要查询的航班号")
	flightNumberInfoCommand.Flag.StringVar(&flightNumberCheckDate, "date", "", "需要搜索的日期（格式: YYYYMMDD 例如: 20191017）")

	// 机场信息
	airportInfoCommand.Run = executeAirportInfoTableFunc
	airportInfoCommand.Flag.StringVar(&airportName, "airportName", "", "需要查询机场名称（例如: 广州）")
	airportInfoCommand.Flag.StringVar(&airportDepOrArr, "depOrArr", "", "进场的进出港类别")
}

// 输出命令的使用方式
func commandUsage() {
	flag.Usage()
	fmt.Println("\n参数(Options):")
	fmt.Println("    schedule <起飞机场> <到达机场> <当前日期(日期格式: YYYY-MM-DD)>")
	fmt.Println("    oversea <起飞地> <到达地> <当前日期(日期格式: YYYY-MM-DD)> <舱位等级>")
	fmt.Println("    code <航班号> <当前日期(日期格式: YYYYMMDD)>")
	fmt.Println("    airport <城市名> <进出港字段(例如,进港: arr; 出港: dep)>")
}