-
Notifications
You must be signed in to change notification settings - Fork 60
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add support for aws managed grafana (#96)
This adds a new login method for AWS Managed Grafana Instances. If the user account has MFA enabled, it waits for the user to input the OTP and then continues with the normal flow of waiting for a non existing element to be visible to exit.
- Loading branch information
Showing
5 changed files
with
85 additions
and
4 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
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,69 @@ | ||
| package kiosk | ||
|
|
||
| import ( | ||
| "context" | ||
| "github.com/chromedp/chromedp" | ||
| "github.com/chromedp/chromedp/kb" | ||
| "log" | ||
| "os" | ||
| "time" | ||
| ) | ||
|
|
||
| func GrafanaKioskAWSLogin(cfg *Config) { | ||
| dir, err := os.MkdirTemp(os.TempDir(), "chromedp-kiosk") | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| log.Println("Using temp dir:", dir) | ||
| defer os.RemoveAll(dir) | ||
|
|
||
| opts := generateExecutorOptions(dir, cfg.General.WindowPosition, cfg.Target.IgnoreCertificateErrors) | ||
|
|
||
| allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...) | ||
| defer cancel() | ||
|
|
||
| // also set up a custom logger | ||
| taskCtx, cancel := chromedp.NewContext(allocCtx, chromedp.WithLogf(log.Printf)) | ||
| defer cancel() | ||
|
|
||
| listenChromeEvents(taskCtx, targetCrashed) | ||
|
|
||
| // ensure that the browser process is started | ||
| if err := chromedp.Run(taskCtx); err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| var generatedURL = GenerateURL(cfg.Target.URL, cfg.General.Mode, cfg.General.AutoFit, cfg.Target.IsPlayList) | ||
|
|
||
| log.Println("Navigating to ", generatedURL) | ||
|
|
||
| // Give browser time to load next page (this can be prone to failure, explore different options vs sleeping) | ||
| time.Sleep(2000 * time.Millisecond) | ||
|
|
||
| if err := chromedp.Run(taskCtx, | ||
| chromedp.Navigate(generatedURL), | ||
| chromedp.WaitVisible(`//a[contains(@href,'login/sso')]`, chromedp.BySearch), | ||
| chromedp.Click(`//a[contains(@href,'login/sso')]`, chromedp.BySearch), | ||
| chromedp.WaitVisible(`input#awsui-input-0`, chromedp.BySearch), | ||
| chromedp.SendKeys(`input#awsui-input-0`, cfg.Target.Username+kb.Enter, chromedp.BySearch), | ||
| chromedp.WaitVisible(`input#awsui-input-1`, chromedp.BySearch), | ||
| chromedp.SendKeys(`input#awsui-input-1`, cfg.Target.Password+kb.Enter, chromedp.BySearch), | ||
| ); err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| if cfg.Target.UseMFA { | ||
| if err := chromedp.Run(taskCtx, | ||
| chromedp.WaitNotVisible(`input#awsui-input-2`, chromedp.BySearch), | ||
| ); err != nil { | ||
| panic(err) | ||
| } | ||
| } | ||
|
|
||
| if err := chromedp.Run(taskCtx, | ||
| chromedp.WaitVisible(`notinputPassword`, chromedp.ByID), | ||
| ); err != nil { | ||
| panic(err) | ||
| } | ||
| } |
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