83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/chromedp/chromedp"
|
|
)
|
|
|
|
const (
|
|
webUrl = "https://gisem.dei.estg.ipleiria.pt"
|
|
)
|
|
|
|
var waitTime int
|
|
|
|
func main() {
|
|
//* Parse the flags
|
|
wait, headless, err := parseFlags()
|
|
if err != nil {
|
|
log.Fatalf("Failed to parse flags: %v", err)
|
|
}
|
|
waitTime = wait
|
|
|
|
fmt.Println("Waiting time:", waitTime)
|
|
|
|
//* Welcome message
|
|
welcomeMessage()
|
|
|
|
creds, err := getCreds()
|
|
if err != nil {
|
|
// If the credentials file doesn't exist, prompt for credentials
|
|
if err.Error() == "credentials file not found: stat "+credFile+": no such file or directory" {
|
|
// The file doesn't exist, create it by prompting for credentials
|
|
creds, err = promptAndSaveCreds(credFile)
|
|
if err != nil {
|
|
log.Fatalf("Failed to get credentials: %v", err)
|
|
}
|
|
} else {
|
|
log.Fatalf("Failed to get credentials: %v", err)
|
|
}
|
|
}
|
|
|
|
opts := append(chromedp.DefaultExecAllocatorOptions[:],
|
|
chromedp.Flag("headless", headless), // Disable headless mode
|
|
chromedp.Flag("disable-gpu", headless), // Enable GPU
|
|
chromedp.Flag("start-maximized", !headless), // Start maximized
|
|
)
|
|
|
|
allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
|
|
defer cancel()
|
|
|
|
ctx, cancel := chromedp.NewContext(allocCtx)
|
|
defer cancel()
|
|
|
|
//* Navigate to the website
|
|
if err := chromedp.Run(ctx, chromedp.Navigate(webUrl)); err != nil {
|
|
log.Fatalf("Failed to navigate: %v", err)
|
|
}
|
|
|
|
//* Check if there's a login button
|
|
var isVisible bool
|
|
if err := chromedp.Run(ctx, chromedp.Tasks{
|
|
chromedp.WaitVisible(loginButton),
|
|
chromedp.Evaluate(fmt.Sprintf("document.querySelector('%s').offsetParent !== null", loginButton), &isVisible),
|
|
}); err != nil {
|
|
log.Fatalf("Failed to evaluate: %v", err)
|
|
}
|
|
|
|
if isVisible {
|
|
log.Println("Logging in...")
|
|
if err := login(ctx, creds); err != nil {
|
|
log.Fatalf("Failed to login: %v", err)
|
|
}
|
|
time.Sleep(1 * time.Second)
|
|
log.Println("Waiting for activation...")
|
|
if err := marcarPresenca(ctx); err != nil {
|
|
log.Fatalf("Failed to mark presence: %v", err)
|
|
}
|
|
}
|
|
}
|