From 3ecb7702720e1e9020a2346d1ee78e87185b2455 Mon Sep 17 00:00:00 2001
From: Thiago Nascimento <nascimenthiago@gmail.com>
Date: Tue, 5 Nov 2024 16:56:58 -0300
Subject: [PATCH] style: Geracao de DV do titulo de eleitor

---
 .../validation/TituloEleitoralValidator.java  | 30 +++++++++++--------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/stella-core/src/main/java/br/com/caelum/stella/validation/TituloEleitoralValidator.java b/stella-core/src/main/java/br/com/caelum/stella/validation/TituloEleitoralValidator.java
index 1438b153..a4a2303f 100644
--- a/stella-core/src/main/java/br/com/caelum/stella/validation/TituloEleitoralValidator.java
+++ b/stella-core/src/main/java/br/com/caelum/stella/validation/TituloEleitoralValidator.java
@@ -179,24 +179,28 @@ private String calculaDigitos(String tituloSemDigito) {
 		return digito1 + digito2;
 	}
 
-    private String geraDigito(boolean ehEstadoSubstitutorDigito, String base) {
+   private String geraDigito(boolean ehEstadoSubstitutorDigito, String base) {
     	
     	String digito = new DigitoPara(base).mod(11).calcula();
     	
-    	if (ehEstadoSubstitutorDigito) {
-    		if (digito.equals("1")) {
-    			digito = "0";
-    		} else if (digito.equals("0")) {
-    			digito = "1";
-    		} else {
-    			digito = new DigitoPara(base).complementarAoModulo().mod(11).calcula();
+    	if (!ehEstadoSubstitutorDigito) {
+    		
+    		if (digito.equals("1") || digito.equals("0")) {
+    			return "0";
     		}
-    	} else {
-    		digito = digito.equals("1") || digito.equals("0") ? "0" :
-    			new DigitoPara(base).complementarAoModulo().mod(11).calcula();
-    	}
+    		
+    		return new DigitoPara(base).complementarAoModulo().mod(11).calcula();
+    	} 
     	
-    	return digito;
+    	if (digito.equals("1")) {
+    		return "0";
+    	} 
+    	
+    	if (digito.equals("0")) {
+    		return "1";
+    	} 
+    	
+    	return new DigitoPara(base).complementarAoModulo().mod(11).calcula();
     }
     
 	private boolean hasCodigoDeEstadoInvalido(String tituloDeEleitor) {