Mercurial > hg > orthanc
annotate Core/Lua/LuaContext.cpp @ 824:a811bdf8b8eb
precompiled headers
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Mon, 02 Jun 2014 11:36:07 +0200 |
| parents | 2d0a347e8cfc |
| children | cf52f3bcb2b3 |
| rev | line source |
|---|---|
| 386 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 689 | 3 * Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, |
| 386 | 4 * Belgium |
| 5 * | |
| 6 * This program is free software: you can redistribute it and/or | |
| 7 * modify it under the terms of the GNU General Public License as | |
| 8 * published by the Free Software Foundation, either version 3 of the | |
| 9 * License, or (at your option) any later version. | |
| 10 * | |
| 11 * In addition, as a special exception, the copyright holders of this | |
| 12 * program give permission to link the code of its release with the | |
| 13 * OpenSSL project's "OpenSSL" library (or with modified versions of it | |
| 14 * that use the same license as the "OpenSSL" library), and distribute | |
| 15 * the linked executables. You must obey the GNU General Public License | |
| 16 * in all respects for all of the code used other than "OpenSSL". If you | |
| 17 * modify file(s) with this exception, you may extend this exception to | |
| 18 * your version of the file(s), but you are not obligated to do so. If | |
| 19 * you do not wish to do so, delete this exception statement from your | |
| 20 * version. If you delete this exception statement from all source files | |
| 21 * in the program, then also delete it here. | |
| 22 * | |
| 23 * This program is distributed in the hope that it will be useful, but | |
| 24 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 26 * General Public License for more details. | |
| 27 * | |
| 28 * You should have received a copy of the GNU General Public License | |
| 29 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 30 **/ | |
| 31 | |
| 32 | |
|
824
a811bdf8b8eb
precompiled headers
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
33 #include "../PrecompiledHeaders.h" |
| 386 | 34 #include "LuaContext.h" |
| 35 | |
| 36 #include <glog/logging.h> | |
| 37 | |
| 38 extern "C" | |
| 39 { | |
| 40 #include <lualib.h> | |
| 41 #include <lauxlib.h> | |
| 42 } | |
| 43 | |
| 44 namespace Orthanc | |
| 45 { | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
46 int LuaContext::PrintToLog(lua_State *state) |
| 386 | 47 { |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
48 // Get the pointer to the "LuaContext" underlying object |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
49 lua_getglobal(state, "_LuaContext"); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
50 assert(lua_type(state, -1) == LUA_TLIGHTUSERDATA); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
51 LuaContext* that = const_cast<LuaContext*>(reinterpret_cast<const LuaContext*>(lua_topointer(state, -1))); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
52 assert(that != NULL); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
53 lua_pop(state, 1); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
54 |
| 386 | 55 // http://medek.wordpress.com/2009/02/03/wrapping-lua-errors-and-print-function/ |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
56 int nArgs = lua_gettop(state); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
57 lua_getglobal(state, "tostring"); |
| 386 | 58 |
| 59 // Make sure you start at 1 *NOT* 0 for arrays in Lua. | |
| 60 std::string result; | |
| 61 | |
| 62 for (int i = 1; i <= nArgs; i++) | |
| 63 { | |
| 64 const char *s; | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
65 lua_pushvalue(state, -1); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
66 lua_pushvalue(state, i); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
67 lua_call(state, 1, 1); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
68 s = lua_tostring(state, -1); |
| 386 | 69 |
| 70 if (result.size() > 0) | |
| 71 result.append(", "); | |
| 72 | |
| 73 if (s == NULL) | |
| 74 result.append("<No conversion to string>"); | |
| 75 else | |
| 76 result.append(s); | |
| 77 | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
78 lua_pop(state, 1); |
| 386 | 79 } |
| 80 | |
|
397
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
81 LOG(INFO) << "Lua says: " << result; |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
82 that->log_.append(result); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
83 that->log_.append("\n"); |
| 386 | 84 |
| 85 return 0; | |
| 86 } | |
| 87 | |
| 88 | |
| 89 LuaContext::LuaContext() | |
| 90 { | |
| 91 lua_ = luaL_newstate(); | |
| 92 if (!lua_) | |
| 93 { | |
| 94 throw LuaException("Unable to create the Lua context"); | |
| 95 } | |
| 96 | |
| 97 luaL_openlibs(lua_); | |
| 98 lua_register(lua_, "print", PrintToLog); | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
99 |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
100 lua_pushlightuserdata(lua_, this); |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
101 lua_setglobal(lua_, "_LuaContext"); |
| 386 | 102 } |
| 103 | |
| 104 | |
| 105 LuaContext::~LuaContext() | |
| 106 { | |
| 107 lua_close(lua_); | |
| 108 } | |
| 109 | |
| 110 | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
111 void LuaContext::Execute(std::string* output, |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
112 const std::string& command) |
| 386 | 113 { |
| 114 boost::mutex::scoped_lock lock(mutex_); | |
| 115 | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
116 log_.clear(); |
| 386 | 117 int error = (luaL_loadbuffer(lua_, command.c_str(), command.size(), "line") || |
| 118 lua_pcall(lua_, 0, 0, 0)); | |
| 119 | |
| 120 if (error) | |
| 121 { | |
| 122 assert(lua_gettop(lua_) >= 1); | |
| 123 | |
| 124 std::string description(lua_tostring(lua_, -1)); | |
| 125 lua_pop(lua_, 1); /* pop error message from the stack */ | |
| 126 throw LuaException(description); | |
| 127 } | |
|
418
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
128 |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
129 if (output != NULL) |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
130 { |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
131 *output = log_; |
|
b79bf2f4ab2e
execution of lua through REST
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
397
diff
changeset
|
132 } |
| 386 | 133 } |
| 134 | |
| 135 | |
| 136 void LuaContext::Execute(EmbeddedResources::FileResourceId resource) | |
| 137 { | |
| 138 std::string command; | |
| 139 EmbeddedResources::GetFileResource(command, resource); | |
| 140 Execute(command); | |
| 141 } | |
|
397
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
142 |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
143 |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
144 bool LuaContext::IsExistingFunction(const char* name) |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
145 { |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
146 boost::mutex::scoped_lock lock(mutex_); |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
147 lua_settop(lua_, 0); |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
148 lua_getglobal(lua_, name); |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
149 return lua_type(lua_, -1) == LUA_TFUNCTION; |
|
941ea46e9e26
lua filter of new instances
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
386
diff
changeset
|
150 } |
| 386 | 151 } |