1 files changed, 65 insertions, 0 deletions
diff --git a/templates/nodemcu-arduino.py b/templates/nodemcu-arduino.py
new file mode 100644
index 0000000..1f258e6
--- /dev/null
+++ b/templates/nodemcu-arduino.py
@@ -0,0 +1,65 @@
+global generateHeader
+
+gpio_map = {
+    # Left side
+    1: None,  # Vin
+    2: None,  # GND
+    3: None,  # RST
+    4: None,  # EN
+    5: None,  # 3V3
+    6: None,  # GND
+    7: None,  # CLK
+    8: None,  # SD0
+    9: None,  # CMD
+    10: None,  # SD1
+    11: '9',  # SD2
+    12: '10',  # SD3
+    13: None,  # RSV
+    14: None,  # RSV
+    15: '0',  # A0
+
+    # Right side
+    16: '16',
+    17: '5',
+    18: '4',
+    19: '0',
+    20: '2',
+    21: None, # 3V3
+    22: None, # Gnd
+    23: '14',
+    24: '12',
+    25: '13',
+    26: '15',
+    27: '3',
+    28: '1',
+    29: None, # GND
+    30: None, # 3V3
+}
+
+usages = generateHeader.netlist.find_nets_using_ref(generateHeader.ref)
+
+entries = []
+
+for usage in usages:
+    if usage.name is None:
+        continue
+
+    node = usage.node_for_ref(generateHeader.ref)
+
+    name = gpio_map.get(node.pin)
+
+    if name is None:
+        continue
+
+    entries.append([usage.name, name])
+
+generateHeader.println("""
+namespace schematic {
+""")
+
+for x in sorted(entries, key=lambda e: e[0]):
+    generateHeader.println("static const int " + x[0]+ " = " + x[1] + ";")
+
+generateHeader.println("""
+} // namespace schematic
+""")