Subversion Repositories HomeAutomation

Rev

Rev 1642 | Rev 1647 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | SVN | RSS feed

Rev 1642 Rev 1644
1
/*
1
/*
2
 *
2
 *
3
 *  Copyright (C) 2010  Mattias Runge
3
 *  Copyright (C) 2010  Mattias Runge
4
 *
4
 *
5
 *  This program is free software; you can redistribute it and/or modify
5
 *  This program is free software; you can redistribute it and/or modify
6
 *  it under the terms of the GNU General Public License as published by
6
 *  it under the terms of the GNU General Public License as published by
7
 *  the Free Software Foundation; either version 2 of the License, or
7
 *  the Free Software Foundation; either version 2 of the License, or
8
 *  (at your option) any later version.
8
 *  (at your option) any later version.
9
 *
9
 *
10
 *  This program is distributed in the hope that it will be useful,
10
 *  This program is distributed in the hope that it will be useful,
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 *  GNU General Public License for more details.
13
 *  GNU General Public License for more details.
14
 *
14
 *
15
 *  You should have received a copy of the GNU General Public License along
15
 *  You should have received a copy of the GNU General Public License along
16
 *  with this program; if not, write to the Free Software Foundation, Inc.,
16
 *  with this program; if not, write to the Free Software Foundation, Inc.,
17
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18
 *
18
 *
19
 */
19
 */
20
 
20
 
21
#include "Manager.h"
21
#include "Manager.h"
22
 
22
 
23
#include <fstream>
23
#include <fstream>
24
 
24
 
25
#include <boost/lexical_cast.hpp>
25
#include <boost/lexical_cast.hpp>
26
#include <boost/bind.hpp>
26
#include <boost/bind.hpp>
27
 
27
 
28
namespace atom {
28
namespace atom {
29
namespace vm {
29
namespace vm {
30
 
30
 
31
Manager::Pointer Manager::instance_;
31
Manager::Pointer Manager::instance_;
32
 
32
 
33
Manager::Manager() : LOG("vm::Manager")
33
Manager::Manager() : LOG("vm::Manager")
34
{
34
{
35
}
35
}
36
 
36
 
37
Manager::~Manager()
37
Manager::~Manager()
38
{
38
{
39
    this->io_service_.stop();
39
    this->io_service_.stop();
40
   
40
   
41
    this->plugins_.clear();
41
    this->plugins_.clear();
42
    this->context_.Dispose();
42
    this->context_.Dispose();
43
 
43
 
44
    for (ImportFunctionList::iterator it = this->functions_.begin(); it != this->functions_.end(); it++)
44
    for (ImportFunctionList::iterator it = this->functions_.begin(); it != this->functions_.end(); it++)
45
    {
45
    {
46
        it->second.Dispose();
46
        it->second.Dispose();
47
    }
47
    }
48
}
48
}
49
 
49
 
50
void Manager::Create()
50
void Manager::Create()
51
{
51
{
52
    Manager::instance_ = Manager::Pointer(new Manager());
52
    Manager::instance_ = Manager::Pointer(new Manager());
53
}
53
}
54
 
54
 
55
Manager::Pointer Manager::Instance()
55
Manager::Pointer Manager::Instance()
56
{
56
{
57
    return Manager::instance_;
57
    return Manager::instance_;
58
}
58
}
59
 
59
 
60
void Manager::Delete()
60
void Manager::Delete()
61
{
61
{
62
    Manager::instance_.reset();
62
    Manager::instance_.reset();
63
}
63
}
64
 
64
 
65
v8::Persistent<v8::Context>& Manager::GetContext()
65
v8::Persistent<v8::Context>& Manager::GetContext()
66
{
66
{
67
    return this->context_;
67
    return this->context_;
68
}
68
}
69
 
69
 
70
void Manager::AddPlugin(Plugin::Pointer plugin)
70
void Manager::AddPlugin(Plugin::Pointer plugin)
71
{
71
{
72
    this->plugins_.push_back(plugin);
72
    this->plugins_.push_back(plugin);
73
}
73
}
74
 
74
 
75
void Manager::Start(std::string script_path)
75
void Manager::Start(std::string script_path)
76
{
76
{
77
    this->script_path_ = script_path;
77
    this->script_path_ = script_path;
78
    this->io_service_.post(boost::bind(&Manager::StartHandler, this));
78
    this->io_service_.post(boost::bind(&Manager::StartHandler, this));
79
}
79
}
80
 
80
 
81
void Manager::SendResponse(std::string plugin_name, unsigned int request_id, std::string response)
81
void Manager::SendResponse(std::string plugin_name, unsigned int request_id, std::string response)
82
{
82
{
83
    for (unsigned int n = 0; n < this->plugins_.size(); n++)
83
    for (unsigned int n = 0; n < this->plugins_.size(); n++)
84
    {
84
    {
85
        if (this->plugins_[n]->GetName() == plugin_name)
85
        if (this->plugins_[n]->GetName() == plugin_name)
86
        {
86
        {
87
            this->plugins_[n]->ExecutionResult(response, request_id);
87
            this->plugins_[n]->ExecutionResult(response, request_id);
88
            return;
88
            return;
89
        }
89
        }
90
    }
90
    }
91
}
91
}
92
 
92
 
93
void Manager::Call(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments)
93
void Manager::Call(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments)
94
{
94
{
95
    this->io_service_.post(boost::bind(&Manager::CallHandler, this, plugin_name, request_id, name, arguments));
95
    this->io_service_.post(boost::bind(&Manager::CallHandler, this, plugin_name, request_id, name, arguments));
96
}
96
}
97
 
97
 
98
void Manager::Execute(std::string plugin_name, unsigned int request_id, std::string code)
98
void Manager::Execute(std::string plugin_name, unsigned int request_id, std::string code)
99
{
99
{
100
    this->io_service_.post(boost::bind(&Manager::ExecuteHandler, this, plugin_name, request_id, code));
100
    this->io_service_.post(boost::bind(&Manager::ExecuteHandler, this, plugin_name, request_id, code));
101
}
101
}
102
 
102
 
103
void Manager::CallHandler(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments)
103
void Manager::CallHandler(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments)
104
{
104
{
105
    v8::Context::Scope context_scope(this->context_);
105
    v8::Context::Scope context_scope(this->context_);
106
    v8::TryCatch try_catch;
106
    v8::TryCatch try_catch;
107
   
107
   
108
    if (this->functions_.find(name) == this->functions_.end())
108
    if (this->functions_.find(name) == this->functions_.end())
109
    {
109
    {
110
        this->SendResponse(plugin_name, request_id, "No such function found, " + name);
110
        this->SendResponse(plugin_name, request_id, "No such function found, " + name);
111
        return;
111
        return;
112
    }
112
    }
113
 
113
 
114
    //LOG.Debug("Call: " + name);
114
    //LOG.Debug("Call: " + name);
115
 
115
 
116
    Value result = this->functions_[name]->Call(this->context_->Global(), arguments->size(), arguments->data());
116
    Value result = this->functions_[name]->Call(this->context_->Global(), arguments->size(), arguments->data());
117
   
117
   
118
    if (result.IsEmpty())
118
    if (result.IsEmpty())
119
    {
119
    {
120
        this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
120
        this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
121
        return;
121
        return;
122
    }
122
    }
123
   
123
   
124
    v8::String::AsciiValue response(result->ToString());
124
    v8::String::AsciiValue response(result->ToString());
125
   
125
   
126
    this->SendResponse(plugin_name, request_id, std::string(*response));
126
    this->SendResponse(plugin_name, request_id, std::string(*response));
127
}
127
}
128
 
128
 
129
void Manager::StartHandler()
129
void Manager::StartHandler()
130
{
130
{
131
    v8::Handle<v8::ObjectTemplate> raw_template = v8::ObjectTemplate::New();
131
    v8::Handle<v8::ObjectTemplate> raw_template = v8::ObjectTemplate::New();
132
   
132
   
133
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
133
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
134
    {
134
    {
135
        ExportFunctionList export_functions = this->plugins_[c]->GetExportFunctions();
135
        ExportFunctionList export_functions = this->plugins_[c]->GetExportFunctions();
136
       
136
       
137
        for (ExportFunctionList::iterator it = export_functions.begin(); it != export_functions.end(); it++)
137
        for (ExportFunctionList::iterator it = export_functions.begin(); it != export_functions.end(); it++)
138
        {
138
        {
139
            raw_template->Set(v8::String::New(it->first.data()), v8::FunctionTemplate::New(it->second));
139
            raw_template->Set(v8::String::New(it->first.data()), v8::FunctionTemplate::New(it->second));
140
        }
140
        }
141
    }
141
    }
142
   
142
   
143
    this->context_ = v8::Context::New(NULL, raw_template);
143
    this->context_ = v8::Context::New(NULL, raw_template);
144
   
144
   
145
    v8::Context::Scope context_scope(this->context_);
145
    v8::Context::Scope context_scope(this->context_);
146
   
146
   
147
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
147
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
148
    {
148
    {
149
        this->LoadScript("plugin/" + this->plugins_[c]->GetName() + ".js");
149
        this->LoadScript("plugin/" + this->plugins_[c]->GetName() + ".js");
150
    }
150
    }
151
   
151
   
152
    this->object_template_ = v8::Persistent<v8::ObjectTemplate>::New(raw_template);
152
    this->object_template_ = v8::Persistent<v8::ObjectTemplate>::New(raw_template);
153
   
153
   
154
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
154
    for (unsigned int c = 0; c < this->plugins_.size(); c++)
155
    {
155
    {
156
        this->plugins_[c]->InitializeDone();
156
        this->plugins_[c]->InitializeDone();
157
    }
157
    }
158
   
158
   
159
    LOG.Info("VM initialized.");
159
    LOG.Info("VM initialized.");
160
}
160
}
161
 
161
 
162
bool Manager::ImportFunction(std::string functionname)
162
bool Manager::ImportFunction(std::string functionname)
163
{
163
{
164
    v8::Context::Scope context_scope(this->context_);
164
    v8::Context::Scope context_scope(this->context_);
165
   
165
   
166
    v8::TryCatch try_catch;
166
    v8::TryCatch try_catch;
167
   
167
   
168
    v8::Handle<v8::String> process_name = v8::String::New(functionname.data());
168
    v8::Handle<v8::String> process_name = v8::String::New(functionname.data());
169
    v8::Handle<v8::Value> process_val = this->context_->Global()->Get(process_name);
169
    v8::Handle<v8::Value> process_val = this->context_->Global()->Get(process_name);
170
   
170
   
171
    if (!process_val->IsFunction())
171
    if (!process_val->IsFunction())
172
    {
172
    {
173
        LOG.Error(functionname + " was not found to be a function!");
173
        LOG.Error(functionname + " was not found to be a function!");
174
        return false;
174
        return false;
175
    }
175
    }
176
   
176
   
177
    v8::Handle<v8::Function> process_fun = v8::Handle<v8::Function>::Cast(process_val);
177
    v8::Handle<v8::Function> process_fun = v8::Handle<v8::Function>::Cast(process_val);
178
   
178
   
179
    this->functions_[functionname] = v8::Persistent<v8::Function>::New(process_fun);
179
    this->functions_[functionname] = v8::Persistent<v8::Function>::New(process_fun);
180
   
180
   
181
    return true;
181
    return true;
182
}
182
}
183
 
183
 
184
bool Manager::LoadScript(std::string scriptname)
184
bool Manager::LoadScript(std::string scriptname)
185
{
185
{
-
 
186
    for (unsigned int n = 0; n < this->loaded_scripts_.size(); n++)
-
 
187
    {
-
 
188
        if (this->loaded_scripts_[n] == scriptname)
-
 
189
        {
-
 
190
            LOG.Info(scriptname + " is already loaded.");
-
 
191
            return true;
-
 
192
        }
-
 
193
    }
-
 
194
   
186
    std::string path = this->script_path_ + scriptname;
195
    std::string path = this->script_path_ + scriptname;
187
    std::ifstream file(path.data());
196
    std::ifstream file(path.data());
188
   
197
   
189
    if (!file.is_open())
198
    if (!file.is_open())
190
    {
199
    {
191
        file.close();
200
        file.close();
192
        LOG.Warning("Could not load " + scriptname + ".");
201
        LOG.Warning("Could not load " + scriptname + ".");
193
        return false;
202
        return false;
194
    }
203
    }
195
   
204
   
196
    std::string code = "";
205
    std::string code = "";
197
    std::string line;
206
    std::string line;
198
   
207
   
199
    while (!file.eof())
208
    while (!file.eof())
200
    {
209
    {
201
        std::getline(file, line);
210
        std::getline(file, line);
202
       
211
       
203
        if (file.eof())
212
        if (file.eof())
204
        {
213
        {
205
            break;
214
            break;
206
        }
215
        }
207
       
216
       
208
        code += line + "\n";
217
        code += line + "\n";
209
    }
218
    }
210
   
219
   
211
    code += "\n";
220
    code += "\n";
212
   
221
   
213
    file.close();
222
    file.close();
214
   
223
   
215
    v8::Context::Scope context_scope(this->context_);
224
    v8::Context::Scope context_scope(this->context_);
216
   
225
   
217
    v8::TryCatch try_catch;
226
    v8::TryCatch try_catch;
218
   
227
   
219
    v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()),
228
    v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()),
220
                                                        v8::String::New(scriptname.data(), scriptname.length()));
229
                                                        v8::String::New(scriptname.data(), scriptname.length()));
221
   
230
   
222
    if (script.IsEmpty())
231
    if (script.IsEmpty())
223
    {
232
    {
224
        LOG.Error(this->FormatException(try_catch));
233
        LOG.Error(this->FormatException(try_catch));
225
        return false;
234
        return false;
226
    }
235
    }
227
    else
236
    else
228
    {
237
    {
229
        Value result = script->Run();
238
        Value result = script->Run();
230
       
239
       
231
        if (result.IsEmpty())
240
        if (result.IsEmpty())
232
        {
241
        {
233
            LOG.Error(this->FormatException(try_catch));
242
            LOG.Error(this->FormatException(try_catch));
234
            return false;
243
            return false;
235
        }
244
        }
236
    }
245
    }
-
 
246
   
-
 
247
    this->loaded_scripts_.push_back(scriptname);
237
   
248
   
238
    LOG.Info("Loaded " + scriptname + " successfully.");
249
    LOG.Info("Loaded " + scriptname + " successfully.");
239
    return true;
250
    return true;
240
}
251
}
241
 
252
 
242
void Manager::ExecuteHandler(std::string plugin_name, unsigned int request_id, std::string code)
253
void Manager::ExecuteHandler(std::string plugin_name, unsigned int request_id, std::string code)
243
{
254
{
244
    std::string scriptname = "execute_" + plugin_name + ".js";
255
    std::string scriptname = "execute_" + plugin_name + ".js";
245
   
256
   
246
    v8::Context::Scope context_scope(this->context_);
257
    v8::Context::Scope context_scope(this->context_);
247
   
258
   
248
    v8::TryCatch try_catch;
259
    v8::TryCatch try_catch;
249
   
260
   
250
    v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()),
261
    v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()),
251
                                                        v8::String::New(scriptname.data(), scriptname.length()));
262
                                                        v8::String::New(scriptname.data(), scriptname.length()));
252
   
263
   
253
    if (script.IsEmpty())
264
    if (script.IsEmpty())
254
    {
265
    {
255
        this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
266
        this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
256
    }
267
    }
257
    else
268
    else
258
    {
269
    {
259
        Value result = script->Run();
270
        Value result = script->Run();
260
       
271
       
261
        if (result.IsEmpty())
272
        if (result.IsEmpty())
262
        {
273
        {
263
            this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
274
            this->SendResponse(plugin_name, request_id, this->FormatException(try_catch));
264
        }
275
        }
265
        else
276
        else
266
        {
277
        {
267
            v8::String::AsciiValue response(result->ToString());
278
            v8::String::AsciiValue response(result->ToString());
268
           
279
           
269
            this->SendResponse(plugin_name, request_id, std::string(*response));
280
            this->SendResponse(plugin_name, request_id, std::string(*response));
270
        }
281
        }
271
    }
282
    }
272
}
283
}
273
 
284
 
274
std::string Manager::FormatException(v8::TryCatch& try_catch)
285
std::string Manager::FormatException(v8::TryCatch& try_catch)
275
{
286
{
276
    v8::Context::Scope context_scope(this->context_);
287
    v8::Context::Scope context_scope(this->context_);
277
   
288
   
278
    std::string result;
289
    std::string result;
279
   
290
   
280
    v8::HandleScope handle_scope;
291
    v8::HandleScope handle_scope;
281
    v8::String::Utf8Value exception(try_catch.Exception());
292
    v8::String::Utf8Value exception(try_catch.Exception());
282
    v8::Handle<v8::Message> message = try_catch.Message();
293
    v8::Handle<v8::Message> message = try_catch.Message();
283
   
294
   
284
    if (message.IsEmpty())
295
    if (message.IsEmpty())
285
    {
296
    {
286
        result = *exception;
297
        result = *exception;
287
    }
298
    }
288
    else
299
    else
289
    {
300
    {
290
        v8::String::Utf8Value filename(message->GetScriptResourceName());
301
        v8::String::Utf8Value filename(message->GetScriptResourceName());
291
       
302
       
292
        result = std::string(*filename) + ":" + boost::lexical_cast<std::string>(message->GetLineNumber()) + ": " + std::string(*exception);
303
        result = std::string(*filename) + ":" + boost::lexical_cast<std::string>(message->GetLineNumber()) + ": " + std::string(*exception);
293
       
304
       
294
        /*v8::String::Utf8Value sourceline(message->GetSourceLine());
305
        /*v8::String::Utf8Value sourceline(message->GetSourceLine());
295
        string strSourceline = *sourceline;
306
        string strSourceline = *sourceline;
296
       
307
       
297
        result += strSourceline + "\n";
308
        result += strSourceline + "\n";
298
       
309
       
299
        string underline = rpad("", message->GetStartColumn(), ' ');
310
        string underline = rpad("", message->GetStartColumn(), ' ');
300
        underline = rpad(underline, message->GetEndColumn(), '^');
311
        underline = rpad(underline, message->GetEndColumn(), '^');
301
       
312
       
302
        result += underline + "\n";*/
313
        result += underline + "\n";*/
303
    }
314
    }
304
   
315
   
305
    return result;
316
    return result;
306
}
317
}
307
   
318
   
308
}; // namespace vm
319
}; // namespace vm
309
}; // namespace atom
320
}; // namespace atom
310
 
321