d4708
/
source-engine
mirror of https://github.com/nillerusr/source-engine.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
193 Commits
25 Branches
6 Tags
221 MiB
 
 
 
 
 
 
Tree: c739b2c5bb
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'c739b2c5bb'
${ noResults }
source-engine/utils/vproject/vproject.cpp

648 lines
15 KiB
Raw Blame History

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// VPROJECT.CPP
//
//=====================================================================================//
#include "vproject.h"
#define MAX_PROJECTS 50
#define ID_PROJECTS_LISTVIEW 100
// column id
#define ID_PROJECT_NAME 0
#define ID_PROJECT_DIRECTORY 1
#define WM_TRAY (WM_APP + 1)
#define ID_TRAY 5000
#define ID_TRAY_ADDVPROJECT 6000
#define ID_TRAY_MODIFYVPROJECT 6001
#define ID_TRAY_MOVEUP 6002
#define ID_TRAY_MOVEDOWN 6003
#define ID_TRAY_EXIT 6004
typedef struct
{
char *pName;
char *pGamedir;
} project_t;
HWND g_hWnd;
char g_project_name[256];
char g_project_gamedir[256];
HINSTANCE g_hInstance;
project_t g_projects[MAX_PROJECTS];
int g_numProjects;
NOTIFYICONDATA g_iconData;
HMENU g_hMenu;
int g_nActiveVProject;
POINT g_cursorPoint;
void TrayMessageHandler( HWND hWnd, UINT uMessageID );
//-----------------------------------------------------------------------------
// SetVProject
//
//-----------------------------------------------------------------------------
void SetVProject( const char *pProjectName )
{
char *pGamedir;
char project[256];
int i;
if ( pProjectName )
{
strcpy( project, pProjectName );
Sys_StripQuotesFromToken( project );
for ( i=0; i<g_numProjects; i++ )
{
if ( !stricmp( g_projects[i].pName, project ) )
{
// found
break;
}
}
if ( i >= g_numProjects )
{
// not found
return;
}
pGamedir = g_projects[i].pGamedir;
}
else
{
pGamedir = "";
}
// Changed to CURRENT_USER to solve security issues in vista!
Sys_SetRegistryString(
//"HKEY_LOCAL_MACHINE\\System\\CurrentControlSet\\Control\\Session Manager\\Environment\\VProject",
"HKEY_CURRENT_USER\\Environment\\VProject"
pGamedir );
DWORD result;
SendMessageTimeout(
HWND_BROADCAST,
WM_SETTINGCHANGE,
0,
(LPARAM)"Environment",
SMTO_ABORTIFHUNG,
0,
&result );
}
//-----------------------------------------------------------------------------
// ModifyVProject
//
//-----------------------------------------------------------------------------
void ModifyVProject( int index, const char *pName, const char *pGamedir )
{
free( g_projects[index].pName );
free( g_projects[index].pGamedir );
if ( !pName || !pName[0] )
{
// delete
if ( g_numProjects-index-1 > 0 )
{
// shift remaining elements
memcpy( &g_projects[index], &g_projects[index+1], (g_numProjects-index-1)*sizeof( project_t ) );
}
g_projects[g_numProjects-1].pName = NULL;
g_projects[g_numProjects-1].pGamedir = NULL;
g_numProjects--;
if ( g_nActiveVProject == index+1 )
{
// deleted current vproject
if ( !g_numProjects )
{
// no more projects
g_nActiveVProject = 0;
SetVProject( NULL );
}
else
{
// set to top
g_nActiveVProject = 1;
SetVProject( g_projects[0].pName );
}
}
}
else
{
g_projects[index].pName = strdup( pName );
g_projects[index].pGamedir = strdup( pGamedir );
}
}
//-----------------------------------------------------------------------------
// AddVProject
//
//-----------------------------------------------------------------------------
void AddVProject( const char *pName, const char *pGamedir )
{
if ( !pName || !pName[0] )
{
// do not add empty projects
return;
}
ModifyVProject( g_numProjects, pName, pGamedir );
g_numProjects++;
}
//-----------------------------------------------------------------------------
// LoadRegistryValues
//
//-----------------------------------------------------------------------------
void LoadRegistryValues()
{
char keyBuff[32];
char valueBuff[256];
char projectName[256];
char gamedirString[256];
char *ptr;
char *token;
int i;
for ( i=0; i<MAX_PROJECTS; i++ )
{
projectName[0] = '\0';
gamedirString[0] = '\0';
sprintf( keyBuff, "project%d", i );
Sys_GetRegistryString( keyBuff, valueBuff, "", sizeof( valueBuff ) );
// parse and populate valid values
ptr = valueBuff;
token = Sys_GetToken( &ptr, false, NULL );
if ( token[0] )
{
strcpy( projectName, token );
}
else
{
continue;
}
token = Sys_GetToken( &ptr, false, NULL );
if ( token[0] )
{
strcpy( gamedirString, token );
}
AddVProject( projectName, gamedirString );
}
}
//-----------------------------------------------------------------------------
// SaveRegistryValues
//
//-----------------------------------------------------------------------------
void SaveRegistryValues()
{
char valueBuff[256];
char keyBuff[32];
char *pProjectName;
char *pGamedir;
int len;
int i;
for ( i=0; i<MAX_PROJECTS; i++ )
{
sprintf( keyBuff, "project%d", i );
pProjectName = g_projects[i].pName;
if ( !pProjectName )
{
pProjectName = "";
}
pGamedir = g_projects[i].pGamedir;
if ( !pGamedir )
{
pGamedir = "";
}
len = _snprintf( valueBuff, sizeof( valueBuff ), "\"%s\" \"%s\"", pProjectName, pGamedir );
if ( len == -1 )
{
// kill it
valueBuff[0] = '\0';
}
Sys_SetRegistryString( keyBuff, valueBuff );
}
}
//-----------------------------------------------------------------------------
// ShiftActiveProjectUp
//
//-----------------------------------------------------------------------------
void ShiftActiveProjectUp()
{
if ( g_numProjects <= 1 || !g_nActiveVProject )
{
// nothing to do
return;
}
int active = g_nActiveVProject-1;
int previous = (active + g_numProjects - 1) % g_numProjects;
project_t tempProject;
tempProject = g_projects[previous];
g_projects[previous] = g_projects[active];
g_projects[active] = tempProject;
g_nActiveVProject = previous + 1;
}
//-----------------------------------------------------------------------------
// ShiftActiveProjectDown
//
//-----------------------------------------------------------------------------
void ShiftActiveProjectDown()
{
if ( g_numProjects <= 1 || !g_nActiveVProject )
{
// nothing to do
return;
}
int active = g_nActiveVProject-1;
int next = (active + g_numProjects + 1) % g_numProjects;
project_t tempProject;
tempProject = g_projects[next];
g_projects[next] = g_projects[active];
g_projects[active] = tempProject;
g_nActiveVProject = next + 1;
}
//-----------------------------------------------------------------------------
// ModifyDlg_Proc
//
//-----------------------------------------------------------------------------
BOOL CALLBACK ModifyDlg_Proc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam )
{
size_t len;
int width;
int height;
RECT rect;
switch ( message )
{
case WM_INITDIALOG:
SetDlgItemText( hWnd, IDC_MODIFY_PROJECT, g_project_name );
SetDlgItemText( hWnd, IDC_MODIFY_GAMEDIR, g_project_gamedir );
// center dialog
GetWindowRect( hWnd, &rect );
width = GetSystemMetrics( SM_CXSCREEN );
height = GetSystemMetrics( SM_CYSCREEN );
SetWindowPos( hWnd, NULL, ( width - ( rect.right - rect.left ) )/2, ( height - ( rect.bottom - rect.top ) )/2, 0, 0, SWP_NOSIZE | SWP_NOZORDER );
return ( TRUE );
case WM_COMMAND:
switch ( LOWORD( wParam ) )
{
case IDC_OK:
GetDlgItemText( hWnd, IDC_MODIFY_PROJECT, g_project_name, sizeof( g_project_name ) );
GetDlgItemText( hWnd, IDC_MODIFY_GAMEDIR, g_project_gamedir, sizeof( g_project_gamedir ) );
// remove trailing seperator
Sys_NormalizePath( g_project_gamedir, false );
len = strlen( g_project_gamedir );
if ( len > 2 && g_project_gamedir[len-1] == '\\' )
{
g_project_gamedir[len-1] = '\0';
}
// fall through
case IDCANCEL:
case IDC_CANCEL:
EndDialog( hWnd, wParam );
return ( TRUE );
}
break;
}
return ( FALSE );
}
//-----------------------------------------------------------------------------
// ModifyDlg_Open
//
//-----------------------------------------------------------------------------
BOOL ModifyDlg_Open()
{
int result;
result = DialogBox( g_hInstance, MAKEINTRESOURCE( IDD_VPROJECT ), g_hWnd, ( DLGPROC )ModifyDlg_Proc );
if ( LOWORD( result ) != IDC_OK )
{
return FALSE;
}
return TRUE;
}
//-----------------------------------------------------------------------------
// ShowPopupMenu
//
//-----------------------------------------------------------------------------
void ShowPopupMenu( HWND hWnd, bool bUseCachedMenuPos )
{
BOOL bDelete = true;
// delete existing entries
for ( int nIndex = 1; nIndex < ID_TRAY_ADDVPROJECT && bDelete; nIndex++ )
{
bDelete = DeleteMenu( g_hMenu, nIndex, MF_BYCOMMAND );
}
// Insert projects
char szMenuItem[MAX_PATH];
for ( int nIndex = 0; nIndex < g_numProjects; nIndex++ )
{
strcpy( szMenuItem, g_projects[nIndex].pName );
strcat( szMenuItem, "\t" );
strcat( szMenuItem, g_projects[nIndex].pGamedir );
strcat( szMenuItem, " " );
InsertMenu( g_hMenu, nIndex, MF_BYPOSITION | MF_STRING, nIndex + 1, szMenuItem );
}
if ( g_nActiveVProject )
{
CheckMenuItem( g_hMenu, g_nActiveVProject, MF_BYCOMMAND|MF_CHECKED );
SetMenuDefaultItem( g_hMenu, g_nActiveVProject, FALSE );
}
// Display the popup menu at the current cursor location
// Use the cached cursor position if set
if ( !bUseCachedMenuPos || ( !g_cursorPoint.x && !g_cursorPoint.y ) )
{
GetCursorPos( &g_cursorPoint );
}
SetForegroundWindow( hWnd );
TrackPopupMenu( g_hMenu, 0, g_cursorPoint.x, g_cursorPoint.y, 0, hWnd, 0 );
PostMessage( hWnd, WM_NULL, 0, 0 );
}
//-----------------------------------------------------------------------------
// TrayMessageHandler
//
//-----------------------------------------------------------------------------
void TrayMessageHandler( HWND hWnd, UINT uMessageID )
{
switch ( uMessageID )
{
case 0:
break;
case ID_TRAY_EXIT:
DestroyWindow( hWnd );
break;
case ID_TRAY_ADDVPROJECT:
SetForegroundWindow( hWnd );
g_project_name[0] = '\0';
g_project_gamedir[0] = '\0';
if ( ModifyDlg_Open() )
{
AddVProject( g_project_name, g_project_gamedir );
SaveRegistryValues();
}
ShowPopupMenu( hWnd, true );
break;
case ID_TRAY_MODIFYVPROJECT:
SetForegroundWindow( hWnd );
if ( g_nActiveVProject )
{
strcpy( g_project_name, g_projects[g_nActiveVProject-1].pName );
strcpy( g_project_gamedir, g_projects[g_nActiveVProject-1].pGamedir );
if ( ModifyDlg_Open() )
{
ModifyVProject( g_nActiveVProject-1, g_project_name, g_project_gamedir );
SaveRegistryValues();
}
ShowPopupMenu( hWnd, true );
}
break;
case ID_TRAY_MOVEUP:
SetForegroundWindow( hWnd );
if ( g_nActiveVProject )
{
ShiftActiveProjectUp();
SaveRegistryValues();
ShowPopupMenu( hWnd, true );
}
break;
case ID_TRAY_MOVEDOWN:
SetForegroundWindow( hWnd );
if ( g_nActiveVProject )
{
ShiftActiveProjectDown();
SaveRegistryValues();
ShowPopupMenu( hWnd, true );
}
break;
default:
if ( uMessageID >= 1 && uMessageID <= MAX_PROJECTS )
{
// set current vproject
g_nActiveVProject = uMessageID;
SetVProject( g_projects[uMessageID-1].pName );
}
break;
}
}
//-----------------------------------------------------------------------------
// Main_WndProc
//
//-----------------------------------------------------------------------------
LRESULT CALLBACK Main_WndProc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam )
{
switch ( message )
{
case WM_DESTROY:
PostQuitMessage( 0 );
return 0L;
case WM_TRAY:
if ( lParam == WM_LBUTTONDOWN || lParam == WM_RBUTTONDOWN )
{
ShowPopupMenu( hWnd, false );
return 0L;
}
break;
case WM_COMMAND:
TrayMessageHandler( hWnd, LOWORD( wParam ) );
break;
}
return ( DefWindowProc( hWnd, message, wParam, lParam ) );
}
//-----------------------------------------------------------------------------
// Startup
//
//-----------------------------------------------------------------------------
bool Startup()
{
int i;
// set up our window class
WNDCLASS wndclass;
memset( &wndclass, 0, sizeof( wndclass ) );
wndclass.style = 0;
wndclass.lpfnWndProc = Main_WndProc;
wndclass.cbClsExtra = 0;
wndclass.cbWndExtra = 0;
wndclass.hInstance = g_hInstance;
wndclass.hIcon = LoadIcon( g_hInstance, (LPCTSTR)IDI_VPROJECT );
wndclass.hCursor = LoadCursor( NULL, IDC_ARROW );
wndclass.hbrBackground = NULL;
wndclass.lpszMenuName = NULL;
wndclass.lpszClassName = VPROJECT_CLASSNAME;
if ( !RegisterClass( &wndclass ) )
{
return false;
}
// create the hidden window
g_hWnd = CreateWindow(
VPROJECT_CLASSNAME,
0,
WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT,
0,
CW_USEDEFAULT,
0,
NULL,
NULL,
g_hInstance,
NULL );
// Create tray icon
g_iconData.cbSize = sizeof( NOTIFYICONDATA );
g_iconData.hIcon = LoadIcon( g_hInstance, (LPCTSTR)IDI_VPROJECT );
g_iconData.hWnd = g_hWnd;
g_iconData.uCallbackMessage = WM_TRAY;
g_iconData.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
g_iconData.uID = ID_TRAY;
strcpy(g_iconData.szTip, "VPROJECT");
Shell_NotifyIcon( NIM_ADD, &g_iconData );
// Create popup menu and add initial items
g_hMenu = CreatePopupMenu();
AppendMenu( g_hMenu, MF_SEPARATOR, 0, 0);
AppendMenu( g_hMenu, MF_STRING, ID_TRAY_ADDVPROJECT, "Add VProject" );
AppendMenu( g_hMenu, MF_STRING, ID_TRAY_MODIFYVPROJECT, "Modify VProject" );
AppendMenu( g_hMenu, MF_STRING, ID_TRAY_MOVEUP, "Move Up" );
AppendMenu( g_hMenu, MF_STRING, ID_TRAY_MOVEDOWN, "Move Down" );
AppendMenu( g_hMenu, MF_SEPARATOR, 0, 0);
AppendMenu( g_hMenu, MF_STRING, ID_TRAY_EXIT, "Remove From Tray" );
// find the current vproject
char* vproject = getenv( "vproject" );
if ( vproject && vproject[0] )
{
char temp[MAX_PATH];
strcpy( temp, vproject );
Sys_NormalizePath( temp, false );
for ( i=0; i<g_numProjects; i++ )
{
if ( !stricmp( g_projects[i].pGamedir, temp ) )
{
// found
g_nActiveVProject = i+1;
break;
}
}
}
return true;
}
//-----------------------------------------------------------------------------
// Shutdown
//
// Free all resources
//-----------------------------------------------------------------------------
void Shutdown()
{
}
//-----------------------------------------------------------------------------
// WinMain
//
// Entry point for program
//-----------------------------------------------------------------------------
int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR pCmdLine, int nCmdShow )
{
bool error = true;
MSG msg = {0};
g_hInstance = hInstance;
// get the project list
LoadRegistryValues();
if ( pCmdLine && pCmdLine[0] )
{
// set directly
SetVProject( pCmdLine );
return 0;
}
HWND hwnd = FindWindow( VPROJECT_CLASSNAME, NULL );
if ( hwnd )
{
// single instance only
return ( FALSE );
}
if ( !Startup() )
{
goto cleanUp;
}
// message pump
while ( GetMessage( &msg, NULL, 0, 0 ) )
{
TranslateMessage( &msg );
DispatchMessage( &msg );
}
// no-error, end of app
error = false;
cleanUp:
if ( error )
{
char str[255];
FormatMessage( FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(), 0, str, 255, NULL );
MessageBox( NULL, str, NULL, MB_OK );
}
Shutdown();
return ( (int)msg.wParam );
}