73479bab26
- OS custom internal package for file system interaction - Remove fileManager external dependency - Closer API to Go's native API on the OS - Create directories at startup - Better testability - Move Unsetenv to os interface
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package dns
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/qdm12/gluetun/internal/constants"
|
|
)
|
|
|
|
func (c *configurator) DownloadRootHints(ctx context.Context, uid, gid int) error {
|
|
return c.downloadAndSave(ctx, "root hints",
|
|
string(constants.NamedRootURL), string(constants.RootHints), uid, gid)
|
|
}
|
|
|
|
func (c *configurator) DownloadRootKey(ctx context.Context, uid, gid int) error {
|
|
return c.downloadAndSave(ctx, "root key",
|
|
string(constants.RootKeyURL), string(constants.RootKey), uid, gid)
|
|
}
|
|
|
|
func (c *configurator) downloadAndSave(ctx context.Context, logName, url, filepath string, uid, gid int) error {
|
|
c.logger.Info("downloading %s from %s", logName, url)
|
|
content, status, err := c.client.Get(ctx, url)
|
|
if err != nil {
|
|
return err
|
|
} else if status != http.StatusOK {
|
|
return fmt.Errorf("HTTP status code is %d for %s", status, url)
|
|
}
|
|
|
|
file, err := c.openFile(filepath, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0400)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = file.Write(content)
|
|
if err != nil {
|
|
_ = file.Close()
|
|
return err
|
|
}
|
|
|
|
err = file.Chown(uid, gid)
|
|
if err != nil {
|
|
_ = file.Close()
|
|
return err
|
|
}
|
|
|
|
return file.Close()
|
|
}
|