-
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.
restructured files and handled index out of bounds errors in flag han…
…dling
- Loading branch information
Showing
5 changed files
with
200 additions
and
178 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,136 +1,9 @@ | ||
| package cam | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "github.com/cloudfoundry/cli/cf/terminal" | ||
| "github.com/cloudfoundry/cli/plugin" | ||
| "github.com/ibmjstart/bluemix-cloudant-replicator/utils" | ||
| "net/http" | ||
| "regexp" | ||
| "strings" | ||
| "time" | ||
| ) | ||
|
|
||
| type CloudantAccount struct { | ||
| Endpoint string | ||
| Username string | ||
| Password string | ||
| Url string | ||
| Cookie string | ||
| } | ||
|
|
||
| type CreateAccountResponse struct { | ||
| account CloudantAccount | ||
| err error | ||
| } | ||
|
|
||
| func init() { | ||
| terminal.InitColorSupport() | ||
| } | ||
|
|
||
| func createAccount(cliConnection plugin.CliConnection, httpClient *http.Client, env []string, endpoint string) CreateAccountResponse { | ||
| account, err := parseCreds(env) | ||
| if err != nil { | ||
| err = errors.New("Problem finding Cloudant credentials for app at '" + terminal.ColorizeBold(endpoint, 36) + | ||
| "'.\nMake sure that there is a valid 'cloudantNoSQLDB' service bound to your app.\nContinuing on with other regions.\n") | ||
| return CreateAccountResponse{account: account, err: err} | ||
| } | ||
| account.Endpoint = endpoint | ||
| account.Cookie = getCookie(account, httpClient) | ||
| return CreateAccountResponse{account: account, err: nil} | ||
| } | ||
|
|
||
| /* | ||
| * Cycles through all endpoints and retrieves the Cloudant | ||
| * credentials for the specified app in each region. | ||
| */ | ||
| func GetCloudantAccounts(cliConnection plugin.CliConnection, httpClient *http.Client, ENDPOINTS []string, appname string, password string) ([]CloudantAccount, error) { | ||
| var cloudantAccounts []CloudantAccount | ||
| _, username, org, space := bcr_utils.GetCurrentTarget(cliConnection) | ||
| ch := make(chan CreateAccountResponse) | ||
| for i := 0; i < len(ENDPOINTS); i++ { | ||
| env, err := getAppEnv(cliConnection, username, password, org, ENDPOINTS[i], appname, space) | ||
| go func(cliConnection plugin.CliConnection, httpClient *http.Client, env []string, endpoint string, envErr error) { | ||
| if envErr == nil { | ||
| ch <- createAccount(cliConnection, httpClient, env, endpoint) | ||
| } else { | ||
| ch <- CreateAccountResponse{account: CloudantAccount{}, err: err} | ||
| } | ||
| }(cliConnection, httpClient, env, ENDPOINTS[i], err) | ||
| } | ||
| responses := 0 | ||
| for { | ||
| select { | ||
| case r := <-ch: | ||
| responses += 1 | ||
| bcr_utils.CheckErrorNonFatal(r.err) | ||
| if r.err == nil { | ||
| cloudantAccounts = append(cloudantAccounts, r.account) | ||
| } | ||
| case <-time.After(50 * time.Millisecond): | ||
| continue | ||
| } | ||
| if responses == len(ENDPOINTS) { | ||
| break | ||
| } | ||
| } | ||
| close(ch) | ||
| return cloudantAccounts, nil | ||
| } | ||
|
|
||
| func parseCreds(env []string) (CloudantAccount, error) { | ||
| var account CloudantAccount | ||
| for i := 0; i < len(env); i++ { | ||
| if strings.Index(env[i], "cloudantNoSQLDB") != -1 { | ||
| user_reg, _ := regexp.Compile("\"username\": \"([\x00-\x7F]+)\"") | ||
| pass_reg, _ := regexp.Compile("\"password\": \"([\x00-\x7F]+)\"") | ||
| url_reg, _ := regexp.Compile("\"url\": \"([\x00-\x7F]+)\"") | ||
| account.Username = strings.Split(user_reg.FindString(env[i]), "\"")[3] | ||
| account.Password = strings.Split(pass_reg.FindString(env[i]), "\"")[3] | ||
| account.Url = strings.Split(url_reg.FindString(env[i]), "\"")[3] | ||
| break | ||
| } | ||
| } | ||
| if account.Username == "" || account.Password == "" || account.Url == "" { | ||
| return account, errors.New("Cloudant credentials incomplete\n") | ||
| } | ||
| return account, nil | ||
| } | ||
|
|
||
| /* | ||
| * Returns the result of "cf env APP" | ||
| */ | ||
| func getAppEnv(cliConnection plugin.CliConnection, username string, password string, org string, endpoint string, appname string, space string) ([]string, error) { | ||
| fmt.Println("Retrieving CloudantNoSQLDB credentials for '" + terminal.ColorizeBold(appname, 36) + "' in '" + terminal.ColorizeBold(endpoint, 36) + "'\n") | ||
| startingEndpoint, _ := cliConnection.ApiEndpoint() | ||
| if startingEndpoint != endpoint { | ||
| _, err := cliConnection.CliCommandWithoutTerminalOutput("login", "-u", username, "-p", password, "-o", org, "-a", endpoint, "-s", space) | ||
| if err != nil { | ||
| fmt.Println("Unable to log in to org '" + terminal.ColorizeBold(org, 36) + "' and/or space '" + terminal.ColorizeBold(space, 36) + "'\n") | ||
| _, err = cliConnection.CliCommand("login", "-u", username, "-p", password, "-a", endpoint) | ||
| bcr_utils.CheckErrorFatal(err) | ||
| } | ||
| } | ||
| output, err := cliConnection.CliCommandWithoutTerminalOutput("env", appname) | ||
| if err != nil { | ||
| return output, errors.New("No '" + terminal.ColorizeBold(appname, 36) + "' in '" + terminal.ColorizeBold(endpoint, 36) + | ||
| "'.\nContinuing on with other regions.\n") | ||
| } | ||
| return output, err | ||
| } | ||
|
|
||
| /* | ||
| * Gets cookie for a specified CloudantAccount. This cookie is | ||
| * used to authenticate all necessary api calls. | ||
| */ | ||
| func getCookie(account CloudantAccount, httpClient *http.Client) string { | ||
| url := "https://" + account.Username + ".cloudant.com/_session" | ||
| body := "name=" + account.Username + "&password=" + account.Password | ||
| headers := map[string]string{"Content-Type": "application/x-www-form-urlencoded"} | ||
| resp, err := bcr_utils.MakeRequest(httpClient, "POST", url, body, headers) | ||
| bcr_utils.CheckErrorFatal(err) | ||
| cookie := resp.Header.Get("Set-Cookie") | ||
| resp.Body.Close() | ||
| return cookie | ||
| } |
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,129 @@ | ||
| package ca | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "github.com/cloudfoundry/cli/cf/terminal" | ||
| "github.com/cloudfoundry/cli/plugin" | ||
| "github.com/ibmjstart/bluemix-cloudant-replicator/CloudantAccountModel" | ||
| "github.com/ibmjstart/bluemix-cloudant-replicator/utils" | ||
| "net/http" | ||
| "regexp" | ||
| "strings" | ||
| "time" | ||
| ) | ||
|
|
||
| type CreateAccountResponse struct { | ||
| account cam.CloudantAccount | ||
| err error | ||
| } | ||
|
|
||
| func init() { | ||
| terminal.InitColorSupport() | ||
| } | ||
|
|
||
| func createAccount(cliConnection plugin.CliConnection, httpClient *http.Client, env []string, endpoint string) CreateAccountResponse { | ||
| account, err := parseCreds(env) | ||
| if err != nil { | ||
| err = errors.New("Problem finding Cloudant credentials for app at '" + terminal.ColorizeBold(endpoint, 36) + | ||
| "'.\nMake sure that there is a valid 'cloudantNoSQLDB' service bound to your app.\nContinuing on with other regions.\n") | ||
| return CreateAccountResponse{account: account, err: err} | ||
| } | ||
| account.Endpoint = endpoint | ||
| account.Cookie = getCookie(account, httpClient) | ||
| return CreateAccountResponse{account: account, err: nil} | ||
| } | ||
|
|
||
| /* | ||
| * Cycles through all endpoints and retrieves the Cloudant | ||
| * credentials for the specified app in each region. | ||
| */ | ||
| func GetCloudantAccounts(cliConnection plugin.CliConnection, httpClient *http.Client, ENDPOINTS []string, appname string, password string) ([]cam.CloudantAccount, error) { | ||
| var cloudantAccounts []cam.CloudantAccount | ||
| _, username, org, space := bcr_utils.GetCurrentTarget(cliConnection) | ||
| ch := make(chan CreateAccountResponse) | ||
| for i := 0; i < len(ENDPOINTS); i++ { | ||
| env, err := getAppEnv(cliConnection, username, password, org, ENDPOINTS[i], appname, space) | ||
| go func(cliConnection plugin.CliConnection, httpClient *http.Client, env []string, endpoint string, envErr error) { | ||
| if envErr == nil { | ||
| ch <- createAccount(cliConnection, httpClient, env, endpoint) | ||
| } else { | ||
| ch <- CreateAccountResponse{account: cam.CloudantAccount{}, err: err} | ||
| } | ||
| }(cliConnection, httpClient, env, ENDPOINTS[i], err) | ||
| } | ||
| responses := 0 | ||
| for { | ||
| select { | ||
| case r := <-ch: | ||
| responses += 1 | ||
| bcr_utils.CheckErrorNonFatal(r.err) | ||
| if r.err == nil { | ||
| cloudantAccounts = append(cloudantAccounts, r.account) | ||
| } | ||
| case <-time.After(50 * time.Millisecond): | ||
| continue | ||
| } | ||
| if responses == len(ENDPOINTS) { | ||
| break | ||
| } | ||
| } | ||
| close(ch) | ||
| return cloudantAccounts, nil | ||
| } | ||
|
|
||
| func parseCreds(env []string) (cam.CloudantAccount, error) { | ||
| var account cam.CloudantAccount | ||
| for i := 0; i < len(env); i++ { | ||
| if strings.Index(env[i], "cloudantNoSQLDB") != -1 { | ||
| user_reg, _ := regexp.Compile("\"username\": \"([\x00-\x7F]+)\"") | ||
| pass_reg, _ := regexp.Compile("\"password\": \"([\x00-\x7F]+)\"") | ||
| url_reg, _ := regexp.Compile("\"url\": \"([\x00-\x7F]+)\"") | ||
| account.Username = strings.Split(user_reg.FindString(env[i]), "\"")[3] | ||
| account.Password = strings.Split(pass_reg.FindString(env[i]), "\"")[3] | ||
| account.Url = strings.Split(url_reg.FindString(env[i]), "\"")[3] | ||
| break | ||
| } | ||
| } | ||
| if account.Username == "" || account.Password == "" || account.Url == "" { | ||
| return account, errors.New("Cloudant credentials incomplete\n") | ||
| } | ||
| return account, nil | ||
| } | ||
|
|
||
| /* | ||
| * Returns the result of "cf env APP" | ||
| */ | ||
| func getAppEnv(cliConnection plugin.CliConnection, username string, password string, org string, endpoint string, appname string, space string) ([]string, error) { | ||
| fmt.Println("Retrieving CloudantNoSQLDB credentials for '" + terminal.ColorizeBold(appname, 36) + "' in '" + terminal.ColorizeBold(endpoint, 36) + "'\n") | ||
| startingEndpoint, _ := cliConnection.ApiEndpoint() | ||
| if startingEndpoint != endpoint { | ||
| _, err := cliConnection.CliCommandWithoutTerminalOutput("login", "-u", username, "-p", password, "-o", org, "-a", endpoint, "-s", space) | ||
| if err != nil { | ||
| fmt.Println("Unable to log in to org '" + terminal.ColorizeBold(org, 36) + "' and/or space '" + terminal.ColorizeBold(space, 36) + "'\n") | ||
| _, err = cliConnection.CliCommand("login", "-u", username, "-p", password, "-a", endpoint) | ||
| bcr_utils.CheckErrorFatal(err) | ||
| } | ||
| } | ||
| output, err := cliConnection.CliCommandWithoutTerminalOutput("env", appname) | ||
| if err != nil { | ||
| return output, errors.New("No '" + terminal.ColorizeBold(appname, 36) + "' in '" + terminal.ColorizeBold(endpoint, 36) + | ||
| "'.\nContinuing on with other regions.\n") | ||
| } | ||
| return output, err | ||
| } | ||
|
|
||
| /* | ||
| * Gets cookie for a specified CloudantAccount. This cookie is | ||
| * used to authenticate all necessary api calls. | ||
| */ | ||
| func getCookie(account cam.CloudantAccount, httpClient *http.Client) string { | ||
| url := "https://" + account.Username + ".cloudant.com/_session" | ||
| body := "name=" + account.Username + "&password=" + account.Password | ||
| headers := map[string]string{"Content-Type": "application/x-www-form-urlencoded"} | ||
| resp, err := bcr_utils.MakeRequest(httpClient, "POST", url, body, headers) | ||
| bcr_utils.CheckErrorFatal(err) | ||
| cookie := resp.Header.Get("Set-Cookie") | ||
| resp.Body.Close() | ||
| return cookie | ||
| } |
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.