47 lines
667 B
Go
47 lines
667 B
Go
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"os/user"
|
|
)
|
|
|
|
// Am I root?
|
|
func AmIRoot() (bool, error) {
|
|
u, err := user.Current()
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return u.Username == "root", nil
|
|
}
|
|
|
|
// Replace `a` with `b` in a given file at `path`. This function is non-atomic
|
|
// so be careful with crashing.
|
|
func ReplaceInFile(path string, a, b []byte) error {
|
|
data, err := os.ReadFile(path)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
data = bytes.Replace(
|
|
data,
|
|
a,
|
|
b,
|
|
1,
|
|
)
|
|
|
|
err = os.WriteFile(path, data, 0644)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func WriteToFile(path string, data []byte) error {
|
|
return os.WriteFile(path, data, 0644)
|
|
}
|