Merge pull request #2 from PhilippePitzClairoux/featrure/add-autofill…

…-username-password

Featrure/add autofill username password

OpenConnectSSO.go

@@ -13,7 +13,9 @@ import (
 )
 
 // flags
-var server = flag.String("server", "", "server to connect to via openconnect")
+var server = flag.String("server", "", "Server to connect to via openconnect")
+var username = flag.String("username", "", "Username to inject in login form")
+var password = flag.String("password", "", "Password to inject in login form")
 
 func main() {
 	flag.Parse()
@@ -24,13 +26,17 @@ func main() {
 	samlAuth := internal.AuthenticationInit(client, targetUrl)
 	ctx, closeBrowser := internal.CreateBrowserContext()
 
+	// generate tasks
+	tasks := generateDefaultBrowserTasks(samlAuth)
+
+	// close browser at the end - no matter what happens
+	defer closeBrowser()
+
 	log.Println("Starting goroutine that searches for authentication cookie ", samlAuth.Auth.SsoV2TokenCookieName)
 	go internal.BrowserCookieFinder(ctx, cookieFound, samlAuth.Auth.SsoV2TokenCookieName)
 
 	log.Println("Open browser and navigate to SSO login page : ", samlAuth.Auth.SsoV2Login)
-	err := chromedp.Run(ctx, chromedp.Tasks{
-		chromedp.Navigate(samlAuth.Auth.SsoV2Login),
-	})
+	err := chromedp.Run(ctx, tasks)
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -39,6 +45,27 @@ func main() {
 	startVpnOnLoginCookie(cookieFound, client, samlAuth, targetUrl, closeBrowser)
 }
 
+func generateDefaultBrowserTasks(samlAuth *internal.AuthenticationInitExpectedResponse) chromedp.Tasks {
+	var tasks chromedp.Tasks
+
+	// create list of tasks to be executed by browser
+	tasks = append(tasks, chromedp.Navigate(samlAuth.Auth.SsoV2Login))
+	addAutofillTaskOnValue(&tasks, *password, "#passwordInput")
+	addAutofillTaskOnValue(&tasks, *username, "#userNameInput")
+
+	return tasks
+}
+
+func addAutofillTaskOnValue(actions *chromedp.Tasks, value, selector string) {
+	if value != "" {
+		*actions = append(
+			*actions,
+			chromedp.WaitVisible(selector, chromedp.ByID),
+			chromedp.SendKeys(selector, value, chromedp.ByID),
+		)
+	}
+}
+
 // startVpnOnLoginCookie waits to get a cookie from the authenticationCookies channel before confirming
 // the authentication process (to get token/cert) and then starting openconnect
 func startVpnOnLoginCookie(authenticationCookies chan string, client *http.Client, auth *internal.AuthenticationInitExpectedResponse, targetUrl string, closeBrowser context.CancelFunc) {