1

/* $Id$ */

     2

     3

/** @file fios.c

     4

 * This file contains functions for building file lists for the save/load dialogs.

     5

 */

     6

     7

#include "stdafx.h"

     8

#include "openttd.h"

     9

#include "hal.h"

    10

#include "string.h"

    11

#include "variables.h"

    12

#include "functions.h"

    13

#include "heightmap.h"

    14

#include "table/strings.h"

    15

#include "fios.h"

    16

#include <sys/types.h>

    17

#include <sys/stat.h>

    18

    19

#ifdef WIN32

    20

# include <tchar.h>

    21

# include <io.h>

    22

#else

    23

# include <unistd.h>

    24

#endif /* WIN32 */

    25

    26

/* Variables to display file lists */

    27

int _fios_num;

    28

    29

static char *_fios_path;

    30

static FiosItem *_fios_items;

    31

static int _fios_count, _fios_alloc;

    32

    33

/* OS-specific functions are taken from their respective files (win32/unix/os2 .c) */

    34

extern bool FiosIsRoot(const char *path);

    35

extern bool FiosIsValidFile(const char *path, const struct dirent *ent, struct stat *sb);

    36

extern bool FiosIsHiddenFile(const struct dirent *ent);

    37

extern void FiosGetDrives(void);

    38

extern bool FiosGetDiskFreeSpace(const char *path, uint32 *tot);

    39

    40

/* get the name of an oldstyle savegame */

    41

extern void GetOldSaveGameName(char *title, const char *path, const char *file);

    42

    43

/**

    44

 * Allocate a new FiosItem.

    45

 * @return A pointer to the newly allocated FiosItem.

    46

 */

    47

FiosItem *FiosAlloc(void)

    48

{

    49

	if (_fios_count == _fios_alloc) {

    50

		_fios_alloc += 256;

    51

		_fios_items = realloc(_fios_items, _fios_alloc * sizeof(FiosItem));

    52

	}

    53

	return &_fios_items[_fios_count++];

    54

}

    55

    56

/**

    57

 * Compare two FiosItem's. Used with qsort when sorting the file list.

    58

 * @param a A pointer to the first FiosItem to compare.

    59

 * @param a A pointer to the second FiosItem to compare.

    60

 * @return -1, 0 or 1, depending on how the two items should be sorted.

    61

 */

    62

int CDECL compare_FiosItems(const void *a, const void *b)

    63

{

    64

	const FiosItem *da = (const FiosItem *)a;

    65

	const FiosItem *db = (const FiosItem *)b;

    66

	int r;

    67

    68

	if (_savegame_sort_order & SORT_BY_NAME) {

    69

		r = strcasecmp(da->title, db->title);

    70

	} else {

    71

		r = da->mtime < db->mtime ? -1 : 1;

    72

	}

    73

    74

	if (_savegame_sort_order & SORT_DESCENDING) r = -r;

    75

	return r;

    76

}

    77

    78

/**

    79

 * Free the list of savegames

    80

 */

    81

void FiosFreeSavegameList(void)

    82

{

    83

	free(_fios_items);

    84

	_fios_items = NULL;

    85

	_fios_alloc = _fios_count = 0;

    86

}

    87

    88

/**

    89

 * Get descriptive texts. Returns the path and free space

    90

 * left on the device

    91

 * @param path string describing the path

    92

 * @param total_free total free space in megabytes, optional (can be NULL)

    93

 * @return StringID describing the path (free space or failure)

    94

 */

    95

StringID FiosGetDescText(const char **path, uint32 *total_free)

    96

{

    97

	*path = _fios_path;

    98

	return FiosGetDiskFreeSpace(*path, total_free) ? STR_4005_BYTES_FREE : STR_4006_UNABLE_TO_READ_DRIVE;

    99

}

   100

   101

/* Browse to a new path based on the passed FiosItem struct

   102

 * @param *item FiosItem object telling us what to do

   103

 * @return a string if we have given a file as a target, otherwise NULL */

   104

char *FiosBrowseTo(const FiosItem *item)

   105

{

   106

	char *s;

   107

	char *path = _fios_path;

   108

   109

	switch (item->type) {

   110

#if defined(WIN32) || defined(__OS2__)

   111

	case FIOS_TYPE_DRIVE: sprintf(path, "%c:" PATHSEP, item->title[0]); break;

   112

#endif

   113

   114

	case FIOS_TYPE_PARENT:

   115

		/* Check for possible NULL ptr (not required for UNIXes, but AmigaOS-alikes) */

   116

		if ((s = strrchr(path, PATHSEPCHAR)) != NULL) {

   117

			s[1] = '\0'; // go up a directory

   118

			if (!FiosIsRoot(path)) s[0] = '\0';

   119

		}

   120

#if defined(__MORPHOS__) || defined(__AMIGAOS__)

   121

		/* On MorphOS or AmigaOS paths look like: "Volume:directory/subdirectory" */

   122

		else if ((s = strrchr(path, ':')) != NULL) s[1] = '\0';

   123

#endif

   124

		break;

   125

   126

	case FIOS_TYPE_DIR:

   127

		if (!FiosIsRoot(path)) strcat(path, PATHSEP);

   128

		strcat(path, item->name);

   129

		break;

   130

   131

	case FIOS_TYPE_DIRECT:

   132

		sprintf(path, "%s" PATHSEP, item->name);

   133

		s = strrchr(path, PATHSEPCHAR);

   134

		if (s != NULL && s[1] == '\0') s[0] = '\0'; // strip trailing slash

   135

		break;

   136

   137

	case FIOS_TYPE_FILE:

   138

	case FIOS_TYPE_OLDFILE:

   139

	case FIOS_TYPE_SCENARIO:

   140

	case FIOS_TYPE_OLD_SCENARIO:

   141

	case FIOS_TYPE_PNG:

   142

	case FIOS_TYPE_BMP:

   143

	{

   144

		static char str_buffr[512];

   145

   146

#if defined(__MORPHOS__) || defined(__AMIGAOS__)

   147

		/* On MorphOS or AmigaOS paths look like: "Volume:directory/subdirectory" */

   148

		if (FiosIsRoot(path)) {

   149

			snprintf(str_buffr, lengthof(str_buffr), "%s:%s", path, item->name);

   150

		} else // XXX - only next line!

   151

#endif

   152

		snprintf(str_buffr, lengthof(str_buffr), "%s" PATHSEP "%s", path, item->name);

   153

   154

		return str_buffr;

   155

	}

   156

	}

   157

   158

	return NULL;

   159

}

   160

   161

void FiosMakeSavegameName(char *buf, const char *name, size_t size)

   162

{

   163

	const char *extension, *period;

   164

   165

	extension = (_game_mode == GM_EDITOR) ? ".scn" : ".sav";

   166

   167

	/* Don't append the extension if it is already there */

   168

	period = strrchr(name, '.');

   169

	if (period != NULL && strcasecmp(period, extension) == 0) extension = "";

   170

   171

	snprintf(buf, size, "%s" PATHSEP "%s%s", _fios_path, name, extension);

   172

}

   173

   174

#if defined(WIN32)

   175

# define unlink _tunlink

   176

#endif

   177

   178

bool FiosDelete(const char *name)

   179

{

   180

	char filename[512];

   181

   182

	FiosMakeSavegameName(filename, name, lengthof(filename));

   183

	return unlink(OTTD2FS(filename)) == 0;

   184

}

   185

   186

bool FileExists(const char *filename)

   187

{

   188

	return access(filename, 0) == 0;

   189

}

   190

   191

typedef byte fios_getlist_callback_proc(int mode, const char *filename, const char *ext, char *title);

   192

   193

/** Create a list of the files in a directory, according to some arbitrary rule.

   194

 *  @param num Will be filled with the amount of items.

   195

 *  @param mode The mode we are in. Some modes don't allow 'parent'.

   196

 *  @param callback The function that is called where you need to do the filtering.

   197

 *  @return Return the list of files. */

   198

static FiosItem *FiosGetFileList(int mode, fios_getlist_callback_proc *callback_proc)

   199

{

   200

	struct stat sb;

   201

	struct dirent *dirent;

   202

	DIR *dir;

   203

	FiosItem *fios;

   204

	int sort_start;

   205

	char d_name[sizeof(fios->name)];

   206

   207

	/* A parent directory link exists if we are not in the root directory */

   208

	if (!FiosIsRoot(_fios_path) && mode != SLD_NEW_GAME) {

   209

		fios = FiosAlloc();

   210

		fios->type = FIOS_TYPE_PARENT;

   211

		fios->mtime = 0;

   212

		ttd_strlcpy(fios->name, "..", lengthof(fios->name));

   213

		ttd_strlcpy(fios->title, ".. (Parent directory)", lengthof(fios->title));

   214

	}

   215

   216

	/* Show subdirectories */

   217

	if (mode != SLD_NEW_GAME && (dir = ttd_opendir(_fios_path)) != NULL) {

   218

		while ((dirent = readdir(dir)) != NULL) {

   219

			ttd_strlcpy(d_name, FS2OTTD(dirent->d_name), sizeof(d_name));

   220

   221

			/* found file must be directory, but not '.' or '..' */

   222

			if (FiosIsValidFile(_fios_path, dirent, &sb) && (sb.st_mode & S_IFDIR) &&

   223

					(!FiosIsHiddenFile(dirent) || strncasecmp(d_name, PERSONAL_DIR, strlen(d_name)) == 0) &&

   224

					strcmp(d_name, ".") != 0 && strcmp(d_name, "..") != 0) {

   225

				fios = FiosAlloc();

   226

				fios->type = FIOS_TYPE_DIR;

   227

				fios->mtime = 0;

   228

				ttd_strlcpy(fios->name, d_name, lengthof(fios->name));

   229

				snprintf(fios->title, lengthof(fios->title), "%s" PATHSEP " (Directory)", d_name);

   230

				str_validate(fios->title);

   231

			}

   232

		}

   233

		closedir(dir);

   234

	}

   235

   236

	/* Sort the subdirs always by name, ascending, remember user-sorting order */

   237

	{

   238

		byte order = _savegame_sort_order;

   239

		_savegame_sort_order = SORT_BY_NAME | SORT_ASCENDING;

   240

		qsort(_fios_items, _fios_count, sizeof(FiosItem), compare_FiosItems);

   241

		_savegame_sort_order = order;

   242

	}

   243

   244

	/* This is where to start sorting for the filenames */

   245

	sort_start = _fios_count;

   246

   247

	/* Show files */

   248

	dir = ttd_opendir(_fios_path);

   249

	if (dir != NULL) {

   250

		while ((dirent = readdir(dir)) != NULL) {

   251

			char fios_title[64];

   252

			char *t;

   253

			byte type;

   254

			ttd_strlcpy(d_name, FS2OTTD(dirent->d_name), sizeof(d_name));

   255

   256

			if (!FiosIsValidFile(_fios_path, dirent, &sb) || !(sb.st_mode & S_IFREG) || FiosIsHiddenFile(dirent)) continue;

   257

   258

			/* File has no extension, skip it */

   259

			if ((t = strrchr(d_name, '.')) == NULL) continue;

   260

			fios_title[0] = '\0'; // reset the title;

   261

   262

			type = callback_proc(mode, d_name, t, fios_title);

   263

			if (type != FIOS_TYPE_INVALID) {

   264

				fios = FiosAlloc();

   265

				fios->mtime = sb.st_mtime;

   266

				fios->type = type;

   267

				ttd_strlcpy(fios->name, d_name, lengthof(fios->name));

   268

   269

				/* Some callbacks want to lookup the title of the file. Allow that.

   270

				 * If we just copy the title from the filename, strip the extension */

   271

				t = (fios_title[0] == '\0') ? *t = '\0', d_name : fios_title;

   272

				ttd_strlcpy(fios->title, t, lengthof(fios->title));

   273

				str_validate(fios->title);

   274

			}

   275

		}

   276

		closedir(dir);

   277

	}

   278

   279

	qsort(_fios_items + sort_start, _fios_count - sort_start, sizeof(FiosItem), compare_FiosItems);

   280

   281

	/* Show drives */

   282

	if (mode != SLD_NEW_GAME) FiosGetDrives();

   283

   284

	_fios_num = _fios_count;

   285

	return _fios_items;

   286

}

   287

   288

/**

   289

 * Callback for FiosGetFileList. It tells if a file is a savegame or not.

   290

 * @param mode Save/load mode.

   291

 * @param file Name of the file to check.

   292

 * @param ext A pointer to the extension identifier inside file

   293

 * @param title Buffer if a callback wants to lookup the title of the file

   294

 * @return a FIOS_TYPE_* type of the found file, FIOS_TYPE_INVALID if not a savegame

   295

 * @see FiosGetFileList

   296

 * @see FiosGetSavegameList

   297

 */

   298

static byte FiosGetSavegameListCallback(int mode, const char *file, const char *ext, char *title)

   299

{

   300

	/* Show savegame files

   301

	 * .SAV OpenTTD saved game

   302

	 * .SS1 Transport Tycoon Deluxe preset game

   303

	 * .SV1 Transport Tycoon Deluxe (Patch) saved game

   304

	 * .SV2 Transport Tycoon Deluxe (Patch) saved 2-player game */

   305

	if (strcasecmp(ext, ".sav") == 0) return FIOS_TYPE_FILE;

   306

   307

	if (mode == SLD_LOAD_GAME || mode == SLD_LOAD_SCENARIO) {

   308

		if (strcasecmp(ext, ".ss1") == 0 || strcasecmp(ext, ".sv1") == 0 ||

   309

				strcasecmp(ext, ".sv2") == 0) {

   310

			GetOldSaveGameName(title, _fios_path, file);

   311

			return FIOS_TYPE_OLDFILE;

   312

		}

   313

	}

   314

   315

	return FIOS_TYPE_INVALID;

   316

}

   317

   318

/**

   319

 * Get a list of savegames.

   320

 * @param mode Save/load mode.

   321

 * @return A pointer to an array of FiosItem representing all the files to be shown in the save/load dialog.

   322

 * @see FiosGetFileList

   323

 */

   324

FiosItem *FiosGetSavegameList(int mode)

   325

{

   326

	static char *_fios_save_path = NULL;

   327

   328

	if (_fios_save_path == NULL) {

   329

		_fios_save_path = malloc(MAX_PATH);

   330

		ttd_strlcpy(_fios_save_path, _paths.save_dir, MAX_PATH);

   331

	}

   332

   333

	_fios_path = _fios_save_path;

   334

   335

	return FiosGetFileList(mode, &FiosGetSavegameListCallback);

   336

}

   337

   338

/**

   339

 * Callback for FiosGetFileList. It tells if a file is a scenario or not.

   340

 * @param mode Save/load mode.

   341

 * @param file Name of the file to check.

   342

 * @param ext A pointer to the extension identifier inside file

   343

 * @param title Buffer if a callback wants to lookup the title of the file

   344

 * @return a FIOS_TYPE_* type of the found file, FIOS_TYPE_INVALID if not a scenario

   345

 * @see FiosGetFileList

   346

 * @see FiosGetScenarioList

   347

 */

   348

static byte FiosGetScenarioListCallback(int mode, const char *file, const char *ext, char *title)

   349

{

   350

	/* Show scenario files

   351

	 * .SCN OpenTTD style scenario file

   352

	 * .SV0 Transport Tycoon Deluxe (Patch) scenario

   353

	 * .SS0 Transport Tycoon Deluxe preset scenario */

   354

	if (strcasecmp(ext, ".scn") == 0) return FIOS_TYPE_SCENARIO;

   355

   356

	if (mode == SLD_LOAD_GAME || mode == SLD_LOAD_SCENARIO || mode == SLD_NEW_GAME) {

   357

		if (strcasecmp(ext, ".sv0") == 0 || strcasecmp(ext, ".ss0") == 0 ) {

   358

			GetOldSaveGameName(title, _fios_path, file);

   359

			return FIOS_TYPE_OLD_SCENARIO;

   360

		}

   361

	}

   362

   363

	return FIOS_TYPE_INVALID;

   364

}

   365

   366

/**

   367

 * Get a list of scenarios.

   368

 * @param mode Save/load mode.

   369

 * @return A pointer to an array of FiosItem representing all the files to be shown in the save/load dialog.

   370

 * @see FiosGetFileList

   371

 */

   372

FiosItem *FiosGetScenarioList(int mode)

   373

{

   374

	static char *_fios_scn_path = NULL;

   375

   376

	if (_fios_scn_path == NULL) {

   377

		_fios_scn_path = malloc(MAX_PATH);

   378

		ttd_strlcpy(_fios_scn_path, _paths.scenario_dir, MAX_PATH);

   379

	}

   380

   381

	_fios_path = _fios_scn_path;

   382

   383

	return FiosGetFileList(mode, &FiosGetScenarioListCallback);

   384

}

   385

   386

static byte FiosGetHeightmapListCallback(int mode, const char *file, const char *ext, char *title)

   387

{

   388

	/* Show heightmap files

   389

	 * .PNG PNG Based heightmap files

   390

	 * .BMP BMP Based heightmap files

   391

	 */

   392

   393

#ifdef WITH_PNG

   394

	if (strcasecmp(ext, ".png") == 0) return FIOS_TYPE_PNG;

   395

#endif /* WITH_PNG */

   396

   397

	if (strcasecmp(ext, ".bmp") == 0) return FIOS_TYPE_BMP;

   398

   399

	return FIOS_TYPE_INVALID;

   400

}

   401

   402

// Get a list of Heightmaps

   403

FiosItem *FiosGetHeightmapList(int mode)

   404

{

   405

	static char *_fios_hmap_path = NULL;

   406

   407

	if (_fios_hmap_path == NULL) {

   408

		_fios_hmap_path = malloc(MAX_PATH);

   409

		strcpy(_fios_hmap_path, _paths.heightmap_dir);

   410

	}

   411

   412

	_fios_path = _fios_hmap_path;

   413

   414

	return FiosGetFileList(mode, &FiosGetHeightmapListCallback);

   415

}