mirror of
https://github.com/davidallendj/magellan.git
synced 2025-12-20 11:37:01 -07:00
* feat: initial implementation of command split * feat: update collect and new send cmd * chore: cleanup unused code * chore: refactored getting username * chore: more refactoring and cleanup * feat: update send cmd implementation * chore: changed/updated example config * chore: made cmd more consistent and added formatting * refactor: removed --host flag from scan * chore: cleaned up and fixed issue with client * chore: cleaned up CLI flags in collect cmd * feat: updated crawl to include managers and output YAML optionally * refactor: updated and improved send implementation * refactor: minor improvements * refactor: added util func to check for empty slices * fix: issue with reading from stdin * refactor: added scheme trimming function for URIs * refactor: changed host arg back to positional * refactor: removed unused vars and added --output-dir flag * fix: make -f for secrets persistent * refactor: removed --host flag and request in collect * refactor: changed --output flag to --output-file * fix: updated flags for collect * fix: typo in crawler error * fix: dir being created when outputDir not set * fix: reading stdin and data args * fix: made output using -v and -o consistent * readme: added info about command split * updated changelog adding missing version entries * chore: updated example to use host as positional arg * fix: issue with reading --data arg * fix: remove unused import from collect pkg Signed-off-by: Devon Bautista <devonb@lanl.gov> --------- Signed-off-by: David Allen <16520934+davidallendj@users.noreply.github.com> Signed-off-by: Devon Bautista <devonb@lanl.gov> Co-authored-by: Devon Bautista <devonb@lanl.gov>
36 lines
675 B
Go
36 lines
675 B
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"os/user"
|
|
"time"
|
|
)
|
|
|
|
// CheckUntil regularly check a predicate until it's true or time out is reached.
|
|
func CheckUntil(interval time.Duration, timeout time.Duration, predicate func() (bool, error)) error {
|
|
timeoutCh := time.After(timeout)
|
|
|
|
for {
|
|
select {
|
|
case <-time.After(interval):
|
|
predTrue, err := predicate()
|
|
if predTrue {
|
|
return nil
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case <-timeoutCh:
|
|
return fmt.Errorf("timeout of %ds reached", int64(timeout/time.Second))
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetCurrentUsername() string {
|
|
currentUser, _ := user.Current()
|
|
if currentUser != nil {
|
|
return currentUser.Username
|
|
}
|
|
return ""
|
|
}
|