/*
SIGUSR2 will toggle trace from level 2 to 6
*/
static void traceHandler(void *ignored, MprSignal *sp)
{
int level;
level = mprGetLogLevel() > 2 ? 2 : 6;
mprLog(0, "Change log level to %d", level);
mprSetLogLevel(level);
}
/*
SIGUSR2 will toggle trace from level 2 to 4
*/
static void traceHandler(void *ignored, MprSignal *sp)
{
int level;
level = mprGetLogLevel() > 2 ? 2 : 4;
mprLog("info appweb", 0, "Change log and trace level to %d", level);
mprSetLogLevel(level);
httpSetTraceLevel(level);
}
/*
function redirect(location: String, level: Number = null): Void
*/
static EjsFile *lf_redirect(Ejs *ejs, EjsObj *unused, int argc, EjsObj **argv)
{
cchar *logSpec;
int level;
logSpec = ejsToMulti(ejs, argv[0]);
level = (argc >= 2) ? ejsGetInt(ejs, argv[1]) : -1;
mprStartLogging(logSpec, 0);
if (level >= 0) {
mprSetLogLevel(level);
}
return 0;
}
/*
* Start error and information logging. Note: this is not per-request access logging
*/
int maStartLogging(MprCtx ctx, cchar *logSpec)
{
Mpr *mpr;
MprFile *file;
MprTime now;
char *levelSpec, *spec, timeText[80];
int level;
level = 0;
mpr = mprGetMpr(ctx);
if (logSpec == 0) {
logSpec = "stdout:2";
}
if (*logSpec && strcmp(logSpec, "none") != 0) {
spec = mprStrdup(mpr, logSpec);
if ((levelSpec = strrchr(spec, ':')) != 0 && isdigit((int) levelSpec[1])) {
*levelSpec++ = '\0';
level = atoi(levelSpec);
}
if (strcmp(spec, "stdout") == 0) {
file = mpr->fileService->console;
} else {
if ((file = mprOpen(mpr, spec, O_CREAT | O_WRONLY | O_TRUNC | O_TEXT, 0664)) == 0) {
mprErrorPrintf(mpr, "Can't open log file %s\n", spec);
return -1;
}
}
mprSetLogLevel(mpr, level);
mprSetLogHandler(mpr, logHandler, (void*) file);
now = mprGetTime(mpr);
mprCtime(mpr, timeText, sizeof(timeText), now);
mprLog(mpr, MPR_CONFIG, "Configuration for %s", mprGetAppTitle(mpr));
mprLog(mpr, MPR_CONFIG, "--------------------------------------------");
mprLog(mpr, MPR_CONFIG, "Host: %s", mprGetHostName(mpr));
mprLog(mpr, MPR_CONFIG, "CPU: %s", BLD_HOST_CPU);
mprLog(mpr, MPR_CONFIG, "OS: %s", BLD_HOST_OS);
mprLog(mpr, MPR_CONFIG, "Distribution: %s %s", BLD_HOST_DIST, BLD_HOST_DIST_VER);
mprLog(mpr, MPR_CONFIG, "OS: %s", BLD_HOST_OS);
mprLog(mpr, MPR_CONFIG, "Version: %s.%d", BLD_VERSION, BLD_NUMBER);
mprLog(mpr, MPR_CONFIG, "BuildType: %s", BLD_TYPE);
mprLog(mpr, MPR_CONFIG, "Started at: %s", timeText);
mprLog(mpr, MPR_CONFIG, "--------------------------------------------");
}
return 0;
}
/*
* Start error and information logging. Note: this is not per-request access logging
*/
int maStartLogging(MprCtx ctx, cchar *logSpec)
{
Mpr *mpr;
MprFile *file;
char *levelSpec, *spec;
int level;
level = 0;
mpr = mprGetMpr(ctx);
if (logSpec == 0) {
logSpec = "stdout:0";
}
if (*logSpec && strcmp(logSpec, "none") != 0) {
spec = mprStrdup(mpr, logSpec);
if ((levelSpec = strrchr(spec, ':')) != 0 && isdigit((int) levelSpec[1])) {
*levelSpec++ = '\0';
level = atoi(levelSpec);
}
if (strcmp(spec, "stdout") == 0) {
file = mpr->fileSystem->stdOutput;
} else {
if ((file = mprOpen(mpr, spec, O_CREAT | O_WRONLY | O_TRUNC | O_TEXT, 0664)) == 0) {
mprPrintfError(mpr, "Can't open log file %s\n", spec);
return -1;
}
}
mprSetLogLevel(mpr, level);
mprSetLogHandler(mpr, logHandler, (void*) file);
mprLog(mpr, MPR_CONFIG, "Configuration for %s", mprGetAppTitle(mpr));
mprLog(mpr, MPR_CONFIG, "---------------------------------------------");
mprLog(mpr, MPR_CONFIG, "Host: %s", mprGetHostName(mpr));
mprLog(mpr, MPR_CONFIG, "CPU: %s", BLD_CPU);
mprLog(mpr, MPR_CONFIG, "OS: %s", BLD_OS);
if (strcmp(BLD_DIST, "Unknown") != 0) {
mprLog(mpr, MPR_CONFIG, "Distribution: %s %s", BLD_DIST, BLD_DIST_VER);
}
mprLog(mpr, MPR_CONFIG, "Version: %s-%s", BLD_VERSION, BLD_NUMBER);
mprLog(mpr, MPR_CONFIG, "BuildType: %s", BLD_TYPE);
mprLog(mpr, MPR_CONFIG, "---------------------------------------------");
}
return 0;
}
static int startLogging(Mpr *mpr, char *logSpec)
{
MprFile *file;
char *levelSpec;
int level;
level = 0;
if ((levelSpec = strchr(logSpec, ':')) != 0) {
*levelSpec++ = '\0';
level = atoi(levelSpec);
}
if (strcmp(logSpec, "stdout") == 0) {
file = mpr->fileSystem->stdOutput;
} else {
if ((file = mprOpen(mpr, logSpec, O_CREAT | O_WRONLY | O_TRUNC | O_TEXT, 0664)) == 0) {
mprPrintfError(mpr, "Can't open log file %s\n", logSpec);
return -1;
}
}
mprSetLogLevel(mpr, level);
mprSetLogHandler(mpr, logHandler, (void*) file);
return 0;
}
MAIN(ejsMain, int argc, char **argv, char **envp)
{
Mpr *mpr;
EcCompiler *cp;
Ejs *ejs;
cchar *cmd, *className, *method, *homeDir;
char *argp, *searchPath, *modules, *name, *tok, *extraFiles;
int nextArg, err, ecFlags, stats, merge, bind, noout, debug, optimizeLevel, warnLevel, strict, i, next;
/*
Initialize Multithreaded Portable Runtime (MPR)
*/
mpr = mprCreate(argc, argv, 0);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddStandardSignals();
if (mprStart(mpr) < 0) {
mprError("Cannot start mpr services");
return EJS_ERR;
}
err = 0;
className = 0;
cmd = 0;
method = 0;
searchPath = 0;
stats = 0;
merge = 0;
bind = 1;
noout = 1;
debug = 1;
warnLevel = 1;
optimizeLevel = 9;
strict = 0;
app->files = mprCreateList(-1, 0);
app->iterations = 1;
argc = mpr->argc;
argv = (char**) mpr->argv;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (smatch(argp, "--bind")) {
bind = 1;
} else if (smatch(argp, "--class")) {
if (nextArg >= argc) {
err++;
} else {
className = argv[++nextArg];
}
} else if (smatch(argp, "--chdir") || smatch(argp, "--home") || smatch(argp, "-C")) {
if (nextArg >= argc) {
err++;
} else {
homeDir = argv[++nextArg];
if (chdir((char*) homeDir) < 0) {
mprError("Cannot change directory to %s", homeDir);
}
}
#if BIT_UNIX_LIKE
} else if (smatch(argp, "--chroot")) {
/* Not documented or supported */
if (nextArg >= argc) {
err++;
} else {
homeDir = mprGetAbsPath(argv[++nextArg]);
if (chroot(homeDir) < 0) {
if (errno == EPERM) {
mprPrintfError("%s: Must be super user to use the --chroot option", mprGetAppName(mpr));
} else {
mprPrintfError("%s: Cannot change change root directory to %s, errno %d",
mprGetAppName(), homeDir, errno);
}
return 4;
}
}
#endif
} else if (smatch(argp, "--cmd") || smatch(argp, "-c")) {
if (nextArg >= argc) {
err++;
} else {
cmd = argv[++nextArg];
}
#if BIT_WIN_LIKE
} else if (smatch(argp, "--cygroot")) {
if (nextArg >= argc) {
err++;
} else {
app->cygroot = sclone(argv[++nextArg]);
}
#endif
} else if (smatch(argp, "--debug")) {
debug = 1;
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--files") || smatch(argp, "-f")) {
/* Compatibility with mozilla shell */
if (nextArg >= argc) {
err++;
} else {
extraFiles = sclone(argv[++nextArg]);
name = stok(extraFiles, " \t", &tok);
while (name != NULL) {
mprAddItem(app->files, sclone(name));
name = stok(NULL, " \t", &tok);
}
}
} else if (smatch(argp, "--iterations") || smatch(argp, "-i")) {
if (nextArg >= argc) {
err++;
} else {
app->iterations = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--log")) {
if (nextArg >= argc) {
err++;
} else {
mprStartLogging(argv[++nextArg], 0);
mprSetCmdlineLogging(1);
}
} else if (smatch(argp, "--method")) {
if (nextArg >= argc) {
err++;
} else {
method = argv[++nextArg];
}
} else if (smatch(argp, "--name")) {
/* Just ignore. Used to tag commands with a unique command line */
nextArg++;
} else if (smatch(argp, "--nobind")) {
bind = 0;
} else if (smatch(argp, "--nodebug")) {
debug = 0;
} else if (smatch(argp, "--optimize")) {
if (nextArg >= argc) {
err++;
} else {
optimizeLevel = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "-s")) {
/* Compatibility with mozilla shell. Just ignore */
} else if (smatch(argp, "--search") || smatch(argp, "--searchpath")) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (smatch(argp, "--standard")) {
strict = 0;
} else if (smatch(argp, "--stats")) {
stats = 1;
} else if (smatch(argp, "--strict")) {
strict = 1;
} else if (smatch(argp, "--require")) {
if (nextArg >= argc) {
err++;
} else {
if (app->modules == 0) {
app->modules = mprCreateList(-1, 0);
}
modules = sclone(argv[++nextArg]);
name = stok(modules, " \t", &tok);
while (name != NULL) {
require(name);
name = stok(NULL, " \t", &tok);
}
}
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
mprStartLogging("stderr:2", 0);
mprSetCmdlineLogging(1);
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s-%s\n", BIT_VERSION, BIT_BUILD_NUMBER);
return 0;
} else if (smatch(argp, "--warn")) {
if (nextArg >= argc) {
err++;
} else {
warnLevel = atoi(argv[++nextArg]);
}
} else {
err++;
break;
}
}
if (err) {
/*
If --method or --class is specified, then the named class.method will be run (method defaults to "main", class
defaults to first class with a "main").
Examples:
ejs
ejs script.es arg1 arg2 arg3
ejs --class "Customer" --method "start" --files "script1.es script2.es" main.es
*/
mprPrintfError("Usage: %s [options] script.es [arguments] ...\n"
" Ejscript shell program options:\n"
" --class className # Name of class containing method to run\n"
" --cmd ejscriptCode # Literal ejscript statements to execute\n"
" --cygroot path # Set cygwin root for resolving script paths\n"
" --debug # Use symbolic debugging information (default)\n"
" --debugger # Disable timeouts to make using a debugger easier\n"
" --files \"files..\" # Extra source to compile\n"
" --log logSpec # Internal compiler diagnostics logging\n"
" --method methodName # Name of method to run. Defaults to main\n"
" --nodebug # Omit symbolic debugging information\n"
" --optimize level # Set the optimization level (0-9 default is 9)\n"
" --require 'module,...' # Required list of modules to pre-load\n"
" --search ejsPath # Module search path\n"
" --standard # Default compilation mode to standard (default)\n"
" --stats # Print memory stats on exit\n"
" --strict # Default compilation mode to strict\n"
" --verbose | -v # Same as --log stderr:2 \n"
" --version # Emit the compiler version information\n"
" --warn level # Set the warning message level (0-9 default is 0)\n\n",
mpr->name);
return -1;
}
if ((ejs = ejsCreateVM(argc - nextArg, (cchar **) &argv[nextArg], 0)) == 0) {
return MPR_ERR_MEMORY;
}
app->ejs = ejs;
if (ejsLoadModules(ejs, searchPath, app->modules) < 0) {
return MPR_ERR_CANT_READ;
}
ecFlags = 0;
ecFlags |= (merge) ? EC_FLAGS_MERGE: 0;
ecFlags |= (bind) ? EC_FLAGS_BIND: 0;
ecFlags |= (noout) ? EC_FLAGS_NO_OUT: 0;
ecFlags |= (debug) ? EC_FLAGS_DEBUG: 0;
cp = app->compiler = ecCreateCompiler(ejs, ecFlags);
if (cp == 0) {
return MPR_ERR_MEMORY;
}
ecSetRequire(cp, app->modules);
ecSetOptimizeLevel(cp, optimizeLevel);
ecSetWarnLevel(cp, warnLevel);
ecSetStrictMode(cp, strict);
if (nextArg < argc) {
mprAddItem(app->files, sclone(argv[nextArg]));
}
if (app->cygroot) {
/*
When cygwin invokes a script with shebang, it passes a cygwin path to the script
The ejs --cygroot option permits ejscript to conver this to a native path
*/
for (next = 0; (name = mprGetNextItem(app->files, &next)) != 0; ) {
if (*name == '/' || *name == '\\') {
mprSetItem(app->files, next - 1, sjoin(app->cygroot, name, NULL));
}
}
}
for (i = 0; !err && i < app->iterations; i++) {
if (cmd) {
if (interpretCommands(cp, cmd) < 0) {
err++;
}
} else if (mprGetListLength(app->files) > 0) {
if (interpretFiles(cp, app->files, argc - nextArg, &argv[nextArg], className, method) < 0) {
err++;
}
} else {
/*
No args - run as an interactive shell
*/
if (interpretCommands(cp, NULL) < 0) {
err++;
}
}
}
if (stats) {
#if BIT_DEBUG
mprSetLogLevel(1);
mprPrintMem("Memory Usage", 1);
#endif
}
if (!err) {
err = mpr->exitStatus;
}
app->ejs = 0;
mprTerminate(MPR_EXIT_DEFAULT, err);
ejsDestroyVM(ejs);
mprDestroy(MPR_EXIT_DEFAULT);
return err;
}
int main(int argc, char **argv)
{
Mpr *mpr;
EcCompiler *cp;
EjsService *vmService;
Ejs *ejs;
MprList *useModules, *files;
cchar *cmd, *className, *methodName;
char *argp, *searchPath, *modules, *name, *tok, *extraFiles, *spec;
int nextArg, err, ejsFlags, ecFlags, stats, run, merge, bind, noout, debug, optimizeLevel, nobind, warnLevel;
int compilerMode, lang;
/*
* Create the Embedthis Portable Runtime (MPR) and setup a memory failure handler
*/
mpr = mprCreate(argc, argv, ejsMemoryFailure);
mprSetAppName(mpr, argv[0], 0, 0);
if (mprStart(mpr, 0) < 0) {
mprError(mpr, "Can't start mpr services");
return EJS_ERR;
}
err = 0;
className = 0;
cmd = 0;
methodName = 0;
searchPath = 0;
stats = 0;
run = 1;
merge = 0;
bind = 1;
nobind = 0;
noout = 1;
debug = 1;
warnLevel = 1;
optimizeLevel = 9;
compilerMode = PRAGMA_MODE_STANDARD;
lang = BLD_FEATURE_EJS_LANG;
useModules = mprCreateList(mpr);
files = mprCreateList(mpr);
ejsFlags = EJS_FLAG_COMPILER;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--bind") == 0) {
/* Ignore. Not required and only here for compatibility with ec */
} else if (strcmp(argp, "--class") == 0) {
if (nextArg >= argc) {
err++;
} else {
className = argv[++nextArg];
}
} else if (strcmp(argp, "--cmd") == 0 || strcmp(argp, "-c") == 0) {
if (nextArg >= argc) {
err++;
} else {
cmd = argv[++nextArg];
}
} else if (strcmp(argp, "--debug") == 0) {
debug = 1;
} else if (strcmp(argp, "--lang") == 0) {
if (nextArg >= argc) {
err++;
} else {
spec = argv[++nextArg];
if (mprStrcmpAnyCase(spec, "ecma") == 0) {
lang = EJS_SPEC_ECMA;
} else if (mprStrcmpAnyCase(spec, "plus") == 0) {
lang = EJS_SPEC_PLUS;
} else if (mprStrcmpAnyCase(spec, "fixed") == 0) {
lang = EJS_SPEC_FIXED;
}
}
} else if (strcmp(argp, "--files") == 0 || strcmp(argp, "-f") == 0) {
if (nextArg >= argc) {
err++;
} else {
extraFiles = mprStrdup(mpr, argv[++nextArg]);
name = mprStrTok(extraFiles, " \t", &tok);
while (name != NULL) {
mprAddItem(files, name);
name = mprStrTok(NULL, " \t", &tok);
}
}
} else if (strcmp(argp, "--log") == 0) {
if (nextArg >= argc) {
err++;
} else {
ejsStartLogging(mpr, argv[++nextArg]);
}
} else if (strcmp(argp, "--method") == 0) {
if (nextArg >= argc) {
err++;
} else {
methodName = argv[++nextArg];
}
} else if (strcmp(argp, "--nobind") == 0) {
/*
* This is a hidden switch just for the compiler developers
*/
nobind = 1;
bind = 0;
} else if (strcmp(argp, "--nodebug") == 0) {
debug = 0;
} else if (strcmp(argp, "--optimize") == 0) {
if (nextArg >= argc) {
err++;
} else {
optimizeLevel = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--searchpath") == 0) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (strcmp(argp, "--standard") == 0) {
compilerMode = PRAGMA_MODE_STANDARD;
} else if (strcmp(argp, "--stats") == 0) {
stats = 1;
} else if (strcmp(argp, "--strict") == 0) {
compilerMode = PRAGMA_MODE_STRICT;
} else if (strcmp(argp, "--use") == 0) {
if (nextArg >= argc) {
err++;
} else {
modules = mprStrdup(mpr, argv[++nextArg]);
name = mprStrTok(modules, " \t", &tok);
while (name != NULL) {
mprAddItem(useModules, name);
name = mprStrTok(NULL, " \t", &tok);
}
}
} else if (strcmp(argp, "--version") == 0 || strcmp(argp, "-V") == 0) {
mprErrorPrintf(mpr, "%s %s\n"
"Copyright (C) Embedthis Software 2003-2009\n"
"Copyright (C) Michael O'Brien 2003-2009\n",
BLD_NAME, BLD_VERSION);
exit(0);
} else if (strcmp(argp, "--warn") == 0) {
if (nextArg >= argc) {
err++;
} else {
warnLevel = atoi(argv[++nextArg]);
}
} else {
err++;
break;
}
}
if (err) {
/*
* If --method or --class is specified, then the named class.method will be run (method defaults to "main", class
* defaults to first class with a "main").
*
* Examples:
* ejs
* ejs script.es arg1 arg2 arg3
* ejs --class "Customer" --method "start" --files "script1.es script2.es" main.es
*/
mprErrorPrintf(mpr,
"Usage: %s [options] script.es [arguments] ...\n"
" Ejscript shell program options:\n"
" --class className # Name of class containing method to run\n"
" --cmd ejscriptCode # Literal ejscript statements to execute\n"
" --debug # Include symbolic debugging information in output\n"
" --lang # Language compliance (ecma|plus|fixed)\n"
" --files \"files..\" # Extra source to compile\n"
" --log logSpec # Internal compiler diagnostics logging\n"
" --method methodName # Name of method to run. Defaults to main\n"
" --nodebug # Omit symbolic debugging information in output\n"
" --optimize level # Set the optimization level (0-9 default is 9)\n"
" --searchpath ejsPath # Module search path\n"
" --standard # Default compilation mode to standard (default)\n"
" --stats # Print stats on exit\n"
" --strict # Default compilation mode to strict\n"
" --use 'module, ...' # List of modules to pre-load\n"
" --version # Emit the compiler version information\n"
" --warn level # Set the warning message level (0-9 default is 0)\n\n",
mpr->name);
return -1;
}
vmService = ejsCreateService(mpr);
if (vmService == 0) {
return MPR_ERR_NO_MEMORY;
}
if (searchPath) {
ejsSetSearchPath(vmService, searchPath);
}
ejs = ejsCreate(vmService, 0, ejsFlags);
if (ejs == 0) {
return MPR_ERR_NO_MEMORY;
}
ecFlags = 0;
ecFlags |= (run) ? EC_FLAGS_RUN: 0;
ecFlags |= (merge) ? EC_FLAGS_MERGE: 0;
ecFlags |= (bind) ? EC_FLAGS_BIND_GLOBALS: 0;
ecFlags |= (nobind) ? EC_FLAGS_NO_BIND: 0;
ecFlags |= (noout) ? EC_FLAGS_NO_OUT: 0;
ecFlags |= (debug) ? EC_FLAGS_DEBUG: 0;
cp = ecCreateCompiler(ejs, ecFlags, lang);
if (cp == 0) {
return MPR_ERR_NO_MEMORY;
}
ecSetOptimizeLevel(cp, optimizeLevel);
ecSetWarnLevel(cp, warnLevel);
ecSetDefaultMode(cp, compilerMode);
if (preloadModules(cp, useModules) < 0) {
return EJS_ERR;
}
if (cmd) {
if (interpretCommands(cp, cmd) < 0) {
err++;
}
} else if (nextArg < argc) {
name = argv[nextArg];
mprAddItem(files, argv[nextArg]);
if (interpretFiles(cp, files, argc - nextArg, &argv[nextArg], className, methodName) < 0) {
err++;
}
} else {
/*
* No args - run as an interactive shell
*/
if (interpretCommands(cp, NULL) < 0) {
err++;
}
}
#if BLD_DEBUG
if (stats) {
mprSetLogLevel(ejs, 1);
ejsPrintAllocReport(ejs);
}
#endif
mprFree(mpr);
return err;
}
int APIENTRY WinMain(HINSTANCE inst, HINSTANCE junk, char *args, int junk2)
{
char **argv, *argp;
int argc, err, nextArg;
mpr = mprCreate(0, NULL, MPR_USER_EVENTS_THREAD);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddTerminator(terminating);
err = 0;
app->appInst = inst;
app->heartBeatPeriod = HEART_BEAT_PERIOD;
setAppDefaults();
mprSetAppName(BLD_PRODUCT "Manager", BLD_NAME "Manager", BLD_VERSION);
app->appName = mprGetAppName();
app->appTitle = mprGetAppTitle(mpr);
mprSetLogHandler(logHandler);
mprSetWinMsgCallback((MprMsgCallback) msgProc);
if ((argc = mprMakeArgv(args, &argv, MPR_ARGV_ARGS_ONLY)) < 0) {
return FALSE;
}
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-' || strcmp(argp, "--") == 0) {
break;
}
if (strcmp(argp, "--args") == 0) {
/*
Args to pass to service when it starts
*/
if (nextArg >= argc) {
err++;
} else {
app->serviceArgs = argv[++nextArg];
}
} else if (strcmp(argp, "--console") == 0) {
/*
Allow the service to interact with the console
*/
app->createConsole++;
} else if (strcmp(argp, "--daemon") == 0) {
/* Ignored on windows */
} else if (strcmp(argp, "--heartBeat") == 0) {
/*
Set the heart beat period
*/
if (nextArg >= argc) {
err++;
} else {
app->heartBeatPeriod = atoi(argv[++nextArg]) * 1000;
}
} else if (strcmp(argp, "--home") == 0) {
/*
Change to this directory before starting the service
*/
if (nextArg >= argc) {
err++;
} else {
app->serviceHome = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--log") == 0) {
/*
Pass the log directive through to the service
*/
if (nextArg >= argc) {
err++;
} else {
app->logSpec = sclone(argv[++nextArg]);
mprStartLogging(app->logSpec, 0);
mprSetCmdlineLogging(1);
}
} else if (strcmp(argp, "--name") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->serviceName = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--program") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->serviceProgram = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--verbose") == 0 || strcmp(argp, "-v") == 0) {
mprSetLogLevel(1);
} else {
err++;
}
if (err) {
mprUserError("Bad command line: %s\n"
" Usage: %s [options] [program args]\n"
" Switches:\n"
" --args # Args to pass to service\n"
" --console # Display the service console\n"
" --heartBeat interval # Heart beat interval period (secs)\n"
" --home path # Home directory for service\n"
" --log logFile:level # Log directive for service\n"
" --name name # Name of the service to manage\n"
" --program path # Service program to start\n"
" --verbose # Show command feedback\n"
" Commands:\n"
" disable # Disable the service\n"
" enable # Enable the service\n"
" start # Start the service\n"
" stop # Start the service\n"
" run # Run and watch over the service\n",
args, app->appName);
return -1;
}
}
if (mprStart() < 0) {
mprUserError("Can't start MPR for %s", mprGetAppName());
} else {
mprStartEventsThread();
if (nextArg >= argc) {
return process("run");
} else for (; nextArg < argc; nextArg++) {
if (!process(argv[nextArg])) {
return FALSE;
}
}
}
return TRUE;
}
int main(int argc, char *argv[])
{
char *argp, *value;
int err, nextArg, status;
err = 0;
mprCreate(argc, argv, MPR_USER_EVENTS_THREAD);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddTerminator(terminating);
mprAddStandardSignals();
setAppDefaults();
for (nextArg = 1; nextArg < argc && !err; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--args") == 0) {
/*
Args to pass to service
*/
if (nextArg >= argc) {
err++;
} else {
app->serviceArgs = argv[++nextArg];
}
} else if (strcmp(argp, "--console") == 0) {
/* Does nothing. Here for compatibility with windows watcher */
} else if (strcmp(argp, "--daemon") == 0) {
app->runAsDaemon++;
#if FUTURE
} else if (strcmp(argp, "--heartBeat") == 0) {
/*
Set the frequency to check on the program.
*/
if (nextArg >= argc) {
err++;
} else {
app->heartBeatPeriod = atoi(argv[++nextArg]);
}
#endif
} else if (strcmp(argp, "--home") == 0) {
/*
Change to this directory before starting the service
*/
if (nextArg >= argc) {
err++;
} else {
app->serviceHome = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--log") == 0) {
/*
Pass the log directive through to the service
*/
if (nextArg >= argc) {
err++;
} else {
app->logSpec = sclone(argv[++nextArg]);
mprStartLogging(app->logSpec, 0);
mprSetCmdlineLogging(1);
}
} else if (strcmp(argp, "--name") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->serviceName = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--pidfile") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->pidPath = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--program") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->serviceProgram = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--retries") == 0) {
if (nextArg >= argc) {
err++;
} else {
app->retries = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--signal") == 0) {
if (nextArg >= argc) {
err++;
} else {
value = argv[++nextArg];
if (smatch(value, "SIGABRT")) {
app->signal = SIGABRT;
} else if (smatch(value, "SIGINT")) {
app->signal = SIGINT;
} else if (smatch(value, "SIGHUP")) {
app->signal = SIGHUP;
} else if (smatch(value, "SIGQUIT")) {
app->signal = SIGQUIT;
} else if (smatch(value, "SIGTERM")) {
app->signal = SIGTERM;
} else if (smatch(value, "SIGUSR1")) {
app->signal = SIGUSR1;
} else if (smatch(value, "SIGUSR2")) {
app->signal = SIGUSR2;
} else {
app->signal = atoi(argv[++nextArg]);
}
}
} else if (strcmp(argp, "--verbose") == 0 || strcmp(argp, "-v") == 0) {
mprSetLogLevel(1);
} else {
err++;
break;
}
}
if (nextArg >= argc) {
err++;
}
if (err) {
mprUserError("Bad command line: \n"
" Usage: %s [commands]\n"
" Switches:\n"
" --args # Args to pass to service\n"
" --daemon # Run manager as a daemon\n"
" --home path # Home directory for service\n"
" --log logFile:level # Log directive for service\n"
" --retries count # Max count of app restarts\n"
" --name name # Name of the service to manage\n"
" --pidfile path # Location of the pid file\n"
" --program path # Service program to start\n"
" --signal signo # Signal number to terminate service\n"
" --verbose # Show command feedback\n"
#if FUTURE
" --heartBeat interval # Heart beat interval period (secs) \n"
#endif
" Commands:\n"
" disable # Disable the service\n"
" enable # Enable the service\n"
" install # Install the service\n"
" run # Run and watch over the service\n"
" start # Start the service\n"
" stop # Start the service\n"
" uninstall # Uninstall the service\n"
, app->appName);
return -1;
}
if (!app->pidPath) {
app->pidPath = sjoin(app->pidDir, "/", app->serviceName, ".pid", NULL);
}
if (app->runAsDaemon) {
makeDaemon();
}
status = 0;
if (getuid() != 0) {
mprUserError("Must run with administrator privilege. Use sudo.");
status = 1;
} else if (mprStart() < 0) {
mprUserError("Can't start MPR for %s", mprGetAppName());
status = 2;
} else {
mprStartEventsThread();
for (; nextArg < argc; nextArg++) {
if (!process(argv[nextArg], 0)) {
status = 3;
break;
}
}
}
mprDestroy(MPR_EXIT_DEFAULT);
return status;
}
/*
function set level(value: Number): Void
*/
static EjsObj *lf_set_level(Ejs *ejs, EjsObj *unused, int argc, EjsObj **argv)
{
mprSetLogLevel(ejsGetInt(ejs, argv[0]));
return 0;
}
