FernandoJVideira/gisem_presencas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1 from FernandoJVideira/helper-flags

Browse Source
This commit is contained in:
Afonso Sousa
2025-03-13 23:28:36 +00:00
committed by GitHub
parent c1dd8c19b1 832ee30b86
commit efb5ae4e45
3 changed files with 126 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4
View File
@@ -87,6 +87,10 @@ gisem_presencas.exe -w value
## Flags
- `-w` or `-wait`: Time to wait in between refreshes (default is 10 seconds).
- `-l` or `-log`: Send the output to a file.
- `-m` or `-maxtime`: Sets a maxmium time for the program to try set Attendance (default is -1 (forever)).
- `-r` or `-replace`: Forces the program into replacing the existing saved credentials.
- `-v` or `-verbose`: Shows addicional info.
## Project Structure
main.go
+72 -4
View File
@@ -4,6 +4,7 @@ import (
"context"
"fmt"
"log"
"os" // added for logging to file
"time"
"github.com/chromedp/chromedp"
@@ -14,21 +15,75 @@ const (
)
var waitTime int
var isVerbose bool
var maxTime int
var logFile string
func main() {
//* Parse the flags
wait, headless, err := parseFlags()
wait, verbose, headless, logFile, timeout, replacePass, err := parseFlags()
if err != nil {
log.Fatalf("Failed to parse flags: %v", err)
}
waitTime = wait
isVerbose = verbose
maxTime = timeout
if logFile != "" {
file, err := os.OpenFile(logFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
if err != nil {
log.Fatalf("Failed to open log file: %v", err)
}
defer file.Close()
log.SetOutput(file)
}
fmt.Println("Waiting time:", waitTime)
if isVerbose {
fmt.Println("Waiting time:", waitTime)
fmt.Println("Verbose mode:", isVerbose)
fmt.Println("Headless mode:", headless)
if logFile == "" {
fmt.Println("Logging to file: disabled")
} else {
fmt.Println("Logging to file:", logFile)
}
if maxTime == -1 {
log.Println("Max time: disabled")
} else {
log.Println("Max time:", maxTime)
}
}
//* Welcome message
welcomeMessage()
if logFile != "" {
fmt.Println("Logging to file:", logFile)
fmt.Println("(No text will be shown because logging is enabled, dont worry, the program is still running)")
log.Print("Starting new program... (logging to file)\n\n")
isVerbose = true
if maxTime == -1 {
log.Println("Max time: disabled")
} else {
log.Println("Max time:", maxTime)
}
}
creds, err := getCreds()
if replacePass {
creds, err = promptAndSaveCreds(credFile)
if err != nil {
log.Fatalf("Failed to get credentials: %v", err)
}
if isVerbose {
log.Println("Credentials replaced")
}
}
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" {
@@ -59,6 +114,9 @@ func main() {
log.Fatalf("Failed to navigate: %v", err)
}
if isVerbose {
log.Println("Navigated to the website")
}
//* Check if there's a login button
var isVisible bool
if err := chromedp.Run(ctx, chromedp.Tasks{
@@ -68,15 +126,25 @@ func main() {
log.Fatalf("Failed to evaluate: %v", err)
}
if isVerbose {
log.Println("Login button visible:", isVisible)
}
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...")
log.Println("Logged in!")
if isVerbose {
log.Println("Waiting for activation...")
}
if err := marcarPresenca(ctx); err != nil {
log.Fatalf("Failed to mark presence: %v", err)
log.Println("Failed to mark presence: %v", err)
} else {
log.Println("Presence marked!")
}
}
log.Print("Program finished successfully! (logging to file) \n\n")
}
utils.go
+50 -18
View File
@@ -9,6 +9,7 @@ import (
"crypto/sha256"
"encoding/hex"
"encoding/json"
"errors"
"flag"
"fmt"
"io"
@@ -106,24 +107,37 @@ func decrypt(encryptedHex, passphrase string) (string, error) {
return string(plaintext), nil
}
func parseFlags() (int, bool, error) {
func parseFlags() (int, bool, bool, string, int, bool, error) {
//* Set the flags
wait := flag.Int("w", 10, "Time to wait in between refreshes")
flag.IntVar(wait, "wait", 10, "Time to wait in between refreshes")
wait := flag.Int("w", 10, "\tTime to wait in between refreshes")
flag.IntVar(wait, "wait", 10, "\tTime to wait in between refreshes")
// Verbose mode
isVerbose := flag.Bool("v", false, "\tEnable verbose mode")
flag.BoolVar(isVerbose, "verbose", false, "\tEnable verbose mode")
// By default, run in headless mode
headless := flag.Bool("h", true, "Run in headless mode")
flag.BoolVar(headless, "headless", true, "Run in headless mode")
headless := flag.Bool("h", true, "\tRun in headless mode")
flag.BoolVar(headless, "headless", true, "\tRun in headless mode")
// Log to file
logFile := flag.String("l", "", "\tLog output to file")
flag.StringVar(logFile, "log", "", "\tLog output to file")
// Maximum time to run the program (minutes)
maxTime := flag.Int("m", -1, "\tMaximum time to run the program (minutes)")
flag.IntVar(maxTime, "maxtime", -1, "\tMaximum time to run the program (minutes)")
replacePass := flag.Bool("r", false, "\tReplace saved credentials")
flag.BoolVar(replacePass, "replace", false, "\tReplace saved credentials")
flag.Parse()
if *wait < 0 {
return 0, false, fmt.Errorf("invalid wait time: %d", *wait)
return 0, false, false, "", -1, false, fmt.Errorf("invalid wait time: %d", *wait)
}
fmt.Println(*headless)
return *wait, *headless, nil
return *wait, *isVerbose, *headless, *logFile, *maxTime, *replacePass, nil
}
func welcomeMessage() {
@@ -253,7 +267,7 @@ func saveCreds(filePath, username, password string) (Credentials, error) {
return Credentials{}, fmt.Errorf("could not write credentials to file: %v", err)
}
fmt.Println("Credentials saved successfully!")
log.Println("Credentials saved successfully!")
return creds, nil
}
@@ -279,6 +293,12 @@ func login(ctx context.Context, creds Credentials) error {
func marcarPresenca(ctx context.Context) error {
var attempt int
startTime := time.Now()
if isVerbose || maxTime != -1 {
log.Printf("Starting attendance registration at %s\n", startTime.Format(time.RFC1123))
}
for attempt = 1; attempt <= maxRetries; attempt++ {
if err := chromedp.Run(ctx,
chromedp.ActionFunc(func(ctx context.Context) error {
@@ -286,23 +306,23 @@ func marcarPresenca(ctx context.Context) error {
waitCtx, cancel := context.WithTimeout(ctx, time.Duration(waitTime)*time.Second)
defer cancel()
//Try to wait for the element
err := chromedp.Run(waitCtx,
chromedp.WaitVisible(menuPresencas))
if err == nil {
fmt.Println("Navigation menu is visible!")
if isVerbose && logFile != "" {
log.Println("Navigation menu is visible!")
}
}
chromedp.Run(waitCtx, chromedp.Click(menuPresencas))
fmt.Println("Navigation menu clicked!")
var currentURL string
if err := chromedp.Location(&currentURL).Do(ctx); err != nil {
return err
}
if strings.Contains(currentURL, "obterAulasMarcarPresenca") {
fmt.Println("Attendance page loaded!")
fmt.Println("Registering attendance...")
log.Println("Attendance page loaded :", time.Now().Format(time.RFC1123))
log.Println("Registering attendance... : ", time.Now().Format(time.RFC1123))
break
}
@@ -310,6 +330,14 @@ func marcarPresenca(ctx context.Context) error {
return err
}
time.Sleep(time.Duration(waitTime) * time.Second)
if maxTime != -1 {
elapsedMinutes := time.Since(startTime).Minutes()
if int(elapsedMinutes) >= maxTime {
log.Println("Maximum time reached, attendance not registered", time.Now().Format(time.RFC1123))
return errors.New("maximum time reached")
}
}
}
return nil
}),
@@ -327,11 +355,11 @@ func marcarPresenca(ctx context.Context) error {
select {
case alertMessage := <-ch:
if strings.Contains(alertMessage, "sucesso") {
fmt.Println("Attendance registered successfully!")
log.Println("Attendance registered successfully! : ", time.Now().Format(time.RFC1123))
} else if strings.Contains(alertMessage, "anteriormente") {
fmt.Println("Attendance previously registered!")
log.Println("Attendance previously registered! : ", time.Now().Format(time.RFC1123))
} else {
fmt.Println("Error registering attendance: ", alertMessage)
log.Println("Error registering attendance: ", alertMessage)
}
case <-time.After(time.Duration(waitTime) * time.Second):
return fmt.Errorf("timeout waiting for alert")
@@ -339,6 +367,10 @@ func marcarPresenca(ctx context.Context) error {
return nil
}),
); err != nil {
// Check if the error was due to maximum time reached
if err.Error() == "maximum time reached" {
return err
}
log.Printf("Attempt %d: Failed to register attendance: %v", attempt, err)
time.Sleep(5 * time.Second)
continue