From 9c072cd8c67c3bc344d2c5014e4a0a5e9971860f Mon Sep 17 00:00:00 2001
From: Edoardo Rosa <6991986+notdodo@users.noreply.github.com>
Date: Mon, 23 Sep 2024 17:56:03 +0200
Subject: [PATCH 1/3] enh: support updates on telegram api

---
 pulumi/__main__.py          |  5 +++
 pulumi/telegram_provider.py | 67 +++++++++++++++++++++++++++++++------
 2 files changed, 61 insertions(+), 11 deletions(-)

diff --git a/pulumi/__main__.py b/pulumi/__main__.py
index 0bb5e85..43b9d9a 100644
--- a/pulumi/__main__.py
+++ b/pulumi/__main__.py
@@ -331,5 +331,10 @@
         f"{RESOURCES_PREFIX}-apigateway-registration",
         token=pulumi.Config().require_secret("telegram-bot-token"),
         authorization_token=telegram_authorization_token,
+        react_on=[
+            "message",
+            "inline_query",
+            "my_chat_member",
+        ],
         url=f"https://{CUSTOM_DOMAIN_NAME}/erfiume_bot",
     )
diff --git a/pulumi/telegram_provider.py b/pulumi/telegram_provider.py
index cf8ac83..cdd90c2 100644
--- a/pulumi/telegram_provider.py
+++ b/pulumi/telegram_provider.py
@@ -8,22 +8,29 @@
 from typing import TYPE_CHECKING, Any
 
 import requests
-from pulumi.dynamic import CreateResult, Resource, ResourceProvider
+from pulumi.dynamic import (
+    CreateResult,
+    ReadResult,
+    Resource,
+    ResourceProvider,
+    UpdateResult,
+)
 
 if TYPE_CHECKING:
     import pulumi
+    from pulumi import ResourceOptions
 
 
 class _TelegramWebhookProvider(ResourceProvider):
+    """Define how to interact with the Telegram API for the Webhook."""
+
     def create(self, props: dict[str, Any]) -> CreateResult:
-        webhook_url = props["url"]
-        token = props["token"]
-        secret_token = props["authorization_token"]
         response = requests.post(
-            f"https://api.telegram.org/bot{token}/setWebhook",
+            f"https://api.telegram.org/bot{props["token"]}/setWebhook",
             json={
-                "url": webhook_url,
-                "secret_token": secret_token,
+                "url": props["url"],
+                "allowed_updates": props["react_on"],
+                "secret_token": props[".authorization_token"],
             },
             timeout=10,
         )
@@ -31,26 +38,64 @@ def create(self, props: dict[str, Any]) -> CreateResult:
             raise requests.RequestException(response.text)
         return CreateResult(id_="-")
 
+    def read(
+        self,
+        id: str,  # noqa: A002
+        props: dict[str, Any],
+    ) -> ReadResult:
+        response = requests.get(
+            f"https://api.telegram.org/bot{props["token"]}/getWebhookInfo",
+            timeout=10,
+        )
+
+        if response.status_code != requests.codes.OK:
+            raise requests.RequestException(response.text)
+        return ReadResult(id, response.json())
+
+    def update(
+        self,
+        _id: str,
+        _oldInputs: dict[str, Any],  # noqa: N803
+        newInputs: dict[str, Any],  # noqa: N803
+    ) -> UpdateResult:
+        response = requests.post(
+            f"https://api.telegram.org/bot{newInputs["token"]}/setWebhook",
+            json={
+                "url": newInputs["url"],
+                "allowed_updates": newInputs["react_on"],
+                "secret_token": newInputs["authorization_token"],
+            },
+            timeout=10,
+        )
+        if response.status_code != requests.codes.OK:
+            raise requests.RequestException(response.text)
+        return UpdateResult(response.json())
+
 
 class Webhook(Resource):
     """
-    Register a Telegram Webhook with a custom URL
+    Pulumi dynamic resource.
     """
 
-    def __init__(
+    def __init__(  # noqa: PLR0913
         self,
         name: str,
         token: str | pulumi.Output[str],
         url: str | pulumi.Output[str],
+        react_on: list[str] | None,
         authorization_token: str | pulumi.Output[str] | None = None,
-    ) -> None:
+        opts: ResourceOptions | None = None,
+    ):
+        if not react_on:
+            react_on = ["message", "inline_query"]
         super().__init__(
             _TelegramWebhookProvider(),
             name,
             {
                 "token": token,
                 "url": url,
+                "react_on": react_on,
                 "authorization_token": authorization_token,
             },
-            None,
+            opts,
         )

From e36a5aac72a6dd6ec26770c29c5d1aa968574f45 Mon Sep 17 00:00:00 2001
From: Edoardo Rosa <6991986+notdodo@users.noreply.github.com>
Date: Mon, 23 Sep 2024 17:56:35 +0200
Subject: [PATCH 2/3] fix: source

---
 pulumi/telegram_provider.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pulumi/telegram_provider.py b/pulumi/telegram_provider.py
index cdd90c2..992fbd7 100644
--- a/pulumi/telegram_provider.py
+++ b/pulumi/telegram_provider.py
@@ -1,6 +1,6 @@
 """
 Custom provider to register the API GW as Telegram webhook.
-source: https://github.com/omerholz/chatbot-example/blob/serverless-telegram-bot/infra/bot_lambda.py
+improved from: https://github.com/omerholz/chatbot-example/blob/serverless-telegram-bot/infra/bot_lambda.py
 """
 
 from __future__ import annotations
@@ -8,6 +8,7 @@
 from typing import TYPE_CHECKING, Any
 
 import requests
+
 from pulumi.dynamic import (
     CreateResult,
     ReadResult,

From f27ebae9659f71fa720dac0c5241d3a0061df7aa Mon Sep 17 00:00:00 2001
From: Edoardo Rosa <6991986+notdodo@users.noreply.github.com>
Date: Mon, 23 Sep 2024 17:58:02 +0200
Subject: [PATCH 3/3] fix: lint

---
 pulumi/telegram_provider.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pulumi/telegram_provider.py b/pulumi/telegram_provider.py
index 992fbd7..9de3793 100644
--- a/pulumi/telegram_provider.py
+++ b/pulumi/telegram_provider.py
@@ -8,7 +8,6 @@
 from typing import TYPE_CHECKING, Any
 
 import requests
-
 from pulumi.dynamic import (
     CreateResult,
     ReadResult,