From c9d5f6b2f6aa36d286f8f02859bb316c5da6e5e6 Mon Sep 17 00:00:00 2001
From: vvzvlad <git@vvzvlad.xyz>
Date: Fri, 18 Apr 2025 05:06:26 +0300
Subject: [PATCH] refactor PostParser and RSS generator to remove header
 generation and adjust body rendering

---
 post_parser.py   | 14 ++++----------
 rss_generator.py |  6 ++----
 2 files changed, 6 insertions(+), 14 deletions(-)

diff --git a/post_parser.py b/post_parser.py
index 18955b4..ed85da8 100644
--- a/post_parser.py
+++ b/post_parser.py
@@ -484,7 +484,6 @@ class PostParser:
             'text': message.text or message.caption or '',
             'html': {
                 'title': self._generate_title(message),
-                'header': self._generate_html_header(message),
                 'body': self._generate_html_body(message),
                 'media': self._generate_html_media(message),
                 'footer': self.generate_html_footer(message)
@@ -542,17 +541,12 @@ class PostParser:
             logger.error(f"html_sanitization_error: {str(e)}")
             return html_raw
 
-    def _generate_html_header(self, message: Message) -> str:
-        content_header = []
-        # Add forwarded from or reply info if present
-        if forward_html := self._format_forward_info(message): content_header.append(forward_html)
-        elif reply_html := self._format_reply_info(message): content_header.append(reply_html)
-        html_header = '\n'.join(content_header)
-        html_header = self._sanitize_html(html_header)
-        return html_header
-
     def _generate_html_body(self, message: Message) -> str:
         content_body = []
+
+        if forward_html := self._format_forward_info(message): content_body.append(forward_html)
+        elif reply_html := self._format_reply_info(message): content_body.append(reply_html)
+
         if message.text: text = message.text.html
         elif message.caption: text = message.caption.html
         else: text = ''
diff --git a/rss_generator.py b/rss_generator.py
index e46aa2f..2e3c335 100644
--- a/rss_generator.py
+++ b/rss_generator.py
@@ -146,9 +146,8 @@ async def _render_messages_groups(messages_groups, post_parser, exclude_flags: s
             if len(group) == 1: # Single message - simple case
                 message_data = post_parser.process_message(group[0])
                 html_parts = [
-                    f'<div class="message-header">{message_data["html"]["header"]}</div>',
                     f'<div class="message-media">{message_data["html"]["media"]}</div>',
-                    f'<div class="message-text">{message_data["html"]["body"]}</div>',
+                    f'<div class="message-body">{message_data["html"]["body"]}</div>',
                     f'<div class="message-footer">{message_data["html"]["footer"]}</div>'
                 ]
                 rendered_posts.append({
@@ -190,9 +189,8 @@ async def _render_messages_groups(messages_groups, post_parser, exclude_flags: s
                 footer_html = post_parser.generate_html_footer(main_message, flags_list=merged_flags)
 
                 html_parts = [
-                    f'<div class="message-header">{main_message["html"]["header"]}</div>',
                     f'<div class="message-media">{combined_media}</div>',
-                    f'<div class="message-text">{combined_html_body}</div>',
+                    f'<div class="message-body">{combined_html_body}</div>',
                     f'<div class="message-footer">{footer_html}</div>'
                 ]