|
|
@@ -5,7 +5,6 @@ import (
|
|
|
"encoding/xml"
|
|
|
"fmt"
|
|
|
"io/ioutil"
|
|
|
- "log"
|
|
|
"net/http"
|
|
|
"net/http/cookiejar"
|
|
|
"strings"
|
|
|
@@ -77,12 +76,23 @@ type Progress struct {
|
|
|
}
|
|
|
|
|
|
// LoadConfig func
|
|
|
-func (api *API) LoadConfig(name string) {
|
|
|
+func (api *API) LoadConfig(name string) error {
|
|
|
cfg, err := ioutil.ReadFile(name)
|
|
|
if err != nil {
|
|
|
- log.Fatal("Unable to load config", err)
|
|
|
+ return err
|
|
|
}
|
|
|
yaml.Unmarshal(cfg, &api)
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// SaveConfig func
|
|
|
+func (api *API) SaveConfig(name string) error {
|
|
|
+ data, err := yaml.Marshal(api)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ err = ioutil.WriteFile(name, data, 0700)
|
|
|
+ return err
|
|
|
}
|
|
|
|
|
|
func (api *API) setHeader(req *http.Request) {
|
