2020-03-03 22:19:22 +00:00
|
|
|
package config
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"io"
|
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
|
|
|
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
)
|
|
|
|
|
|
|
|
type LocalPath struct {
|
|
|
|
Path string
|
|
|
|
}
|
|
|
|
|
|
|
|
// .lotusstorage/storage.json
|
|
|
|
type StorageConfig struct {
|
|
|
|
StoragePaths []LocalPath
|
|
|
|
}
|
|
|
|
|
|
|
|
func StorageFromFile(path string, def *StorageConfig) (*StorageConfig, error) {
|
|
|
|
file, err := os.Open(path)
|
|
|
|
switch {
|
|
|
|
case os.IsNotExist(err):
|
|
|
|
if def == nil {
|
|
|
|
return nil, xerrors.Errorf("couldn't load storage config: %w", err)
|
|
|
|
}
|
|
|
|
return def, nil
|
|
|
|
case err != nil:
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer file.Close() //nolint:errcheck // The file is RO
|
2020-03-04 02:24:08 +00:00
|
|
|
return StorageFromReader(file)
|
2020-03-03 22:19:22 +00:00
|
|
|
}
|
|
|
|
|
2020-03-04 02:24:08 +00:00
|
|
|
func StorageFromReader(reader io.Reader) (*StorageConfig, error) {
|
|
|
|
var cfg StorageConfig
|
2020-03-03 22:19:22 +00:00
|
|
|
err := json.NewDecoder(reader).Decode(&cfg)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &cfg, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func WriteStorageFile(path string, config StorageConfig) error {
|
2020-03-04 02:24:08 +00:00
|
|
|
b, err := json.MarshalIndent(config, "", " ")
|
2020-03-03 22:19:22 +00:00
|
|
|
if err != nil {
|
|
|
|
return xerrors.Errorf("marshaling storage config: %w", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := ioutil.WriteFile(path, b, 0644); err != nil {
|
|
|
|
return xerrors.Errorf("persisting storage config (%s): %w", path, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|