134 lines
2.5 KiB
Go
134 lines
2.5 KiB
Go
package mozilla
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"regexp"
|
|
)
|
|
|
|
const (
|
|
PrefsFile = "prefs.js"
|
|
|
|
// Parses vales in prefs.js under the form:
|
|
// user_pref("my.pref.option", value);
|
|
REFirefoxPrefs = `user_pref\("(?P<option>%s)",\s+"*(?P<value>.*[^"])"*\)\s*;\s*(\n|$)`
|
|
)
|
|
|
|
var (
|
|
ErrPrefNotFound = errors.New("pref not defined")
|
|
ErrPrefNotBool = errors.New("pref is not bool")
|
|
)
|
|
|
|
// Finds and returns a prefernce definition.
|
|
// Returns empty string ("") if no pref found
|
|
func FindPref(path string, name string) (string, error) {
|
|
text, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
re := regexp.MustCompile(fmt.Sprintf(REFirefoxPrefs, name))
|
|
match := re.FindSubmatch(text)
|
|
|
|
if match == nil {
|
|
return "", nil
|
|
}
|
|
|
|
results := map[string]string{}
|
|
for i, name := range match {
|
|
results[re.SubexpNames()[i]] = string(name)
|
|
}
|
|
|
|
return results["value"], nil
|
|
}
|
|
|
|
// Returns true if the `name` preference is found in `prefs.js`
|
|
func HasPref(path string, name string) (bool, error) {
|
|
res, err := FindPref(path, name)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if res != "" {
|
|
return true, nil
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
|
|
//TODO: Pass profile name and use standard `prefs.js` file name in base
|
|
// directory of profile
|
|
func GetPrefBool(path string, name string) (bool, error) {
|
|
val, err := FindPref(path, name)
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if val == "" {
|
|
return false, ErrPrefNotFound
|
|
}
|
|
|
|
if val == "true" {
|
|
return true, nil
|
|
} else if val == "false" {
|
|
return false, nil
|
|
}
|
|
|
|
return false, ErrPrefNotBool
|
|
|
|
}
|
|
|
|
// Set a preference in the preference file under `path`
|
|
func SetPrefBool(path string, name string, val bool) error {
|
|
// Get file mode
|
|
info, err := os.Stat(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mode := info.Mode()
|
|
|
|
// Pref already defined, replace it
|
|
if v, _ := HasPref(path, name); v {
|
|
|
|
f, err := os.OpenFile(path, os.O_RDWR, mode)
|
|
defer f.Sync()
|
|
defer f.Close()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
re := regexp.MustCompile(fmt.Sprintf(REFirefoxPrefs, name))
|
|
template := []byte(fmt.Sprintf("user_pref(\"$option\", %t) ;\n", val))
|
|
text, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = f.Seek(0, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
output := string(re.ReplaceAll(text, template))
|
|
fmt.Fprint(f, output)
|
|
|
|
} else {
|
|
f, err := os.OpenFile(path, os.O_RDWR|os.O_APPEND, mode)
|
|
defer f.Sync()
|
|
defer f.Close()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Append pref
|
|
fmt.Fprintf(f, "user_pref(\"%s\", %t);\n", name, val)
|
|
}
|
|
|
|
return nil
|
|
}
|