From 8a98b4940d4502ee5f49ae7240bc8d054280079f Mon Sep 17 00:00:00 2001
From: Brian Buller <brian@bullercodeworks.com>
Date: Thu, 20 Jun 2024 12:04:42 -0500
Subject: [PATCH] Working on v2 algorithm

---
 passage.go | 63 +++++++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 53 insertions(+), 10 deletions(-)

diff --git a/passage.go b/passage.go
index 1637336..25d5408 100644
--- a/passage.go
+++ b/passage.go
@@ -16,18 +16,20 @@ type Passage struct {
 	valids int
 	length int
 	pin    string
+
+	requireEachCharSet bool
 }
 
 func CreatePassage(pin string) *Passage {
 	p := new(Passage)
 	p.length = 20
-	p.valids = UseLower | UseUpper | UseNumber
+	p.valids = UseLower | UseUpper | UseNumber | UseSymbol
 	p.pin = pin
 	return p
 }
 
-func (p *Passage) SetPin(pin string) {
-	p.pin = pin
+func (p *Passage) SetRequireEachCharSet(r bool) {
+	p.requireEachCharSet = r
 }
 
 func (p *Passage) EnableChars(which int) {
@@ -42,34 +44,75 @@ func (p *Passage) SetLength(length int) {
 	p.length = length
 }
 
+func (p *Passage) GetLowers() string {
+	return "abcdefghijklmnopqrstuvwxyz"
+}
+func (p *Passage) GetUppers() string {
+	return "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+}
+func (p *Passage) GetNumbers() string {
+	return "1234567890"
+}
+func (p *Passage) GetSymbols() string {
+	return "~!@#$%^&*()[]{}|;:,<.>/?"
+}
+
 func (p *Passage) GetValidChars() string {
 	validChars := ""
 	if p.valids&UseLower == UseLower {
-		validChars += "abcdefghijklmnopqrstuvwxyz"
+		validChars += p.GetLowers()
 	}
 	if p.valids&UseUpper == UseUpper {
-		validChars += "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+		validChars += p.GetUppers()
 	}
 	if p.valids&UseNumber == UseNumber {
-		validChars += "1234567890"
+		validChars += p.GetNumbers()
 	}
 	if p.valids&UseSymbol == UseSymbol {
-		validChars += "~!@#$%^&*()[]{}|;:,<.>/?"
+		validChars += p.GetSymbols()
+	}
+	return validChars
+}
+func (p *Passage) GetValidCharSets() []string {
+	var validChars []string
+	if p.valids&UseLower == UseLower {
+		validChars = append(validChars, p.GetLowers())
+	}
+	if p.valids&UseUpper == UseUpper {
+		validChars = append(validChars, p.GetUppers())
+	}
+	if p.valids&UseNumber == UseNumber {
+		validChars = append(validChars, p.GetNumbers())
+	}
+	if p.valids&UseSymbol == UseSymbol {
+		validChars = append(validChars, p.GetSymbols())
 	}
 	return validChars
 }
 
+func (p *Passage) SetPin(pin string) {
+	p.pin = pin
+}
+
 func (p *Passage) GetPassword(door string) string {
 	ret := ""
-	validChars := p.GetValidChars()
 	hashing := p.pin + " - " + door
 	shaArr := sha256.Sum256([]byte(hashing))
 	var sha []byte
 	for i := range shaArr {
 		sha = append(sha, shaArr[i])
 	}
-	for i := 0; i < p.length; i++ {
-		ret += string(validChars[int(sha[i])%len(validChars)])
+	if p.requireEachCharSet {
+		validChars := p.GetValidCharSets()
+		for i := 0; i < p.length; i++ {
+			set := i % len(validChars)
+			ret += string(validChars[set][int(sha[i])%len(validChars)])
+		}
+	} else {
+		validChars := p.GetValidChars()
+		for i := 0; i < p.length; i++ {
+			ret += string(validChars[int(sha[i])%len(validChars)])
+		}
 	}
 	return ret
 }