/refresh command

src/controllers/_bot_controller.py

@@ -1,12 +1,12 @@
 import os
-
+import traceback
+import aiohttp
 from fastapi import APIRouter, BackgroundTasks, HTTPException, Request
 from fastapi.responses import JSONResponse, RedirectResponse
 from slack_sdk.web.async_client import AsyncWebClient
-
-from src.models import ContentDeliveryFrequency
+from src.models import ContentDeliveryFrequency, User
 from src.services import FileService, SlackBotService, SlackTeamService, UserService
-from src.utils import logger
+from src.utils import logger, Helper
 
 file_set = set()
 
@@ -19,6 +19,7 @@ class BotController:
         self.router.add_api_route("/auth", self.authenticate, methods=["GET"])
         self.router.add_api_route("/frequency", self.update_frequency, methods=["POST"])
         self.router.add_api_route("/install", self.install, methods=["GET"])
+        self.router.add_api_route("/refresh", self.refreshContent, methods=["POST"])
 
     def get_welcome_message(self, firstname):
 
@@ -297,3 +298,256 @@ class BotController:
 
     def ephemeral_response(self, text):
         return {"response_type": "ephemeral", "text": text}
+
+    async def refreshContent(
+          self, request: Request, background_tasks: BackgroundTasks
+      ):
+          print(">>>>>>>>>>>>>>>>>>>>>")
+          form_data = await request.form()
+          payload = dict(form_data)
+          print(">>>>>>>>>>>>>>>>>>>>>",payload, payload.get("command") )
+          if payload.get("command") != "/refresh":
+               return {
+                   "response_type": "ephemeral",
+                   "text": "Invalid command. Please use /refresh command.",
+               }
+          file_url = payload.get("text", "").strip()
+
+          if not file_url:
+            return {
+                "response_type": "ephemeral",
+                "text": "Please provide a valid file URL with the /refresh command.",
+            }
+          async with Helper as helper:    
+               if not helper.is_valid_url(file_url):
+                    return {
+                        "response_type": "ephemeral",
+                        "text": "Please provide a valid URL (e.g., https://example.com/file.pdf).",
+                    }
+          background_tasks.add_task(self.refreshUserContent, payload)
+          return {
+              "response_type": "ephemeral",
+              "text": "You're very welcome! 🎉 We'll be sharing these insights with you gradually over the next few weeks. 😊"
+          }
+          
+    async def refreshUserContent(self, payload: dict):
+        try:
+            user_id = payload.get("user_id")
+            async with UserService() as user_service:
+                user = await user_service.get_user_by_slack_id(user_id)
+                if not user:
+                    return {
+                        "response_type": "ephemeral",
+                        "text": "User not found. Please make sure you're registered.",
+                    }
+
+                async with SlackTeamService() as slack_team_service:
+                    team = await slack_team_service.get_team_by_slack_team_id(
+                        payload.get("team_id")
+                    )
+                    if not team:
+                        return {
+                            "response_type": "ephemeral",
+                            "text": "Team not found",
+                        }
+
+                    async with FileService() as file_service:
+                        try:
+                            # Download file from URL with timeout
+                            file_url = payload.get("text", "").strip()
+                            async with aiohttp.ClientSession() as session:
+                                try:
+                                    async with session.get(file_url, timeout=30) as response:
+                                        if response.status != 200:
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": f"Failed to download file. Server returned status code: {response.status}",
+                                            }
+
+                                        # Check content type
+                                        content_type = response.headers.get("content-type", "")
+                                        if not any(supported_type in content_type.lower() 
+                                                 for supported_type in ['pdf', 'powerpoint', 'document', 'application']):
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": "Invalid file type. Please provide a valid document file URL.",
+                                            }
+
+                                        file_content = await response.read()
+                                        if not file_content:
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": "The file appears to be empty. Please check the URL and try again.",
+                                            }
+
+                                        file_name = file_url.split('/')[-1]
+
+                                        uploaded_file = {
+                                            "filename": file_name,
+                                            "content": file_content,
+                                            "mimetype": content_type
+                                        }
+
+                                except aiohttp.ClientTimeout:
+                                    return {
+                                        "response_type": "ephemeral",
+                                        "text": "Request timed out. Please check the URL and try again.",
+                                    }
+                                except aiohttp.ClientError as e:
+                                    logger.error(f"Error downloading file: {e}")
+                                    return {
+                                        "response_type": "ephemeral",
+                                        "text": "Failed to download file. Please check the URL and try again.",
+                                    }
+
+                            # Process the file
+                            await file_service.process_file(
+                                file=uploaded_file,
+                                user=user,
+                                slack_file_id=None,
+                                slack_token=team.token,
+                            )
+
+                            await user_service.send_message(
+                                user.id,
+                                "You're very welcome! 🎉 We'll be refreshing your insights based on the new file. You'll receive updated content in your next scheduled delivery. 😊",
+                            )
+
+                        except Exception as e:
+                            logger.error(f"Error processing file: {e}")
+                            return {
+                                "response_type": "ephemeral",
+                                "text": "Failed to process the file. Please check the URL and try again.",
+                            }
+
+        except Exception as e:
+            logger.error(f"Error processing refresh command: {e}")
+            traceback.print_exc()
+
+    async def refreshContent(
+          self, request: Request, background_tasks: BackgroundTasks
+      ):
+          print(">>>>>>>>>>>>>>>>>>>>>")
+          form_data = await request.form()
+          payload = dict(form_data)
+          print(">>>>>>>>>>>>>>>>>>>>>",payload, payload.get("command") )
+          if payload.get("command") != "/refresh":
+               return {
+                   "response_type": "ephemeral",
+                   "text": "Invalid command. Please use /refresh command.",
+               }
+          file_url = payload.get("text", "").strip()
+
+          if not file_url:
+            return {
+                "response_type": "ephemeral",
+                "text": "Please provide a valid file URL with the /refresh command.",
+            }
+          async with Helper as helper:    
+               if not helper.is_valid_url(file_url):
+                    return {
+                        "response_type": "ephemeral",
+                        "text": "Please provide a valid URL (e.g., https://example.com/file.pdf).",
+                    }
+          background_tasks.add_task(self.refreshUserContent, payload)
+          return {
+              "response_type": "ephemeral",
+              "text": "You're very welcome! 🎉 We'll be sharing these insights with you gradually over the next few weeks. 😊"
+          }
+          
+    async def refreshUserContent(self, payload: dict):
+        try:
+            async with UserService() as user_service:
+                user = await user_service.get_user_by_slack_id(user_id)
+                if not user:
+                    return {
+                        "response_type": "ephemeral",
+                        "text": "User not found. Please make sure you're registered.",
+                    }
+
+                async with SlackTeamService() as slack_team_service:
+                    team = await slack_team_service.get_team_by_slack_team_id(
+                        payload.get("team_id")
+                    )
+                    if not team:
+                        return {
+                            "response_type": "ephemeral",
+                            "text": "Team not found",
+                        }
+
+                    async with FileService() as file_service:
+                        try:
+                            # Download file from URL with timeout
+                            file_url = payload.get("text", "").strip()
+                            async with aiohttp.ClientSession() as session:
+                                try:
+                                    async with session.get(file_url, timeout=30) as response:
+                                        if response.status != 200:
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": f"Failed to download file. Server returned status code: {response.status}",
+                                            }
+
+                                        # Check content type
+                                        content_type = response.headers.get("content-type", "")
+                                        if not any(supported_type in content_type.lower() 
+                                                 for supported_type in ['pdf', 'powerpoint', 'document', 'application']):
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": "Invalid file type. Please provide a valid document file URL.",
+                                            }
+
+                                        file_content = await response.read()
+                                        if not file_content:
+                                            return {
+                                                "response_type": "ephemeral",
+                                                "text": "The file appears to be empty. Please check the URL and try again.",
+                                            }
+
+                                        file_name = file_url.split('/')[-1]
+
+                                        uploaded_file = {
+                                            "filename": file_name,
+                                            "content": file_content,
+                                            "mimetype": content_type
+                                        }
+
+                                except aiohttp.ClientTimeout:
+                                    return {
+                                        "response_type": "ephemeral",
+                                        "text": "Request timed out. Please check the URL and try again.",
+                                    }
+                                except aiohttp.ClientError as e:
+                                    logger.error(f"Error downloading file: {e}")
+                                    return {
+                                        "response_type": "ephemeral",
+                                        "text": "Failed to download file. Please check the URL and try again.",
+                                    }
+
+                            # Process the file
+                            await file_service.process_file(
+                                file=uploaded_file,
+                                user=user,
+                                slack_file_id=None,
+                                slack_token=team.token,
+                            )
+
+                            await user_service.send_message(
+                                user.id,
+                                "You're very welcome! 🎉 We'll be refreshing your insights based on the new file. You'll receive updated content in your next scheduled delivery. 😊",
+                            )
+
+                        except Exception as e:
+                            logger.error(f"Error processing file: {e}")
+                            return {
+                                "response_type": "ephemeral",
+                                "text": "Failed to process the file. Please check the URL and try again.",
+                            }
+
+        except Exception as e:
+            logger.error(f"Error processing refresh command: {e}")
+            traceback.print_exc()
+            return {
+                "response_type": "ephemeral",
+                "text": "An error occurred while processing your request. Please try again.",
+            }