1//
2//  Copyright (C) 2016 Google, Inc.
3//
4//  Licensed under the Apache License, Version 2.0 (the "License");
5//  you may not use this file except in compliance with the License.
6//  You may obtain a copy of the License at:
7//
8//  http://www.apache.org/licenses/LICENSE-2.0
9//
10//  Unless required by applicable law or agreed to in writing, software
11//  distributed under the License is distributed on an "AS IS" BASIS,
12//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13//  See the License for the specific language governing permissions and
14//  limitations under the License.
15//
16
17#include <rapidjson/document.h>
18#include <rapidjson/writer.h>
19#include <rapidjson/stringbuffer.h>
20#include <map>
21#include <string>
22#include <tuple>
23
24#include <base.h>
25#include <utils/command_receiver.h>
26#include <utils/common_utils.h>
27
28typedef std::map<std::string, MFP> function_map;
29function_map* _funcMap = NULL;
30
31void _clean_result(rapidjson::Document &doc) {
32  doc.RemoveMember(sl4n::kMethodStr);
33  doc.RemoveMember(sl4n::kParamsStr);
34}
35
36void initiate(rapidjson::Document &doc) {
37  doc.AddMember(sl4n::kStatusStr, sl4n::kSuccessStr, doc.GetAllocator());
38}
39
40CommandReceiver::CommandReceiver() {
41  if (_funcMap == NULL) {
42    _funcMap = new function_map();
43  }
44  _funcMap->insert(std::make_pair("initiate", &initiate));
45  _funcMap->insert(std::make_pair("continue", &initiate));
46}
47
48void CommandReceiver::Call(rapidjson::Document& doc) {
49  std::string cmd;
50  if (doc.HasMember(sl4n::kCmdStr)) {
51    cmd = doc[sl4n::kCmdStr].GetString();
52  } else if (doc.HasMember(sl4n::kMethodStr)) {
53    cmd = doc[sl4n::kMethodStr].GetString();
54  }
55
56  function_map::const_iterator iter = _funcMap->find(cmd);
57  if (iter != _funcMap->end()) {
58    iter->second(doc);
59  }
60  _clean_result(doc);
61}
62
63void CommandReceiver::RegisterCommand(std::string name, MFP command) {
64  if (_funcMap == NULL) {
65    _funcMap = new function_map();
66  }
67
68  _funcMap->insert(std::make_pair(name, command));
69}
70