wneessen/apg-go
1
1
Fork 0
mirror of https://github.com/wneessen/apg-go.git synced 2024-11-22 13:50:49 +01:00
Code Issues Projects Releases Packages Wiki Activity

Reduce complexity in main() function

Browse source
This commit is contained in:
Winni Neessen 2024-03-14 22:45:20 +01:00
parent f153754dc4
commit cc45ec1119
Signed by: wneessen
GPG key ID: 5F3AF39B820C119D
1 changed files with 42 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
cmd/apg/apg.go
View file

@ -65,26 +65,8 @@ func main() {
} }
// Old style character modes // Old style character modes
if humanReadable { configOldStyle(config, humanReadable, lowerCase, upperCase, numeric,
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeHumanReadable) special, complexPass)
}
if lowerCase {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeLowerCase)
}
if upperCase {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeUpperCase)
}
if numeric {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeNumeric)
}
if special {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeSpecial)
}
if complexPass {
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeLowerCase|apg.ModeNumeric|
apg.ModeSpecial|apg.ModeUpperCase)
config.Mode = apg.MaskClearMode(config.Mode, apg.ModeHumanReadable)
}
// New style character modes (has higher priority than the old style modes) // New style character modes (has higher priority than the old style modes)
if modeString != "" { if modeString != "" {
@ -93,6 +75,21 @@ func main() {
// For the "minimum amount of" modes we need to imply at the type // For the "minimum amount of" modes we need to imply at the type
// of character mode is set // of character mode is set
configMinRequirement(config)
// Check if algorithm is supported
config.Algorithm = apg.IntToAlgo(algorithm)
if config.Algorithm == apg.AlgoUnsupported {
_, _ = fmt.Fprintf(os.Stderr, "unsupported algorithm value: %d\n", algorithm)
os.Exit(1)
}
// Generate the password based on the given flags and print it to stdout
generate(config)
}
// configMinRequirement configures the "minimum amount" feature
func configMinRequirement(config *apg.Config) {
if config.MinLowerCase > 0 { if config.MinLowerCase > 0 {
if float64(config.MinLength)/2 < float64(config.MinNumeric) { if float64(config.MinLength)/2 < float64(config.MinNumeric) {
_, _ = os.Stderr.WriteString(MinimumAmountTooHigh) _, _ = os.Stderr.WriteString(MinimumAmountTooHigh)
@ -117,15 +114,34 @@ func main() {
} }
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeUpperCase) config.Mode = apg.MaskSetMode(config.Mode, apg.ModeUpperCase)
} }
// Check if algorithm is supported
config.Algorithm = apg.IntToAlgo(algorithm)
if config.Algorithm == apg.AlgoUnsupported {
_, _ = fmt.Fprintf(os.Stderr, "unsupported algorithm value: %d\n", algorithm)
os.Exit(1)
} }
// Generate the password based on the given flags // configOldStyle configures the old style character modes
func configOldStyle(config *apg.Config, humanReadable, lowerCase, upperCase,
numeric, special, complexPass bool) {
if humanReadable {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeHumanReadable)
}
if lowerCase {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeLowerCase)
}
if upperCase {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeUpperCase)
}
if numeric {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeNumeric)
}
if special {
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeSpecial)
}
if complexPass {
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeLowerCase|apg.ModeNumeric|
apg.ModeSpecial|apg.ModeUpperCase)
config.Mode = apg.MaskClearMode(config.Mode, apg.ModeHumanReadable)
}
}
func generate(config *apg.Config) {
generator := apg.New(config) generator := apg.New(config)
for i := int64(0); i < config.NumberPass; i++ { for i := int64(0); i < config.NumberPass; i++ {
password, err := generator.Generate() password, err := generator.Generate()