style: Geracao de DV do titulo de eleitor (#303)

Co-authored-by: Rogerio Angeliski <[email protected]>
caelum · Feb 13, 2025 · f1d635c · f1d635c
1 parent 81f41e2
commit f1d635c
Showing 1 changed file with 17 additions and 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
@@ -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) {