Add AddVararg step

Author: levno-710

src/prometheus/steps.lua

@@ -8,4 +8,5 @@ return {
 	BreakBeautify  = require("prometheus.steps.BreakBeautify");
 	EncryptStrings = require("prometheus.steps.EncryptStrings");
 	NumbersToExpressions = require("prometheus.steps.NumbersToExpressions");
+	AddVararg 	= require("prometheus.steps.AddVararg");
 }

src/prometheus/steps/AddVararg.lua

@@ -0,0 +1,33 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- AddVararg.lua
+--
+-- This Script provides a Simple Obfuscation Step that wraps the entire Script into a function
+
+local Step = require("prometheus.step");
+local Ast = require("prometheus.ast");
+local visitast = require("prometheus.visitast");
+local AstKind = Ast.AstKind;
+
+local AddVararg = Step:extend();
+AddVararg.Description = "This Step Adds Vararg to all Functions";
+AddVararg.Name = "Add Vararg";
+
+AddVararg.SettingsDescriptor = {
+}
+
+function AddVararg:init(settings)
+	
+end
+
+function AddVararg:apply(ast)
+	visitast(ast, nil, function(node)
+        if node.kind == AstKind.FunctionDeclaration or node.kind == AstKind.LocalFunctionDeclaration or node.kind == AstKind.FunctionLiteralExpression then
+            if #node.args < 1 or node.args[#node.args].kind ~= AstKind.VarargExpression then
+                node.args[#node.args + 1] = Ast.VarargExpression();
+            end
+        end
+    end)
+end
+
+return AddVararg;