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