Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add option to show live user notifications when the app starts #8

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Add option to show live user notifications when the app starts #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
4156914
Add option to show live user notifications when the app starts
sir-pinecone Aug 6, 2023
6bfc0d4
Call a user supplied script when a Twitch user goes live
sir-pinecone Sep 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
98 changes: 67 additions & 31 deletions TwitchNotify.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,13 +63,14 @@
#define WM_TWITCH_NOTIFY_FOLLOWED_USERS (WM_USER + 9)

// command id's for popup menu items
#define CMD_OPEN_HOMEPAGE 10 // "Twitch Notify" item selected
#define CMD_USE_MPV 20 // "Use mpv" checkbox
#define CMD_QUALITY 30 // +N for Quality submenu items
#define CMD_EDIT_USERS 40 // "Edit List" in user submenu
#define CMD_DOWNLOAD_USERS 50 // "Download List" in user submenu
#define CMD_EXIT 60 // "Exit"
#define CMD_USER 70 // +N for one of users (indexed into State.Users[] array)
#define CMD_OPEN_HOMEPAGE 10 // "Twitch Notify" item selected
#define CMD_USE_MPV 20 // "Use mpv" checkbox
#define CMD_QUALITY 30 // +N for Quality submenu items
#define CMD_EDIT_USERS 40 // "Edit List" in user submenu
#define CMD_DOWNLOAD_USERS 50 // "Download List" in user submenu
#define CMD_NOTIFY_ON_STARTUP 60 // Check for live users at app startup
#define CMD_EXIT 70 // "Exit"
#define CMD_USER 80 // +N for one of users (indexed into State.Users[] array)

#define TWITCH_NOTIFY_NAME L"Twitch Notify"
#define TWITCH_NOTIFY_INI L"TwitchNotify.ini"
Expand Down Expand Up @@ -154,6 +155,9 @@ struct
// global settings
int Quality;
bool UseMpv;
bool NotifyOnStartup;

WCHAR ScriptToRunWhenUserIsLive[128];

// user list
User Users[MAX_USER_COUNT];
Expand Down Expand Up @@ -277,7 +281,7 @@ static void GetTwitchIcon(LPWSTR ImagePath)

// showing & updating Windows Toast notification

static void ShowUserNotification(User* User)
static void ShowUserNotificationAndMaybeRunScript(User* User)
{
WCHAR ImagePath[MAX_PATH];

Expand Down Expand Up @@ -334,6 +338,12 @@ static void ShowUserNotification(User* User)
void* Notification = WindowsToast_Create(&State.Toast, Xml, XmlLength, Data, ARRAYSIZE(Data));
WindowsToast_Show(&State.Toast, Notification);
User->Notification = Notification;

// Run the user-supplied batch script and pass it the username that just went live.
// We're assuming it's a valid file path!
if (State.ScriptToRunWhenUserIsLive[0] != 0) {
ShellExecuteW(NULL, L"open", State.ScriptToRunWhenUserIsLive, User->Name, NULL, SW_SHOWNORMAL);
}
}

// updates & releases user notification
Expand Down Expand Up @@ -669,7 +679,7 @@ static void LoadUsers(void)

// no notification is shown initially
User->Notification = NULL;

QuerySize = StrCatChainW(Query, ARRAYSIZE(Query), QuerySize, Delim);
QuerySize = StrCatChainW(Query, ARRAYSIZE(Query), QuerySize, L"\\\"");
QuerySize = StrCatChainW(Query, ARRAYSIZE(Query), QuerySize, NewUsers[NewIndex]);
Expand All @@ -681,7 +691,8 @@ static void LoadUsers(void)

// allowed image widths: 28, 50, 70, 150, 300, 600
// use 70 because for 100% dpi scale the toast size is 48 pixels, for 200% it is 96
QuerySize = StrCatChainW(Query, ARRAYSIZE(Query), QuerySize, L"]){id,displayName,profileImageURL(width:70),stream{viewersCount}}}\"}");
// including game info in the stream query so that we can immediately show notifications for users that are already live when the app is starting.
QuerySize = StrCatChainW(Query, ARRAYSIZE(Query), QuerySize, L"]){id,displayName,profileImageURL(width:70),stream{title,viewersCount,game{displayName}}}}\"}");

char QueryBytes[4096];
QuerySize = WideCharToMultiByte(CP_UTF8, 0, Query, QuerySize, QueryBytes, ARRAYSIZE(QueryBytes), NULL, NULL);
Expand Down Expand Up @@ -872,7 +883,8 @@ static void ShowTrayMenu(HWND Window)
AppendMenuW(Menu, MF_SEPARATOR, 0, NULL);

AppendMenuW(Menu, MF_STRING | (CanUpdateUsers ? 0 : MF_GRAYED), CMD_DOWNLOAD_USERS, L"Download User List");
AppendMenuW(Menu, MF_STRING, CMD_EDIT_USERS, L"Edit User List");
AppendMenuW(Menu, MF_STRING, CMD_EDIT_USERS, L"Edit Config");
AppendMenuW(Menu, (State.NotifyOnStartup ? MF_CHECKED : MF_STRING), CMD_NOTIFY_ON_STARTUP, L"Notify on Startup");

AppendMenuW(Menu, MF_SEPARATOR, 0, NULL);

Expand Down Expand Up @@ -907,6 +919,10 @@ static void ShowTrayMenu(HWND Window)
{
DownloadFollowedUsers();
}
else if (Command == CMD_NOTIFY_ON_STARTUP)
{
State.NotifyOnStartup = !State.NotifyOnStartup;
}
else if (Command == CMD_EXIT)
{
DestroyWindow(Window);
Expand Down Expand Up @@ -1052,10 +1068,10 @@ static void OnUserStatus(int UserId, bool IsLive)
User->IsLive = true;

// show initial notification
ShowUserNotification(User);
ShowUserNotificationAndMaybeRunScript(User);

// start download of game/stream title
DownloadUserStream(UserId, 0);
DownloadUserStream(User->UserId, 0);
}
else
{
Expand All @@ -1080,20 +1096,8 @@ static void OnUserViewerCount(int UserId, int ViewerCount)
User->ViewerCount = ViewerCount;
}

static void OnUserStream(int UserId, JsonObject* Json)
static void ShowGameInfoFromStream(User* User, JsonObject* Stream)
{
User* User = GetUser(UserId);
if (!User)
{
// no problem if user is not found
// probably user list was edited right before gql download finished
return;
}

JsonObject* Data = JsonObject_GetObject(Json, JsonCSTR("data"));
JsonArray* Users = JsonObject_GetArray(Data, JsonCSTR("users"));
JsonObject* UserData = JsonArray_GetObject(Users, 0);
JsonObject* Stream = JsonObject_GetObject(UserData, JsonCSTR("stream"));
JsonObject* Game = JsonObject_GetObject(Stream, JsonCSTR("game"));
HSTRING GameName = JsonObject_GetString(Game, JsonCSTR("displayName"));
HSTRING StreamName = JsonObject_GetString(Stream, JsonCSTR("title"));
Expand All @@ -1102,7 +1106,7 @@ static void OnUserStream(int UserId, JsonObject* Json)
{
// if stream name is available in json response
// update notification with actual game/stream title
LPCWSTR GameNameStr = WindowsGetStringRawBuffer(GameName, NULL);
LPCWSTR GameNameStr = WindowsGetStringRawBuffer(GameName, NULL);
LPCWSTR StreamNameStr = WindowsGetStringRawBuffer(StreamName, NULL);
UpdateUserNotification(User, GameNameStr, StreamNameStr);
}
Expand All @@ -1112,13 +1116,32 @@ static void OnUserStream(int UserId, JsonObject* Json)
// if notification is still up then retry a bit later - after 1 second
if (User->Notification != NULL)
{
DownloadUserStream(UserId, 1000);
DownloadUserStream(User->UserId, 1000);
}
}

WindowsDeleteString(StreamName);
WindowsDeleteString(GameName);
JsonRelease(Game);
}

static void OnUserStream(int UserId, JsonObject* Json)
{
User* User = GetUser(UserId);
if (!User)
{
// no problem if user is not found
// probably user list was edited right before gql download finished
return;
}

JsonObject* Data = JsonObject_GetObject(Json, JsonCSTR("data"));
JsonArray* Users = JsonObject_GetArray(Data, JsonCSTR("users"));
JsonObject* UserData = JsonArray_GetObject(Users, 0);
JsonObject* Stream = JsonObject_GetObject(UserData, JsonCSTR("stream"));

ShowGameInfoFromStream(User, Stream);

JsonRelease(Stream);
JsonRelease(UserData);
JsonRelease(Users);
Expand Down Expand Up @@ -1177,6 +1200,13 @@ static void OnUserInfo(JsonObject* Json)

User->IsLive = Stream != NULL;

if (State.NotifyOnStartup && User->IsLive)
{
// show initial notification and then update it with game info if available.
ShowUserNotificationAndMaybeRunScript(User);
ShowGameInfoFromStream(User, Stream);
}

// subscribe to user live events on websocket
// if websocket is not connected yet, this subscription will
// happen on WM_TWITCH_NOTIFY_WEBSOCKET message
Expand Down Expand Up @@ -1277,7 +1307,7 @@ static void OnFollowedUsers(JsonObject* Json)

WCHAR Users[32768];
int UsersLength = 0;

int Count = JsonArray_GetCount(Edges);
for (int Index = 0; Index < Count; Index++)
{
Expand Down Expand Up @@ -1320,6 +1350,11 @@ static LRESULT CALLBACK WindowProc(HWND Window, UINT Message, WPARAM WParam, LPA
State.Quality = GetPrivateProfileIntW(L"player", L"quality", 0, State.IniPath);
State.UseMpv = GetPrivateProfileIntW(L"player", L"mpv", 1, State.IniPath);

State.NotifyOnStartup = GetPrivateProfileIntW(L"twitch", L"notifyOnStartup", 0, State.IniPath);

// We could verify that this is a valid file path, but whatever.
GetPrivateProfileStringW(L"twitch", L"scriptToRunWhenUserIsLive", L"", State.ScriptToRunWhenUserIsLive, ARRAYSIZE(State.ScriptToRunWhenUserIsLive), State.IniPath);

if (GetPrivateProfileIntW(L"twitch", L"autoupdate", 0, State.IniPath) == 0)
{
// load initial user list
Expand All @@ -1334,9 +1369,10 @@ static LRESULT CALLBACK WindowProc(HWND Window, UINT Message, WPARAM WParam, LPA
}
else if (Message == WM_DESTROY)
{
WCHAR Str[2] = { L'0' + State.Quality, 0 };
WritePrivateProfileStringW(L"player", L"quality", Str, State.IniPath);
WCHAR StateStr[2] = { L'0' + State.Quality, 0 };
WritePrivateProfileStringW(L"player", L"quality", StateStr, State.IniPath);
WritePrivateProfileStringW(L"player", L"mpv", State.UseMpv ? L"1" : L"0", State.IniPath);
WritePrivateProfileStringW(L"twitch", L"notifyOnStartup", State.NotifyOnStartup ? L"1" : L"0", State.IniPath);
RemoveTrayIcon(Window);
if (State.Websocket)
{
Expand Down
7 changes: 7 additions & 0 deletions TwitchNotify.ini
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,13 @@
; WARNING: this will replace contents of [users] section
autoupdate=0

; set to 1 to show notifications for live users when the app starts
notifyOnStartup=0

; set to the full path of a file that you want to run when a user goes live. The file is passed
; the Twitch username that went live. This can be used to automate stream downloads.
;scriptToRunWhenUserIsLive=C:\\Users\\Example\\download_twitch_stream.bat

[users]
; list of users to notify about, each on separate line
molly_rocket
Expand Down