blob: ef8dfda9e046653f15f68e1b498f04cc39f7f54f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
package mountain_pass
import (
"context"
"database/sql"
"os"
"github.com/martinlehoux/kagamigo/kcore"
"github.com/schollz/progressbar/v3"
)
// ProgressBarEnabled reports whether stderr is attached to a terminal, so the
// progress bar only renders for interactive runs and stays out of piped or
// redirected output.
func ProgressBarEnabled() bool {
fi, err := os.Stderr.Stat()
return err == nil && fi.Mode()&os.ModeCharDevice != 0
}
func FetchOSM() {
err := FetchFrancePBF("france-latest.osm.pbf")
kcore.Expect(err, "failed to download France OSM PBF")
}
func ExtractOSM(db *sql.DB, pbfPath string) {
var bar *progressbar.ProgressBar
_, err := ExtractMountainPasses(context.Background(), pbfPath, db, func(current, total int64) {
if bar == nil && ProgressBarEnabled() {
bar = progressbar.DefaultBytes(total, "Extracting mountain passes")
}
if bar != nil {
bar.Set64(current)
}
})
if bar != nil {
bar.Finish()
}
kcore.Expect(err, "failed to extract mountain passes from OSM")
}
|
