1

/* $Id$ */

     2

     3

/** @file console.cpp */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "gui.h"

     8

#include "textbuf_gui.h"

     9

#include "window_gui.h"

    10

#include "variables.h"

    11

#include <stdarg.h>

    12

#include <string.h>

    13

#include "console.h"

    14

#include "network/network.h"

    15

#include "network/network_data.h"

    16

#include "network/network_server.h"

    17

#include "core/alloc_func.hpp"

    18

#include "window_func.h"

    19

#include "string_func.h"

    20

#include "gfx_func.h"

    21

    22

#include "table/strings.h"

    23

    24

#define ICON_BUFFER 79

    25

#define ICON_HISTORY_SIZE 20

    26

#define ICON_LINE_HEIGHT 12

    27

#define ICON_RIGHT_BORDERWIDTH 10

    28

#define ICON_BOTTOM_BORDERWIDTH 12

    29

#define ICON_MAX_ALIAS_LINES 40

    30

#define ICON_TOKEN_COUNT 20

    31

    32

/* console parser */

    33

IConsoleCmd   *_iconsole_cmds;    ///< list of registred commands

    34

IConsoleVar   *_iconsole_vars;    ///< list of registred vars

    35

IConsoleAlias *_iconsole_aliases; ///< list of registred aliases

    36

    37

/* console colors/modes */

    38

byte _icolour_def;

    39

byte _icolour_err;

    40

byte _icolour_warn;

    41

byte _icolour_dbg;

    42

byte _icolour_cmd;

    43

IConsoleModes _iconsole_mode;

    44

    45

/* ** main console ** */

    46

static char *_iconsole_buffer[ICON_BUFFER + 1];

    47

static uint16 _iconsole_cbuffer[ICON_BUFFER + 1];

    48

static Textbuf _iconsole_cmdline;

    49

    50

/* ** stdlib ** */

    51

byte _stdlib_developer = 1;

    52

bool _stdlib_con_developer = false;

    53

FILE *_iconsole_output_file;

    54

    55

/* ** main console cmd buffer ** */

    56

static char *_iconsole_history[ICON_HISTORY_SIZE];

    57

static byte _iconsole_historypos;

    58

    59

/* *************** *

    60

 *  end of header  *

    61

 * *************** */

    62

    63

static void IConsoleClearCommand()

    64

{

    65

	memset(_iconsole_cmdline.buf, 0, ICON_CMDLN_SIZE);

    66

	_iconsole_cmdline.length = 0;

    67

	_iconsole_cmdline.width = 0;

    68

	_iconsole_cmdline.caretpos = 0;

    69

	_iconsole_cmdline.caretxoffs = 0;

    70

	SetWindowDirty(FindWindowById(WC_CONSOLE, 0));

    71

}

    72

    73

static inline void IConsoleResetHistoryPos() {_iconsole_historypos = ICON_HISTORY_SIZE - 1;}

    74

    75

    76

static void IConsoleHistoryAdd(const char *cmd);

    77

static void IConsoleHistoryNavigate(int direction);

    78

    79

/* ** console window ** */

    80

static void IConsoleWndProc(Window *w, WindowEvent *e)

    81

{

    82

	static byte iconsole_scroll = ICON_BUFFER;

    83

    84

	switch (e->event) {

    85

		case WE_PAINT: {

    86

			int i = iconsole_scroll;

    87

			int max = (w->height / ICON_LINE_HEIGHT) - 1;

    88

			int delta = 0;

    89

			GfxFillRect(w->left, w->top, w->width, w->height - 1, 0);

    90

			while ((i > 0) && (i > iconsole_scroll - max) && (_iconsole_buffer[i] != NULL)) {

    91

				DoDrawString(_iconsole_buffer[i], 5,

    92

					w->height - (iconsole_scroll + 2 - i) * ICON_LINE_HEIGHT, _iconsole_cbuffer[i]);

    93

				i--;

    94

			}

    95

			/* If the text is longer than the window, don't show the starting ']' */

    96

			delta = w->width - 10 - _iconsole_cmdline.width - ICON_RIGHT_BORDERWIDTH;

    97

			if (delta > 0) {

    98

				DoDrawString("]", 5, w->height - ICON_LINE_HEIGHT, _icolour_cmd);

    99

				delta = 0;

   100

			}

   101

   102

			DoDrawString(_iconsole_cmdline.buf, 10 + delta, w->height - ICON_LINE_HEIGHT, _icolour_cmd);

   103

   104

			if (_iconsole_cmdline.caret)

   105

				DoDrawString("_", 10 + delta + _iconsole_cmdline.caretxoffs, w->height - ICON_LINE_HEIGHT, TC_WHITE);

   106

			break;

   107

		}

   108

		case WE_MOUSELOOP:

   109

			if (HandleCaret(&_iconsole_cmdline))

   110

				SetWindowDirty(w);

   111

			break;

   112

		case WE_DESTROY:

   113

			_iconsole_mode = ICONSOLE_CLOSED;

   114

			break;

   115

		case WE_KEYPRESS:

   116

			e->we.keypress.cont = false;

   117

			switch (e->we.keypress.keycode) {

   118

				case WKC_UP:

   119

					IConsoleHistoryNavigate(+1);

   120

					SetWindowDirty(w);

   121

					break;

   122

				case WKC_DOWN:

   123

					IConsoleHistoryNavigate(-1);

   124

					SetWindowDirty(w);

   125

					break;

   126

				case WKC_SHIFT | WKC_PAGEUP:

   127

					if (iconsole_scroll - (w->height / ICON_LINE_HEIGHT) - 1 < 0) {

   128

						iconsole_scroll = 0;

   129

					} else {

   130

						iconsole_scroll -= (w->height / ICON_LINE_HEIGHT) - 1;

   131

					}

   132

					SetWindowDirty(w);

   133

					break;

   134

				case WKC_SHIFT | WKC_PAGEDOWN:

   135

					if (iconsole_scroll + (w->height / ICON_LINE_HEIGHT) - 1 > ICON_BUFFER) {

   136

						iconsole_scroll = ICON_BUFFER;

   137

					} else {

   138

						iconsole_scroll += (w->height / ICON_LINE_HEIGHT) - 1;

   139

					}

   140

					SetWindowDirty(w);

   141

					break;

   142

				case WKC_SHIFT | WKC_UP:

   143

					if (iconsole_scroll <= 0) {

   144

						iconsole_scroll = 0;

   145

					} else {

   146

						--iconsole_scroll;

   147

					}

   148

					SetWindowDirty(w);

   149

					break;

   150

				case WKC_SHIFT | WKC_DOWN:

   151

					if (iconsole_scroll >= ICON_BUFFER) {

   152

						iconsole_scroll = ICON_BUFFER;

   153

					} else {

   154

						++iconsole_scroll;

   155

					}

   156

					SetWindowDirty(w);

   157

					break;

   158

				case WKC_BACKQUOTE:

   159

					IConsoleSwitch();

   160

					break;

   161

				case WKC_RETURN: case WKC_NUM_ENTER:

   162

					IConsolePrintF(_icolour_cmd, "] %s", _iconsole_cmdline.buf);

   163

					IConsoleHistoryAdd(_iconsole_cmdline.buf);

   164

   165

					IConsoleCmdExec(_iconsole_cmdline.buf);

   166

					IConsoleClearCommand();

   167

					break;

   168

				case WKC_CTRL | WKC_RETURN:

   169

					_iconsole_mode = (_iconsole_mode == ICONSOLE_FULL) ? ICONSOLE_OPENED : ICONSOLE_FULL;

   170

					IConsoleResize(w);

   171

					MarkWholeScreenDirty();

   172

					break;

   173

				case (WKC_CTRL | 'V'):

   174

					if (InsertTextBufferClipboard(&_iconsole_cmdline)) {

   175

						IConsoleResetHistoryPos();

   176

						SetWindowDirty(w);

   177

					}

   178

					break;

   179

				case (WKC_CTRL | 'L'):

   180

					IConsoleCmdExec("clear");

   181

					break;

   182

				case (WKC_CTRL | 'U'):

   183

					DeleteTextBufferAll(&_iconsole_cmdline);

   184

					SetWindowDirty(w);

   185

					break;

   186

				case WKC_BACKSPACE: case WKC_DELETE:

   187

					if (DeleteTextBufferChar(&_iconsole_cmdline, e->we.keypress.keycode)) {

   188

						IConsoleResetHistoryPos();

   189

						SetWindowDirty(w);

   190

					}

   191

					break;

   192

				case WKC_LEFT: case WKC_RIGHT: case WKC_END: case WKC_HOME:

   193

					if (MoveTextBufferPos(&_iconsole_cmdline, e->we.keypress.keycode)) {

   194

						IConsoleResetHistoryPos();

   195

						SetWindowDirty(w);

   196

					}

   197

					break;

   198

				default:

   199

					if (IsValidChar(e->we.keypress.key, CS_ALPHANUMERAL)) {

   200

						iconsole_scroll = ICON_BUFFER;

   201

						InsertTextBufferChar(&_iconsole_cmdline, e->we.keypress.key);

   202

						IConsoleResetHistoryPos();

   203

						SetWindowDirty(w);

   204

					} else {

   205

						e->we.keypress.cont = true;

   206

					}

   207

			break;

   208

		}

   209

	}

   210

}

   211

   212

static const Widget _iconsole_window_widgets[] = {

   213

	{WIDGETS_END}

   214

};

   215

   216

static const WindowDesc _iconsole_window_desc = {

   217

	0, 0, 2, 2, 2, 2,

   218

	WC_CONSOLE, WC_NONE,

   219

	WDF_STD_TOOLTIPS | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS,

   220

	_iconsole_window_widgets,

   221

	IConsoleWndProc,

   222

};

   223

   224

void IConsoleInit()

   225

{

   226

	extern const char _openttd_revision[];

   227

	_iconsole_output_file = NULL;

   228

	_icolour_def  =  1;

   229

	_icolour_err  =  3;

   230

	_icolour_warn = 13;

   231

	_icolour_dbg  =  5;

   232

	_icolour_cmd  = TC_GOLD;

   233

	_iconsole_historypos = ICON_HISTORY_SIZE - 1;

   234

	_iconsole_mode = ICONSOLE_CLOSED;

   235

   236

#ifdef ENABLE_NETWORK /* Initialize network only variables */

   237

	_redirect_console_to_client = 0;

   238

#endif

   239

   240

	memset(_iconsole_history, 0, sizeof(_iconsole_history));

   241

	memset(_iconsole_buffer, 0, sizeof(_iconsole_buffer));

   242

	memset(_iconsole_cbuffer, 0, sizeof(_iconsole_cbuffer));

   243

	_iconsole_cmdline.buf = CallocT<char>(ICON_CMDLN_SIZE); // create buffer and zero it

   244

	_iconsole_cmdline.maxlength = ICON_CMDLN_SIZE;

   245

   246

	IConsolePrintF(13, "OpenTTD Game Console Revision 7 - %s", _openttd_revision);

   247

	IConsolePrint(12,  "------------------------------------");

   248

	IConsolePrint(12,  "use \"help\" for more information");

   249

	IConsolePrint(12,  "");

   250

	IConsoleStdLibRegister();

   251

	IConsoleClearCommand();

   252

	IConsoleHistoryAdd("");

   253

}

   254

   255

void IConsoleClearBuffer()

   256

{

   257

	uint i;

   258

	for (i = 0; i <= ICON_BUFFER; i++) {

   259

		free(_iconsole_buffer[i]);

   260

		_iconsole_buffer[i] = NULL;

   261

	}

   262

}

   263

   264

static void IConsoleClear()

   265

{

   266

	free(_iconsole_cmdline.buf);

   267

	IConsoleClearBuffer();

   268

}

   269

   270

static void IConsoleWriteToLogFile(const char *string)

   271

{

   272

	if (_iconsole_output_file != NULL) {

   273

		/* if there is an console output file ... also print it there */

   274

		fwrite(string, strlen(string), 1, _iconsole_output_file);

   275

		fwrite("\n", 1, 1, _iconsole_output_file);

   276

	}

   277

}

   278

   279

bool CloseConsoleLogIfActive()

   280

{

   281

	if (_iconsole_output_file != NULL) {

   282

		IConsolePrintF(_icolour_def, "file output complete");

   283

		fclose(_iconsole_output_file);

   284

		_iconsole_output_file = NULL;

   285

		return true;

   286

	}

   287

   288

	return false;

   289

}

   290

   291

void IConsoleFree()

   292

{

   293

	IConsoleClear();

   294

	CloseConsoleLogIfActive();

   295

}

   296

   297

void IConsoleResize(Window *w)

   298

{

   299

	switch (_iconsole_mode) {

   300

		case ICONSOLE_OPENED:

   301

			w->height = _screen.height / 3;

   302

			w->width = _screen.width;

   303

			break;

   304

		case ICONSOLE_FULL:

   305

			w->height = _screen.height - ICON_BOTTOM_BORDERWIDTH;

   306

			w->width = _screen.width;

   307

			break;

   308

		default: return;

   309

	}

   310

   311

	MarkWholeScreenDirty();

   312

}

   313

   314

void IConsoleSwitch()

   315

{

   316

	switch (_iconsole_mode) {

   317

		case ICONSOLE_CLOSED: {

   318

			Window *w = AllocateWindowDesc(&_iconsole_window_desc);

   319

			w->height = _screen.height / 3;

   320

			w->width = _screen.width;

   321

			_iconsole_mode = ICONSOLE_OPENED;

   322

			SetBit(_no_scroll, SCROLL_CON); // override cursor arrows; the gamefield will not scroll

   323

		} break;

   324

		case ICONSOLE_OPENED: case ICONSOLE_FULL:

   325

			DeleteWindowById(WC_CONSOLE, 0);

   326

			_iconsole_mode = ICONSOLE_CLOSED;

   327

			ClrBit(_no_scroll, SCROLL_CON);

   328

			break;

   329

	}

   330

   331

	MarkWholeScreenDirty();

   332

}

   333

   334

void IConsoleClose() {if (_iconsole_mode == ICONSOLE_OPENED) IConsoleSwitch();}

   335

void IConsoleOpen()  {if (_iconsole_mode == ICONSOLE_CLOSED) IConsoleSwitch();}

   336

   337

/**

   338

 * Add the entered line into the history so you can look it back

   339

 * scroll, etc. Put it to the beginning as it is the latest text

   340

 * @param cmd Text to be entered into the 'history'

   341

 */

   342

static void IConsoleHistoryAdd(const char *cmd)

   343

{

   344

	free(_iconsole_history[ICON_HISTORY_SIZE - 1]);

   345

   346

	memmove(&_iconsole_history[1], &_iconsole_history[0], sizeof(_iconsole_history[0]) * (ICON_HISTORY_SIZE - 1));

   347

	_iconsole_history[0] = strdup(cmd);

   348

	IConsoleResetHistoryPos();

   349

}

   350

   351

/**

   352

 * Navigate Up/Down in the history of typed commands

   353

 * @param direction Go further back in history (+1), go to recently typed commands (-1)

   354

 */

   355

static void IConsoleHistoryNavigate(int direction)

   356

{

   357

	int i = _iconsole_historypos + direction;

   358

   359

	/* watch out for overflows, just wrap around */

   360

	if (i < 0) i = ICON_HISTORY_SIZE - 1;

   361

	if (i >= ICON_HISTORY_SIZE) i = 0;

   362

   363

	if (direction > 0)

   364

		if (_iconsole_history[i] == NULL) i = 0;

   365

   366

	if (direction < 0) {

   367

		while (i > 0 && _iconsole_history[i] == NULL) i--;

   368

	}

   369

   370

	_iconsole_historypos = i;

   371

	IConsoleClearCommand();

   372

	/* copy history to 'command prompt / bash' */

   373

	assert(_iconsole_history[i] != NULL && IsInsideMM(i, 0, ICON_HISTORY_SIZE));

   374

	ttd_strlcpy(_iconsole_cmdline.buf, _iconsole_history[i], _iconsole_cmdline.maxlength);

   375

	UpdateTextBufferSize(&_iconsole_cmdline);

   376

}

   377

   378

/**

   379

 * Handle the printing of text entered into the console or redirected there

   380

 * by any other means. Text can be redirected to other players in a network game

   381

 * as well as to a logfile. If the network server is a dedicated server, all activities

   382

 * are also logged. All lines to print are added to a temporary buffer which can be

   383

 * used as a history to print them onscreen

   384

 * @param color_code the colour of the command. Red in case of errors, etc.

   385

 * @param string the message entered or output on the console (notice, error, etc.)

   386

 */

   387

void IConsolePrint(uint16 color_code, const char *string)

   388

{

   389

	char *str;

   390

#ifdef ENABLE_NETWORK

   391

	if (_redirect_console_to_client != 0) {

   392

		/* Redirect the string to the client */

   393

		SEND_COMMAND(PACKET_SERVER_RCON)(NetworkFindClientStateFromIndex(_redirect_console_to_client), color_code, string);

   394

		return;

   395

	}

   396

#endif

   397

   398

	/* Create a copy of the string, strip if of colours and invalid

   399

	 * characters and (when applicable) assign it to the console buffer */

   400

	str = strdup(string);

   401

	str_strip_colours(str);

   402

	str_validate(str);

   403

   404

	if (_network_dedicated) {

   405

		fprintf(stdout, "%s\n", str);

   406

		fflush(stdout);

   407

		IConsoleWriteToLogFile(str);

   408

		free(str); // free duplicated string since it's not used anymore

   409

		return;

   410

	}

   411

   412

	/* move up all the strings in the buffer one place and do the same for colour

   413

	 * to accomodate for the new command/message */

   414

	free(_iconsole_buffer[0]);

   415

	memmove(&_iconsole_buffer[0], &_iconsole_buffer[1], sizeof(_iconsole_buffer[0]) * ICON_BUFFER);

   416

	_iconsole_buffer[ICON_BUFFER] = str;

   417

   418

	memmove(&_iconsole_cbuffer[0], &_iconsole_cbuffer[1], sizeof(_iconsole_cbuffer[0]) * ICON_BUFFER);

   419

	_iconsole_cbuffer[ICON_BUFFER] = color_code;

   420

   421

	IConsoleWriteToLogFile(_iconsole_buffer[ICON_BUFFER]);

   422

   423

	SetWindowDirty(FindWindowById(WC_CONSOLE, 0));

   424

}

   425

   426

/**

   427

 * Handle the printing of text entered into the console or redirected there

   428

 * by any other means. Uses printf() style format, for more information look

   429

 * at IConsolePrint()

   430

 */

   431

void CDECL IConsolePrintF(uint16 color_code, const char *s, ...)

   432

{

   433

	va_list va;

   434

	char buf[ICON_MAX_STREAMSIZE];

   435

   436

	va_start(va, s);

   437

	vsnprintf(buf, sizeof(buf), s, va);

   438

	va_end(va);

   439

   440

	IConsolePrint(color_code, buf);

   441

}

   442

   443

/**

   444

 * It is possible to print debugging information to the console,

   445

 * which is achieved by using this function. Can only be used by

   446

 * debug() in debug.cpp. You need at least a level 2 (developer) for debugging

   447

 * messages to show up

   448

 * @param dbg debugging category

   449

 * @param string debugging message

   450

 */

   451

void IConsoleDebug(const char *dbg, const char *string)

   452

{

   453

	if (_stdlib_developer > 1)

   454

		IConsolePrintF(_icolour_dbg, "dbg: [%s] %s", dbg, string);

   455

}

   456

   457

/**

   458

 * It is possible to print warnings to the console. These are mostly

   459

 * errors or mishaps, but non-fatal. You need at least a level 1 (developer) for

   460

 * debugging messages to show up

   461

 */

   462

void IConsoleWarning(const char *string)

   463

{

   464

	if (_stdlib_developer > 0)

   465

		IConsolePrintF(_icolour_warn, "WARNING: %s", string);

   466

}

   467

   468

/**

   469

 * It is possible to print error information to the console. This can include

   470

 * game errors, or errors in general you would want the user to notice

   471

 */

   472

void IConsoleError(const char *string)

   473

{

   474

	IConsolePrintF(_icolour_err, "ERROR: %s", string);

   475

}

   476

   477

/**

   478

 * Change a string into its number representation. Supports