Merge pull request #42 from brazzcore/feature/26-voter-registration-card-mock-pt-br

[feature/26] :: voter registration card mock pt br

Author: wellfernandes

examples/mock_example.go

@@ -9,6 +9,7 @@ import (
 	"github.com/brazzcore/mocai/pkg/mocai/entities/company"
 	"github.com/brazzcore/mocai/pkg/mocai/entities/person"
 	"github.com/brazzcore/mocai/pkg/mocai/entities/phone"
+	"github.com/brazzcore/mocai/pkg/mocai/entities/vote_registration"
 	"github.com/brazzcore/mocai/pkg/mocai/translations"
 )
 
@@ -44,6 +45,12 @@ func GenerateMockExample() {
 		return
 	}
 
+	vote_registration_mock, err := vote_registration.GenerateVoteRegistration(false)
+	if err != nil {
+		fmt.Print(err)
+		return
+	}
+
 	fmt.Println(constants.HeaderMain)
 	fmt.Println(constants.SubHeader)
 
@@ -62,5 +69,7 @@ func GenerateMockExample() {
 		address_mock.Street, address_mock.Number, address_mock.City, address_mock.State, address_mock.UF, address_mock.ZIP)
 	fmt.Printf("Phone: (%s) %s\n", phone_mock.AreaCode, phone_mock.Number)
 
+	fmt.Printf("Voter Registration Card: Section: %s, Zone: %s, Registration: %s\n", vote_registration_mock.BrazilianVoteRegistration.Section, vote_registration_mock.BrazilianVoteRegistration.Zone, vote_registration_mock.BrazilianVoteRegistration.Number)
+
 	fmt.Println(constants.Footer)
 }

pkg/mocai/entities/certificate/countries/validator.go

@@ -0,0 +1,127 @@
+package countries
+
+import (
+	"fmt"
+	"math/rand"
+	"strconv"
+	"time"
+)
+
+var (
+	globalRand = rand.New(rand.NewSource(time.Now().UnixNano()))
+)
+
+// BrazilianVoteRegistration represents a Brazilian vote registration.
+type BrazilianVoteRegistration struct {
+	Section string
+	Zone    string
+	Number  string
+}
+
+// GenerateBrazilianVoteRegistration generates a valid Brazilian vote registration number.
+// If formatted is true, the Brazilian vote registration number will be returned in the format XXX XXX XXX.
+// If formatted is false, the Brazilian vote registration number will be returned as a plain string.
+func GenerateBrazilianVoteRegistration(formatted bool) (*BrazilianVoteRegistration, error) {
+	section := randomInt3Digits()
+	zone := randomInt3Digits()
+
+	// Generate an 8 digit sequence number
+	sequenceNumber := randomInt(1, 99999999)
+	sequenceNumberStr := fmt.Sprintf("%08d", sequenceNumber)
+
+	// Generate a random state code 01 to 28
+	stateCode := randomInt(1, 28)
+	stateCodeStr := fmt.Sprintf("%02d", stateCode)
+
+	// Calculate the first check digit
+	checkDigit1, err := calculateCheckDigit1(sequenceNumberStr)
+	if err != nil {
+		return nil, err
+	}
+
+	// Calculate the second check digit
+	checkDigit2, err := calculateCheckDigit2(stateCodeStr, checkDigit1, stateCode)
+	if err != nil {
+		return nil, err
+	}
+
+	// Combine everything to form the complete number
+	number := sequenceNumberStr + stateCodeStr + checkDigit1 + checkDigit2
+	if number == "" || len(number) != 12 {
+		return nil, ErrInvalidVoteRegistration
+	}
+
+	if formatted {
+		number = fmt.Sprintf("%s %s %s", number[:4], number[4:8], number[8:])
+	}
+
+	createdBrazilianVoteRegistration := &BrazilianVoteRegistration{
+		Section: section,
+		Zone:    zone,
+		Number:  number,
+	}
+
+	return createdBrazilianVoteRegistration, nil
+}
+
+// randomInt generates a random integer between min and max
+func randomInt(min, max int) int {
+	return globalRand.Intn(max-min+1) + min
+}
+
+// randomInt3Digits generates a random 3 digit number
+func randomInt3Digits() string {
+	return fmt.Sprintf("%03d", globalRand.Intn(1000))
+}
+
+// calculateCheckDigit1 calculates the first check digit.
+// It depends on the sequence number.
+func calculateCheckDigit1(sequenceNumber string) (string, error) {
+	sum := 0
+	weights := []int{2, 3, 4, 5, 6, 7, 8, 9}
+	for i := 0; i < len(sequenceNumber); i++ {
+		digit, err := strconv.Atoi(string(sequenceNumber[i]))
+		if err != nil {
+			return "", ErrInvalidCheckDigit1
+		}
+		sum += digit * weights[i]
+	}
+
+	checkDigit1 := sum % 11
+	if checkDigit1 == 10 {
+		checkDigit1 = 0
+	}
+
+	return strconv.Itoa(checkDigit1), nil
+}
+
+// calculateCheckDigit2 calculates the second check digit.
+// It depends on the state code and the first check digit.
+func calculateCheckDigit2(stateCode, checkDigit1 string, stateCodeInt int) (string, error) {
+	sum := 0
+	weights := []int{7, 8}
+	for i := 0; i < len(stateCode); i++ {
+		digit, err := strconv.Atoi(string(stateCode[i]))
+		if err != nil {
+			return "", ErrInvalidCheckDigit2
+		}
+		sum += digit * weights[i]
+	}
+	checkDigit1Int, err := strconv.Atoi(checkDigit1)
+	if err != nil {
+		return "", ErrInvalidCheckDigit2
+	}
+	sum += checkDigit1Int * 9
+
+	checkDigit2 := sum % 11
+	if checkDigit2 == 10 {
+		checkDigit2 = 0
+	}
+
+	// Special case for states 01 SP and 02 MG
+	if (stateCodeInt == 1 || stateCodeInt == 2) && checkDigit2 == 0 {
+		checkDigit2 = 1
+	}
+
+	return strconv.Itoa(checkDigit2), nil
+}

pkg/mocai/entities/vote_registration/countries/brazilian_vote_registration.go

@@ -0,0 +1,12 @@
+package countries
+
+import "errors"
+
+// Errors for Vote Registration data generation failures.
+// These errors represent specific failure scenarios and should be wrapped with additional
+// context when returned.
+var (
+	ErrInvalidVoteRegistration = errors.New("invalid vote registration")
+	ErrInvalidCheckDigit1      = errors.New("invalid check digit 1")
+	ErrInvalidCheckDigit2      = errors.New("invalid check digit 2")
+)

pkg/mocai/entities/vote_registration/countries/errors.go

@@ -0,0 +1,20 @@
+package vote_registration
+
+import "github.com/brazzcore/mocai/pkg/mocai/entities/vote_registration/countries"
+
+// VoteRegistration represents a Brazilian vote registration.
+type VoteRegistration struct {
+	BrazilianVoteRegistration *countries.BrazilianVoteRegistration
+}
+
+// GenerateVoteRegistration generates a Brazilian vote registration.
+// If formatted is true, the Brazilian vote registration will be returned in the format XXX XXX XXX.
+// If formatted is false, the Brazilian vote registration will be returned as a plain string.
+func GenerateVoteRegistration(formatted bool) (*VoteRegistration, error) {
+	brazilianVoteRegistration, err := countries.GenerateBrazilianVoteRegistration(formatted)
+	if err != nil {
+		return nil, err
+	}
+
+	return &VoteRegistration{BrazilianVoteRegistration: brazilianVoteRegistration}, nil
+}