Update main.py

Author: Jason2866

builder/main.py

@@ -561,6 +561,19 @@ def check_lib_archive_exists():
     sys.stderr.write("Warning! Unknown upload protocol %s\n" % upload_protocol)
 
 
+def print_firmware_metrics(target, source, env):
+    map_file = os.path.join(env.subst("$BUILD_DIR"), "firmware.map")
+    if os.path.isfile(map_file):
+        try:
+            import esp_idf_size
+            print("[INFO] Running esp-idf-size on %s" % map_file)
+            env.Execute("$PYTHONEXE -m esp-idf-size \" \"{map_file}\"")
+        except:
+            pass
+
+env.AlwaysBuild(env.Alias("firmware-metrics", None, print_firmware_metrics))
+
+
 env.AddPlatformTarget("upload", target_firm, upload_actions, "Upload")
 env.AddPlatformTarget("uploadfs", target_firm, upload_actions, "Upload Filesystem Image")
 env.AddPlatformTarget(