Update tool to generate small icons table

Author: cedelavergne-ledger

tools/gen_networks.py

@@ -18,104 +18,124 @@ def __init__(self, chain_id: int, name: str, ticker: str):
         self.ticker = ticker
 
 
-def get_network_glyph_name(net: Network) -> str:
-    return "stax_chain_%u_64px" % (net.chain_id)
+def get_network_glyph_name(net: Network, small: bool) -> str:
+    if small:
+        return f"stax_app_chain_{net.chain_id}"
+    return f"stax_chain_{net.chain_id}_64px"
 
 
 def get_header() -> str:
-    return """\
+    return f"""\
 /*
- * Generated by %s
+ * Generated by {sys.argv[0]}
  */
 
-""" % (sys.argv[0])
+"""
 
 
-def gen_icons_array_inc(networks: list[Network], path: str) -> bool:
-    with open(path + ".h", "w") as out:
-        print(get_header() + """\
+def gen_icons_array_inc(networks: int, small_networks: int, path: str) -> None:
+    with open(path + ".h", "w", encoding="utf-8") as out:
+        print(get_header() + f"""\
 #ifndef NETWORK_ICONS_GENERATED_H_
 #define NETWORK_ICONS_GENERATED_H_
 
 #include <stdint.h>
 #include "nbgl_types.h"
 
-typedef struct {
+typedef struct {{
     uint64_t chain_id;
     const nbgl_icon_details_t *icon;
-} network_icon_t;
+}} network_icon_t;
 
-extern const network_icon_t g_network_icons[%u];
+extern const network_icon_t g_network_icons[{networks}];
+extern const network_icon_t g_network_small_icons[{small_networks}];
 
 #endif // NETWORK_ICONS_GENERATED_H_ \
-""" % (len(networks)), file=out)
-    return True
+""", file=out)
 
 
-def gen_icons_array_src(networks: list[Network], path: str) -> bool:
-    with open(path + ".c", "w") as out:
-        print(get_header() + """\
+def gen_network_array_src(networks: list[Network], small: bool, out) -> None:
+
+    if small:
+        print(f"const network_icon_t g_network_small_icons[{len(networks)}] = {{", file=out)
+    else:
+        print(f"const network_icon_t g_network_icons[{len(networks)}] = {{", file=out)
+
+    for net in networks:
+        glyph_name = get_network_glyph_name(net, small)
+        glyph_file = f"glyphs/{glyph_name}.gif"
+        if os.path.isfile(glyph_file):
+            if os.path.islink(glyph_file):
+                glyph_name = Path(os.path.realpath(glyph_file)).stem
+            print(f"    {{.chain_id = {net.chain_id}, .icon = &C_{glyph_name}}}, // {net.name}",
+                    file=out)
+
+    print("};", file=out)
+
+
+def gen_icons_array_src(networks: list[Network], small_networks: list[Network], path: str) -> None:
+    with open(path + ".c", "w", encoding="utf-8") as out:
+        print(get_header() + f"""\
 #include "glyphs.h"
-#include "%s.h"
+#include "{os.path.basename(path)}.h"
+""", file=out)
+# const network_icon_t g_network_icons[{len(networks)}] = {{""", file=out)
 
-const network_icon_t g_network_icons[%u] = {\
-""" % (os.path.basename(path), len(networks)), file=out)
+        gen_network_array_src(networks, False, out)
+        gen_network_array_src(small_networks, True, out)
 
-        for net in networks:
-            glyph_name = get_network_glyph_name(net)
-            glyph_file = "glyphs/%s.gif" % (glyph_name)
-            if os.path.isfile(glyph_file):
-                if os.path.islink(glyph_file):
-                    glyph_name = Path(os.path.realpath(glyph_file)).stem
-                print(" "*4, end="", file=out)
-                print("{.chain_id = %u, .icon = &C_%s}, // %s" % (net.chain_id,
-                                                                  glyph_name,
-                                                                  net.name),
-                      file=out)
+        # for net in networks:
+        #     glyph_name = get_network_glyph_name(net, False)
+        #     glyph_file = f"glyphs/{glyph_name}.gif"
+        #     if os.path.isfile(glyph_file):
+        #         if os.path.islink(glyph_file):
+        #             glyph_name = Path(os.path.realpath(glyph_file)).stem
+        #         print(f"    {{.chain_id = {net.chain_id}, .icon = &C_{glyph_name}}}, // {net.name}",
+        #               file=out)
 
-        print("};", file=out)
-    return True
+        # print("};", file=out)
 
 
-def gen_icons_array(networks: list[Network], path: str) -> bool:
+def gen_icons_array(networks: list[Network], small_networks: list[Network], path: str) -> None:
     path += "/net_icons.gen"
-    if not gen_icons_array_inc(networks, path) or \
-       not gen_icons_array_src(networks, path):
-        return False
-    return True
+    gen_icons_array_inc(len(networks), len(small_networks), path)
+    gen_icons_array_src(networks, small_networks, path)
 
 
 def network_icon_exists(net: Network) -> bool:
-    return os.path.isfile("glyphs/%s.gif" % (get_network_glyph_name(net)))
+    return os.path.isfile(f"glyphs/{get_network_glyph_name(net, False)}.gif")
+
+
+def network_small_icon_exists(net: Network) -> bool:
+    return os.path.isfile(f"glyphs/{get_network_glyph_name(net, True)}.gif")
 
 
 def main(output_dir: str) -> bool:
-    networks: list[Network] = list()
+    networks: list[Network] = []
 
     # get chain IDs and network names
     expr = r"{\.chain_id = ([0-9]*), \.name = \"(.*)\", \.ticker = \"(.*)\"},"
-    with open("src/network.c") as f:
+    with open("src/network.c", encoding="utf-8") as f:
         for line in f.readlines():
             line = line.strip()
             if line.startswith("{") and line.endswith("},"):
                 m = re.search(expr,
                               line)
-                assert(m.lastindex == 3)
+                assert m.lastindex == 3
                 networks.append(Network(int(m.group(1)),
                                         m.group(2),
                                         m.group(3)))
 
     networks.sort(key=lambda x: x.chain_id)
 
-    if not gen_icons_array(list(filter(network_icon_exists, networks)),
-                           output_dir):
-        return False
-    return True
+    gen_icons_array(list(filter(network_icon_exists, networks)),
+                    list(filter(network_small_icon_exists, networks)),
+                    output_dir)
 
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("OUTPUT_DIR")
     args = parser.parse_args()
     assert os.path.isdir(args.OUTPUT_DIR)
-    quit(0 if main(args.OUTPUT_DIR) else 1)
+    main(args.OUTPUT_DIR)