mirror of https://github.com/Icinga/icinga2.git
Add source code for i2eval to the contrib directory
This commit is contained in:
Gunnar Beutner
parent
21b11f9c9a
commit
5f821f8560
|
|
@ -0,0 +1,3 @@
|
|||
i2tcl.so: i2tcl.cpp i2tcl.hpp
|
||||
swig -c++ -o i2tcl_wrap.cpp i2tcl.hpp
|
||||
g++ -g -DI2_DEBUG -I/usr/include/tcl8.5 -shared -fpic -Iicinga2/lib -Iicinga2/build -Iicinga2/build/lib -L/opt/icinga2/lib/icinga2 -lbase -lconfig -Wl,-rpath=/opt/icinga2/lib/icinga2 -o i2tcl.so i2tcl_wrap.cpp i2tcl.cpp
|
||||
|
|
@ -0,0 +1 @@
|
|||
This is the source code for the 'i2eval' IRC bot that's on #icinga and #icinga-devel.
|
||||
|
|
@ -0,0 +1,172 @@
|
|||
#include "i2tcl.hpp"
|
||||
#include "config/configcompiler.hpp"
|
||||
#include "config/configcompilercontext.hpp"
|
||||
#include "base/function.hpp"
|
||||
#include "base/json.hpp"
|
||||
#include "base/application.hpp"
|
||||
#include <boost/algorithm/string.hpp>
|
||||
|
||||
using namespace icinga;
|
||||
|
||||
static bool l_Init_Called;
|
||||
static Tcl_Interp *l_Interp;
|
||||
static Tcl_Encoding l_Encoding;
|
||||
static std::map<String, String> l_Lines;
|
||||
static int l_NextLine = 1;
|
||||
|
||||
static Value i2_call_tcl(const String& command, const String& mtype, const std::vector<Value>& args)
|
||||
{
|
||||
Tcl_Obj **objv = new Tcl_Obj *[args.size() + 1];
|
||||
objv[0] = Tcl_NewStringObj(command.CStr(), -1);
|
||||
Tcl_IncrRefCount(objv[0]);
|
||||
|
||||
for (size_t i = 0; i < args.size(); i++) {
|
||||
Tcl_DString dsText;
|
||||
String arg = static_cast<String>(args[i]);
|
||||
Tcl_ExternalToUtfDString(l_Encoding, arg.CStr(), -1, &dsText);
|
||||
objv[i + 1] = Tcl_NewStringObj(Tcl_DStringValue(&dsText), Tcl_DStringLength(&dsText));
|
||||
Tcl_DStringFree(&dsText);
|
||||
Tcl_IncrRefCount(objv[i + 1]);
|
||||
}
|
||||
|
||||
int code = Tcl_EvalObjv(l_Interp, args.size() + 1, objv, TCL_EVAL_GLOBAL);
|
||||
|
||||
Tcl_Obj *result = Tcl_GetObjResult(l_Interp);
|
||||
|
||||
for (size_t i = 0; i < args.size() + 1; i++)
|
||||
Tcl_DecrRefCount(objv[i]);
|
||||
|
||||
delete [] objv;
|
||||
|
||||
if (code == TCL_ERROR)
|
||||
BOOST_THROW_EXCEPTION(std::runtime_error("An error occured in the TCL script"));
|
||||
|
||||
Value vresult;
|
||||
|
||||
if (mtype == "list") {
|
||||
Array::Ptr arr = new Array();
|
||||
|
||||
int len;
|
||||
if (Tcl_ListObjLength(l_Interp, result, &len) != TCL_OK)
|
||||
BOOST_THROW_EXCEPTION(std::invalid_argument("TCL proc returned something that is not a list"));
|
||||
|
||||
for (size_t i = 0; i < len; i++) {
|
||||
Tcl_Obj *obj;
|
||||
Tcl_ListObjIndex(l_Interp, result, i, &obj);
|
||||
|
||||
const char* strObj = Tcl_GetString(obj);
|
||||
|
||||
Tcl_DString dsObj;
|
||||
arr->Add(Tcl_UtfToExternalDString(l_Encoding, strObj, -1, &dsObj));
|
||||
Tcl_DStringFree(&dsObj);
|
||||
}
|
||||
|
||||
vresult = arr;
|
||||
} else if (mtype == "null") {
|
||||
/* Nothing to do here */
|
||||
} else if (mtype == "number") {
|
||||
const char* strResult = Tcl_GetString(result);
|
||||
vresult = Convert::ToDouble(strResult);
|
||||
} else if (mtype == "bool") {
|
||||
const char* strResult = Tcl_GetString(result);
|
||||
vresult = Convert::ToBool(Convert::ToLong(strResult));
|
||||
} else {
|
||||
const char* strResult = Tcl_GetString(result);
|
||||
|
||||
Tcl_DString dsResult;
|
||||
vresult = Tcl_UtfToExternalDString(l_Encoding, strResult, -1, &dsResult);
|
||||
Tcl_DStringFree(&dsResult);
|
||||
}
|
||||
|
||||
|
||||
return vresult;
|
||||
}
|
||||
|
||||
void i2_register_command(const char *icmd, const char *tcmd, const char *mtype, Tcl_Interp *interp)
|
||||
{
|
||||
Function::Ptr sf = new Function(boost::bind(i2_call_tcl, String(tcmd), String(mtype), _1));
|
||||
ScriptGlobal::Set(icmd, sf);
|
||||
}
|
||||
|
||||
void *i2_new_frame(Tcl_Interp *interp)
|
||||
{
|
||||
if (!l_Init_Called) {
|
||||
l_Init_Called = true;
|
||||
l_Encoding = Tcl_GetEncoding(l_Interp, "ISO8859-1");
|
||||
Application::InitializeBase();
|
||||
}
|
||||
|
||||
return new ScriptFrame();
|
||||
}
|
||||
|
||||
void i2_free_frame(void *frame, Tcl_Interp *interp)
|
||||
{
|
||||
delete reinterpret_cast<ScriptFrame *>(frame);
|
||||
}
|
||||
|
||||
char *i2_eval(void *uframe, const char *text, Tcl_Interp *interp)
|
||||
{
|
||||
std::ostringstream msgbuf;
|
||||
Expression *expr;
|
||||
ScriptFrame *frame = reinterpret_cast<ScriptFrame *>(uframe);
|
||||
|
||||
l_Interp = interp;
|
||||
|
||||
try {
|
||||
String lineNum = Convert::ToString(l_NextLine);
|
||||
l_NextLine++;
|
||||
|
||||
String fileName = "<" + lineNum + ">";
|
||||
l_Lines[fileName] = text;
|
||||
|
||||
expr = ConfigCompiler::CompileText(fileName, text);
|
||||
|
||||
if (expr) {
|
||||
Value result = expr->Evaluate(*frame);
|
||||
if (!result.IsObject() || result.IsObjectType<Array>() || result.IsObjectType<Dictionary>())
|
||||
msgbuf << JsonEncode(result);
|
||||
else
|
||||
msgbuf << result;
|
||||
}
|
||||
} catch (const ScriptError& ex) {
|
||||
DebugInfo di = ex.GetDebugInfo();
|
||||
|
||||
String text = l_Lines[di.Path];
|
||||
|
||||
std::vector<String> lines;
|
||||
boost::algorithm::split(lines, text, boost::is_any_of("\n"));
|
||||
|
||||
for (int i = di.FirstLine; i <= di.LastLine; i++) {
|
||||
int start, len;
|
||||
|
||||
if (i == di.FirstLine)
|
||||
start = di.FirstColumn;
|
||||
else
|
||||
start = 0;
|
||||
|
||||
if (i == di.LastLine)
|
||||
len = di.LastColumn - di.FirstColumn + 1;
|
||||
else
|
||||
len = lines[i].GetLength();
|
||||
|
||||
String pathInfo = di.Path;
|
||||
if (i != 1)
|
||||
pathInfo += "(" + Convert::ToString(i) + ")";
|
||||
pathInfo += ": ";
|
||||
|
||||
msgbuf << pathInfo << lines[i - 1] << "\n";
|
||||
msgbuf << String(pathInfo.GetLength(), ' ');
|
||||
msgbuf << String(start, ' ') << String(len, '^') << "\n";
|
||||
}
|
||||
|
||||
msgbuf << ex.what();
|
||||
} catch (const std::exception& ex) {
|
||||
msgbuf << "Error: " << DiagnosticInformation(ex);
|
||||
}
|
||||
|
||||
delete expr;
|
||||
|
||||
std::string str = msgbuf.str();
|
||||
return strdup(str.c_str());
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,23 @@
|
|||
#ifdef SWIG
|
||||
%module i2tcl
|
||||
%{
|
||||
#include "i2tcl.hpp"
|
||||
%}
|
||||
|
||||
%typemap(in,numinputs=0) Tcl_Interp *interp {
|
||||
$1 = interp;
|
||||
}
|
||||
|
||||
#endif /* SWIG */
|
||||
|
||||
#include <tcl.h>
|
||||
|
||||
#ifndef I2TCL_H
|
||||
#define I2TCL_H
|
||||
|
||||
void i2_register_command(const char *icmd, const char *tcmd, const char *mtype, Tcl_Interp *interp);
|
||||
void *i2_new_frame(Tcl_Interp *interp);
|
||||
void i2_free_frame(void *frame, Tcl_Interp *interp);
|
||||
char *i2_eval(void *frame, const char *text, Tcl_Interp *interp);
|
||||
|
||||
#endif /* I2TCL_H */
|
||||
|
|
@ -0,0 +1,110 @@
|
|||
package require http
|
||||
package require tls
|
||||
http::register https 443 [list ::tls::socket -tls1 1]
|
||||
|
||||
load /home/gunnar/i2tcl.so i2tcl
|
||||
|
||||
bind pub - > i2tcl
|
||||
bind pub - >u i2tcl_url
|
||||
bind pub - ^ i2tcl
|
||||
bind pub - ^u i2tcl_url
|
||||
|
||||
if {![info exists ::i2frame]} {
|
||||
set ::i2frame [i2_new_frame]
|
||||
}
|
||||
|
||||
set ::i2chan ""
|
||||
set ::i2nick ""
|
||||
|
||||
i2_register_command irc i2_irc null
|
||||
i2_register_command channels channels list
|
||||
i2_register_command chanlist internalchanlist list
|
||||
i2_register_command getnick getcurrentnick string
|
||||
i2_register_command onchan onchan bool
|
||||
i2_register_command topic topic string
|
||||
i2_register_command topicnick topicnick string
|
||||
i2_register_command topicstamp topicstamp number
|
||||
i2_register_command chanmodes getchanmode string
|
||||
i2_register_command isop isop bool
|
||||
i2_register_command isvoice isvoice bool
|
||||
i2_register_command ishop ishop bool
|
||||
i2_register_command chanhost getchanhost string
|
||||
i2_register_command chanbans chanbans list
|
||||
i2_register_command getnick i2_getnick string
|
||||
i2_register_command getchan i2_getchan string
|
||||
i2_register_command __commit i2_null null
|
||||
i2_register_command commit_objects i2_null null
|
||||
i2_register_command exit i2_null null
|
||||
|
||||
proc i2_null {} {
|
||||
}
|
||||
|
||||
proc i2_getnick {} {
|
||||
global i2nick
|
||||
return $i2nick
|
||||
}
|
||||
|
||||
proc i2_getchan {} {
|
||||
global i2chan
|
||||
return $i2chan
|
||||
}
|
||||
|
||||
proc i2_irc {message} {
|
||||
global i2chan
|
||||
|
||||
if {[string first "\n" $message] != -1 || [string first "\r" $message] != -1} {
|
||||
return
|
||||
}
|
||||
|
||||
putserv "PRIVMSG $i2chan :$message"
|
||||
}
|
||||
|
||||
proc i2tcl {nick host hand chan arg} {
|
||||
global i2frame i2chan i2nick
|
||||
|
||||
set i2chan $chan
|
||||
set i2nick $nick
|
||||
|
||||
set result [i2_eval $i2frame $arg]
|
||||
|
||||
if {$result == ""} { set result "<no error>" }
|
||||
foreach sline [split $result \n] {
|
||||
putserv "PRIVMSG $chan :( $arg ) = $sline"
|
||||
}
|
||||
}
|
||||
|
||||
proc i2tcl_url {nick host hand chan arg} {
|
||||
global i2frame i2chan i2nick
|
||||
|
||||
set i2chan $chan
|
||||
set i2nick $nick
|
||||
|
||||
if {[catch {set token [http::geturl $arg]} msg]} {
|
||||
putserv "PRIVMSG $chan :HTTP request failed: $msg"
|
||||
http::cleanup $token
|
||||
return
|
||||
}
|
||||
|
||||
if {[http::status $token] != "ok"} {
|
||||
putserv "PRIVMSG $chan :HTTP request failed: [http::error $token]"
|
||||
http::cleanup $token
|
||||
return
|
||||
}
|
||||
|
||||
set rpl [split [http::code $token] " "]
|
||||
|
||||
if {[lindex $rpl 1] != 200} {
|
||||
putserv "PRIVMSG $chan :HTTP request failed: [join [lrange $rpl 1 end]]"
|
||||
http::cleanup $token
|
||||
return
|
||||
}
|
||||
|
||||
set code [http::data $token]
|
||||
http::cleanup $token
|
||||
set result [i2_eval $i2frame $code]
|
||||
|
||||
if {$result == ""} { set result "<no error>" }
|
||||
foreach sline [split $result \n] {
|
||||
putserv "PRIVMSG $chan :( $arg ) = $sline"
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue