json-as 0.9.17 → 0.9.18

package/transform/lib/index.js

@@ -1,460 +1,514 @@
-import { FieldDeclaration } from "assemblyscript/dist/assemblyscript.js";
+import {
+  FieldDeclaration,
+  StringLiteralExpression,
+} from "assemblyscript/dist/assemblyscript.js";
 import { toString, isStdlib } from "visitor-as/dist/utils.js";
 import { BaseVisitor, SimpleParser } from "visitor-as/dist/index.js";
 import { Transform } from "assemblyscript/dist/transform.js";
 class JSONTransform extends BaseVisitor {
-    constructor() {
-        super(...arguments);
-        this.schemasList = [];
-        this.sources = new Set();
+  constructor() {
+    super(...arguments);
+    this.schemasList = [];
+    this.sources = new Set();
+  }
+  visitMethodDeclaration() {}
+  visitClassDeclaration(node) {
+    if (!node.decorators?.length) return;
+    let found = false;
+    for (const decorator of node.decorators) {
+      const name = decorator.name.text;
+      if (name === "json" || name === "serializable") {
+        found = true;
+        break;
+      }
     }
-    visitMethodDeclaration() { }
-    visitClassDeclaration(node) {
-        if (!node.decorators?.length)
-            return;
-        let found = false;
-        for (const decorator of node.decorators) {
-            const name = decorator.name.text;
-            if (name === "json" || name === "serializable") {
-                found = true;
-                break;
-            }
-        }
-        if (!found)
-            return;
-        const schema = new SchemaData();
-        schema.node = node;
-        schema.name = node.name.text;
-        const members = [
-            ...node.members.filter(v => v.kind === 54 /* NodeKind.FieldDeclaration */)
-        ];
-        if (node.extendsType) {
-            schema.parent = this.schemasList.find((v) => v.name == node.extendsType?.name.identifier.text);
-            if (schema.parent?.members) {
-                for (let i = schema.parent.members.length - 1; i >= 0; i--) {
-                    const replace = schema.members.find((v) => v.name == schema.parent?.members[i]?.name);
-                    if (!replace) {
-                        //schema.members.unshift(schema.parent?.members[i]!);
-                        members.unshift(schema.parent?.members[i].node);
-                    }
-                }
-            }
-        }
-        if (!members.length) {
-            let SERIALIZE_RAW_EMPTY = "__SERIALIZE(): string {\n  return \"{}\";\n}";
-            //let SERIALIZE_PRETTY_EMPTY = "__SERIALIZE_PRETTY(): string {\n  return \"{}\";\n}";
-            let INITIALIZE_EMPTY = "__INITIALIZE(): this {\n  return this;\n}";
-            let DESERIALIZE_EMPTY = "__DESERIALIZE(data: string, key_start: i32, key_end: i32, value_start: i32, value_end: i32): boolean {\n  return false;\n}";
-            if (process.env["JSON_DEBUG"]) {
-                console.log(SERIALIZE_RAW_EMPTY);
-                //console.log(SERIALIZE_PRETTY_EMPTY);
-                console.log(INITIALIZE_EMPTY);
-                console.log(DESERIALIZE_EMPTY);
-            }
-            const SERIALIZE_RAW_METHOD_EMPTY = SimpleParser.parseClassMember(SERIALIZE_RAW_EMPTY, node);
-            //const SERIALIZE_PRETTY_METHOD = SimpleParser.parseClassMember(SERIALIZE_PRETTY, node);
-            const INITIALIZE_METHOD_EMPTY = SimpleParser.parseClassMember(INITIALIZE_EMPTY, node);
-            const DESERIALIZE_METHOD_EMPTY = SimpleParser.parseClassMember(DESERIALIZE_EMPTY, node);
-            if (!node.members.find(v => v.name.text == "__SERIALIZE"))
-                node.members.push(SERIALIZE_RAW_METHOD_EMPTY);
-            if (!node.members.find(v => v.name.text == "__INITIALIZE"))
-                node.members.push(INITIALIZE_METHOD_EMPTY);
-            if (!node.members.find(v => v.name.text == "__DESERIALIZE"))
-                node.members.push(DESERIALIZE_METHOD_EMPTY);
-            this.schemasList.push(schema);
+    if (!found) return;
+    const schema = new SchemaData();
+    schema.node = node;
+    schema.name = node.name.text;
+    const members = [
+      ...node.members.filter(
+        (v) => v.kind === 54 /* NodeKind.FieldDeclaration */,
+      ),
+    ];
+    if (node.extendsType) {
+      schema.parent = this.schemasList.find(
+        (v) => v.name == node.extendsType?.name.identifier.text,
+      );
+      if (schema.parent?.members) {
+        for (let i = schema.parent.members.length - 1; i >= 0; i--) {
+          const replace = schema.members.find(
+            (v) => v.name == schema.parent?.members[i]?.name,
+          );
+          if (!replace) {
+            members.unshift(schema.parent?.members[i].node);
+          }
         }
-        for (const member of members) {
-            const name = member.name;
-            if (!(member instanceof FieldDeclaration))
-                continue;
-            if (!member.type) {
-                throw new Error("Fields must be strongly typed! Found " + toString(member) + " at " + node.range.source.normalizedPath);
-            }
-            const type = toString(member.type);
-            if (type.startsWith("(") && type.includes("=>"))
-                continue;
-            const value = member.initializer ? toString(member.initializer) : null;
-            if (member.flags == 32 /* CommonFlags.Static */)
-                continue;
-            if (member.flags === 512 /* CommonFlags.Private */)
-                continue;
-            if (member.flags === 1024 /* CommonFlags.Protected */)
-                continue;
-            if (member.decorators && member.decorators.find((v) => v.name.text == "omit"))
-                continue;
-            const mem = new Property();
-            mem.name = name.text;
-            mem.type = type;
-            mem.value = value;
-            mem.node = member;
-            if (member.decorators) {
-                let decorator = null;
-                if (decorator = member.decorators.find(v => v.name.text == "alias")) {
-                    if (decorator.name.text == "alias") {
-                        if (!decorator.args?.length)
-                            throw new Error("Expected 1 argument but got zero at @alias in " + node.range.source.normalizedPath);
-                        mem.flags.push(PropertyFlags.Alias);
-                        mem.alias = decorator.args[0].value;
-                    }
-                }
-                for (let i = 0; i < (member.decorators).length; i++) {
-                    const decorator = member.decorators[i];
-                    if (decorator.name.text == "omitnull") {
-                        mem.flags.push(PropertyFlags.OmitNull);
-                    }
-                    else if (decorator.name.text == "omitif") {
-                        if (!decorator.args?.length)
-                            throw new Error("Expected 1 argument but got zero at @omitif in " + node.range.source.normalizedPath);
-                        mem.args?.push(decorator.args[0].value);
-                        mem.flags.push(PropertyFlags.OmitIf);
-                    }
-                    else if (decorator.name.text == "flatten") {
-                        if (!decorator.args?.length)
-                            throw new Error("Expected 1 argument but got zero at @flatten in " + node.range.source.normalizedPath);
-                        mem.flags.push(PropertyFlags.Flatten);
-                        mem.args = [decorator.args[0].value];
-                    }
-                }
-            }
-            if (!mem.flags.length) {
-                mem.flags = [PropertyFlags.None];
-                if (type == "JSON.Raw") {
-                    mem.serialize = escapeString(JSON.stringify(mem.alias || mem.name)) + ":${this." + name.text + "}";
-                    mem.deserialize = "this." + name.text + " = " + "data.substring(value_start, value_end);";
-                }
-                else {
-                    mem.serialize = escapeString(JSON.stringify(mem.alias || mem.name)) + ":${__SERIALIZE<" + type + ">(this." + name.text + ")}";
-                    mem.deserialize = "this." + name.text + " = " + "__DESERIALIZE<" + type + ">(data.substring(value_start, value_end));";
-                }
-            }
-            if (mem.flags.includes(PropertyFlags.OmitNull)) {
-                mem.serialize = "${changetype<usize>(this." + mem.name + ") == <usize>0" + " ? \"\" : '" + escapeString(JSON.stringify(mem.alias || mem.name)) + ":' + __SERIALIZE<" + type + ">(this." + name.text + ") + \",\"}";
-                mem.deserialize = "this." + name.text + " = " + "__DESERIALIZE<" + type + ">(data.substring(value_start, value_end));";
-            }
-            else if (mem.flags.includes(PropertyFlags.OmitIf)) {
-                mem.serialize = "${" + mem.args[0] + " ? \"\" : '" + escapeString(JSON.stringify(mem.alias || mem.name)) + ":' + __SERIALIZE<" + type + ">(this." + name.text + ") + \",\"}";
-                mem.deserialize = "this." + name.text + " = " + "__DESERIALIZE<" + type + ">(data.substring(value_start, value_end));";
-            }
-            else if (mem.flags.includes(PropertyFlags.Alias)) {
-                mem.serialize = escapeString(JSON.stringify(mem.alias || mem.name)) + ":${__SERIALIZE<" + type + ">(this." + name.text + ")}";
-                mem.deserialize = "this." + name.text + " = " + "__DESERIALIZE<" + type + ">(data.substring(value_start, value_end));";
-                mem.name = name.text;
-            }
-            else if (mem.flags.includes(PropertyFlags.Flatten)) {
-                const nullable = mem.node.type.isNullable;
-                if (nullable) {
-                    mem.serialize = escapeString(JSON.stringify(mem.alias || mem.name)) + ":${this." + name.text + " ? __SERIALIZE(changetype<nonnull<" + type + ">>(this." + name.text + ")" + (mem.args?.length ? '.' + mem.args.join(".") : '') + ") : \"null\"}";
-                    mem.deserialize = "if (value_end - value_start == 4 && load<u64>(changetype<usize>(data) + <usize>(value_start << 1)) == " + charCodeAt64("null", 0) + ") {\n        this." + name.text + " = null;\n      } else {\n        this." + name.text + " = " + "__DESERIALIZE<" + type + ">('{\"" + mem.args[0] + "\":' + data.substring(value_start, value_end) + \"}\");\n      }";
-                }
-                else {
-                    mem.serialize = escapeString(JSON.stringify(mem.alias || mem.name)) + ":${this." + name.text + " ? __SERIALIZE(this." + name.text + (mem.args?.length ? '.' + mem.args.join(".") : '') + ") : \"null\"}";
-                    mem.deserialize = "this." + name.text + " = " + "__DESERIALIZE<" + type + ">('{\"" + mem.args[0] + "\":' + data.substring(value_start, value_end) + \"}\");";
-                }
-                mem.name = name.text;
-            }
-            const t = mem.node.type.name?.identifier.text;
-            if (this.schemasList.find(v => v.name == t)) {
-                mem.initialize = "this." + name.text + " = changetype<nonnull<" + mem.type + ">>(__new(offsetof<nonnull<" + mem.type + ">>(), idof<nonnull<" + mem.type + ">>()));\n  changetype<nonnull<" + mem.type + ">>(this." + name.text + ").__INITIALIZE()";
-            }
-            else if (mem.value) {
-                mem.initialize = "this." + name.text + " = " + mem.value;
-            }
-            else if (t === "Map") {
-                mem.initialize = "this." + name.text + " = new " + mem.type + "()";
-            }
-            else if (t === "string") {
-                mem.initialize = "this." + name.text + " = \"\"";
-            }
-            else if (t === "Array") {
-                mem.initialize = "this." + name.text + " = instantiate<" + mem.type + ">()";
-            }
-            else if (t === "bool" || t === "boolean") {
-                mem.initialize = "this." + name.text + " = false";
-            }
-            else if (t === "JSON.Raw") {
-                mem.initialize = "this." + name.text + " = \"\"";
-            }
-            else if (t === "u8" ||
-                t === "u16" ||
-                t === "u32" ||
-                t === "u64" ||
-                t === "i8" ||
-                t === "i16" ||
-                t === "i32" ||
-                t === "i64") {
-                mem.initialize = "this." + name.text + " = 0";
-            }
-            else if (t === "f32" ||
-                t === "f64") {
-                mem.initialize = "this." + name.text + " = 0.0";
-            }
-            schema.members.push(mem);
-        }
-        let SERIALIZE_RAW = "__SERIALIZE(): string {\n  let out = `{";
-        let SERIALIZE_PRETTY = "__SERIALIZE_PRETTY(): string {\n  let out = `{";
-        let INITIALIZE = "__INITIALIZE(): this {\n";
-        let DESERIALIZE = "__DESERIALIZE(data: string, key_start: i32, key_end: i32, value_start: i32, value_end: i32): boolean {\n  const len = key_end - key_start;\n";
-        let indent = "  ";
-        if (!schema.members.length)
-            return;
-        found = false;
-        if (schema.members[0]?.flags.includes(PropertyFlags.OmitNull)
-            || schema.members[0]?.flags.includes(PropertyFlags.OmitIf)) {
-            SERIALIZE_RAW += schema.members[0]?.serialize;
-            SERIALIZE_PRETTY += "\\n" + schema.members[0]?.serialize;
-        }
-        else {
-            SERIALIZE_RAW += schema.members[0]?.serialize + ",";
-            SERIALIZE_PRETTY += "\\n" + schema.members[0]?.serialize + ",\\n";
-            found = true;
-        }
-        if (schema.members[0]?.initialize)
-            INITIALIZE += "  " + schema.members[0]?.initialize + ";\n";
-        for (let i = 1; i < schema.members.length; i++) {
-            const member = schema.members[i];
-            if (member.initialize)
-                INITIALIZE += "  " + member.initialize + ";\n";
-            if (member.flags.includes(PropertyFlags.OmitNull)
-                || member.flags.includes(PropertyFlags.OmitIf)) {
-                SERIALIZE_RAW += member.serialize;
-                SERIALIZE_PRETTY += member.serialize;
-            }
-            else {
-                SERIALIZE_RAW += member.serialize + ",";
-                SERIALIZE_PRETTY += indent + member.serialize + ",\\n";
-                found = true;
-            }
+      }
+    }
+    if (!members.length) {
+      let SERIALIZE_RAW_EMPTY = '__SERIALIZE(): string {\n  return "{}";\n}';
+      //let SERIALIZE_PRETTY_EMPTY = "__SERIALIZE_PRETTY(): string {\n  return \"{}\";\n}";
+      let INITIALIZE_EMPTY = "__INITIALIZE(): this {\n  return this;\n}";
+      let DESERIALIZE_EMPTY =
+        "__DESERIALIZE(data: string, key_start: i32, key_end: i32, value_start: i32, value_end: i32): boolean {\n  return false;\n}";
+      if (process.env["JSON_DEBUG"]) {
+        console.log(SERIALIZE_RAW_EMPTY);
+        //console.log(SERIALIZE_PRETTY_EMPTY);
+        console.log(INITIALIZE_EMPTY);
+        console.log(DESERIALIZE_EMPTY);
+      }
+      const SERIALIZE_RAW_METHOD_EMPTY = SimpleParser.parseClassMember(
+        SERIALIZE_RAW_EMPTY,
+        node,
+      );
+      //const SERIALIZE_PRETTY_METHOD = SimpleParser.parseClassMember(SERIALIZE_PRETTY, node);
+      const INITIALIZE_METHOD_EMPTY = SimpleParser.parseClassMember(
+        INITIALIZE_EMPTY,
+        node,
+      );
+      const DESERIALIZE_METHOD_EMPTY = SimpleParser.parseClassMember(
+        DESERIALIZE_EMPTY,
+        node,
+      );
+      if (!node.members.find((v) => v.name.text == "__SERIALIZE"))
+        node.members.push(SERIALIZE_RAW_METHOD_EMPTY);
+      if (!node.members.find((v) => v.name.text == "__INITIALIZE"))
+        node.members.push(INITIALIZE_METHOD_EMPTY);
+      if (!node.members.find((v) => v.name.text == "__DESERIALIZE"))
+        node.members.push(DESERIALIZE_METHOD_EMPTY);
+      this.schemasList.push(schema);
+    }
+    for (const member of members) {
+      const name = member.name;
+      if (!(member instanceof FieldDeclaration)) continue;
+      if (!member.type) {
+        throw new Error(
+          "Fields must be strongly typed! Found " +
+            toString(member) +
+            " at " +
+            node.range.source.normalizedPath,
+        );
+      }
+      const type = toString(member.type);
+      if (type.startsWith("(") && type.includes("=>")) continue;
+      const value = member.initializer ? toString(member.initializer) : null;
+      if (member.flags == 32 /* CommonFlags.Static */) continue;
+      if (member.flags === 512 /* CommonFlags.Private */) continue;
+      if (member.flags === 1024 /* CommonFlags.Protected */) continue;
+      const mem = new Property();
+      mem.name = name.text;
+      mem.type = type;
+      mem.value = value;
+      mem.node = member;
+      if (type == "JSON.Raw") {
+        mem.flags.set(PropertyFlags.JSON_Raw, []);
+      }
+      if (member.decorators) {
+        for (const decorator of member.decorators) {
+          const decoratorName = decorator.name.text;
+          const args = getArgs(decorator.args);
+          switch (decoratorName) {
+            case "alias": {
+              if (!args.length)
+                throw new Error(
+                  "Expected 1 argument but got zero at @alias in " +
+                    node.range.source.normalizedPath,
+                );
+              mem.alias = args[0];
+              mem.flags.set(PropertyFlags.Alias, args);
+              break;
+            }
+            case "omit": {
+              mem.flags.set(PropertyFlags.Omit, args);
+              break;
+            }
+            case "omitif": {
+              if (!decorator.args?.length)
+                throw new Error(
+                  "Expected 1 argument but got zero at @omitif in " +
+                    node.range.source.normalizedPath,
+                );
+              mem.flags.set(PropertyFlags.OmitIf, args);
+              break;
+            }
+            case "omitnull": {
+              mem.flags.set(PropertyFlags.OmitNull, args);
+              break;
+            }
+          }
         }
-        if (found) {
-            SERIALIZE_RAW += "`;\n  store<u16>(changetype<usize>(out) + ((out.length - 1) << 1), 125);\n  return out;\n}";
-            SERIALIZE_PRETTY += "`;\n  store<u32>(changetype<usize>(out) + ((out.length - 2) << 1), 8192010);\n  return out;\n}";
+      }
+      mem.generate();
+      if (this.schemasList.find((v) => v.name == type)) {
+        mem.initialize =
+          "this." +
+          name.text +
+          " = changetype<nonnull<" +
+          mem.type +
+          ">>(__new(offsetof<nonnull<" +
+          mem.type +
+          ">>(), idof<nonnull<" +
+          mem.type +
+          ">>()));\n  changetype<nonnull<" +
+          mem.type +
+          ">>(this." +
+          name.text +
+          ").__INITIALIZE()";
+      } else if (mem.value) {
+        mem.initialize = "this." + name.text + " = " + mem.value;
+      } else if (type === "Map") {
+        mem.initialize = "this." + name.text + " = new " + mem.type + "()";
+      } else if (type === "string") {
+        mem.initialize = "this." + name.text + ' = ""';
+      } else if (type === "Array") {
+        mem.initialize =
+          "this." + name.text + " = instantiate<" + mem.type + ">()";
+      } else if (type === "bool" || type === "boolean") {
+        mem.initialize = "this." + name.text + " = false";
+      } else if (type === "JSON.Raw") {
+        mem.initialize = "this." + name.text + ' = ""';
+      } else if (
+        type === "u8" ||
+        type === "u16" ||
+        type === "u32" ||
+        type === "u64" ||
+        type === "i8" ||
+        type === "i16" ||
+        type === "i32" ||
+        type === "i64"
+      ) {
+        mem.initialize = "this." + name.text + " = 0";
+      } else if (type === "f32" || type === "f64") {
+        mem.initialize = "this." + name.text + " = 0.0";
+      }
+      schema.members.push(mem);
+    }
+    let SERIALIZE_RAW = "__SERIALIZE(): string {\n  let out = `{";
+    let SERIALIZE_PRETTY = "__SERIALIZE_PRETTY(): string {\n  let out = `{";
+    let INITIALIZE = "__INITIALIZE(): this {\n";
+    let DESERIALIZE =
+      "__DESERIALIZE(data: string, key_start: i32, key_end: i32, value_start: i32, value_end: i32): boolean {\n  const len = key_end - key_start;\n";
+    let indent = "  ";
+    if (!schema.members.length) return;
+    found = false;
+    if (
+      schema.members[0]?.flags.has(PropertyFlags.OmitNull) ||
+      schema.members[0]?.flags.has(PropertyFlags.OmitIf)
+    ) {
+      SERIALIZE_RAW += schema.members[0]?.serialize;
+      SERIALIZE_PRETTY += "\\n" + schema.members[0]?.serialize;
+    } else {
+      SERIALIZE_RAW += schema.members[0]?.serialize + ",";
+      SERIALIZE_PRETTY += "\\n" + schema.members[0]?.serialize + ",\\n";
+      found = true;
+    }
+    if (schema.members[0]?.initialize)
+      INITIALIZE += "  " + schema.members[0]?.initialize + ";\n";
+    for (let i = 1; i < schema.members.length; i++) {
+      const member = schema.members[i];
+      if (member.initialize) INITIALIZE += "  " + member.initialize + ";\n";
+      if (
+        member.flags.has(PropertyFlags.OmitNull) ||
+        member.flags.has(PropertyFlags.OmitIf)
+      ) {
+        SERIALIZE_RAW += member.serialize;
+        SERIALIZE_PRETTY += member.serialize;
+      } else {
+        SERIALIZE_RAW += member.serialize + ",";
+        SERIALIZE_PRETTY += indent + member.serialize + ",\\n";
+        found = true;
+      }
+    }
+    if (found) {
+      SERIALIZE_RAW +=
+        "`;\n  store<u16>(changetype<usize>(out) + ((out.length - 1) << 1), 125);\n  return out;\n}";
+      SERIALIZE_PRETTY +=
+        "`;\n  store<u32>(changetype<usize>(out) + ((out.length - 2) << 1), 8192010);\n  return out;\n}";
+    } else {
+      SERIALIZE_RAW += "`;\n};";
+      SERIALIZE_PRETTY += "`;\n};";
+    }
+    INITIALIZE += "  return this;\n}";
+    const sortedMembers = [];
+    const _sorted = schema.members.sort(
+      (a, b) => a.name.length - b.name.length,
+    );
+    let len = 0;
+    let offset = 0;
+    sortedMembers.push([_sorted[0]]);
+    len = _sorted[0]?.name.length;
+    for (let i = 1; i < _sorted.length; i++) {
+      const member = _sorted[i];
+      if (member.alias?.length || member.name.length > len) {
+        sortedMembers.push([member]);
+        len = member.alias?.length || member.name.length;
+        offset++;
+      } else {
+        sortedMembers[offset].push(member);
+      }
+    }
+    let first = true;
+    for (const memberSet of sortedMembers) {
+      const firstMember = memberSet[0];
+      const name = encodeKey(firstMember.alias || firstMember.name);
+      if (name.length === 1) {
+        if (first) {
+          DESERIALIZE +=
+            "  if (1 === len) {\n    switch (load<u16>(changetype<usize>(data) + (key_start << 1))) {\n";
+          first = false;
+        } else {
+          DESERIALIZE +=
+            "else if (1 === len) {\n    switch (load<u16>(changetype<usize>(data) + (key_start << 1))) {\n";
         }
-        else {
-            SERIALIZE_RAW += "`;\n};";
-            SERIALIZE_PRETTY += "`;\n};";
+      } else if (name.length === 2) {
+        if (first) {
+          DESERIALIZE +=
+            "  if (2 === len) {\n    switch (load<u32>(changetype<usize>(data) + (key_start << 1))) {\n";
+          first = false;
+        } else {
+          DESERIALIZE +=
+            "else if (2 === len) {\n    switch (load<u32>(changetype<usize>(data) + (key_start << 1))) {\n";
         }
-        INITIALIZE += "  return this;\n}";
-        const sortedMembers = [];
-        const _sorted = schema.members.sort((a, b) => a.name.length - b.name.length);
-        let len = 0;
-        let offset = 0;
-        sortedMembers.push([_sorted[0]]);
-        len = _sorted[0]?.name.length;
-        for (let i = 1; i < _sorted.length; i++) {
-            const member = _sorted[i];
-            if (member.alias?.length || member.name.length > len) {
-                sortedMembers.push([member]);
-                len = member.alias?.length || member.name.length;
-                offset++;
-            }
-            else {
-                sortedMembers[offset].push(member);
-            }
+      } else if (name.length === 4) {
+        if (first) {
+          DESERIALIZE +=
+            "  if (4 === len) {\n    const code = load<u64>(changetype<usize>(data) + (key_start << 1));\n";
+          first = false;
+        } else {
+          DESERIALIZE +=
+            "else if (4 === len) {\n    const code = load<u64>(changetype<usize>(data) + (key_start << 1));\n";
         }
-        let first = true;
-        for (const memberSet of sortedMembers) {
-            const firstMember = memberSet[0];
-            const name = encodeKey(firstMember.alias || firstMember.name);
-            if (name.length === 1) {
-                if (first) {
-                    DESERIALIZE += "  if (1 === len) {\n    switch (load<u16>(changetype<usize>(data) + (key_start << 1))) {\n";
-                    first = false;
-                }
-                else {
-                    DESERIALIZE += "else if (1 === len) {\n    switch (load<u16>(changetype<usize>(data) + (key_start << 1))) {\n";
-                }
-            }
-            else if (name.length === 2) {
-                if (first) {
-                    DESERIALIZE += "  if (2 === len) {\n    switch (load<u32>(changetype<usize>(data) + (key_start << 1))) {\n";
-                    first = false;
-                }
-                else {
-                    DESERIALIZE += "else if (2 === len) {\n    switch (load<u32>(changetype<usize>(data) + (key_start << 1))) {\n";
-                }
-            }
-            else if (name.length === 4) {
-                if (first) {
-                    DESERIALIZE += "  if (4 === len) {\n    const code = load<u64>(changetype<usize>(data) + (key_start << 1));\n";
-                    first = false;
-                }
-                else {
-                    DESERIALIZE += "else if (4 === len) {\n    const code = load<u64>(changetype<usize>(data) + (key_start << 1));\n";
-                }
-            }
-            else {
-                if (first) {
-                    DESERIALIZE += "  if (" + name.length + " === len) {\n";
-                    first = false;
-                }
-                else {
-                    DESERIALIZE += "else if (" + name.length + " === len) {\n";
-                }
-            }
-            let f = true;
-            for (let i = 0; i < memberSet.length; i++) {
-                const member = memberSet[i];
-                if (!member.deserialize)
-                    continue;
-                const name = encodeKey(member.alias || member.name);
-                if (name.length === 1) {
-                    DESERIALIZE += `      case ${name.charCodeAt(0)}: {\n        ${member.deserialize}\n        return true;\n      }\n`;
-                }
-                else if (name.length === 2) {
-                    DESERIALIZE += `      case ${charCodeAt32(name, 0)}: {\n        ${member.deserialize}\n        return true;\n      }\n`;
-                }
-                else if (name.length === 4) {
-                    if (f) {
-                        f = false;
-                        DESERIALIZE += `    if (${charCodeAt64(name, 0)} === code) {\n      ${member.deserialize}\n      return true;\n    }\n`;
-                    }
-                    else {
-                        DESERIALIZE = DESERIALIZE.slice(0, DESERIALIZE.length - 1) + `else if (${charCodeAt64(name, 0)} === code) {\n      ${member.deserialize}\n      return true;\n    }\n`;
-                    }
-                }
-                else {
-                    if (f) {
-                        f = false;
-                        DESERIALIZE += `    if (0 == memory.compare(changetype<usize>("${escapeQuote(escapeSlash(name))}"), changetype<usize>(data) + (key_start << 1), ${name.length << 1})) {\n      ${member.deserialize}\n      return true;\n    }\n`;
-                    }
-                    else {
-                        DESERIALIZE = DESERIALIZE.slice(0, DESERIALIZE.length - 1) + ` else if (0 == memory.compare(changetype<usize>("${escapeQuote(escapeSlash(name))}"), changetype<usize>(data) + (key_start << 1), ${name.length << 1})) {\n      ${member.deserialize}\n      return true;\n    }\n`;
-                    }
-                }
-            }
-            if (name.length < 3) {
-                DESERIALIZE += `      default: {\n        return false;\n      }\n    }\n`;
-            }
-            else if (name.length == 4) {
-                DESERIALIZE = DESERIALIZE.slice(0, DESERIALIZE.length - 1) + ` else {\n      return false;\n    }\n`;
-            }
-            else {
-                DESERIALIZE = DESERIALIZE.slice(0, DESERIALIZE.length - 1) + ` else {\n      return false;\n    }\n`;
-            }
-            DESERIALIZE += "  } ";
+      } else {
+        if (first) {
+          DESERIALIZE += "  if (" + name.length + " === len) {\n";
+          first = false;
+        } else {
+          DESERIALIZE += "else if (" + name.length + " === len) {\n";
         }
-        DESERIALIZE += "\n  return false;\n}";
-        //console.log(sortedMembers);
-        if (process.env["JSON_DEBUG"]) {
-            console.log(SERIALIZE_RAW);
-            //console.log(SERIALIZE_PRETTY);
-            console.log(INITIALIZE);
-            console.log(DESERIALIZE);
+      }
+      let f = true;
+      for (let i = 0; i < memberSet.length; i++) {
+        const member = memberSet[i];
+        if (!member.deserialize) continue;
+        const name = encodeKey(member.alias || member.name);
+        if (name.length === 1) {
+          DESERIALIZE += `      case ${name.charCodeAt(0)}: {\n        ${member.deserialize}\n        return true;\n      }\n`;
+        } else if (name.length === 2) {
+          DESERIALIZE += `      case ${charCodeAt32(name, 0)}: {\n        ${member.deserialize}\n        return true;\n      }\n`;
+        } else if (name.length === 4) {
+          if (f) {
+            f = false;
+            DESERIALIZE += `    if (${charCodeAt64(name, 0)} === code) {\n      ${member.deserialize}\n      return true;\n    }\n`;
+          } else {
+            DESERIALIZE =
+              DESERIALIZE.slice(0, DESERIALIZE.length - 1) +
+              `else if (${charCodeAt64(name, 0)} === code) {\n      ${member.deserialize}\n      return true;\n    }\n`;
+          }
+        } else {
+          if (f) {
+            f = false;
+            DESERIALIZE += `    if (0 == memory.compare(changetype<usize>("${escapeQuote(escapeSlash(name))}"), changetype<usize>(data) + (key_start << 1), ${name.length << 1})) {\n      ${member.deserialize}\n      return true;\n    }\n`;
+          } else {
+            DESERIALIZE =
+              DESERIALIZE.slice(0, DESERIALIZE.length - 1) +
+              ` else if (0 == memory.compare(changetype<usize>("${escapeQuote(escapeSlash(name))}"), changetype<usize>(data) + (key_start << 1), ${name.length << 1})) {\n      ${member.deserialize}\n      return true;\n    }\n`;
+          }
         }
-        const SERIALIZE_RAW_METHOD = SimpleParser.parseClassMember(SERIALIZE_RAW, node);
-        //const SERIALIZE_PRETTY_METHOD = SimpleParser.parseClassMember(SERIALIZE_PRETTY, node);
-        const INITIALIZE_METHOD = SimpleParser.parseClassMember(INITIALIZE, node);
-        const DESERIALIZE_METHOD = SimpleParser.parseClassMember(DESERIALIZE, node);
-        if (!node.members.find(v => v.name.text == "__SERIALIZE"))
-            node.members.push(SERIALIZE_RAW_METHOD);
-        if (!node.members.find(v => v.name.text == "__INITIALIZE"))
-            node.members.push(INITIALIZE_METHOD);
-        if (!node.members.find(v => v.name.text == "__DESERIALIZE"))
-            node.members.push(DESERIALIZE_METHOD);
-        this.schemasList.push(schema);
+      }
+      if (name.length < 3) {
+        DESERIALIZE += `      default: {\n        return false;\n      }\n    }\n`;
+      } else if (name.length == 4) {
+        DESERIALIZE =
+          DESERIALIZE.slice(0, DESERIALIZE.length - 1) +
+          ` else {\n      return false;\n    }\n`;
+      } else {
+        DESERIALIZE =
+          DESERIALIZE.slice(0, DESERIALIZE.length - 1) +
+          ` else {\n      return false;\n    }\n`;
+      }
+      DESERIALIZE += "  } ";
     }
-    visitSource(node) {
-        super.visitSource(node);
-        // Only add the import statement to sources that have JSON decorated classes.
-        if (!this.sources.has(node)) {
-            return;
-        }
+    DESERIALIZE += "\n  return false;\n}";
+    //console.log(sortedMembers);
+    if (process.env["JSON_DEBUG"]) {
+      console.log(SERIALIZE_RAW);
+      //console.log(SERIALIZE_PRETTY);
+      console.log(INITIALIZE);
+      console.log(DESERIALIZE);
+    }
+    const SERIALIZE_RAW_METHOD = SimpleParser.parseClassMember(
+      SERIALIZE_RAW,
+      node,
+    );
+    //const SERIALIZE_PRETTY_METHOD = SimpleParser.parseClassMember(SERIALIZE_PRETTY, node);
+    const INITIALIZE_METHOD = SimpleParser.parseClassMember(INITIALIZE, node);
+    const DESERIALIZE_METHOD = SimpleParser.parseClassMember(DESERIALIZE, node);
+    if (!node.members.find((v) => v.name.text == "__SERIALIZE"))
+      node.members.push(SERIALIZE_RAW_METHOD);
+    if (!node.members.find((v) => v.name.text == "__INITIALIZE"))
+      node.members.push(INITIALIZE_METHOD);
+    if (!node.members.find((v) => v.name.text == "__DESERIALIZE"))
+      node.members.push(DESERIALIZE_METHOD);
+    this.schemasList.push(schema);
+  }
+  visitSource(node) {
+    super.visitSource(node);
+    // Only add the import statement to sources that have JSON decorated classes.
+    if (!this.sources.has(node)) {
+      return;
     }
+  }
 }
 export default class Transformer extends Transform {
-    // Trigger the transform after parse.
-    afterParse(parser) {
-        // Create new transform
-        const transformer = new JSONTransform();
-        // Sort the sources so that user scripts are visited last
-        const sources = parser.sources
-            .filter((source) => !isStdlib(source))
-            .sort((_a, _b) => {
-            const a = _a.internalPath;
-            const b = _b.internalPath;
-            if (a[0] === "~" && b[0] !== "~") {
-                return -1;
-            }
-            else if (a[0] !== "~" && b[0] === "~") {
-                return 1;
-            }
-            else {
-                return 0;
-            }
-        });
-        // Loop over every source
-        for (const source of sources) {
-            // Ignore all lib and std. Visit everything else.
-            if (!isStdlib(source)) {
-                transformer.visit(source);
-            }
-        }
-        // Check that every parent and child class is hooked up correctly
-        const schemas = transformer.schemasList;
-        for (const schema of schemas) {
-            if (schema.parent) {
-                const parent = schemas.find((v) => v.name === schema.parent?.name);
-                if (!parent)
-                    throw new Error(`Class ${schema.name} extends its parent class ${schema.parent}, but ${schema.parent} does not include a @json or @serializable decorator! Add the decorator and rebuild.`);
-            }
+  // Trigger the transform after parse.
+  afterParse(parser) {
+    // Create new transform
+    const transformer = new JSONTransform();
+    // Sort the sources so that user scripts are visited last
+    const sources = parser.sources
+      .filter((source) => !isStdlib(source))
+      .sort((_a, _b) => {
+        const a = _a.internalPath;
+        const b = _b.internalPath;
+        if (a[0] === "~" && b[0] !== "~") {
+          return -1;
+        } else if (a[0] !== "~" && b[0] === "~") {
+          return 1;
+        } else {
+          return 0;
         }
+      });
+    // Loop over every source
+    for (const source of sources) {
+      // Ignore all lib and std. Visit everything else.
+      if (!isStdlib(source)) {
+        transformer.visit(source);
+      }
+    }
+    // Check that every parent and child class is hooked up correctly
+    const schemas = transformer.schemasList;
+    for (const schema of schemas) {
+      if (schema.parent) {
+        const parent = schemas.find((v) => v.name === schema.parent?.name);
+        if (!parent)
+          throw new Error(
+            `Class ${schema.name} extends its parent class ${schema.parent}, but ${schema.parent} does not include a @json or @serializable decorator! Add the decorator and rebuild.`,
+          );
+      }
     }
+  }
 }
 var PropertyFlags;
 (function (PropertyFlags) {
-    PropertyFlags[PropertyFlags["None"] = 0] = "None";
-    PropertyFlags[PropertyFlags["Omit"] = 1] = "Omit";
-    PropertyFlags[PropertyFlags["OmitNull"] = 2] = "OmitNull";
-    PropertyFlags[PropertyFlags["OmitIf"] = 3] = "OmitIf";
-    PropertyFlags[PropertyFlags["Alias"] = 4] = "Alias";
-    PropertyFlags[PropertyFlags["Flatten"] = 5] = "Flatten";
+  PropertyFlags[(PropertyFlags["Omit"] = 0)] = "Omit";
+  PropertyFlags[(PropertyFlags["OmitNull"] = 1)] = "OmitNull";
+  PropertyFlags[(PropertyFlags["OmitIf"] = 2)] = "OmitIf";
+  PropertyFlags[(PropertyFlags["Alias"] = 3)] = "Alias";
+  PropertyFlags[(PropertyFlags["JSON_Raw"] = 4)] = "JSON_Raw";
 })(PropertyFlags || (PropertyFlags = {}));
 class Property {
-    constructor() {
-        this.name = "";
-        this.alias = null;
-        this.type = "";
-        this.value = null;
-        this.flags = [];
-        this.args = [];
-        this.serialize = null;
-        this.deserialize = null;
-        this.initialize = null;
+  constructor() {
+    this.name = "";
+    this.alias = null;
+    this.type = "";
+    this.value = null;
+    this.flags = new Map();
+    this.args = [];
+    this.serialize = null;
+    this.deserialize = null;
+    this.initialize = null;
+    this.right_s = "";
+    this.right_d = "";
+  }
+  generate() {
+    const name = this.name;
+    const escapedName = escapeString(JSON.stringify(this.alias || this.name));
+    const type = this.type;
+    if (this.flags.has(PropertyFlags.Omit)) return;
+    if (this.flags.has(PropertyFlags.JSON_Raw)) {
+      this.right_s = "this." + name;
+      this.right_d = "data.substring(value_start, value_end);";
+    } else {
+      this.right_s = "__SERIALIZE<" + type + ">(this." + name + ")";
+      this.right_d =
+        "__DESERIALIZE<" + type + ">(data.substring(value_start, value_end))";
     }
+    if (this.flags.has(PropertyFlags.OmitIf)) {
+      const condition = this.args[0];
+      this.serialize =
+        "${" +
+        condition +
+        ' ? "" : \'' +
+        escapedName +
+        ":' + " +
+        this.right_s +
+        ' + ","}';
+      this.deserialize = "this." + name + " = " + this.right_d + ";";
+    } else if (this.flags.has(PropertyFlags.OmitNull)) {
+      this.serialize =
+        "${changetype<usize>(this." +
+        name +
+        ") == <usize>0" +
+        ' ? "" : \'' +
+        escapedName +
+        ":' + " +
+        this.right_s +
+        ' + ","}';
+      this.deserialize = "this." + name + " = " + this.right_d + ";";
+    } else {
+      this.serialize = escapedName + ":${" + this.right_s + "}";
+      this.deserialize = "this." + name + " = " + this.right_d + ";";
+    }
+  }
 }
 class SchemaData {
-    constructor() {
-        this.name = "";
-        this.members = [];
-        this.parent = null;
-    }
+  constructor() {
+    this.name = "";
+    this.members = [];
+    this.parent = null;
+  }
 }
 function charCodeAt32(data, offset) {
-    return (data.charCodeAt(offset + 1) << 16) | data.charCodeAt(offset);
+  return (data.charCodeAt(offset + 1) << 16) | data.charCodeAt(offset);
 }
 function charCodeAt64(data, offset) {
-    if (offset + 3 >= data.length) {
-        throw new Error("The string must have at least 4 characters from the specified offset.");
-    }
-    const firstCharCode = BigInt(data.charCodeAt(offset));
-    const secondCharCode = BigInt(data.charCodeAt(offset + 1));
-    const thirdCharCode = BigInt(data.charCodeAt(offset + 2));
-    const fourthCharCode = BigInt(data.charCodeAt(offset + 3));
-    const u64Value = (fourthCharCode << 48n) | (thirdCharCode << 32n) | (secondCharCode << 16n) | firstCharCode;
-    return u64Value;
+  if (offset + 3 >= data.length) {
+    throw new Error(
+      "The string must have at least 4 characters from the specified offset.",
+    );
+  }
+  const firstCharCode = BigInt(data.charCodeAt(offset));
+  const secondCharCode = BigInt(data.charCodeAt(offset + 1));
+  const thirdCharCode = BigInt(data.charCodeAt(offset + 2));
+  const fourthCharCode = BigInt(data.charCodeAt(offset + 3));
+  const u64Value =
+    (fourthCharCode << 48n) |
+    (thirdCharCode << 32n) |
+    (secondCharCode << 16n) |
+    firstCharCode;
+  return u64Value;
 }
 function encodeKey(key) {
-    const data = JSON.stringify(key);
-    return data.slice(1, data.length - 1);
+  const data = JSON.stringify(key);
+  return data.slice(1, data.length - 1);
 }
 function escapeString(data) {
-    return data.replace(/\\/g, "\\\\")
-        .replace(/\`/g, '\\`');
+  return data.replace(/\\/g, "\\\\").replace(/\`/g, "\\`");
 }
 function escapeSlash(data) {
-    return data.replace(/\\/g, "\\\\")
-        .replace(/\`/g, '\\`');
+  return data.replace(/\\/g, "\\\\").replace(/\`/g, "\\`");
 }
 function escapeQuote(data) {
-    return data.replace(/\"/g, "\\\"");
+  return data.replace(/\"/g, '\\"');
+}
+function getArgs(args) {
+  if (!args) return [];
+  let out = [];
+  for (const arg of args) {
+    if (arg instanceof StringLiteralExpression) {
+      out.push(arg.value);
+    }
+  }
+  return out;
 }