v2 send_upload codegen

Author: blast-hardcheese

src/replit_river/codegen/client.py

@@ -914,18 +914,17 @@ async def {name}(
             ]
         )
     elif procedure.type == "upload":
-        assert input_meta
         assert output_meta
         assert error_meta
-        _, input_type, render_input_method = input_meta
         _, output_type, parse_output_method = output_meta
         _, error_type, parse_error_method = error_meta
         error_type_name = extract_inner_type(error_type)
 
         output_or_error_type = UnionTypeExpr([output_type, error_type_name])
 
-        if init_meta:
+        if init_meta and input_meta:
             _, init_type, render_init_method = init_meta
+            _, input_type, render_input_method = input_meta
             current_chunks.extend(
                 [
                     reindent(
@@ -950,8 +949,9 @@ async def {name}(
                     )
                 ]
             )
-        else:
-            assert protocol_version == "v1.1", "Protocol v2 requires init to be defined"
+        elif protocol_version == "v1.1":
+            assert input_meta, "Protocol v1 requires input to be defined"
+            _, input_type, render_input_method = input_meta
             current_chunks.extend(
                 [
                     reindent(
@@ -977,6 +977,36 @@ async def {name}(
                     )
                 ]
             )
+        elif protocol_version == "v2.0":
+            assert init_meta, "Protocol v2 requires init to be defined"
+            _, init_type, render_init_method = init_meta
+            current_chunks.extend(
+                [
+                    reindent(
+                        "  ",
+                        f"""\
+        async def {name}(
+          self,
+          init: {render_type_expr(init_type)},
+        ) -> {  # TODO(dstewart) This should just be output_type
+                            render_type_expr(output_or_error_type)
+                        }:
+          return await self.client.send_upload(
+            {repr(schema_name)},
+            {repr(name)},
+            init,
+            None,
+            {reindent("                    ", render_init_method)},
+            None,
+            {reindent("                    ", parse_output_method)},
+            {reindent("                    ", parse_error_method)},
+          )
+                    """,
+                    )
+                ]
+            )
+        else:
+            raise ValueError("Precondition failed")
     elif procedure.type == "stream":
         assert output_meta
         assert error_meta