From 13b3bdcb97da839a2fa0dbb1efd199bfd3c892fb Mon Sep 17 00:00:00 2001 From: alexbcberio Date: Fri, 18 Jun 2021 18:35:34 +0200 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Extract=20pubSubClient?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- index.ts | 136 ++---------------------------- src/backend/helpers/twitch.ts | 14 ++- src/backend/pubSubClient/index.ts | 122 +++++++++++++++++++++++++++ 3 files changed, 143 insertions(+), 129 deletions(-) create mode 100644 src/backend/pubSubClient/index.ts diff --git a/index.ts b/index.ts index 03a7c17..66a261c 100644 --- a/index.ts +++ b/index.ts @@ -1,140 +1,20 @@ -import { PubSubClient, PubSubRedemptionMessage } from "twitch-pubsub-client"; -import { broadcast, chatClient, connect, say, } from "./src/backend/chatClient"; -import { getApiClient, getAuthProvider } from "./src/backend/helpers/twitch"; -import { saveScheduledActions, scheduledActions } from "./src/backend/helpers/scheduledActions"; +import { getApiClient, getAuthProvider, } from "./src/backend/helpers/twitch"; -import { ApiClient } from "twitch"; +import { connect } from "./src/backend/chatClient"; import { listen, } from "./src/backend/webServer"; +import { registerUserListener } from "./src/backend/pubSubClient"; const channel = "alexbcberio"; -let apiClient: ApiClient; - async function init() { + + const apiClient = await getApiClient(); + await registerUserListener(apiClient, channel); + const authProvider = await getAuthProvider(); - - apiClient = await getApiClient(); - - const pubSubClient = new PubSubClient(); - const userId = await pubSubClient.registerUserListener(apiClient, channel); - /*const listener = */ await pubSubClient.onRedemption(userId, onRedemption); - - console.log("[Twitch PubSub] Connected & registered"); - await connect(authProvider, [channel]); listen(); } -init(); - -async function onRedemption(message: PubSubRedemptionMessage) { - console.log( - `Reward: "${message.rewardName}" (${message.rewardId}) redeemed by ${message.userDisplayName}` - ); - // @ts-ignore - const reward = message._data.data.redemption.reward; - - let msg: any = { - id: message.id, - channelId: message.channelId, - rewardId: message.rewardId, - rewardName: message.rewardName, - rewardImage: message.rewardImage - ? message.rewardImage.url_4x - : "https://static-cdn.jtvnw.net/custom-reward-images/default-4.png", - message: message.message, - userDisplayName: message.userDisplayName, - // non directly available values from PubSubRedemptionMessage - backgroundColor: reward.background_color - }; - - switch (msg.rewardId) { - // robar vip - case "ac750bd6-fb4c-4259-b06d-56953601243b": - msg = await stealVip(msg); - break; - } - - if (msg) { - console.log(msg); - - broadcast(msg); - } -} - -// adds a user to vips -async function addVip(channel: string, username: string, message?: string) { - if (!message) { - message = `Otorgado VIP a @${username}.`; - } - - await chatClient.addVip(channel, username); - say(channel, message); -} - -async function hasVip(channel: string, username: string) { - if (!username) { - return false; - } - - const vips = await chatClient.getVips(channel); - return vips.includes(username); -} - -// removes a user from vips -async function removeVip(channel: string, username: string, message?: string) { - if (!message) { - message = `Se ha acabado el chollo, VIP de @${username} eliminado.`; - } - - await chatClient.removeVip(channel, username); - say(channel, message); -} - -async function getUsernameFromId(userId: number) { - const user = await apiClient.helix.users.getUserById(userId); - - if (!user) { - return null; - } - - return user.displayName; -} - -// remove vip from a user to grant it to yourself -async function stealVip(msg: { - channelId: string; - userDisplayName: string; - message: string; -}) { - const channel = await getUsernameFromId(parseInt(msg.channelId)); - - if (!channel) { - console.log("No channel found"); - return; - } - - const addVipUser = msg.userDisplayName; - const removeVipUser = msg.message; - - if (await hasVip(channel, removeVipUser)) { - await removeVip(channel, removeVipUser); - await addVip(channel, addVipUser); - - const scheduledRemoveVipIndex = scheduledActions.findIndex( - s => s.action === "removeVip" && s.username === removeVipUser - ); - - if (scheduledRemoveVipIndex > -1) { - scheduledActions[scheduledRemoveVipIndex].username = addVipUser; - saveScheduledActions(); - } - - msg.message = `@${addVipUser} ha robado el VIP a @${removeVipUser}.`; - - return msg; - } - - return null; -} \ No newline at end of file +init(); \ No newline at end of file diff --git a/src/backend/helpers/twitch.ts b/src/backend/helpers/twitch.ts index 8125d8c..25a3946 100644 --- a/src/backend/helpers/twitch.ts +++ b/src/backend/helpers/twitch.ts @@ -10,7 +10,8 @@ let refreshAuthProvider: RefreshableAuthProvider; export { getAuthProvider, - getApiClient + getApiClient, + getUsernameFromId } interface ClientCredentials { @@ -94,3 +95,14 @@ async function getApiClient() { return new ApiClient({ authProvider }); } + +async function getUsernameFromId(userId: number) { + const apiClient = await getApiClient(); + const user = await apiClient.helix.users.getUserById(userId); + + if (!user) { + return null; + } + + return user.displayName; +} \ No newline at end of file diff --git a/src/backend/pubSubClient/index.ts b/src/backend/pubSubClient/index.ts new file mode 100644 index 0000000..b7e9745 --- /dev/null +++ b/src/backend/pubSubClient/index.ts @@ -0,0 +1,122 @@ +import { ApiClient, UserIdResolvable } from "twitch"; +import { PubSubClient, PubSubRedemptionMessage } from "twitch-pubsub-client"; +import { broadcast, chatClient, say } from "../chatClient"; +import { saveScheduledActions, scheduledActions } from "../helpers/scheduledActions"; + +import { getUsernameFromId } from "../helpers/twitch"; + +export { + registerUserListener +} + +async function registerUserListener(apiClient: ApiClient, user?: UserIdResolvable) { + const pubSubClient = new PubSubClient(); + const userId = await pubSubClient.registerUserListener(apiClient, user); + /*const listener = */ await pubSubClient.onRedemption(userId, onRedemption); + + console.log("[Twitch PubSub] Connected & registered"); +} + +async function onRedemption(message: PubSubRedemptionMessage) { + console.log( + `Reward: "${message.rewardName}" (${message.rewardId}) redeemed by ${message.userDisplayName}` + ); + // @ts-ignore + const reward = message._data.data.redemption.reward; + + let msg: any = { + id: message.id, + channelId: message.channelId, + rewardId: message.rewardId, + rewardName: message.rewardName, + rewardImage: message.rewardImage + ? message.rewardImage.url_4x + : "https://static-cdn.jtvnw.net/custom-reward-images/default-4.png", + message: message.message, + userDisplayName: message.userDisplayName, + // non directly available values from PubSubRedemptionMessage + backgroundColor: reward.background_color + }; + + switch (msg.rewardId) { + // robar vip + case "ac750bd6-fb4c-4259-b06d-56953601243b": + msg = await stealVip(msg); + break; + } + + if (msg) { + console.log(msg); + + if (typeof msg !== "string") { + msg = JSON.stringify(msg); + } + broadcast(msg); + } +} + +// remove vip from a user to grant it to yourself +async function stealVip(msg: { + channelId: string; + userDisplayName: string; + message: string; +}) { + const channel = await getUsernameFromId(parseInt(msg.channelId)); + + if (!channel) { + console.log("No channel found"); + return; + } + + const addVipUser = msg.userDisplayName; + const removeVipUser = msg.message; + + if (await hasVip(channel, removeVipUser)) { + await removeVip(channel, removeVipUser); + await addVip(channel, addVipUser); + + const scheduledRemoveVipIndex = scheduledActions.findIndex( + s => s.action === "removeVip" && s.username === removeVipUser + ); + + if (scheduledRemoveVipIndex > -1) { + scheduledActions[scheduledRemoveVipIndex].username = addVipUser; + saveScheduledActions(); + } + + msg.message = `@${addVipUser} ha robado el VIP a @${removeVipUser}.`; + + return msg; + } + + return null; +} + +// adds a user to vips +async function addVip(channel: string, username: string, message?: string) { + if (!message) { + message = `Otorgado VIP a @${username}.`; + } + + await chatClient.addVip(channel, username); + say(channel, message); +} + +async function hasVip(channel: string, username: string) { + if (!username) { + return false; + } + + const vips = await chatClient.getVips(channel); + return vips.includes(username); +} + +// removes a user from vips +async function removeVip(channel: string, username: string, message?: string) { + if (!message) { + message = `Se ha acabado el chollo, VIP de @${username} eliminado.`; + } + + await chatClient.removeVip(channel, username); + say(channel, message); +} \ No newline at end of file