diff options
Diffstat (limited to 'cli/cli.go')
| -rw-r--r-- | cli/cli.go | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/cli/cli.go b/cli/cli.go new file mode 100644 index 0000000..a25d35d --- /dev/null +++ b/cli/cli.go @@ -0,0 +1,169 @@ +package cli + +import ( + "context" + "database/sql" + "flag" + "log/slog" + "os" + "path/filepath" + "runtime/pprof" + "strings" + "time" + + "github.com/martinlehoux/biking_home/chart" + "github.com/martinlehoux/biking_home/config" + "github.com/martinlehoux/biking_home/mountain_pass" + "github.com/martinlehoux/biking_home/osmpass" + "github.com/martinlehoux/biking_home/ride" + "github.com/martinlehoux/biking_home/web" + "github.com/martinlehoux/kagamigo/kcore" +) + +var ( + download = flag.Bool("download", false, "download mountain passes into the database") + resume = flag.Bool("resume", false, "skip departments already cached on disk") + importCache = flag.Bool("import-cached", false, "import cached department CSVs into the database") + fetchOSM = flag.Bool("fetch-osm", false, "download the France OSM PBF (resumable) into france-latest.osm.pbf") + extractOSM = flag.String("extract-osm", "", "extract mountain passes from an OSM PBF file into the database") + enrich = flag.Bool("enrich", false, "backfill mountain pass coordinates from OSM data") + demo = flag.Bool("demo", false, "run the climb similarity demo") + chartFile = flag.String("chart", "", "render a climb/pass chart for a GPX file") + cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file") + parser = ride.GPXRideParser{} +) + +func Run(db *sql.DB, configPath string, appConfig config.Config) { + switch { + case *download: + err := mountain_pass.DownloadMountainPasses(db, 5*time.Second, *resume) + kcore.Expect(err, "failed to download mountain passes") + case *importCache: + _, err := mountain_pass.ImportCachedDepartments(db, []string{"06", "13"}) + kcore.Expect(err, "failed to import cached departments") + case *fetchOSM: + err := osmpass.FetchFrancePBF("france-latest.osm.pbf") + kcore.Expect(err, "failed to download France OSM PBF") + case *extractOSM != "": + _, err := osmpass.ExtractMountainPasses(context.Background(), *extractOSM, db) + kcore.Expect(err, "failed to extract mountain passes from OSM") + case *enrich: + _, err := osmpass.EnrichMountainPasses(db) + kcore.Expect(err, "failed to enrich mountain passes") + case *demo: + runDemo(db) + case *chartFile != "": + runChart(db, *chartFile) + default: + runServer(db, configPath, appConfig) + } +} + +func runServer(db *sql.DB, configPath string, appConfig config.Config) { + server := web.NewServer(db, configPath) + slog.Info("Starting web server", "address", appConfig.Server.PublicURL) + kcore.Expect(server.ListenAndServe(appConfig.Server.Address), "web server stopped") +} + +func runChart(db *sql.DB, filename string) { + r, err := ride.ParseFile(parser, filename) + kcore.Expect(err, "failed to parse ride") + passes, err := mountain_pass.LoadMountainPasses(db) + kcore.Expect(err, "failed to load mountain passes") + + climbs := r.AllClimbs() + for i := range climbs { + if matched, ok := mountain_pass.MatchClimb(climbs[i], passes, 300, 50); ok { + climbs[i].Name = matched.Name + } + } + crossings := mountain_pass.DetectCrossings(r, passes, 100, 25) + + output := strings.TrimSuffix(filename, filepath.Ext(filename)) + "-chart.png" + chart.Render(r, climbs, crossings, output) + slog.Info("Chart written", "file", output) +} + +func runDemo(db *sql.DB) { + if *cpuprofile != "" { + f, err := os.Create(*cpuprofile) + kcore.Expect(err, "failed to create CPU profile") + pprof.StartCPUProfile(f) + defer pprof.StopCPUProfile() + } + ride19Jan, _ := ride.ParseFile(parser, "examples/activity_18043356988.gpx") + ride30Mar, _ := ride.ParseFile(parser, "examples/activity_18679866717.gpx") + ride20Apr, _ := ride.ParseFile(parser, "examples/activity_18880605641.gpx") + for _, r := range []*ride.Ride{&ride19Jan, &ride30Mar, &ride20Apr} { + slog.Info("Ride", "difficulty", r.DifficultyScore()) + } + passes, passesErr := mountain_pass.LoadMountainPasses(db) + if passesErr != nil { + slog.Warn("No mountain passes in database, skipping crossings and climb naming", "error", passesErr) + passes = nil + } else { + rides := []struct { + name string + ride ride.Ride + }{ + {"2022-07-21.Pogacar", func() ride.Ride { + r, _ := ride.ParseFile(parser, "examples/2022-07-21.Pogacar.gpx") + return r + }()}, + {"2023-06-17.AlpesVerdonTour", func() ride.Ride { + r, _ := ride.ParseFile(parser, "examples/2023-06-17.AlpesVerdonTour.gpx") + return r + }()}, + {"2024-12-29.MimetArbois", func() ride.Ride { + r, _ := ride.ParseFile(parser, "examples/2024-12-29.MimetArbois.gpx") + return r + }()}, + {"activity_18043356988", ride19Jan}, + {"activity_18679866717", ride30Mar}, + {"activity_18880605641", ride20Apr}, + } + for _, entry := range rides { + crossings := mountain_pass.DetectCrossings(entry.ride, passes, 100, 25) + if len(crossings) > 0 { + slog.Info("Crossings", "ride", entry.name, "count", len(crossings)) + for _, crossing := range crossings { + slog.Info("Crossing", "pass", crossing.String()) + } + } + } + } + + nameClimbs := func(climbs []ride.Climb) { + for i := range climbs { + if matched, ok := mountain_pass.MatchClimb(climbs[i], passes, 300, 50); ok { + climbs[i].Name = matched.Name + } + } + } + verdon, _ := ride.ParseFile(parser, "examples/2023-06-17.AlpesVerdonTour.gpx") + verdonClimbs := verdon.AllClimbs() + nameClimbs(verdonClimbs) + for _, climb := range verdonClimbs { + slog.Info("Climb", "climb", climb, "elevation", climb.Top().ElevationM, "duration", climb.Duration(), "speed", climb.Speed()*3.6) + } + + index := ride.NewFlatRideClimbsIndex(30) + index.Insert(ride19Jan) + index.Insert(ride30Mar) + ride20AprilClimbs := ride20Apr.AllClimbs() + nameClimbs(ride20AprilClimbs) + for _, climb := range ride20AprilClimbs { + slog.Info("Climb", "climb", climb, "duration", climb.Duration(), "speed", climb.Speed()*3.6) + if len(index.Similar(ride20Apr, climb)) > 0 { + for _, similar := range index.Similar(ride20Apr, climb) { + slog.Info("Similar", "climb", similar, "duration", similar.Duration(), "speed", similar.Speed()*3.6) + } + } + } + ride30MarchClimbs := ride30Mar.AllClimbs() + nameClimbs(ride30MarchClimbs) + for _, climb := range ride30MarchClimbs { + slog.Info("Climb", "climb", climb, "duration", climb.Duration(), "speed", climb.Speed()*3.6) + } + ride.PlotScore(&ride30Mar, 61.2, 66.7, "Ride 30 Mar.png") +} |