3

#include <sqstdio.h>

     5

#include "debug.h"

    22

SQInteger SquirrelStd::require(HSQUIRRELVM vm)

    23

{

    24

	SQInteger top = sq_gettop(vm);

    25

	const SQChar *filename;

    26

	SQChar *real_filename;

    27

    28

	sq_getstring(vm, 2, &filename);

    29

    30

	/* Get the script-name of the current file, so we can work relative from it */

    31

	SQStackInfos si;

    32

	sq_stackinfos(vm, 1, &si);

    33

	if (si.source == NULL) {

    34

		DEBUG(misc, 0, "[squirrel] Couldn't detect the script-name of the 'require'-caller; this should never happen!");

    35

		return SQ_ERROR;

    36

	}

    37

	real_filename = strdup(si.source);

    38

	/* Keep the dir, remove the rest */

    39

	SQChar *s = strrchr(real_filename, PATHSEPCHAR);

    40

	if (s != NULL) {

    41

		/* Keep the PATHSEPCHAR there, remove the rest */

    42

		*s++;

    43

		*s = '\0';

    44

	}

    45

	/* And now we concat, so we are relative from the current script */

    46

	strcat(real_filename, filename);

    47

    48

	/* Make sure we are in the root-table, so everything is included in the root-level */

    49

	sq_pushroottable(vm);

    50

	/* Compile and load the file */

    51

	if (!SQ_SUCCEEDED(sqstd_dofile(vm, real_filename, SQFalse, SQTrue))) {

    52

		DEBUG(misc, 0, "[squirrel] Failed to compile '%s'", filename);

    53

		sq_settop(vm, top);

    54

		return SQ_ERROR;

    55

	}

    56

	/* Reset the top, so the stack stays correct */

    57

	sq_settop(vm, top);

    58

	return 0;

    59

}

    60

    61

void squirrel_register_global_std(Squirrel *engine)

    62

{

    63

	/* We don't use squirrel_helper here, as we want to register to the global

    64

	 *  scope and not to a class. */

    65

	engine->AddMethod("require", &SquirrelStd::require, 2, "?s");

    66

}

    67