prompt.c File Reference

#include "precomp.h"

Include dependency graph for prompt.c:

Go to the source code of this file.

Macros
#define	FOREGROUND_WHITE   (FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY)

Functions
VOID	InitPrompt (VOID)
BOOL	HasInfoLine (VOID)
VOID	PrintInfoLine (VOID)
VOID	PrintPrompt (VOID)
INT	cmd_prompt (LPTSTR param)

Variables
static TCHAR	DefaultPrompt [] = _T("$P$G")

Macro Definition Documentation

FOREGROUND_WHITE

#define FOREGROUND_WHITE   (FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY)

Function Documentation

cmd_prompt()

INT cmd_prompt

(

LPTSTR

param

)

Definition at line 260 of file prompt.c.

{
    INT retval = 0;
 
    if (!_tcsncmp(param, _T("/?"), 2))
    {
        ConOutResPaging(TRUE, STRING_PROMPT_HELP1);
#ifdef FEATURE_DIRECTORY_STACK
        ConOutResPaging(FALSE, STRING_PROMPT_HELP2);
#endif
        ConOutResPaging(FALSE, STRING_PROMPT_HELP3);
        return 0;
    }
 
    /*
     * Set the PROMPT environment variable. If 'param' is NULL or is
     * an empty string (the user entered "prompt" only), then remove
     * the environment variable and therefore use the default prompt.
     * Otherwise, use the new prompt.
     */
    if (!SetEnvironmentVariable(_T("PROMPT"),
                                (param && param[0] != _T('\0') ? param : NULL)))
    {
        retval = 1;
    }
 
    if (BatType != CMD_TYPE)
    {
        if (retval != 0)
            nErrorLevel = retval;
    }
    else
    {
        nErrorLevel = retval;
    }
 
    return retval;
}

HasInfoLine()

BOOL HasInfoLine

(

VOID

)

Definition at line 72 of file prompt.c.

{
    LPTSTR pr;
    TCHAR szPrompt[256];
 
    if (GetEnvironmentVariable(_T("PROMPT"), szPrompt, _countof(szPrompt)))
    {
        pr = szPrompt;
        while (*pr)
        {
            if (*pr++ != _T('$'))
                continue;
            if (!*pr || _totupper(*pr++) != _T('I'))
                continue;
 
            return TRUE;
        }
    }
 
    return FALSE;
}

Referenced by Initialize(), and PrintPrompt().

InitPrompt()

VOID InitPrompt

(

VOID

)

Definition at line 57 of file prompt.c.

{
    TCHAR Buffer[2];
 
    /*
     * Set the PROMPT environment variable if it doesn't exist already.
     * You can change the PROMPT environment variable before cmd starts.
     */
    if (GetEnvironmentVariable(_T("PROMPT"), Buffer, _countof(Buffer)) == 0)
        SetEnvironmentVariable(_T("PROMPT"), DefaultPrompt);
}

Referenced by Initialize().

PrintInfoLine()

VOID PrintInfoLine

(

VOID

)

Definition at line 97 of file prompt.c.

{
#define FOREGROUND_WHITE (FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY)
 
    HANDLE hOutput = GetStdHandle(STD_OUTPUT_HANDLE);
    CONSOLE_SCREEN_BUFFER_INFO csbi;
    COORD coPos;
    DWORD dwWritten;
 
    PTSTR pszInfoLine = NULL;
    INT iInfoLineLen;
 
    /* Return directly if the output handle is not a console handle */
    if (!GetConsoleScreenBufferInfo(hOutput, &csbi))
        return;
 
    iInfoLineLen = LoadString(CMD_ModuleHandle, STRING_CMD_INFOLINE, (PTSTR)&pszInfoLine, 0);
    if (!pszInfoLine || iInfoLineLen == 0)
        return;
 
    /* Display the localized information line */
    coPos.X = 0;
    coPos.Y = 0;
    FillConsoleOutputAttribute(hOutput, BACKGROUND_BLUE | FOREGROUND_WHITE,
                               csbi.dwSize.X,
                               coPos, &dwWritten);
    FillConsoleOutputCharacter(hOutput, _T(' '),
                               csbi.dwSize.X,
                               coPos, &dwWritten);
 
    WriteConsoleOutputCharacter(hOutput, pszInfoLine, iInfoLineLen,
                                coPos, &dwWritten);
}

Referenced by PrintPrompt().

PrintPrompt()

VOID PrintPrompt

(

VOID

)

Definition at line 134 of file prompt.c.

{
    LPTSTR pr, Prompt;
    TCHAR szPrompt[256];
    TCHAR szPath[MAX_PATH];
 
    if (GetEnvironmentVariable(_T("PROMPT"), szPrompt, _countof(szPrompt)))
        Prompt = szPrompt;
    else
        Prompt = DefaultPrompt;
 
    /*
     * Special pre-handling for $I: If the information line is displayed
     * on top of the screen, ensure that the prompt won't be hidden below it.
     */
    if (HasInfoLine() && GetCursorY() == 0)
        ConOutChar(_T('\n'));
 
    /* Parse the prompt string */
    for (pr = Prompt; *pr; ++pr)
    {
        if (*pr != _T('$'))
        {
            ConOutChar(*pr);
        }
        else
        {
            ++pr;
            if (!*pr) break;
            switch (_totupper(*pr))
            {
                case _T('A'):
                    ConOutChar(_T('&'));
                    break;
 
                case _T('B'):
                    ConOutChar(_T('|'));
                    break;
 
                case _T('C'):
                    ConOutChar(_T('('));
                    break;
 
                case _T('D'):
                    ConOutPrintf(_T("%s"), GetDateString());
                    break;
 
                case _T('E'):
                    ConOutChar(_T('\x1B'));
                    break;
 
                case _T('F'):
                    ConOutChar(_T(')'));
                    break;
 
                case _T('G'):
                    ConOutChar(_T('>'));
                    break;
 
                case _T('H'):
                    ConOutPuts(_T("\x08 \x08"));
                    break;
 
                case _T('I'):
                    PrintInfoLine();
                    break;
 
                case _T('L'):
                    ConOutChar(_T('<'));
                    break;
 
                case _T('N'):
                {
                    GetCurrentDirectory(_countof(szPath), szPath);
                    ConOutChar(szPath[0]);
                    break;
                }
 
                case _T('P'):
                {
                    GetCurrentDirectory(_countof(szPath), szPath);
                    ConOutPrintf(_T("%s"), szPath);
                    break;
                }
 
                case _T('Q'):
                    ConOutChar(_T('='));
                    break;
 
                case _T('S'):
                    ConOutChar(_T(' '));
                    break;
 
                case _T('T'):
                    ConOutPrintf(_T("%s"), GetTimeString());
                    break;
 
                case _T('V'):
                    PrintOSVersion();
                    break;
 
                case _T('_'):
                    ConOutChar(_T('\n'));
                    break;
 
                case _T('$'):
                    ConOutChar(_T('$'));
                    break;
 
#ifdef FEATURE_DIRECTORY_STACK
                case _T('+'):
                {
                    INT i;
                    for (i = 0; i < GetDirectoryStackDepth(); i++)
                        ConOutChar(_T('+'));
                    break;
                }
#endif
            }
        }
    }
}

Referenced by Batch(), ExecuteCommandWithEcho(), ReadCommand(), and ReadLine().

Variable Documentation

DefaultPrompt

TCHAR DefaultPrompt[] = _T("$P$G")

static

Definition at line 52 of file prompt.c.

Referenced by InitPrompt(), and PrintPrompt().