Add server generation option

Author: AdrienVannson

src/betterproto2_compiler/plugin/parser.py

@@ -14,7 +14,7 @@
     CodeGeneratorResponseFeature,
     CodeGeneratorResponseFile,
 )
-from betterproto2_compiler.settings import ClientGeneration, Settings
+from betterproto2_compiler.settings import ClientGeneration, ServerGeneration, Settings
 
 from .compiler import outputfile_compiler
 from .models import (
@@ -62,22 +62,27 @@ def _traverse(
 def get_settings(plugin_options: list[str]) -> Settings:
     # Synchronous clients are suitable for most users
     client_generation = ClientGeneration.SYNC
+    server_generation = ServerGeneration.NONE
 
     for opt in plugin_options:
         if opt.startswith("client_generation="):
             name = opt.split("=")[1]
-
-            # print(ClientGeneration.__members__, file=sys.stderr)
-            # print([member.value for member in ClientGeneration])
-
             try:
                 client_generation = ClientGeneration(name)
             except ValueError:
                 raise ValueError(f"Invalid client_generation option: {name}")
 
+        if opt.startswith("server_generation="):
+            name = opt.split("=")[1]
+            try:
+                server_generation = ServerGeneration(name)
+            except ValueError:
+                raise ValueError(f"Invalid server_generation option: {name}")
+
     return Settings(
         pydantic_dataclasses="pydantic_dataclasses" in plugin_options,
         client_generation=client_generation,
+        server_generation=server_generation,
     )
 
 

src/betterproto2_compiler/settings.py

@@ -20,17 +20,17 @@ class ClientGeneration(StrEnum):
     SYNC_ASYNC = "sync_async"
     """Both synchronous and asynchronous clients are generated.
 
-    The asynchronous client is generated with the Async suffix."""
+    Asynchronous clients are generated with the Async suffix."""
 
     ASYNC_SYNC = "async_sync"
     """Both synchronous and asynchronous clients are generated.
 
-    The synchronous client is generated with the Sync suffix."""
+    Synchronous clients are generated with the Sync suffix."""
 
     SYNC_ASYNC_NO_DEFAULT = "sync_async_no_default"
     """Both synchronous and asynchronous clients are generated.
 
-    The synchronous client is generated with the Sync suffix, and the asynchronous client is generated with the Async
+    Synchronous clients are generated with the Sync suffix, and asynchronous clients are generated with the Async
     suffix."""
 
     @property
@@ -60,8 +60,14 @@ def is_async_prefixed(self) -> bool:
         return self in {ClientGeneration.SYNC_ASYNC, ClientGeneration.SYNC_ASYNC_NO_DEFAULT}
 
 
+class ServerGeneration(StrEnum):
+    NONE = "none"
+    ASYNC = "async"
+
+
 @dataclass
 class Settings:
     pydantic_dataclasses: bool
 
     client_generation: ClientGeneration
+    server_generation: ServerGeneration

src/betterproto2_compiler/templates/template.py.j2

@@ -102,6 +102,7 @@ default_message_pool.register_message("{{ output_file.package }}", "{{ message.p
 {{ i }}
 {% endfor %}
 
+{% if output_file.settings.server_generation == "async" %}
 {% for _, service in output_file.services|dictsort(by="key") %}
 class {{ service.py_name }}Base(ServiceBase):
     {% if service.comment %}
@@ -170,3 +171,4 @@ class {{ service.py_name }}Base(ServiceBase):
         }
 
 {% endfor %}
+{% endif %}