From fed24b52ec70e33decf063f9866e6c9d9fc7243b Mon Sep 17 00:00:00 2001
From: Cyborus <cyborus@cyborus.xyz>
Date: Tue, 16 Jan 2024 18:18:38 -0500
Subject: [PATCH] only check 200 responses for if optional

---
 generator/src/main.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/generator/src/main.rs b/generator/src/main.rs
index 94c1c87..49e540f 100644
--- a/generator/src/main.rs
+++ b/generator/src/main.rs
@@ -495,8 +495,11 @@ fn create_method_response(
 ) -> eyre::Result<String> {
     let mut has_empty = false;
     let mut only_empty = true;
-    for (_, res) in &op.responses.http_codes {
+    for (code, res) in &op.responses.http_codes {
         let name = response_ref_type_name(spec, res)?;
+        if !code.starts_with("2") {
+            continue;
+        }
         if name == "()" || name == "empty" {
             has_empty = true;
         } else {
@@ -509,7 +512,7 @@ fn create_method_response(
     out.push_str("match response.status().as_u16() {\n");
     for (code, res) in &op.responses.http_codes {
         let (_, res) = deref_response(spec, res)?;
-        if code.starts_with("4") {
+        if !code.starts_with("2") {
             continue;
         }
         out.push_str(code);