Draw update reminder, but only for 5 seconds, if watermark is off

Author: sabianroberts

cl_dll/hud_watermark.cpp

@@ -3,6 +3,8 @@
 #include "parsemsg.h"
 #include "update_checker.h"
 
+extern cvar_t* hud_watermark;
+
 int CHudWatermark::Init()
 {
 	m_iFlags = 0;
@@ -22,9 +24,20 @@ int CHudWatermark::VidInit()
 
 int CHudWatermark::Draw(float time)
 {
-	if (refresh_draw_until || (draw_until > gHUD.m_flTime + 15.0f)) {
+	if (hud_watermark->value != 0)
+	{
+		
+		if (refresh_draw_until || (draw_until > gHUD.m_flTime + 15.0f)) {
 		refresh_draw_until = false;
 		draw_until = gHUD.m_flTime + 15.0f;
+		}
+	}
+	else
+	{
+		if (refresh_draw_until || (draw_until > gHUD.m_flTime + 5.0f)) {
+		refresh_draw_until = false;
+		draw_until = gHUD.m_flTime + 5.0f;
+		}
 	}
 
 	if (gHUD.m_flTime >= draw_until) {
@@ -35,14 +48,13 @@ int CHudWatermark::Draw(float time)
 	int r, g, b;
 	UnpackRGB(r, g, b, gHUD.m_iDefaultHUDColor);
 
-	extern cvar_t* hud_watermark;
-	if (hud_watermark->value == 1)
+	if (hud_watermark->value != 0)
 	{
 		gEngfuncs.pfnDrawString(ScreenWidth / 20, gHUD.m_scrinfo.iCharHeight, "OpenAG client build " __DATE__, r, g, b);
 		gEngfuncs.pfnDrawString(ScreenWidth / 20, gHUD.m_scrinfo.iCharHeight * 2, "j.mp/OpenAG", r, g, b);
-
-		if (update_is_available)
-			gEngfuncs.pfnDrawString(ScreenWidth / 20, gHUD.m_scrinfo.iCharHeight / 2 * 7, "An update is available.", r, g, b);
 	}
+
+	if (update_is_available)
+		gEngfuncs.pfnDrawString(ScreenWidth / 20, gHUD.m_scrinfo.iCharHeight / 2 * 7, "An update is available.", r, g, b);
 	return 0;
 }