-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- includes support for CSRF token - includes new modules for L3 configuration - includes new modules for FullConfig operations - refactors modules to use Client object - improves error handling and reporting
- Loading branch information
1 parent
28f54e4
commit 946f51b
Showing
9 changed files
with
1,545 additions
and
54 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,251 @@ | ||
| package aoscxgo | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "errors" | ||
| "fmt" | ||
| "io/ioutil" | ||
| "log" | ||
| "net/http" | ||
| "time" | ||
|
|
||
| "github.com/google/go-cmp/cmp" | ||
| ) | ||
|
|
||
| type FullConfig struct { | ||
|
|
||
| // Connection properties. | ||
| FileName string `json:"filename"` | ||
| //Hash hash.Hash `json:"hash"` | ||
| Config string `json:"config"` | ||
| uri string `json:"uri"` | ||
| } | ||
|
|
||
| // Create performs POST to create VLAN configuration on the given Client object. | ||
| func (fc *FullConfig) Create(c *Client) (*http.Response, error) { | ||
| if fc.FileName == "" { | ||
| return nil, &RequestError{ | ||
| StatusCode: "Missing FileName", | ||
| Err: errors.New("Create Error"), | ||
| } | ||
| } | ||
|
|
||
| config_str, err := fc.ReadConfigFile(fc.FileName) | ||
|
|
||
| if err != nil { | ||
| return nil, &RequestError{ | ||
| StatusCode: "Error in reading config file", | ||
| Err: err, | ||
| } | ||
| } | ||
|
|
||
| res, body := fc.ValidateConfig(c, config_str) | ||
|
|
||
| if body == nil { | ||
| return res, &RequestError{ | ||
| StatusCode: res.Status, | ||
| Err: errors.New("Validation Error"), | ||
| } | ||
|
|
||
| } else if body["state"] == "success" { | ||
| res2, body2 := fc.ApplyConfig(c, config_str) | ||
| if body2["state"] != "success" { | ||
| errors_dict := body2["errors"].([]interface{}) | ||
| error_str := convert_errors(errors_dict) | ||
|
|
||
| return res2, &RequestError{ | ||
| StatusCode: "Error in applying config error : \n" + error_str, | ||
| Err: errors.New("Apply Error"), | ||
| } | ||
| } else { | ||
| log.Println("New Config Applied Successfully") | ||
| fc.Get(c) | ||
| return res2, nil | ||
| } | ||
| } else if res != nil && body != nil { | ||
| errors_dict := body["errors"].([]interface{}) | ||
| errors_dict = errors_dict | ||
| error_str := convert_errors(errors_dict) | ||
|
|
||
| return res, &RequestError{ | ||
| StatusCode: "Error in validating config error : \n" + error_str, | ||
| Err: errors.New("Apply Error"), | ||
| } | ||
|
|
||
| } | ||
|
|
||
| return res, &RequestError{ | ||
| StatusCode: res.Status, | ||
| Err: errors.New("Validation Error"), | ||
| } | ||
| } | ||
|
|
||
| // Get performs GET to retrieve Running configuration for the given Client object. | ||
| func (fc *FullConfig) Get(c *Client) error { | ||
| base_uri := "configs/running-config" | ||
| url := "https://" + c.Hostname + "/rest/" + c.Version + "/" + base_uri | ||
| res, _ := get_accept_text(c, url) | ||
|
|
||
| if res.Status != "200 OK" { | ||
| return &RequestError{ | ||
| StatusCode: res.Status + url, | ||
| Err: errors.New("Retrieval Error"), | ||
| } | ||
| } | ||
|
|
||
| // Read the content | ||
| var bodyBytes []byte | ||
| if res.Body != nil { | ||
| bodyBytes, _ = ioutil.ReadAll(res.Body) | ||
| } | ||
| // Restore the io.ReadCloser to its original state | ||
| res.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes)) | ||
| // Use the content | ||
| bodyString := string(bodyBytes) | ||
|
|
||
| // config := bytes.NewBuffer(nil) | ||
| // config, _ = ioutil.ReadAll(res.Body) | ||
| fc.Config = bodyString | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (fc *FullConfig) ReadConfigFile(filename string) (string, error) { | ||
| config_contents, err := ioutil.ReadFile(filename) | ||
|
|
||
| if err != nil { | ||
| err_str := "Unable to read file " + filename | ||
| return "", &RequestError{ | ||
| StatusCode: err_str, | ||
| Err: err, | ||
| } | ||
| } | ||
| config_str := string(config_contents) | ||
| return config_str, nil | ||
|
|
||
| } | ||
|
|
||
| // Formats the errors provided by dryrun for user | ||
| func convert_errors(errors_list []interface{}) string { | ||
| errors_str := "" | ||
| for _, error_dict := range errors_list { | ||
| tmp_dict := error_dict.(map[string]interface{}) | ||
| line_num := tmp_dict["line"] | ||
| line_float := line_num.(float64) | ||
| line_int := int(line_float) | ||
| errors_str += "line " | ||
| line_num_str := fmt.Sprintf("%d", line_int) | ||
| errors_str += line_num_str | ||
| errors_str += " | " | ||
| errors_str += tmp_dict["message"].(string) | ||
| errors_str += "\n" | ||
| } | ||
| return errors_str | ||
| } | ||
|
|
||
| // Sets Config Attribute | ||
| func (fc *FullConfig) SetConfig(config string) error { | ||
| fc.Config = config | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // Compares supplied string Config to stored Object | ||
| func (fc *FullConfig) CompareConfig(new_config string) string { | ||
|
|
||
| return cmp.Diff(fc.Config, new_config) | ||
| } | ||
|
|
||
| // Compares supplied string Config to stored Object | ||
| func (fc *FullConfig) DownloadConfig(c *Client, filename string) error { | ||
| base_uri := "configs/running-config" | ||
| url := "https://" + c.Hostname + "/rest/" + c.Version + "/" + base_uri | ||
| res, _ := get_accept_text(c, url) | ||
|
|
||
| if res.Status != "200 OK" { | ||
| return &RequestError{ | ||
| StatusCode: res.Status + url, | ||
| Err: errors.New("Retrieval Error"), | ||
| } | ||
| } | ||
|
|
||
| // Read the content | ||
| var bodyBytes []byte | ||
| if res.Body != nil { | ||
| bodyBytes, _ = ioutil.ReadAll(res.Body) | ||
| } | ||
| // Restore the io.ReadCloser to its original state | ||
| res.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes)) | ||
| // Use the content | ||
| bodyString := string(bodyBytes) | ||
|
|
||
| err := ioutil.WriteFile(filename, []byte(bodyString), 0644) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| return err | ||
| } | ||
|
|
||
| // Validates supplied CLI configuration as string using dryrun | ||
| func (fc *FullConfig) ValidateConfig(c *Client, config string) (*http.Response, map[string]interface{}) { | ||
| base_uri := "configs/running-config" | ||
| url := "https://" + c.Hostname + "/rest/" + c.Version + "/" + base_uri | ||
| dryrun_url := url + "?dryrun=validate" | ||
|
|
||
| json_body := bytes.NewBufferString(config) | ||
|
|
||
| res := post(c, dryrun_url, json_body) | ||
|
|
||
| if res.Status != "200 OK" && res.Status != "202 Accepted" { | ||
| return res, nil | ||
| } | ||
|
|
||
| dryrun_url = url + "?dryrun" | ||
|
|
||
| res2, body := get(c, dryrun_url) | ||
|
|
||
| iterations := 10 | ||
|
|
||
| for iterations > 0 { | ||
| iterations -= 1 | ||
| if body["state"] == "success" || body["state"] == "error" { | ||
| break | ||
| } | ||
| time.Sleep(2 * time.Second) | ||
| res2, body = get(c, dryrun_url) | ||
| } | ||
|
|
||
| return res2, body | ||
| } | ||
|
|
||
| func (fc *FullConfig) ApplyConfig(c *Client, config string) (*http.Response, map[string]interface{}) { | ||
| base_uri := "configs/running-config" | ||
| url := "https://" + c.Hostname + "/rest/" + c.Version + "/" + base_uri | ||
| dryrun_url := url + "?dryrun=apply" | ||
|
|
||
| json_body := bytes.NewBufferString(config) | ||
|
|
||
| res := post(c, dryrun_url, json_body) | ||
|
|
||
| if res.Status != "200 OK" && res.Status != "202 Accepted" { | ||
| return res, nil | ||
| } | ||
|
|
||
| dryrun_url = url + "?dryrun" | ||
|
|
||
| res2, body := get(c, dryrun_url) | ||
|
|
||
| iterations := 10 | ||
|
|
||
| for iterations > 0 { | ||
| iterations -= 1 | ||
| if body["state"] == "success" || body["state"] == "error" { | ||
| break | ||
| } | ||
| time.Sleep(2 * time.Second) | ||
| res2, body = get(c, dryrun_url) | ||
| } | ||
|
|
||
| return res2, body | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,5 @@ | ||
| module github.com/aruba/aoscxgo | ||
|
|
||
| go 1.17 | ||
| go 1.18 | ||
|
|
||
| require golang.org/x/exp v0.0.0-20240103183307-be819d1f06fc // indirect |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.