From: Scott Worley Date: Thu, 3 Nov 2011 23:09:13 +0000 (-0700) Subject: Read --end into a map X-Git-Url: http://git.scottworley.com/planeteer/commitdiff_plain/1c1ede6836ed424330981ce160a8dd38e6759f48?ds=sidebyside Read --end into a map --- diff --git a/planeteer.go b/planeteer.go index d5c359a..c207c13 100644 --- a/planeteer.go +++ b/planeteer.go @@ -32,7 +32,7 @@ var start = flag.String("start", "", var flight_plan_string = flag.String("flight_plan", "", "Your hidey-holes for the day, comma-separated.") -var end = flag.String("end", "", +var end_string = flag.String("end", "", "A comma-separated list of acceptable ending planets.") var planet_data_file = flag.String("planet_data_file", "planet-data", @@ -60,18 +60,29 @@ var visit_string = flag.String("visit", "", func visit() []string { if *visit_string == "" { - return []string{} + return nil } return strings.Split(*visit_string, ",") } func flight_plan() []string { if *flight_plan_string == "" { - return []string{} + return nil } return strings.Split(*flight_plan_string, ",") } +func end() map[string]bool { + if *end_string == "" { + return nil + } + m := make(map[string]bool) + for _, p := range strings.Split(*flight_plan_string, ",") { + m[p] = true + } + return m +} + type Commodity struct { BasePrice int CanSell bool