support more data category

Author: xiaoqi

sample/src/main/java/com/xiaoqi/autosavesample/Example.java

@@ -0,0 +1,10 @@
+package com.xiaoqi.autosavesample;
+
+import java.io.Serializable;
+
+/**
+ * Created by xiaoqi on 2017/1/10.
+ */
+
+public class Example implements Serializable{
+}

sample/src/main/java/com/xiaoqi/autosavesample/MainActivity.java

@@ -6,12 +6,31 @@
 import com.processor.NeedSave;
 import com.xqand.savehelper.SaveHelper;
 
+import java.util.ArrayList;
+
 public class MainActivity extends AppCompatActivity {
 	@NeedSave
 	public int a;
 	@NeedSave
 	public String test;
-
+	@NeedSave
+	public boolean b;
+	@NeedSave
+	public Boolean c;
+	@NeedSave
+	public ArrayList<String> t;
+	@NeedSave
+	public Integer i;
+	@NeedSave
+	public Example example;
+	@NeedSave
+	public Float f1;
+	@NeedSave
+	public float f2;
+	@NeedSave
+	public char achar;
+	@NeedSave
+	public char achars[];
 	@Override
 	protected void onCreate(Bundle savedInstanceState) {
 		super.onCreate(savedInstanceState);

xqauto-processor/src/main/java/com/xqand/helper/HelperClass.java

@@ -59,21 +59,24 @@ public JavaFile generateCode(){
 
 			for (HelperSavedValues value : elementArrayList) {
 				Name fieldName = value.getSimpleName();
-				if(value.getFieldType().toString().equals("java.lang.String")){
-					saveMethodBuilder.addStatement("outState.putString($S,recoverActivity.$N)", fieldName.toString
-							().toUpperCase(),fieldName);
-					recoverMethodBuilder.addStatement("recoverActivity.$N = savedInstanceState.getString($S)",fieldName,
-							fieldName.toString().toUpperCase());
-				}else if(value.getFieldType().toString().equals("int")){
-					saveMethodBuilder.addStatement("outState.putInt($S,recoverActivity.$N)", fieldName.toString
-							().toUpperCase(),fieldName);
-					recoverMethodBuilder.addStatement("recoverActivity.$N = savedInstanceState.getInt($S)",fieldName,
-							fieldName.toString().toUpperCase());
-				}else if(value.getFieldType() instanceof Serializable){
-					saveMethodBuilder.addStatement("outState.putSerializable($S,recoverActivity.$S)", fieldName.toString
-							().toUpperCase(),fieldName);
-					recoverMethodBuilder.addStatement("recoverActivity.$N = savedInstanceState.getSerializable($S)",
-							fieldName, fieldName.toString().toUpperCase());
+				String fieldType = value.getFieldType().toString();
+				String type = getFieldType(fieldType);
+				if(!type.equals("unKnow")){
+					addMethodStatement(saveMethodBuilder,recoverMethodBuilder, type, fieldName);
+				}else {
+//					Class c = Class.forName(fieldType);
+//					Class[] interfaces = c.getInterfaces();
+//					for (Class inter : interfaces) {
+//						saveMethodBuilder.addStatement("//$N", inter.getName());
+//					}
+					if( value.getFieldType() instanceof Serializable){
+						saveMethodBuilder.addStatement("outState.putSerializable($S,recoverActivity.$S)", fieldName.toString
+								().toUpperCase(),fieldName);
+						recoverMethodBuilder.addStatement("recoverActivity.$N = savedInstanceState.getSerializable($S)",
+								fieldName, fieldName.toString().toUpperCase());
+					}else {
+						saveMethodBuilder.addStatement("//$S", fieldType);
+					}
 				}
 			}
 			recoverMethodBuilder.endControlFlow();
@@ -102,4 +105,56 @@ private String getPackageName() {
 	private String getCacheClassName(){
 		return encloseElement.getSimpleName().toString() + "_Cache";
 	}
+
+	private String upperFirstWord(String str){
+		if(str != null){
+			char[] ch = str.toCharArray();
+			if (ch[0] >= 'a' && ch[0] <= 'z') {
+				ch[0] = (char) (ch[0] - 32);
+			}
+			return new String(ch);
+		}else {
+			return "";
+		}
+	}
+
+	private void addSaveMethodStatement(MethodSpec.Builder saveMethodBuilder, String str, Name fieldName){
+		saveMethodBuilder.addStatement(String.format("outState.put%s($S,recoverActivity.$N)",upperFirstWord(str)),
+				fieldName.toString().toUpperCase(),fieldName);
+	}
+
+	private void addRecoverMethodStatement(MethodSpec.Builder recoverMethodBuilder, String str, Name fieldName){
+		recoverMethodBuilder.addStatement(String.format("recoverActivity.$N = savedInstanceState.get%s($S)",upperFirstWord(str)),
+				fieldName, fieldName.toString().toUpperCase());
+	}
+
+	private void addMethodStatement(MethodSpec.Builder saveMethodBuilder,MethodSpec.Builder recoverMethodBuilder, String str, Name fieldName){
+		addSaveMethodStatement(saveMethodBuilder,str,fieldName);
+		addRecoverMethodStatement(recoverMethodBuilder,str,fieldName);
+	}
+
+	private String getFieldType(String fieldType){
+		String type;
+		if(fieldType.equals("java.lang.String")){
+			type = "String";
+		}else if(fieldType.equals("int") || fieldType.equals("java.lang.Integer")){
+			type = "Int";
+		}else if(fieldType.equals("short") || fieldType.equals("java.lang.Short")){
+			type = "Short";
+		}else if(fieldType.equals("double") || fieldType.equals("java.lang.Double")){
+			type = "Double";
+		}else if(fieldType.equals("float") || fieldType.equals("java.lang.Float")){
+			type = "Float";
+		}else if(fieldType.equals("boolean") || fieldType.equals("java.lang.Boolean")){
+			type = "Boolean";
+		}else if(fieldType.equals("char")){
+			type = "Char";
+		}else if(fieldType.equals("char[]")){
+			type = "CharArray";
+		}else {
+			type = "unKnow";
+		}
+		return type;
+	}
+
 }