{

Console out;

if (argc <= 1)

{

fprintf(stderr, "Usage: dfhack-run <command> [args...]\n\n");

fprintf(stderr, "Note: this command does not start DFHack; it is intended to connect\n"

"to a running DFHack instance. If you were trying to start DFHack, run\n"

#ifdef _WIN32

" Dwarf Fortress.exe\n"

#else

" ./dfhack\n"

#endif

"or see the documentation in hack/docs/index.html for more help.\n"

);

#ifdef _WIN32

fprintf(stderr, "\nPress Enter to quit.\n");

fgetc(stdin);

#endif

return 2;

}

// Connect to DFHack

RemoteClient client(&out);

if (!client.connect())

return 2;

out.init(true);

command_result rv;

if (strcmp(argv[1], "--lua") == 0)

{

if (argc <= 3)

{

out.shutdown();

fprintf(stderr, "Usage: dfhack-run --lua <module> <function> [args...]\n");

return 2;

}

RemoteFunction<dfproto::CoreRunLuaRequest,dfproto::StringListMessage> run_call;

if (!run_call.bind(&client, "RunLua"))

{

out.shutdown();

fprintf(stderr, "No RunLua protocol function found.");

return 3;

}

run_call.in()->set_module(argv[2]);

run_call.in()->set_function(argv[3]);

for (int i = 4; i < argc; i++)

run_call.in()->add_arguments(argv[i]);

rv = run_call();

out.flush();

if (rv == CR_OK)

{

for (int i = 0; i < run_call.out()->value_size(); i++)

printf("%s%s", (i>0?"\t":""), run_call.out()->value(i).c_str());

printf("\n");

}

}

else

{

// Call the command

std::vector<std::string> args;

for (int i = 2; i < argc; i++)

args.push_back(argv[i]);

rv = client.run_command(argv[1], args);

}

out.flush();

out.shutdown();

if (rv != CR_OK)

return 1;

return 0;