277 lines
7.5 KiB
Go
277 lines
7.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/chromedp/chromedp"
|
|
)
|
|
|
|
// Standalone script to extract JWT token from Kosmi using browser automation
|
|
// Usage: go run get-token.go --email "email@email.com" --password "password"
|
|
|
|
var (
|
|
emailFlag = flag.String("email", "", "Email for Kosmi login")
|
|
passwordFlag = flag.String("password", "", "Password for Kosmi login")
|
|
headlessFlag = flag.Bool("headless", true, "Run browser in headless mode")
|
|
verboseFlag = flag.Bool("verbose", false, "Enable verbose logging")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if *emailFlag == "" || *passwordFlag == "" {
|
|
log.Fatal("Error: --email and --password are required")
|
|
}
|
|
|
|
// Set up logging
|
|
if !*verboseFlag {
|
|
log.SetOutput(os.Stderr)
|
|
}
|
|
|
|
log.Println("Starting browser automation to extract JWT token...")
|
|
log.Printf("Email: %s", *emailFlag)
|
|
log.Printf("Headless mode: %v", *headlessFlag)
|
|
|
|
token, err := extractToken(*emailFlag, *passwordFlag, *headlessFlag)
|
|
if err != nil {
|
|
log.Fatalf("Failed to extract token: %v", err)
|
|
}
|
|
|
|
// Output token to stdout (so it can be captured)
|
|
fmt.Println(token)
|
|
|
|
if *verboseFlag {
|
|
log.Printf("✓ Successfully extracted token (length: %d)", len(token))
|
|
}
|
|
}
|
|
|
|
func extractToken(email, password string, headless bool) (string, error) {
|
|
// Create context with timeout
|
|
ctx, cancel := context.WithTimeout(context.Background(), 90*time.Second)
|
|
defer cancel()
|
|
|
|
// Set up chromedp options
|
|
opts := []chromedp.ExecAllocatorOption{
|
|
chromedp.NoFirstRun,
|
|
chromedp.NoDefaultBrowserCheck,
|
|
chromedp.DisableGPU,
|
|
chromedp.NoSandbox,
|
|
}
|
|
|
|
if headless {
|
|
opts = append(opts, chromedp.Headless)
|
|
}
|
|
|
|
// Create allocator context
|
|
allocCtx, cancel := chromedp.NewExecAllocator(ctx, opts...)
|
|
defer cancel()
|
|
|
|
// Create browser context - suppress cookie errors unless verbose
|
|
logFunc := func(string, ...interface{}) {}
|
|
if *verboseFlag {
|
|
logFunc = log.Printf
|
|
}
|
|
ctx, cancel = chromedp.NewContext(allocCtx, chromedp.WithLogf(logFunc))
|
|
defer cancel()
|
|
|
|
var token string
|
|
|
|
// Run the automation tasks
|
|
err := chromedp.Run(ctx,
|
|
// Step 1: Navigate to Kosmi
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Navigating to https://app.kosmi.io")
|
|
return chromedp.Navigate("https://app.kosmi.io").Do(ctx)
|
|
}),
|
|
|
|
// Step 2: Wait for page to load
|
|
chromedp.WaitReady("body"),
|
|
chromedp.Sleep(3*time.Second),
|
|
|
|
// Step 3: Find and click Login button
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Looking for Login button...")
|
|
|
|
// Log all buttons we can see
|
|
if *verboseFlag {
|
|
var buttonTexts []string
|
|
chromedp.Evaluate(`
|
|
Array.from(document.querySelectorAll('button')).map(el => el.textContent.trim())
|
|
`, &buttonTexts).Do(ctx)
|
|
log.Printf(" Found buttons: %v", buttonTexts)
|
|
}
|
|
|
|
var found bool
|
|
if err := chromedp.Evaluate(`
|
|
(() => {
|
|
const buttons = Array.from(document.querySelectorAll('button'));
|
|
const btn = buttons.find(el => {
|
|
const text = el.textContent.trim();
|
|
return text === 'Login' || text === 'Log in';
|
|
});
|
|
if (btn) {
|
|
btn.click();
|
|
return true;
|
|
}
|
|
return false;
|
|
})()
|
|
`, &found).Do(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
if !found {
|
|
return fmt.Errorf("Login button not found")
|
|
}
|
|
|
|
log.Println("✓ Clicked Login button")
|
|
return nil
|
|
}),
|
|
|
|
// Step 4: Wait for login modal
|
|
chromedp.Sleep(3*time.Second),
|
|
|
|
// Step 5: Click "Login with Email" button
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Looking for 'Login with Email' button...")
|
|
|
|
if *verboseFlag {
|
|
var buttonTexts []string
|
|
chromedp.Evaluate(`
|
|
Array.from(document.querySelectorAll('button')).map(el => el.textContent.trim())
|
|
`, &buttonTexts).Do(ctx)
|
|
log.Printf(" Found buttons: %v", buttonTexts)
|
|
}
|
|
|
|
var found bool
|
|
if err := chromedp.Evaluate(`
|
|
(() => {
|
|
const btn = Array.from(document.querySelectorAll('button')).find(el =>
|
|
el.textContent.includes('Email')
|
|
);
|
|
if (btn) {
|
|
btn.click();
|
|
return true;
|
|
}
|
|
return false;
|
|
})()
|
|
`, &found).Do(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
if !found {
|
|
return fmt.Errorf("'Login with Email' button not found")
|
|
}
|
|
|
|
log.Println("✓ Clicked 'Login with Email' button")
|
|
return nil
|
|
}),
|
|
|
|
// Step 6: Wait for email form
|
|
chromedp.Sleep(3*time.Second),
|
|
chromedp.WaitVisible(`input[type="password"]`, chromedp.ByQuery),
|
|
|
|
// Step 7: Fill in email
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Filling email field...")
|
|
return nil
|
|
}),
|
|
chromedp.Click(`input[placeholder*="Email"], input[placeholder*="Username"]`, chromedp.ByQuery),
|
|
chromedp.Sleep(200*time.Millisecond),
|
|
chromedp.SendKeys(`input[placeholder*="Email"], input[placeholder*="Username"]`, email, chromedp.ByQuery),
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("✓ Email entered")
|
|
return nil
|
|
}),
|
|
|
|
chromedp.Sleep(500*time.Millisecond),
|
|
|
|
// Step 8: Fill in password
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Filling password field...")
|
|
return nil
|
|
}),
|
|
chromedp.Click(`input[type="password"]`, chromedp.ByQuery),
|
|
chromedp.Sleep(200*time.Millisecond),
|
|
chromedp.SendKeys(`input[type="password"]`, password+"\n", chromedp.ByQuery),
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("✓ Password entered and form submitted")
|
|
return nil
|
|
}),
|
|
|
|
chromedp.Sleep(500*time.Millisecond),
|
|
|
|
// Step 10: Wait for login to complete - check for modal to close
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
log.Println("→ Waiting for login to complete...")
|
|
|
|
// Wait for the login modal to disappear (indicates successful login)
|
|
maxAttempts := 30 // 15 seconds total
|
|
for i := 0; i < maxAttempts; i++ {
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
// Check if login modal is gone (successful login)
|
|
var modalGone bool
|
|
chromedp.Evaluate(`
|
|
(() => {
|
|
// Check if the email/password form is still visible
|
|
const emailInput = document.querySelector('input[placeholder*="Email"], input[placeholder*="Username"]');
|
|
const passwordInput = document.querySelector('input[type="password"]');
|
|
return !emailInput && !passwordInput;
|
|
})()
|
|
`, &modalGone).Do(ctx)
|
|
|
|
if modalGone {
|
|
log.Println("✓ Login modal closed - login successful")
|
|
// Modal is gone, wait a bit more for token to be set
|
|
time.Sleep(2 * time.Second)
|
|
chromedp.Evaluate(`localStorage.getItem('token')`, &token).Do(ctx)
|
|
if token != "" {
|
|
log.Println("✓ Authenticated token received")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// Check for error messages
|
|
var errorText string
|
|
chromedp.Evaluate(`
|
|
(() => {
|
|
const errorEl = document.querySelector('[role="alert"], .error, .error-message');
|
|
return errorEl ? errorEl.textContent.trim() : '';
|
|
})()
|
|
`, &errorText).Do(ctx)
|
|
|
|
if errorText != "" && errorText != "null" {
|
|
return fmt.Errorf("login failed: %s", errorText)
|
|
}
|
|
}
|
|
|
|
// Timeout - get whatever token is there
|
|
log.Println("⚠ Login timeout - extracting current token")
|
|
chromedp.Evaluate(`localStorage.getItem('token')`, &token).Do(ctx)
|
|
|
|
if token == "" {
|
|
return fmt.Errorf("login timeout: no token found")
|
|
}
|
|
|
|
log.Println("✓ Token extracted (may be anonymous)")
|
|
return nil
|
|
}),
|
|
)
|
|
|
|
if err != nil {
|
|
return "", fmt.Errorf("browser automation failed: %w", err)
|
|
}
|
|
|
|
if token == "" {
|
|
return "", fmt.Errorf("failed to extract token from localStorage")
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|