Clean up main func

This commit is contained in:
2023-12-29 18:19:57 -06:00
parent 88e3cbec24
commit 1a6ef0230a
2 changed files with 7 additions and 20 deletions

View File

@@ -217,7 +217,6 @@ func GetDirectoryCached(letter string) ([]Entry, error) {
// Check if cached // Check if cached
var entries []Entry var entries []Entry
err := db.View(func(txn *badger.Txn) error { err := db.View(func(txn *badger.Txn) error {
log.Debug().Str("key", key).Msg("Accessing Directory Cache")
directoryItem, err := txn.Get([]byte(key)) directoryItem, err := txn.Get([]byte(key))
// Check if key was found // Check if key was found

26
main.go
View File

@@ -6,7 +6,6 @@ import (
"net/http/cookiejar" "net/http/cookiejar"
"net/url" "net/url"
"os" "os"
"os/signal"
badger "github.com/dgraph-io/badger/v4" badger "github.com/dgraph-io/badger/v4"
"github.com/joho/godotenv" "github.com/joho/godotenv"
@@ -102,21 +101,24 @@ func LoadCookies() {
} }
func main() { func main() {
stop := make(chan os.Signal, 1) // stop := make(chan os.Signal, 1)
signal.Notify(stop, os.Interrupt) // signal.Notify(stop, os.Interrupt)
// Load .env // Load .env
godotenv.Load() godotenv.Load()
username := os.Getenv("UTSA_USERNAME") username := os.Getenv("UTSA_USERNAME")
password := os.Getenv("UTSA_PASSWORD") password := os.Getenv("UTSA_PASSWORD")
defer db.Close()
defer SaveCookies()
// Check if logged in
log.Debug().Msg("Checking Login State") log.Debug().Msg("Checking Login State")
loggedIn, err := CheckLoggedIn() loggedIn, err := CheckLoggedIn()
if err != nil { if err != nil {
log.Fatal().Err(err).Msg("Failed to check login state") log.Fatal().Err(err).Msg("Failed to check login state")
} }
// Login if required
if !loggedIn { if !loggedIn {
log.Info().Str("username", username).Msg("Attempting Login") log.Info().Str("username", username).Msg("Attempting Login")
err := Login(username, password) err := Login(username, password)
@@ -124,7 +126,7 @@ func main() {
log.Fatal().Err(err).Msg("Failed to login") log.Fatal().Err(err).Msg("Failed to login")
} }
} else { } else {
log.Info().Msg("Login not required") log.Info().Msg("Login Not Required")
} }
// Get the directory // Get the directory
@@ -133,18 +135,4 @@ func main() {
log.Fatal().Err(err).Msg("Failed to get directory") log.Fatal().Err(err).Msg("Failed to get directory")
} }
log.Info().Int("count", len(directory)).Msg("Directory Loaded") log.Info().Int("count", len(directory)).Msg("Directory Loaded")
defer db.Close()
defer SaveCookies()
// email := strings.ToLower(fmt.Sprintf("%s.%s@my.utsa.edu", fake.FirstName(), fake.LastName()))
// log.Debug().Str("email", email).Msg("Unsubscribing")
// conf, err := Unsubscribe(email)
// if err != nil {
// log.Panic().Err(err).Msg("Failed to Unsubscribe")
// }
// log.Info().Str("formId", conf.FormId).Str("followUpUrl", conf.FollowUpUrl).Str("deliveryType", conf.DeliveryType).Str("followUpStreamValue", conf.FollowUpStreamValue).Str("aliId", conf.AliId).Msg("Unsubscribed")
} }