From 52b52aa53623d535689d71c033eeafc6a18ece26 Mon Sep 17 00:00:00 2001
From: Quantum <quantum2048@gmail.com>
Date: Tue, 29 Oct 2024 01:57:28 -0400
Subject: [PATCH] Use bird sets instead of cases

---
 aspa/generate.py | 15 ++++-----------
 1 file changed, 4 insertions(+), 11 deletions(-)

diff --git a/aspa/generate.py b/aspa/generate.py
index ae2bc65..b2c92a3 100644
--- a/aspa/generate.py
+++ b/aspa/generate.py
@@ -8,18 +8,11 @@ def generate_bird(aspas: list[ASPA]) -> str:
     ]
 
     for aspa in aspas:
-        if not aspa.providers:
+        if aspa.providers:
+            asns = ', '.join(map(str, aspa.providers))
+            lines.append(f'        {aspa.customer}: if upstream_asn !~ [{asns}] then return true;')
+        else:
             lines.append(f'        {aspa.customer}: return true;')
-            continue
-
-        lines.append(f'        {aspa.customer}: case upstream_asn {{')
-        for provider in aspa.providers:
-            lines.append(f'            {provider}: {{}}')
-
-        lines += [
-            '            else: return true;',
-            '        }'
-        ]
 
     lines += [
         '    }',