Add context variables to compiler

* new `addVariable` to ProtodefCompiler, `setVariable` to CompiledProtodef * / at top of switch field will notate a context variable
ProtoDef-io · May 24, 2021 · ee40de9 · ee40de9
1 parent 4431eef
commit ee40de9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/src/compiler.js b/src/compiler.js
@@ -32,6 +32,12 @@ class ProtoDefCompiler {
     this.sizeOfCompiler.addProtocol(protocolData, path)
   }
 
+  addVariable (key, val) {
+    this.readCompiler.addContextType(key, val)
+    this.writeCompiler.addContextType(key, val)
+    this.sizeOfCompiler.addContextType(key, val)
+  }
+
   compileProtoDefSync (options = { printCode: false }) {
     const sizeOfCode = this.sizeOfCompiler.generate()
     const writeCode = this.writeCompiler.generate()
@@ -70,6 +76,12 @@ class CompiledProtodef {
     return writeFn(value, buffer, cursor)
   }
 
+  setVariable (key, val) {
+    this.sizeOfCtx[key] = val
+    this.readCtx[key] = val
+    this.writeCtx[key] = val
+  }
+
   sizeOf (value, type) {
     const sizeFn = this.sizeOfCtx[type]
     if (!sizeFn) { throw new Error('missing data type: ' + type) }

diff --git a/src/datatypes/compiler-conditional.js b/src/datatypes/compiler-conditional.js
@@ -10,7 +10,8 @@ module.exports = {
       let code = `switch (${compare}) {\n`
       for (const key in struct.fields) {
         let val = key
-        if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
+        if (val.startsWith('/')) val = 'ctx.' + val.slice(1) // Root context variable
+        else if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
         code += compiler.indent(`case ${val}: return ` + compiler.callType(struct.fields[key])) + '\n'
       }
       code += compiler.indent('default: return ' + compiler.callType(struct.default ? struct.default : 'void')) + '\n'
@@ -39,7 +40,8 @@ module.exports = {
       let code = `switch (${compare}) {\n`
       for (const key in struct.fields) {
         let val = key
-        if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
+        if (val.startsWith('/')) val = 'ctx.' + val.slice(1) // Root context variable
+        else if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
         code += compiler.indent(`case ${val}: return ` + compiler.callType('value', struct.fields[key])) + '\n'
       }
       code += compiler.indent('default: return ' + compiler.callType('value', struct.default ? struct.default : 'void')) + '\n'
@@ -69,7 +71,8 @@ module.exports = {
       let code = `switch (${compare}) {\n`
       for (const key in struct.fields) {
         let val = key
-        if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
+        if (val.startsWith('/')) val = 'ctx.' + val.slice(1) // Root context variable
+        else if (isNaN(val) && val !== 'true' && val !== 'false') val = `"${val}"`
         code += compiler.indent(`case ${val}: return ` + compiler.callType('value', struct.fields[key])) + '\n'
       }
       code += compiler.indent('default: return ' + compiler.callType('value', struct.default ? struct.default : 'void')) + '\n'