From 1a95131c3cfa3cd87a5d3448814e633ea2dfae22 Mon Sep 17 00:00:00 2001
From: Piotr Rzysko <piotr.rzysko@gmail.com>
Date: Mon, 8 Sep 2025 06:16:31 +0200
Subject: [PATCH] Include windows in ToTokens for SelectPlan

---
 core/translate/display.rs | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/core/translate/display.rs b/core/translate/display.rs
index 8a83f9a76..36eed99eb 100644
--- a/core/translate/display.rs
+++ b/core/translate/display.rs
@@ -459,6 +459,40 @@ impl ToTokens for SelectPlan {
             }
         }
 
+        if let Some(window) = &self.window {
+            if let Some(window_name) = &window.name {
+                s.append(TokenType::TK_WINDOW, None)?;
+                s.append(TokenType::TK_ID, Some(window_name))?;
+                s.append(TokenType::TK_AS, None)?;
+
+                s.append(TokenType::TK_LP, None)?;
+
+                if !window.partition_by.is_empty() {
+                    s.append(TokenType::TK_PARTITION, None)?;
+                    s.append(TokenType::TK_BY, None)?;
+                    s.comma(window.partition_by.iter(), context)?;
+                }
+
+                if !window.order_by.is_empty() {
+                    s.append(TokenType::TK_ORDER, None)?;
+                    s.append(TokenType::TK_BY, None)?;
+                    s.comma(
+                        window
+                            .order_by
+                            .iter()
+                            .map(|(expr, order)| ast::SortedColumn {
+                                expr: Box::new(expr.clone()),
+                                order: Some(*order),
+                                nulls: None,
+                            }),
+                        context,
+                    )?;
+                }
+
+                s.append(TokenType::TK_RP, None)?;
+            }
+        }
+
         if !self.order_by.is_empty() {
             s.append(TokenType::TK_ORDER, None)?;
             s.append(TokenType::TK_BY, None)?;