feat(company): Add postal flag

cmd/company.go

@@ -27,6 +27,7 @@ var (
 	includeYearRange      int
 	includeNumbers        bool
 	includeNumbersRange   int
+	includePostal         int
 )
 
 var companyCmd = &cobra.Command{
@@ -55,6 +56,11 @@ var companyCmd = &cobra.Command{
 			fmt.Println("You cannot use both --year and --numbers")
 			return
 		}
+
+		if includeYear && includePostal != 0 {
+			fmt.Println("You cannot use both --year and --postal")
+			return
+		}
 		// End Temp
 
 		wordlist := generateCompanyPasslist(companyName)
@@ -100,6 +106,7 @@ func init() {
 	companyCmd.Flags().StringVarP(&includeMask, "mask", "m", "", "Apply a custom mask to the passwords")
 	companyCmd.Flags().BoolVar(&includeNumbers, "numbers", false, "Include numbers to the passwords")
 	companyCmd.Flags().IntVar(&includeNumbersRange, "numbers-range", 20, "Apply a range for numbers option")
+	companyCmd.Flags().IntVarP(&includePostal, "postal", "p", 0, "Include postal code to the passwords")
 
 }
 
@@ -124,6 +131,10 @@ func generateCompanyPasslist(name string) []string {
 		wordlist = generate.WithUppercase(wordlist)
 	}
 
+	if includePostal != 0 {
+		wordlist = generate.WithPostal(wordlist, includePostal)
+	}
+
 	if includeNumbers && includeNumbersRange != 0 {
 		wordlist = generate.WithNumbers(wordlist, includeNumbersRange)
 	}

cmd/company_test.go

@@ -16,6 +16,9 @@ func cleanup() {
 	includeUppercase = false
 	includeMask = ""
 	includeYearRange = 0
+	includePostal = 0
+	includeNumbers = false
+	includeNumbersRange = 20
 }
 
 func TestWithoutFlag(t *testing.T) {
@@ -226,3 +229,19 @@ func TestNumbersRangeFlag(t *testing.T) {
 
 	t.Cleanup(cleanup)
 }
+
+func TestPostalFlag(t *testing.T) {
+	includePostal = 75000
+
+	wordlist := generateCompanyPasslist("demo")
+
+	got := strings.Join(wordlist, "\n")
+
+	expected := "demo\ndemo75000"
+
+	if got != expected {
+		t.Errorf("Expected %q, but got %q", expected, got)
+	}
+
+	t.Cleanup(cleanup)
+}

lib/generate/generate.go

@@ -194,3 +194,10 @@ func WithUppercase(wordlist []string) []string {
 	}
 	return wordlist
 }
+
+func WithPostal(wordlist []string, includePostal int) []string {
+	for _, word := range wordlist {
+		wordlist = append(wordlist, word+strconv.Itoa(includePostal))
+	}
+	return wordlist
+}