initial commit

src/main/java/org/bdware/mockjava/JsonUtil.java

@@ -0,0 +1,64 @@
+package org.bdware.mockjava;
+
+import com.google.gson.*;
+import jdk.nashorn.api.scripting.ScriptObjectMirror;
+
+import java.util.HashSet;
+import java.util.Set;
+
+public class JsonUtil {
+    private static Set<Object> recorded = null;
+    public static JsonElement toJson(jdk.nashorn.api.scripting.ScriptObjectMirror res) {
+        recorded = new HashSet<>();
+        JsonElement jsonElement = copyInternal(res);
+        recorded.clear();
+        ;
+        return jsonElement;
+    }
+    private static JsonElement copyInternal(Object obj) {
+        if (recorded.contains(obj)) return JsonNull.INSTANCE;
+        if (obj == null) return JsonNull.INSTANCE;
+        if (obj.getClass() == jdk.nashorn.internal.runtime.Undefined.class)
+            return JsonNull.INSTANCE;
+        if (obj.getClass() == ScriptObjectMirror.class) {
+            recorded.add(obj);
+            ScriptObjectMirror som = (ScriptObjectMirror) obj;
+            if (som.isFunction()) {
+                return JsonNull.INSTANCE;
+            }
+            if (som.isArray()) {
+                JsonArray jarray = new JsonArray();
+
+                for (String str : som.getOwnKeys(true)) {
+                    try {
+                        if (Integer.parseInt(str) >= 0)
+                            jarray.add(copyInternal(som.getMember(str)));
+                    } catch (Exception e) {
+                        // System.out.println("[JSONTool] ignore key:"+str);
+                    }
+                }
+                return jarray;
+            } else {
+                JsonObject jo = new JsonObject();
+                for (String str : som.getOwnKeys(true)) {
+                    jo.add(str, copyInternal(som.getMember(str)));
+                }
+                return jo;
+            }
+        } else if (obj.getClass() == jdk.internal.dynalink.beans.StaticClass.class) {
+            return JsonNull.INSTANCE;
+        } else if (obj instanceof Number) {
+            return new JsonPrimitive((Number) obj);
+
+        } else if (obj instanceof String) {
+            return new JsonPrimitive((String) obj);
+
+        } else if (obj instanceof Character) {
+            return new JsonPrimitive((Character) obj);
+        }
+        if (obj instanceof Boolean) {
+            return new JsonPrimitive((Boolean) obj);
+        }
+        return JsonNull.INSTANCE;
+    }
+}

src/main/java/org/bdware/mockjava/JsonVisitor.java

@@ -0,0 +1,27 @@
+package org.bdware.mockjava;
+
+import com.google.gson.JsonArray;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
+
+public abstract class JsonVisitor {
+    public JsonVisitor visit(JsonElement ele) {
+        if (ele == null) return null;
+        if (ele.isJsonArray()) {
+            return visitJsonArray((JsonArray) ele);
+        } else if (ele.isJsonObject()) {
+            return visitObject(ele.getAsJsonObject());
+        } else {
+            JsonPrimitive primitive = ele.getAsJsonPrimitive();
+            return visitPrimitive(primitive);
+        }
+    }
+
+    public abstract JsonVisitor visitObject(JsonObject asJsonObject);
+
+    public abstract JsonVisitor visitJsonArray(JsonArray ele);
+
+    public abstract JsonVisitor visitPrimitive(JsonPrimitive primitive);
+
+}

src/main/java/org/bdware/mockjava/MockGenerator.java

@@ -0,0 +1,71 @@
+package org.bdware.mockjava;
+
+import com.google.gson.JsonArray;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
+import org.bdware.mockjava.generator.IntegerGenerator;
+
+public class MockGenerator extends JsonVisitor {
+    public static MockGenerator instance = new MockGenerator();
+    JsonElement result;
+
+    public JsonElement get() {
+        return result;
+    }
+
+    @Override
+    public JsonVisitor visitObject(JsonObject jsonObject) {
+        JsonObject jo = new JsonObject();
+        for (String key : jsonObject.keySet()) {
+            JsonElement origin = jsonObject.get(key);
+            visit(origin);
+            JsonElement je = get();
+            if (origin.isJsonArray()) {
+                key += "|" + Math.min(((JsonArray) origin).size(), 20);
+            }
+            jo.add(key, je);
+        }
+        result = jo;
+        return this;
+    }
+
+    @Override
+    public JsonVisitor visitJsonArray(JsonArray array) {
+        JsonArray ret = new JsonArray();
+        if (array.size() > 0) {
+            visit(array.get(0));
+            ret.add(result);
+        }
+        result = ret;
+        return this;
+    }
+
+    @Override
+    public JsonVisitor visitPrimitive(JsonPrimitive primitive) {
+        if (!primitive.isString()) result = primitive.deepCopy();
+
+        ValueGenerator generator = lookupGenerator(primitive.getAsString());
+        result = generator.generate(null, primitive);
+        return this;
+    }
+
+    private ValueGenerator lookupGenerator(final String str) {
+        if (!str.startsWith("@")) {
+            return new ValueGenerator() {
+                @Override
+                public JsonElement generate(String preField, JsonElement previous) {
+                    return new JsonPrimitive(str);
+                }
+            };
+        }
+        String[] args = null;
+        int from = str.indexOf("(");
+        if (from != -1) {
+            int to = str.indexOf(")");
+            String subStr = str.substring(from + 1, to);
+            args = subStr.split(",");
+        }
+        return IntegerGenerator.create(args);
+    }
+}

src/main/java/org/bdware/mockjava/MockSchemaParser.java

@@ -0,0 +1,53 @@
+package org.bdware.mockjava;
+
+import com.google.gson.JsonArray;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
+
+public class MockSchemaParser extends JsonVisitor {
+    public static MockSchemaParser instance = new MockSchemaParser();
+    JsonElement result;
+
+    public JsonElement get() {
+        return result;
+    }
+
+    @Override
+    public JsonVisitor visitObject(JsonObject jsonObject) {
+        JsonObject jo = new JsonObject();
+        for (String key : jsonObject.keySet()) {
+            JsonElement origin = jsonObject.get(key);
+            visit(origin);
+            JsonElement je = get();
+            if (origin.isJsonArray()) {
+                key += "|" + Math.min(((JsonArray) origin).size(), 20);
+            }
+            jo.add(key, je);
+        }
+        result = jo;
+        return this;
+    }
+
+    @Override
+    public JsonVisitor visitJsonArray(JsonArray array) {
+         JsonArray ret = new JsonArray();
+        if (array.size() > 0) {
+            visit(array.get(0));
+            ret.add(result);
+        }
+        result = ret;
+        return this;
+    }
+
+    @Override
+    public JsonVisitor visitPrimitive(JsonPrimitive primitive) {
+        if (primitive.isBoolean()) {
+            result = new JsonPrimitive("@boolean");
+        } else if (primitive.isNumber()) {
+            result = new JsonPrimitive("@integer");
+        }
+        result = new JsonPrimitive("@string");
+        return this;
+    }
+}

src/main/java/org/bdware/mockjava/MockUtil.java

@@ -0,0 +1,97 @@
+package org.bdware.mockjava;
+
+import com.google.gson.*;
+import jdk.nashorn.api.scripting.ScriptObjectMirror;
+import jdk.nashorn.internal.runtime.Context;
+
+import javax.script.ScriptEngine;
+import javax.script.ScriptEngineManager;
+import javax.script.ScriptException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+
+public class MockUtil {
+    private static final String MOCK_JS_PATH = "org/bdware/mockjava/mock.js";
+    public static ScriptEngine MOCK_JS_ENGINE;
+
+    static {
+        MOCK_JS_ENGINE = new ScriptEngineManager().getEngineByName("JavaScript");
+        // System.out.println(MOCK_JS_ENGINE);
+        try (InputStream mockJs =
+                        MockUtil.class.getClassLoader().getResourceAsStream(MOCK_JS_PATH);
+                InputStreamReader reader = new InputStreamReader(mockJs); ) {
+            MOCK_JS_ENGINE.eval(reader);
+        } catch (ScriptException | IOException e) {
+            // log.error("执行MockJs错误", e);
+            System.out.println("执行MockJs错误1");
+            System.out.println(e);
+        }
+    }
+
+    private static boolean isJson(String str) {
+        try {
+            JsonObject jsonStr = JsonParser.parseString(str).getAsJsonObject();
+            return true;
+        } catch (Exception e) {
+            return false;
+        }
+    }
+
+    public static Object mock(String template) {
+        Object result = (ScriptObjectMirror) ScriptObjectMirror.wrap(null, Context.getGlobal());
+        if (template.length() > 0) {
+            try {
+                //
+                // InputStreamReader reader = new InputStreamReader();
+                // System.out.println(MOCK_JS_ENGINE.getContext());
+                result = MOCK_JS_ENGINE.eval("JSON.stringify(Mock.mock(" + template + "))");
+                System.out.println(result);
+                // result = MOCK_JS_ENGINE.eval("JSON.stringify(Mock.mock())").toString();
+            } catch (ScriptException e) {
+                // log.error("执行Mock.mock错误", e);
+                e.printStackTrace();
+                System.out.println("执行MockJs错误2");
+            }
+        }
+        return result;
+    }
+
+    public static JsonElement parseJsonSchema(JsonElement ele) {
+        if (ele == null) return null;
+        if (ele.isJsonPrimitive()) {
+            JsonPrimitive primitive = ele.getAsJsonPrimitive();
+            if (primitive.isBoolean()) {
+                return new JsonPrimitive("@boolean");
+            } else if (primitive.isNumber()) {
+                return new JsonPrimitive("@integer");
+            }
+            return new JsonPrimitive("@string");
+            // TODO
+        } else if (ele.isJsonArray()) {
+            JsonArray array = (JsonArray) ele;
+            JsonArray ret = new JsonArray();
+            if (array.size() > 0) {
+                ret.add(parseJsonSchema(array.get(0)));
+            }
+            return ret;
+            // [{"score":1},,,,] --> [{"score":"@integer"}]
+        } else if (ele.isJsonObject()) {
+            JsonObject jo = new JsonObject();
+            JsonObject eleObj = ele.getAsJsonObject();
+            for (String key : eleObj.keySet()) {
+                JsonElement origin = eleObj.get(key);
+                JsonElement je = parseJsonSchema(origin);
+                if (origin.isJsonArray()) {
+                    key += "|" + Math.min(((JsonArray) origin).size(), 20);
+                }
+                jo.add(key, je);
+            }
+
+            return jo;
+        } // else if(ele.)
+        return new JsonPrimitive("@string");
+    }
+
+
+}

src/main/java/org/bdware/mockjava/ValueGenerator.java

@@ -0,0 +1,7 @@
+package org.bdware.mockjava;
+
+import com.google.gson.JsonElement;
+
+public abstract class ValueGenerator {
+    public abstract JsonElement generate(String preField,JsonElement previous);
+}

src/main/java/org/bdware/mockjava/generator/Generator.java

@@ -0,0 +1,5 @@
+package org.bdware.mockjava.generator;
+
+public @interface Generator {
+    String tag() default "";
+}

src/main/java/org/bdware/mockjava/generator/IntegerGenerator.java

@@ -0,0 +1,23 @@
+package org.bdware.mockjava.generator;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonPrimitive;
+import org.bdware.mockjava.ValueGenerator;
+
+import java.security.SecureRandom;
+import java.util.Random;
+
+@Generator(tag = "@integer")
+public class IntegerGenerator extends ValueGenerator {
+    // TODO
+    public static IntegerGenerator create(String[] args) {
+        return new IntegerGenerator();
+    }
+
+    static Random secureRandom = new SecureRandom();
+
+    @Override
+    public JsonElement generate(String preField, JsonElement previous) {
+        return new JsonPrimitive(secureRandom.nextInt());
+    }
+}

src/test/java/org/bdware/mockjava/MockUtilTest.java

@@ -0,0 +1,44 @@
+package org.bdware.mockjava;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonParser;
+import org.junit.Test;
+
+public class MockUtilTest {
+    //TODO @zhanghongwei 把这个测试用例拆多几个
+    @Test
+    public void main() {
+        // "{'result|min-max':1}":min-max之间的一个数字 如："{'result|1-100':1}" 返回: {"result":21}
+        // "{'result':'@integer(1,100)'}"; 返回一个整数：{"result":5171830293164278}
+        // "{'result':'@string'}" 返回一个字符串 {"result":"2ejKET"}
+        // "{'result':'@datetime'}" "{'result':'@date'}" "{'result':'@time'}"
+        // "{'result':'@cname'}" "{'result':'@name'}" "{'result':'@first'}" "{'result':'@last'}"
+        // "{'result':'@email'}" "{'result':'@ip'}" {'result':'@url'}"
+        // "{'result':'@province'}"  "{'result':'@city'}"  "{'result':'@county'}"
+        // {'id':'@integer','email':'@email','password':'@string','name':'@name'}
+        // String template="{'result|1-100':1}";
+        ///
+        //   Object res = mock("'@email'");
+        // if (res instanceof ScriptObjectMirror)
+        //   System.out.println(JSONTool.copy((ScriptObjectMirror) res));
+        // else System.out.println(res);
+        // System.out.println(res);
+        // "{'list|1-5':[{'id|+1':1,'data':'@datetime','nickname': '@cname','email':'@email'}]}
+        String str = "\"{\\\"score\\\":1}\"";
+        str = "{\"status\":\"success\", \"data\":[{\"score\":1, \"list\":[4,5,6]}]}";
+        String str1 = "[{\"score\":1}]";
+        MockSchemaParser parser = new MockSchemaParser();
+        JsonElement ele = JsonParser.parseString(str);
+        System.out.println(ele.toString());
+        parser.visit(ele);
+        JsonElement ele2 = parser.get();
+        System.out.println(ele2);
+        Object ret = MockUtil.mock(ele2.toString());
+
+        //        System.out.println(
+        //                (JSONTool.copy((jdk.nashorn.api.scripting.ScriptObjectMirror)
+        // mock(ele2.toString()))).toString());
+
+        System.out.println(MockUtil.mock("'@string'"));
+    }
+}