c/ooni-probe-cli
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

On first run also download the geoip data files

Browse Source
This commit is contained in:
Arturo Filastò 2018-03-23 15:38:25 +01:00
parent 749cc665af
commit 9fe917f294
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ooni.go
View File

@ -77,7 +77,7 @@ func (c *Context) Init() error {
return errors.Wrap(err, "migrating home") return errors.Wrap(err, "migrating home")
} }
if err = CreateHomeDirs(c.Home); err != nil { if err = MaybeInitializeHome(c.Home); err != nil {
return err return err
} }
@ -212,16 +212,28 @@ func ParseConfig(b []byte) (*Config, error) {
return c, nil return c, nil
} }
// CreateHomeDirs creates the OONI home subdirectories // MaybeInitializeHome does the setup for a new OONI Home
func CreateHomeDirs(home string) error { func MaybeInitializeHome(home string) error {
firstRun := false
requiredDirs := []string{"db", "msmts", "geoip"} requiredDirs := []string{"db", "msmts", "geoip"}
for _, d := range requiredDirs { for _, d := range requiredDirs {
if _, e := os.Stat(filepath.Join(home, d)); e != nil { if _, e := os.Stat(filepath.Join(home, d)); e != nil {
firstRun = true
if err := os.MkdirAll(filepath.Join(home, d), 0700); err != nil { if err := os.MkdirAll(filepath.Join(home, d), 0700); err != nil {
return err return err
} }
} }
} }
if firstRun == true {
log.Info("This is the first time you are running OONI Probe. Downloading some files.")
geoipDir := utils.GeoIPDir(home)
if err := utils.DownloadGeoIPDatabaseFiles(geoipDir); err != nil {
return err
}
if err := utils.DownloadLegacyGeoIPDatabaseFiles(geoipDir); err != nil {
return err
}
}
return nil return nil
} }