From 358b55e23cf1a5eda5c333ca8cc84b8bee51b6af Mon Sep 17 00:00:00 2001 From: Eroax Date: Fri, 19 Jan 2024 16:54:57 -0700 Subject: [PATCH] Updated NoTwitch Addon to add EventSub Integration --- addons/no_twitch/chat_socket.gd | 4 +- addons/no_twitch/demo/Chat_Join.gd | 2 +- addons/no_twitch/demo/demo_scene.gd | 40 +++++ addons/no_twitch/demo/demo_scene.tscn | 15 ++ addons/no_twitch/demo/test_button.gd | 2 +- addons/no_twitch/demo/token_saver.gd | 4 + addons/no_twitch/eventsub_socket.gd | 70 +++++++++ addons/no_twitch/twitch_connection.gd | 211 ++++++++++++++++++++++++-- 8 files changed, 332 insertions(+), 16 deletions(-) create mode 100644 addons/no_twitch/demo/token_saver.gd create mode 100644 addons/no_twitch/eventsub_socket.gd diff --git a/addons/no_twitch/chat_socket.gd b/addons/no_twitch/chat_socket.gd index eb7f1dc..bdc52cf 100644 --- a/addons/no_twitch/chat_socket.gd +++ b/addons/no_twitch/chat_socket.gd @@ -1,4 +1,5 @@ -extends "res://addons/no_twitch/websocket_client.gd" +extends Websocket_Client +class_name Chat_Socket ## Wrapper class around [Websocket_Client] that handles Twitch Chat @@ -77,7 +78,6 @@ func parse_chat_msg(msg : String, tags : Dictionary): return - var msg_notice = msg.split(" ")[1] match msg_notice: diff --git a/addons/no_twitch/demo/Chat_Join.gd b/addons/no_twitch/demo/Chat_Join.gd index 6bc2608..aa3998a 100644 --- a/addons/no_twitch/demo/Chat_Join.gd +++ b/addons/no_twitch/demo/Chat_Join.gd @@ -29,5 +29,5 @@ func join_chat(): func send_chat(chat : String): - %Twitch_Connection.send_chat_to_channel(chat, %Channel.text) + %Twitch_Connection.send_chat(chat, %Channel.text) diff --git a/addons/no_twitch/demo/demo_scene.gd b/addons/no_twitch/demo/demo_scene.gd index 1af75dd..e03a31d 100644 --- a/addons/no_twitch/demo/demo_scene.gd +++ b/addons/no_twitch/demo/demo_scene.gd @@ -1,8 +1,38 @@ extends Control +var demo_events : Array[Twitch_Connection.EventSub_Subscription] = [ + + Twitch_Connection.EventSub_Subscription.new("channel.update", {"broadcaster_user_id" : ""}, "", "2") + +] + func _ready(): $Twitch_Connection.token_received.connect(save_token) + %Get_User.pressed.connect(func(): + + var resp = %Twitch_Connection.request_user_info() + resp.response_received.connect(print_http_result) + + ) + + %Get_Channel.pressed.connect(func(): + + var resp = %Twitch_Connection.request_channel_info() + resp.response_received.connect(print_http_result) + + ) + + %Connect_Channel_Points.pressed.connect(func(): + + await %Twitch_Connection.setup_eventsub_connection() + print("Passed Await") + %Twitch_Connection.eventsub_socket.notif_received.connect(eventsub_notif_received) + + var resp = %Twitch_Connection.subscribe_to_channel_points(%Twitch_Connection.user_info.id) + resp.response_received.connect(print_http_result) + + ) load_token() @@ -23,5 +53,15 @@ func load_token(): var res = ResourceLoader.load("user://token.tres") $Twitch_Connection.token = res.token + %Twitch_Connection.cache_user_data() +func print_http_result(data): + + print(data) + + +func eventsub_notif_received(info): + print(info) + printt(info.payload.event.reward.id, info.payload.event.user_input, info.payload.event.reward.title) + diff --git a/addons/no_twitch/demo/demo_scene.tscn b/addons/no_twitch/demo/demo_scene.tscn index a636685..c3435cd 100644 --- a/addons/no_twitch/demo/demo_scene.tscn +++ b/addons/no_twitch/demo/demo_scene.tscn @@ -76,6 +76,21 @@ unique_name_in_owner = true layout_mode = 2 text = "Send" +[node name="Get_User" type="Button" parent="VBoxContainer"] +unique_name_in_owner = true +layout_mode = 2 +text = "Get User Info" + +[node name="Get_Channel" type="Button" parent="VBoxContainer"] +unique_name_in_owner = true +layout_mode = 2 +text = "Get Channel Info" + +[node name="Connect_Channel_Points" type="Button" parent="VBoxContainer"] +unique_name_in_owner = true +layout_mode = 2 +text = "Connect Channel Points" + [node name="Twitch_Connection" type="Node" parent="."] unique_name_in_owner = true script = ExtResource("1_13a4v") diff --git a/addons/no_twitch/demo/test_button.gd b/addons/no_twitch/demo/test_button.gd index 7592510..61a1962 100644 --- a/addons/no_twitch/demo/test_button.gd +++ b/addons/no_twitch/demo/test_button.gd @@ -5,5 +5,5 @@ extends Button func _pressed(): - twitch_connection.authenticate_with_twitch() + OS.shell_open(twitch_connection.authenticate_with_twitch(["channel:read:redemptions", "chat:read", "chat:edit"])) diff --git a/addons/no_twitch/demo/token_saver.gd b/addons/no_twitch/demo/token_saver.gd new file mode 100644 index 0000000..b7e1d43 --- /dev/null +++ b/addons/no_twitch/demo/token_saver.gd @@ -0,0 +1,4 @@ +extends Resource +class_name TokenSaver + +@export var token : String diff --git a/addons/no_twitch/eventsub_socket.gd b/addons/no_twitch/eventsub_socket.gd new file mode 100644 index 0000000..682db60 --- /dev/null +++ b/addons/no_twitch/eventsub_socket.gd @@ -0,0 +1,70 @@ +extends Websocket_Client + +const eventsub_url := "wss://eventsub.wss.twitch.tv/ws" + +## Stores the "session id" for this EventSub connection. +var session_id + +var connection : Twitch_Connection + +signal notif_received(data) + +signal welcome_received() + + +func _init(owner): + + connection = owner + + packet_received.connect(data_received) + + +func connect_to_eventsub(events : Array[Twitch_Connection.EventSub_Subscription] = []): + + connect_to_url(eventsub_url) + await welcome_received + + if events.is_empty(): + + return + + + var responses : Array[Twitch_Connection.HTTPResponse] + + for all in events: + + responses.append(connection.add_eventsub_subscription(all)) + + + return responses + + +func data_received(packet : PackedByteArray): + + var info = JSON.parse_string(packet.get_string_from_utf8()) + + match info.metadata.message_type: + + "session_welcome": + + session_id = info.payload.session.id + welcome_received.emit() + + + "session_ping": + + send_pong(info) + + + "notification": + + notif_received.emit(info) + + + + +func send_pong(pong): + + pong.metadata.message_type = "session_pong" + send_text(str(pong)) + diff --git a/addons/no_twitch/twitch_connection.gd b/addons/no_twitch/twitch_connection.gd index 437dfb0..65d8c30 100644 --- a/addons/no_twitch/twitch_connection.gd +++ b/addons/no_twitch/twitch_connection.gd @@ -12,45 +12,74 @@ signal token_received(token : String) ## Port that the redirect_uri will head to on your local system. Defaults to 80 for most cases. Linux tends to prefer 8000 or possibly 1338 @export var redirect_port := "8000" -var twitch_url := "https://id.twitch.tv/oauth2/authorize?response_type=token&" +var auth_url := "https://id.twitch.tv/oauth2/authorize?response_type=token&" var auth_server : TCPServer -## Websocket used for handling the chat connection. -var chat_socket = preload("res://addons/no_twitch/chat_socket.gd").new() + +const chat_socket_class = preload("res://addons/no_twitch/chat_socket.gd") +const eventsub_socket_class = preload("res://addons/no_twitch/eventsub_socket.gd") +## [Websocket_Client] used for receiving all data from Twitch pertaining to chat (Ex. IRC Chat Messages) +var chat_socket : Websocket_Client +## [Websocket_Client] handles all the data from Twitch pertaining to EventSub (Ex. General Alerts) +var eventsub_socket : Websocket_Client var state := make_state() var token : String +## The User data of the account that the authorized [member token] is connected to. +var user_info : Dictionary + +var responses : Array + func _ready(): - chat_socket.chat_received.connect(check_chat_socket) + token_received.connect(cache_user_data.unbind(1)) + + +## Function that handles Caching the data of the Account tied to the Connections Token +func cache_user_data(): + + if token.is_empty(): + + return + + + var resp = request_user_info() + resp.response_received.connect(func(data): + + user_info = data + print("User Info Cached") + + ) + ## Handles the basic Twitch Authentication process to request and then later receive a Token (using [method check_auth_peer]). ## Returns the authentication URL. -func authenticate_with_twitch(client_id = client_id) -> String: +func authenticate_with_twitch(scopes : Array[String] = ["chat:read", "chat:edit"], client_id = client_id) -> String: auth_server = TCPServer.new() - var url := create_auth_url() + var url := create_auth_url(scopes) auth_server.listen(int(redirect_port)) return url -## Sets up a single chat connection. Joining 1 room with [param token] as it's "PASS" specifying what account it's on. And the optional [param default_chat] specifying the default room to join. While [param nick] specifies the "nickname" (Not the username on Twitch) +## Sets up the chat connection. Joining 1 room with [param token] as it's "PASS" specifying what account it's on. And the optional [param default_chat] specifying the default room to join. While [param nick] specifies the "nickname" (Not the username on Twitch) func setup_chat_connection(default_chat : String = "", token : String = token, request_twitch_info = true, nick = "terribletwitch"): - # Temporary, closes the socket to allow connecting multiple times without needing to reset. - chat_socket.close() + chat_socket = chat_socket_class.new() + chat_socket.chat_received_rich.connect(check_chat_socket) + # Connects to the Twitch IRC server. chat_socket.connect_to_chat(token, request_twitch_info) await chat_socket.socket_open if !default_chat.is_empty(): - Connections.save_credentials({"token" : token, "channel" : default_chat}, "twitch") chat_socket.join_chat(default_chat) + ## Joins the given [param channel] over IRC. Essentially just sending JOIN #[param channel] func join_channel(channel : String): @@ -62,6 +91,30 @@ func send_chat(msg : String, channel : String = ""): chat_socket.send_chat(msg, channel) +## Sets up an EventSub connection to allow subscribing to EventSub events. Ex. Alerts, Channel Point Redemptions etc. +func setup_eventsub_connection(events : Array[EventSub_Subscription] = [], timeout_duration : int = 20): + + eventsub_socket = eventsub_socket_class.new(self) + var ret = await eventsub_socket.connect_to_eventsub(events) + return ret + + +func add_eventsub_subscription(sub_type : EventSub_Subscription): + + return twitch_request("https://api.twitch.tv/helix/eventsub/subscriptions", HTTPClient.METHOD_POST, str(sub_type.return_request_dictionary())) + + +func subscribe_to_channel_points(user_id : String): + + if !eventsub_socket.socket_open: + + push_error("NoTwitch Error: No EventSub Connection, please use setup_eventsub_connection first") + return + + + return add_eventsub_subscription(EventSub_Subscription.new("channel.channel_points_custom_reward_redemption.add", {"broadcaster_user_id" : user_id}, eventsub_socket.session_id)) + + func _process(delta): if auth_server and auth_server.is_listening() and auth_server.is_connection_available(): @@ -74,12 +127,17 @@ func _process(delta): chat_socket.poll_socket() + if eventsub_socket: + + eventsub_socket.poll_socket() + + ## Utility function for creating a Twitch Authentication URL with the given ## [param scopes], Twitch Client ID ([param id]) and [param redirect_uri]. ## [param id] defaults to [member client_id] and both [param redirect] and ## [param redirect_port] -func create_auth_url(scopes : Array[String] = ["chat:read", "chat:edit"], port := redirect_port, id : String = client_id, redirect : String = redirect_uri) -> String: +func create_auth_url(scopes : Array[String], port := redirect_port, id : String = client_id, redirect : String = redirect_uri) -> String: var str_scopes : String @@ -92,10 +150,78 @@ func create_auth_url(scopes : Array[String] = ["chat:read", "chat:edit"], port : if !port.is_empty(): full_redirect_uri += ":" + port - var url := twitch_url + "client_id=" + id + "&redirect_uri=" + full_redirect_uri + "&scope=" + str_scopes + "&state=" + str(state) + var url := auth_url + "client_id=" + id + "&redirect_uri=" + full_redirect_uri + "&scope=" + str_scopes + "&state=" + str(state) return url +## Utility Function for making a generic HTTP Request to Twitch +func twitch_request(url : String, method : HTTPClient.Method = HTTPClient.METHOD_GET, body := ""): + + var headers : Array = ["Authorization: Bearer " + token, "Client-Id: " + client_id] + + if !body.is_empty(): + + headers.append("Content-Type: application/json") + + +# Adds the Content type to the headers if we're actually sending some data along + if method != HTTPClient.METHOD_GET: + + headers.append("Content-Type: application/json") + + + var http := HTTPRequest.new() + + var resp = HTTPResponse.new(http, responses) + add_child(http) + http.request(url, headers, method, body) + + return resp + + +## Wrapper function around [method twitch_request] to handle doing a Twitch "Get Users" request. +func request_user_info(users : Array[String] = []): + + var req_url := "https://api.twitch.tv/helix/users" + + if users.is_empty(): + + return twitch_request(req_url) + + + var user_list : String = "?login=" + users[0] + for all in users: + + user_list += "&login=" + all + + + return twitch_request(req_url + user_list) + + +## Wrapper function around [method twitch_request] to grab information about a set of channels based off an Array of Channel IDs Ex. stream title, current category, current tags, etc. +func request_channel_info(channels : Array[String] = []): + + var req_url := "https://api.twitch.tv/helix/channels?broadcaster_id=" + + if channels.is_empty(): + + if user_info.is_empty(): + + return + + + channels.append(user_info.id) + + + var id_string := channels[0] + for all in channels: + + id_string += "&broadcaster_id=" + all + + + return twitch_request(req_url + id_string) + + ## Utility function for creating a "state" used for different requests for some extra security as a semi password. func make_state(len : int = 16) -> String: @@ -140,3 +266,64 @@ func check_chat_socket(dict): + +## "Promise" Class used for relaying the info received from the supplied [HTTPRequest] node. +class HTTPResponse: + + signal response_received + + func _init(http : HTTPRequest, storage : Array): + + storage.append(self) + http.request_completed.connect(request_complete.bind(http, storage)) + + + func request_complete(result: int, response_code: int, headers: PackedStringArray, body: PackedByteArray, http, storage): + + var info = JSON.parse_string(body.get_string_from_utf8()) + + if info.has("error"): + + push_error("NoTwitch Twitch API Error: " + info.error + " " + info.message) + return + + info = info.data[0] + + print("Response Received") + response_received.emit(info) + + http.queue_free() + storage.erase(self) + + + +## Data handler class for making it easier to send the data used for EventSub Subscriptions over [member eventsub_socket] +class EventSub_Subscription: + + ## Specifies the type of subscription this is representing (Ex. "Channel Update" is channel.update) + var subscription_type : String + ## Used for setting the "version" the subscription that we're currently using. + var version : String + ## Stores the "condition", AKA the condition for being able to subscribe to this Event. (Ex. The User ID of the channel you're listening to) + var condition : Dictionary + ## Holds the "session" ID for this EventSub Subscription. Which is held in [member eventsub_socket.session_id] + var session_id : String + + var method = "websocket" + + func _init(sub_type : String, cond : Dictionary, sess_id : String, vers : String = "1"): + + subscription_type = sub_type + version = vers + condition = cond + session_id = sess_id + + + ## Returns back the information pertaining to the Subscription in the format needed for twitch. + func return_request_dictionary(): + + var dict = {"type" : subscription_type, "version" : version, "condition" : condition, "transport" : {"method" : method, "session_id" : session_id}} + + return dict + +