Merge pull request #45 from DeepBlueRobotics/only-import-types-when-n…

…ecessary Only Import `wpiws.types.*` When Necessary
DeepBlueRobotics · May 29, 2024 · 7a60199 · 7a60199
2 parents fe36fbd + 4837463
commit 7a60199
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/asyncapi-template/filters/PropertyFilter.js b/asyncapi-template/filters/PropertyFilter.js
@@ -112,4 +112,22 @@ function isRobotOutput(prop) {
     return prop.startsWith("<");
 }
 
-module.exports = { formatPropName, cap1, formatPropType, formatPropObjType, formatPropCallbackType, formatPropParser, formatPropInitialValue, isRobotInput, isRobotOutput }
+function usesCustomTypes(props, schemaName) {
+    for (const [propName, prop] of Object.entries(props)) {
+        if (formatPropParser(prop, schemaName, propName) !== "null") return true;
+    }
+    return false;
+}
+
+module.exports = {
+    formatPropName,
+    cap1,
+    formatPropType,
+    formatPropObjType,
+    formatPropCallbackType,
+    formatPropParser,
+    formatPropInitialValue,
+    isRobotInput,
+    isRobotOutput,
+    usesCustomTypes
+}
diff --git a/asyncapi-template/template/$$schema$$.java b/asyncapi-template/template/$$schema$$.java
@@ -35,7 +35,9 @@
 import org.team199.wpiws.connection.ConnectionProcessor;
 import org.team199.wpiws.connection.WSValue;
 import org.team199.wpiws.interfaces.*;
+{%- if (props | usesCustomTypes(name)) %}
 import org.team199.wpiws.types.*;
+{%- endif %}
 
 /**
  * Represents a simulated {{ name | lower }}