From 3d5d9574e359616edea21d325c403174f58707fc Mon Sep 17 00:00:00 2001
From: HumanoidSandvichDispenser <humanoidsandvichdispenser@gmail.com>
Date: Fri, 29 Nov 2024 00:04:46 -0800
Subject: [PATCH] feat: Add webhook profile picture support

- Added `webhookBotProfilePicture` field to
  `TeamDiscordIntegrationSchema`.
- Updated `DiscordIntegrationForm.vue` to include profile picture input.
- Created migration to add `webhook_bot_profile_picture` column.
- Updated `Team` model to handle new profile picture field.
- Modified `TeamDiscordIntegration` and `TeamDiscordIntegrationSchema`
  to include the new profile picture attribute.
---
 .../models/TeamDiscordIntegrationSchema.ts    |  1 +
 .../src/components/DiscordIntegrationForm.vue |  5 +++
 ...242e3f99c64_add_webhook_profile_picture.py | 32 +++++++++++++++++++
 backend-flask/models/team.py                  |  5 ++-
 backend-flask/models/team_integration.py      |  4 +++
 5 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 backend-flask/migrations/versions/c242e3f99c64_add_webhook_profile_picture.py

diff --git a/availabili.tf/src/client/models/TeamDiscordIntegrationSchema.ts b/availabili.tf/src/client/models/TeamDiscordIntegrationSchema.ts
index 32b5241..3657f8e 100644
--- a/availabili.tf/src/client/models/TeamDiscordIntegrationSchema.ts
+++ b/availabili.tf/src/client/models/TeamDiscordIntegrationSchema.ts
@@ -4,6 +4,7 @@
 /* eslint-disable */
 export type TeamDiscordIntegrationSchema = {
     webhookBotName: string;
+    webhookBotProfilePicture: (string | null);
     webhookUrl: string;
 };
 
diff --git a/availabili.tf/src/components/DiscordIntegrationForm.vue b/availabili.tf/src/components/DiscordIntegrationForm.vue
index 0eee1b8..af68f47 100644
--- a/availabili.tf/src/components/DiscordIntegrationForm.vue
+++ b/availabili.tf/src/components/DiscordIntegrationForm.vue
@@ -16,6 +16,7 @@ function enableIntegration() {
   model.value = {
     webhookUrl: "",
     webhookBotName: "",
+    webhookBotProfilePicture: null,
   };
   saveIntegration();
 }
@@ -38,6 +39,10 @@ function disableIntegration() {
       <h3>Webhook Bot Name</h3>
       <input v-model="model.webhookBotName">
     </div>
+    <div class="form-group margin">
+      <h3>Webhook Bot Profile Picture URL (optional)</h3>
+      <input v-model="model.webhookBotProfilePicture">
+    </div>
     <div class="form-group margin">
       <div class="action-buttons">
         <button class="destructive-on-hover" @click="disableIntegration">
diff --git a/backend-flask/migrations/versions/c242e3f99c64_add_webhook_profile_picture.py b/backend-flask/migrations/versions/c242e3f99c64_add_webhook_profile_picture.py
new file mode 100644
index 0000000..1e93c0e
--- /dev/null
+++ b/backend-flask/migrations/versions/c242e3f99c64_add_webhook_profile_picture.py
@@ -0,0 +1,32 @@
+"""Add webhook profile picture
+
+Revision ID: c242e3f99c64
+Revises: 286ee26b9e5d
+Create Date: 2024-11-27 10:40:39.027786
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'c242e3f99c64'
+down_revision = '286ee26b9e5d'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('team_discord_integrations', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('webhook_bot_profile_picture', sa.String(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('team_discord_integrations', schema=None) as batch_op:
+        batch_op.drop_column('webhook_bot_profile_picture')
+
+    # ### end Alembic commands ###
diff --git a/backend-flask/models/team.py b/backend-flask/models/team.py
index bf2893b..6cf5042 100644
--- a/backend-flask/models/team.py
+++ b/backend-flask/models/team.py
@@ -36,7 +36,6 @@ class Team(app_db.BaseModel):
 
     def update_integrations(self, integrations: "TeamIntegrationSchema"):
         if integrations.discord_integration:
-            print("DISCORD!!!")
             discord_integration = self.discord_integration \
                 or TeamDiscordIntegration()
             discord_integration.webhook_url = integrations \
@@ -44,6 +43,10 @@ class Team(app_db.BaseModel):
             discord_integration.webhook_bot_name = integrations \
                 .discord_integration.webhook_bot_name
 
+            if integrations.discord_integration.webhook_bot_profile_picture:
+                discord_integration.webhook_bot_profile_picture = integrations \
+                    .discord_integration.webhook_bot_profile_picture
+
             if discord_integration.team_id is None:
                 discord_integration.team_id = self.id
                 app_db.db.session.add(discord_integration)
diff --git a/backend-flask/models/team_integration.py b/backend-flask/models/team_integration.py
index d860d28..a40a729 100644
--- a/backend-flask/models/team_integration.py
+++ b/backend-flask/models/team_integration.py
@@ -1,3 +1,4 @@
+from typing import Optional
 from sqlalchemy.orm import mapped_column, relationship
 from sqlalchemy.orm.attributes import Mapped
 from sqlalchemy.orm.properties import ForeignKey
@@ -12,18 +13,21 @@ class TeamDiscordIntegration(app_db.BaseModel):
     team_id: Mapped[int] = mapped_column(ForeignKey("teams.id"), primary_key=True)
     webhook_url: Mapped[str] = mapped_column(String)
     webhook_bot_name: Mapped[str] = mapped_column(String)
+    webhook_bot_profile_picture: Mapped[str] = mapped_column(String(255), nullable=True)
 
     team: Mapped["Team"] = relationship("Team", back_populates="discord_integration")
 
 class TeamDiscordIntegrationSchema(spec.BaseModel):
     webhook_url: str
     webhook_bot_name: str
+    webhook_bot_profile_picture: Optional[str]
 
     @classmethod
     def from_model(cls, model: TeamDiscordIntegration) -> "TeamDiscordIntegrationSchema":
         return cls(
             webhook_url=model.webhook_url,
             webhook_bot_name=model.webhook_bot_name,
+            webhook_bot_profile_picture=model.webhook_bot_profile_picture,
         )
 
 class TeamLogsTfIntegration(app_db.BaseModel):