diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 11:32:39 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 11:32:39 +0000 |
commit | 56ae875861ab260b80a030f50c4aff9f9dc8fff0 (patch) | |
tree | 531412110fc901a5918c7f7442202804a83cada9 /lib/base/array-script.cpp | |
parent | Initial commit. (diff) | |
download | icinga2-56ae875861ab260b80a030f50c4aff9f9dc8fff0.tar.xz icinga2-56ae875861ab260b80a030f50c4aff9f9dc8fff0.zip |
Adding upstream version 2.14.2.upstream/2.14.2upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | lib/base/array-script.cpp | 260 |
1 files changed, 260 insertions, 0 deletions
diff --git a/lib/base/array-script.cpp b/lib/base/array-script.cpp new file mode 100644 index 0000000..a976683 --- /dev/null +++ b/lib/base/array-script.cpp @@ -0,0 +1,260 @@ +/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */ + +#include "base/array.hpp" +#include "base/function.hpp" +#include "base/functionwrapper.hpp" +#include "base/scriptframe.hpp" +#include "base/objectlock.hpp" +#include "base/exception.hpp" + +using namespace icinga; + +static double ArrayLen() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->GetLength(); +} + +static void ArraySet(int index, const Value& value) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + self->Set(index, value); +} + +static Value ArrayGet(int index) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->Get(index); +} + +static void ArrayAdd(const Value& value) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + self->Add(value); +} + +static void ArrayRemove(int index) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + self->Remove(index); +} + +static bool ArrayContains(const Value& value) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->Contains(value); +} + +static void ArrayClear() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + self->Clear(); +} + +static Array::Ptr ArraySort(const std::vector<Value>& args) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + + Array::Ptr arr = self->ShallowClone(); + + if (args.empty()) { + ObjectLock olock(arr); + std::sort(arr->Begin(), arr->End()); + } else { + Function::Ptr function = args[0]; + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Sort function must be side-effect free.")); + + ObjectLock olock(arr); + std::sort(arr->Begin(), arr->End(), [&args](const Value& a, const Value& b) -> bool { + Function::Ptr cmp = args[0]; + return cmp->Invoke({ a, b }); + }); + } + + return arr; +} + +static Array::Ptr ArrayShallowClone() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->ShallowClone(); +} + +static Value ArrayJoin(const Value& separator) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->Join(separator); +} + +static Array::Ptr ArrayReverse() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->Reverse(); +} + +static Array::Ptr ArrayMap(const Function::Ptr& function) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Map function must be side-effect free.")); + + ArrayData result; + + ObjectLock olock(self); + for (const Value& item : self) { + result.push_back(function->Invoke({ item })); + } + + return new Array(std::move(result)); +} + +static Value ArrayReduce(const Function::Ptr& function) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Reduce function must be side-effect free.")); + + if (self->GetLength() == 0) + return Empty; + + Value result = self->Get(0); + + ObjectLock olock(self); + for (size_t i = 1; i < self->GetLength(); i++) { + result = function->Invoke({ result, self->Get(i) }); + } + + return result; +} + +static Array::Ptr ArrayFilter(const Function::Ptr& function) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Filter function must be side-effect free.")); + + ArrayData result; + + ObjectLock olock(self); + for (const Value& item : self) { + if (function->Invoke({ item })) + result.push_back(item); + } + + return new Array(std::move(result)); +} + +static bool ArrayAny(const Function::Ptr& function) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Filter function must be side-effect free.")); + + ObjectLock olock(self); + for (const Value& item : self) { + if (function->Invoke({ item })) + return true; + } + + return false; +} + +static bool ArrayAll(const Function::Ptr& function) +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + REQUIRE_NOT_NULL(function); + + if (vframe->Sandboxed && !function->IsSideEffectFree()) + BOOST_THROW_EXCEPTION(ScriptError("Filter function must be side-effect free.")); + + ObjectLock olock(self); + for (const Value& item : self) { + if (!function->Invoke({ item })) + return false; + } + + return true; +} +static Array::Ptr ArrayUnique() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + REQUIRE_NOT_NULL(self); + return self->Unique(); +} + +static void ArrayFreeze() +{ + ScriptFrame *vframe = ScriptFrame::GetCurrentFrame(); + Array::Ptr self = static_cast<Array::Ptr>(vframe->Self); + self->Freeze(); +} + +Object::Ptr Array::GetPrototype() +{ + static Dictionary::Ptr prototype = new Dictionary({ + { "len", new Function("Array#len", ArrayLen, {}, true) }, + { "set", new Function("Array#set", ArraySet, { "index", "value" }) }, + { "get", new Function("Array#get", ArrayGet, { "index" }) }, + { "add", new Function("Array#add", ArrayAdd, { "value" }) }, + { "remove", new Function("Array#remove", ArrayRemove, { "index" }) }, + { "contains", new Function("Array#contains", ArrayContains, { "value" }, true) }, + { "clear", new Function("Array#clear", ArrayClear) }, + { "sort", new Function("Array#sort", ArraySort, { "less_cmp" }, true) }, + { "shallow_clone", new Function("Array#shallow_clone", ArrayShallowClone, {}, true) }, + { "join", new Function("Array#join", ArrayJoin, { "separator" }, true) }, + { "reverse", new Function("Array#reverse", ArrayReverse, {}, true) }, + { "map", new Function("Array#map", ArrayMap, { "func" }, true) }, + { "reduce", new Function("Array#reduce", ArrayReduce, { "reduce" }, true) }, + { "filter", new Function("Array#filter", ArrayFilter, { "func" }, true) }, + { "any", new Function("Array#any", ArrayAny, { "func" }, true) }, + { "all", new Function("Array#all", ArrayAll, { "func" }, true) }, + { "unique", new Function("Array#unique", ArrayUnique, {}, true) }, + { "freeze", new Function("Array#freeze", ArrayFreeze, {}) } + }); + + return prototype; +} |