Consolidate config related functionality into the config package
This commit is contained in:
parent
ba056694f1
commit
2653a3f67f
|
@ -1,7 +0,0 @@
|
||||||
package config
|
|
||||||
|
|
||||||
// Advanced settings
|
|
||||||
type Advanced struct {
|
|
||||||
IncludeCountry bool `json:"include_country"`
|
|
||||||
UseDomainFronting bool `json:"use_domain_fronting"`
|
|
||||||
}
|
|
|
@ -1,8 +0,0 @@
|
||||||
package config
|
|
||||||
|
|
||||||
// AutomatedTesting settings
|
|
||||||
type AutomatedTesting struct {
|
|
||||||
Enabled bool `json:"enabled"`
|
|
||||||
EnabledTests []string `json:"enabled_tests"`
|
|
||||||
MonthlyAllowance string `json:"monthly_allowance"`
|
|
||||||
}
|
|
|
@ -1,8 +0,0 @@
|
||||||
package config
|
|
||||||
|
|
||||||
// Notifications settings
|
|
||||||
type Notifications struct {
|
|
||||||
Enabled bool `json:"enabled"`
|
|
||||||
NotifyOnTestCompletion bool `json:"notify_on_test_completion"`
|
|
||||||
NotifyOnNews bool `json:"notify_on_news"`
|
|
||||||
}
|
|
119
config/parser.go
Normal file
119
config/parser.go
Normal file
|
@ -0,0 +1,119 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/ooni/probe-cli/utils"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ReadConfig reads the configuration from the path
|
||||||
|
func ReadConfig(path string) (*Config, error) {
|
||||||
|
b, err := ioutil.ReadFile(path)
|
||||||
|
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
c := &Config{}
|
||||||
|
|
||||||
|
if err = c.Default(); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "defaulting")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = c.Validate(); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "validating")
|
||||||
|
}
|
||||||
|
|
||||||
|
return c, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "reading file")
|
||||||
|
}
|
||||||
|
|
||||||
|
c, err := ParseConfig(b)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "parsing config")
|
||||||
|
}
|
||||||
|
c.path = path
|
||||||
|
return c, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseConfig returns config from JSON bytes.
|
||||||
|
func ParseConfig(b []byte) (*Config, error) {
|
||||||
|
c := &Config{}
|
||||||
|
|
||||||
|
if err := json.Unmarshal(b, c); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "parsing json")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.Default(); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "defaulting")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.Validate(); err != nil {
|
||||||
|
return nil, errors.Wrap(err, "validating")
|
||||||
|
}
|
||||||
|
|
||||||
|
return c, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Config for the OONI Probe installation
|
||||||
|
type Config struct {
|
||||||
|
// Private settings
|
||||||
|
Comment string `json:"_"`
|
||||||
|
Version int64 `json:"_version"`
|
||||||
|
InformedConsent bool `json:"_informed_consent"`
|
||||||
|
|
||||||
|
AutoUpdate bool `json:"auto_update"`
|
||||||
|
Sharing Sharing `json:"sharing"`
|
||||||
|
Notifications Notifications `json:"notifications"`
|
||||||
|
AutomatedTesting AutomatedTesting `json:"automated_testing"`
|
||||||
|
NettestGroups NettestGroups `json:"test_settings"`
|
||||||
|
Advanced Advanced `json:"advanced"`
|
||||||
|
|
||||||
|
mutex sync.Mutex
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write the config file in json to the path
|
||||||
|
func (c *Config) Write() error {
|
||||||
|
c.Lock()
|
||||||
|
configJSON, _ := json.MarshalIndent(c, "", " ")
|
||||||
|
if c.path == "" {
|
||||||
|
return errors.New("config file path is empty")
|
||||||
|
}
|
||||||
|
if err := ioutil.WriteFile(c.path, configJSON, 0644); err != nil {
|
||||||
|
return errors.Wrap(err, "writing config JSON")
|
||||||
|
}
|
||||||
|
c.Unlock()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Lock acquires the write mutex
|
||||||
|
func (c *Config) Lock() {
|
||||||
|
c.mutex.Lock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unlock releases the write mutex
|
||||||
|
func (c *Config) Unlock() {
|
||||||
|
c.mutex.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Default config settings
|
||||||
|
func (c *Config) Default() error {
|
||||||
|
home, err := utils.GetOONIHome()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
c.path = filepath.Join(home, "config.json")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate the config file
|
||||||
|
func (c *Config) Validate() error {
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -82,7 +82,6 @@ func (s *InstantMessaging) NettestConfigs() []NettestConfig {
|
||||||
|
|
||||||
// Performance nettest group
|
// Performance nettest group
|
||||||
type Performance struct {
|
type Performance struct {
|
||||||
EnabledTests []string `json:"enabled_tests"`
|
|
||||||
NDTServer string `json:"ndt_server"`
|
NDTServer string `json:"ndt_server"`
|
||||||
NDTServerPort string `json:"ndt_server_port"`
|
NDTServerPort string `json:"ndt_server_port"`
|
||||||
DashServer string `json:"dash_server"`
|
DashServer string `json:"dash_server"`
|
||||||
|
@ -101,3 +100,32 @@ type NettestGroups struct {
|
||||||
Performance Performance `json:"performance"`
|
Performance Performance `json:"performance"`
|
||||||
Middlebox Middlebox `json:"middlebox"`
|
Middlebox Middlebox `json:"middlebox"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Notifications settings
|
||||||
|
type Notifications struct {
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
NotifyOnTestCompletion bool `json:"notify_on_test_completion"`
|
||||||
|
NotifyOnNews bool `json:"notify_on_news"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sharing settings
|
||||||
|
type Sharing struct {
|
||||||
|
IncludeIP bool `json:"include_ip"`
|
||||||
|
IncludeASN bool `json:"include_asn"`
|
||||||
|
IncludeGPS bool `json:"include_gps"`
|
||||||
|
UploadResults bool `json:"upload_results"`
|
||||||
|
SendCrashReports bool `json:"send_crash_reports"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Advanced settings
|
||||||
|
type Advanced struct {
|
||||||
|
IncludeCountry bool `json:"include_country"`
|
||||||
|
UseDomainFronting bool `json:"use_domain_fronting"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AutomatedTesting settings
|
||||||
|
type AutomatedTesting struct {
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
EnabledTests []string `json:"enabled_tests"`
|
||||||
|
MonthlyAllowance string `json:"monthly_allowance"`
|
||||||
|
}
|
|
@ -1,10 +0,0 @@
|
||||||
package config
|
|
||||||
|
|
||||||
// Sharing settings
|
|
||||||
type Sharing struct {
|
|
||||||
IncludeIP bool `json:"include_ip"`
|
|
||||||
IncludeASN bool `json:"include_asn"`
|
|
||||||
IncludeGPS bool `json:"include_gps"`
|
|
||||||
UploadResults bool `json:"upload_results"`
|
|
||||||
SendCrashReports bool `json:"send_crash_reports"`
|
|
||||||
}
|
|
|
@ -1,5 +1,7 @@
|
||||||
{
|
{
|
||||||
"_": "This is your OONI Probe config file. See https://ooni.io/help/ooniprobe-cli for help",
|
"_": "This is your OONI Probe config file. See https://ooni.io/help/probe-cli for help",
|
||||||
|
"_version": 0,
|
||||||
|
"_informed_consent": false,
|
||||||
"auto_update": true,
|
"auto_update": true,
|
||||||
"sharing": {
|
"sharing": {
|
||||||
"include_ip": false,
|
"include_ip": false,
|
||||||
|
@ -56,8 +58,6 @@
|
||||||
},
|
},
|
||||||
"advanced": {
|
"advanced": {
|
||||||
"include_country": true,
|
"include_country": true,
|
||||||
"use_domain_fronting": true
|
"use_domain_fronting": false
|
||||||
},
|
},
|
||||||
"_config_version": "0.0.1",
|
|
||||||
"_informed_consent": true
|
|
||||||
}
|
}
|
||||||
|
|
125
ooni.go
125
ooni.go
|
@ -1,11 +1,9 @@
|
||||||
package ooni
|
package ooni
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"sync"
|
|
||||||
|
|
||||||
"github.com/apex/log"
|
"github.com/apex/log"
|
||||||
"github.com/jmoiron/sqlx"
|
"github.com/jmoiron/sqlx"
|
||||||
|
@ -17,7 +15,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
// Onboarding process
|
// Onboarding process
|
||||||
func Onboarding(c *Config) error {
|
func Onboarding(c *config.Config) error {
|
||||||
log.Info("Onboarding starting")
|
log.Info("Onboarding starting")
|
||||||
|
|
||||||
// To prevent races we always must acquire the config file lock before
|
// To prevent races we always must acquire the config file lock before
|
||||||
|
@ -35,7 +33,7 @@ func Onboarding(c *Config) error {
|
||||||
|
|
||||||
// Context for OONI Probe
|
// Context for OONI Probe
|
||||||
type Context struct {
|
type Context struct {
|
||||||
Config *Config
|
Config *config.Config
|
||||||
DB *sqlx.DB
|
DB *sqlx.DB
|
||||||
Location *utils.LocationInfo
|
Location *utils.LocationInfo
|
||||||
|
|
||||||
|
@ -82,7 +80,7 @@ func (c *Context) Init() error {
|
||||||
|
|
||||||
if c.configPath != "" {
|
if c.configPath != "" {
|
||||||
log.Debugf("Reading config file from %s", c.configPath)
|
log.Debugf("Reading config file from %s", c.configPath)
|
||||||
c.Config, err = ReadConfig(c.configPath)
|
c.Config, err = config.ReadConfig(c.configPath)
|
||||||
} else {
|
} else {
|
||||||
log.Debug("Reading default config file")
|
log.Debug("Reading default config file")
|
||||||
c.Config, err = ReadDefaultConfigPaths(c.Home)
|
c.Config, err = ReadDefaultConfigPaths(c.Home)
|
||||||
|
@ -118,88 +116,11 @@ func (c *Context) Init() error {
|
||||||
func NewContext(configPath string, homePath string) *Context {
|
func NewContext(configPath string, homePath string) *Context {
|
||||||
return &Context{
|
return &Context{
|
||||||
Home: homePath,
|
Home: homePath,
|
||||||
Config: &Config{},
|
Config: &config.Config{},
|
||||||
configPath: configPath,
|
configPath: configPath,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Config for the OONI Probe installation
|
|
||||||
type Config struct {
|
|
||||||
// Private settings
|
|
||||||
Comment string `json:"_"`
|
|
||||||
ConfigVersion string `json:"_config_version"`
|
|
||||||
InformedConsent bool `json:"_informed_consent"`
|
|
||||||
|
|
||||||
AutoUpdate bool `json:"auto_update"`
|
|
||||||
Sharing config.Sharing `json:"sharing"`
|
|
||||||
Notifications config.Notifications `json:"notifications"`
|
|
||||||
AutomatedTesting config.AutomatedTesting `json:"automated_testing"`
|
|
||||||
NettestGroups config.NettestGroups `json:"test_settings"`
|
|
||||||
Advanced config.Advanced `json:"advanced"`
|
|
||||||
|
|
||||||
mutex sync.Mutex
|
|
||||||
path string
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write the config file in json to the path
|
|
||||||
func (c *Config) Write() error {
|
|
||||||
c.Lock()
|
|
||||||
configJSON, _ := json.MarshalIndent(c, "", " ")
|
|
||||||
if c.path == "" {
|
|
||||||
return errors.New("config file path is empty")
|
|
||||||
}
|
|
||||||
if err := ioutil.WriteFile(c.path, configJSON, 0644); err != nil {
|
|
||||||
return errors.Wrap(err, "writing config JSON")
|
|
||||||
}
|
|
||||||
c.Unlock()
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Lock acquires the write mutex
|
|
||||||
func (c *Config) Lock() {
|
|
||||||
c.mutex.Lock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Unlock releases the write mutex
|
|
||||||
func (c *Config) Unlock() {
|
|
||||||
c.mutex.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Default config settings
|
|
||||||
func (c *Config) Default() error {
|
|
||||||
home, err := utils.GetOONIHome()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
c.path = filepath.Join(home, "config.json")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Validate the config file
|
|
||||||
func (c *Config) Validate() error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ParseConfig returns config from JSON bytes.
|
|
||||||
func ParseConfig(b []byte) (*Config, error) {
|
|
||||||
c := &Config{}
|
|
||||||
|
|
||||||
if err := json.Unmarshal(b, c); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "parsing json")
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := c.Default(); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "defaulting")
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := c.Validate(); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "validating")
|
|
||||||
}
|
|
||||||
|
|
||||||
return c, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// MaybeInitializeHome does the setup for a new OONI Home
|
// MaybeInitializeHome does the setup for a new OONI Home
|
||||||
func MaybeInitializeHome(home string) error {
|
func MaybeInitializeHome(home string) error {
|
||||||
firstRun := false
|
firstRun := false
|
||||||
|
@ -214,9 +135,11 @@ func MaybeInitializeHome(home string) error {
|
||||||
if firstRun == true {
|
if firstRun == true {
|
||||||
log.Info("This is the first time you are running OONI Probe. Downloading some files.")
|
log.Info("This is the first time you are running OONI Probe. Downloading some files.")
|
||||||
geoipDir := utils.GeoIPDir(home)
|
geoipDir := utils.GeoIPDir(home)
|
||||||
|
log.Debugf("Downloading GeoIP database files")
|
||||||
if err := utils.DownloadGeoIPDatabaseFiles(geoipDir); err != nil {
|
if err := utils.DownloadGeoIPDatabaseFiles(geoipDir); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
log.Debugf("Downloading legacy GeoIP database Files")
|
||||||
if err := utils.DownloadLegacyGeoIPDatabaseFiles(geoipDir); err != nil {
|
if err := utils.DownloadLegacyGeoIPDatabaseFiles(geoipDir); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -225,44 +148,14 @@ func MaybeInitializeHome(home string) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReadConfig reads the configuration from the path
|
|
||||||
func ReadConfig(path string) (*Config, error) {
|
|
||||||
b, err := ioutil.ReadFile(path)
|
|
||||||
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
c := &Config{}
|
|
||||||
|
|
||||||
if err = c.Default(); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "defaulting")
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = c.Validate(); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "validating")
|
|
||||||
}
|
|
||||||
|
|
||||||
return c, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.Wrap(err, "reading file")
|
|
||||||
}
|
|
||||||
|
|
||||||
c, err := ParseConfig(b)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.Wrap(err, "parsing config")
|
|
||||||
}
|
|
||||||
c.path = path
|
|
||||||
return c, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReadDefaultConfigPaths from common locations.
|
// ReadDefaultConfigPaths from common locations.
|
||||||
func ReadDefaultConfigPaths(home string) (*Config, error) {
|
func ReadDefaultConfigPaths(home string) (*config.Config, error) {
|
||||||
var paths = []string{
|
var paths = []string{
|
||||||
filepath.Join(home, "config.json"),
|
filepath.Join(home, "config.json"),
|
||||||
}
|
}
|
||||||
for _, path := range paths {
|
for _, path := range paths {
|
||||||
if _, err := os.Stat(path); err == nil {
|
if _, err := os.Stat(path); err == nil {
|
||||||
c, err := ReadConfig(path)
|
c, err := config.ReadConfig(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -271,5 +164,5 @@ func ReadDefaultConfigPaths(home string) (*Config, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run from the default config
|
// Run from the default config
|
||||||
return ReadConfig(paths[0])
|
return config.ReadConfig(paths[0])
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user