From 3c89f0f8fde867a8533f0b42c2071a68d5e51530 Mon Sep 17 00:00:00 2001
From: Csaba Kozak <kozakcsabi@gmail.com>
Date: Tue, 12 Apr 2016 21:35:07 +0200
Subject: [PATCH] Better generated source encoding support

The default encoding is now UTF-8, and the encoding can be configured
with the `encoding` annotation processing option.
---
 .../internal/AndroidAnnotationProcessor.java  |  2 +-
 .../androidannotations/internal/Options.java  |  2 ++
 .../generation/CodeModelGenerator.java        | 30 +++++++++++++++++--
 .../generation/ResourceCodeWriter.java        |  5 ++--
 .../internal/generation/SourceCodeWriter.java |  5 ++--
 5 files changed, 36 insertions(+), 8 deletions(-)

diff --git a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/AndroidAnnotationProcessor.java b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/AndroidAnnotationProcessor.java
index c5558a0a35..a6265b7c87 100644
--- a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/AndroidAnnotationProcessor.java
+++ b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/AndroidAnnotationProcessor.java
@@ -229,7 +229,7 @@ private ModelProcessor.ProcessResult processAnnotations(AnnotationElements valid
 	private void generateSources(ModelProcessor.ProcessResult processResult) throws IOException {
 		timeStats.start("Generate Sources");
 		LOGGER.info("Number of files generated by AndroidAnnotations: {}", processResult.codeModel.countArtifacts());
-		CodeModelGenerator modelGenerator = new CodeModelGenerator(processingEnv.getFiler(), coreVersion);
+		CodeModelGenerator modelGenerator = new CodeModelGenerator(processingEnv.getFiler(), coreVersion, androidAnnotationsEnv.getOptionValue(CodeModelGenerator.OPTION_ENCODING));
 		modelGenerator.generate(processResult);
 		timeStats.stop("Generate Sources");
 	}
diff --git a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/Options.java b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/Options.java
index 756694962c..7169578641 100644
--- a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/Options.java
+++ b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/Options.java
@@ -25,6 +25,7 @@
 import org.androidannotations.Option;
 import org.androidannotations.helper.ModelConstants;
 import org.androidannotations.holder.BaseGeneratedClassHolder;
+import org.androidannotations.internal.generation.CodeModelGenerator;
 import org.androidannotations.internal.helper.AndroidManifestFinder;
 import org.androidannotations.internal.rclass.ProjectRClassFinder;
 import org.androidannotations.logger.LoggerContext;
@@ -46,6 +47,7 @@ public Options(ProcessingEnvironment processingEnvironment) {
 		addSupportedOption(LoggerContext.OPTION_LOG_APPENDER_CONSOLE);
 		addSupportedOption(LoggerContext.OPTION_LOG_APPENDER_FILE);
 		addSupportedOption(BaseGeneratedClassHolder.OPTION_GENERATE_FINAL_CLASSES);
+		addSupportedOption(CodeModelGenerator.OPTION_ENCODING);
 	}
 
 	public void addAllSupportedOptions(List<Option> options) {
diff --git a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/CodeModelGenerator.java b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/CodeModelGenerator.java
index ebe12ff60e..a552349a88 100644
--- a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/CodeModelGenerator.java
+++ b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/CodeModelGenerator.java
@@ -16,31 +16,55 @@
 package org.androidannotations.internal.generation;
 
 import java.io.IOException;
+import java.nio.charset.Charset;
+import java.nio.charset.UnsupportedCharsetException;
 
 import javax.annotation.processing.Filer;
 
+import org.androidannotations.Option;
 import org.androidannotations.internal.process.ModelProcessor;
+import org.androidannotations.logger.Logger;
+import org.androidannotations.logger.LoggerFactory;
 
 import com.helger.jcodemodel.writer.PrologCodeWriter;
 
 public class CodeModelGenerator {
 
+	public static final Option OPTION_ENCODING = new Option("encoding", "UTF-8");
+
+	private static final Logger LOGGER = LoggerFactory.getLogger(CodeModelGenerator.class);
+
 	private final Filer filer;
 	private final String header;
+	private final String encoding;
 
-	public CodeModelGenerator(Filer filer, String aaVersion) {
+	public CodeModelGenerator(Filer filer, String aaVersion, String encoding) {
 		this.filer = filer;
 		this.header = "DO NOT EDIT THIS FILE.\n"
 			+ "Generated using AndroidAnnotations " + aaVersion + ".\n\n"
 			+ "You can create a larger work that contains this file and distribute that work under terms of your choice.\n";
+		this.encoding = encoding;
 	}
 
 	public void generate(ModelProcessor.ProcessResult processResult) throws IOException {
+		Charset charset = getCharset();
 
-		SourceCodeWriter sourceCodeWriter = new SourceCodeWriter(filer, processResult.originatingElements);
+		SourceCodeWriter sourceCodeWriter = new SourceCodeWriter(filer, processResult.originatingElements, charset);
 
 		PrologCodeWriter prologCodeWriter = new PrologCodeWriter(sourceCodeWriter, header);
 
-		processResult.codeModel.build(prologCodeWriter, new ResourceCodeWriter(filer));
+		processResult.codeModel.build(prologCodeWriter, new ResourceCodeWriter(filer, charset));
+	}
+
+	private Charset getCharset() {
+		try {
+			return Charset.forName(encoding);
+		} catch (UnsupportedCharsetException exception) {
+			Charset defaultCharset = Charset.defaultCharset();
+
+			LOGGER.warn("The requested charset ({}) is not available, falling back to platform default ({}).", encoding, defaultCharset);
+
+			return defaultCharset;
+		}
 	}
 }
diff --git a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/ResourceCodeWriter.java b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/ResourceCodeWriter.java
index 1d0a0b3515..3825ef60c2 100644
--- a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/ResourceCodeWriter.java
+++ b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/ResourceCodeWriter.java
@@ -17,6 +17,7 @@
 
 import java.io.IOException;
 import java.io.OutputStream;
+import java.nio.charset.Charset;
 
 import javax.annotation.processing.Filer;
 import javax.tools.FileObject;
@@ -29,8 +30,8 @@ public class ResourceCodeWriter extends AbstractCodeWriter {
 
 	private final Filer filer;
 
-	public ResourceCodeWriter(Filer filer) {
-		super(null);
+	public ResourceCodeWriter(Filer filer, Charset charset) {
+		super(charset);
 		this.filer = filer;
 	}
 
diff --git a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/SourceCodeWriter.java b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/SourceCodeWriter.java
index 02d87b22eb..01d04429de 100644
--- a/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/SourceCodeWriter.java
+++ b/AndroidAnnotations/androidannotations-core/androidannotations/src/main/java/org/androidannotations/internal/generation/SourceCodeWriter.java
@@ -17,6 +17,7 @@
 
 import java.io.IOException;
 import java.io.OutputStream;
+import java.nio.charset.Charset;
 
 import javax.annotation.processing.Filer;
 import javax.annotation.processing.FilerException;
@@ -44,8 +45,8 @@ public void write(int arg0) throws IOException {
 		}
 	}
 
-	public SourceCodeWriter(Filer filer, OriginatingElements originatingElements) {
-		super(null);
+	public SourceCodeWriter(Filer filer, OriginatingElements originatingElements, Charset charset) {
+		super(charset);
 		this.filer = filer;
 		this.originatingElements = originatingElements;
 	}