fix: do not emit graph if empty

This commit is contained in:
Simone Margaritelli 2021-04-09 12:45:06 +02:00
parent 1d306e6cd2
commit 0042b77c36
2 changed files with 33 additions and 22 deletions

View file

@ -20,18 +20,24 @@ func (mod *Module) generateJSONGraph(bssid string) error {
return err
}
if mod.settings.privacy {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx")
if size > 0 {
if mod.settings.privacy {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx")
}
if err := ioutil.WriteFile(mod.settings.json.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.json.output,
time.Since(start),
size,
discarded)
}
} else {
mod.Info("graph is empty")
}
if err := ioutil.WriteFile(mod.settings.json.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.json.output,
time.Since(start),
size,
discarded)
}
return nil
}