Merge pull request #102 from oxince/master

Add Watermark Feature

Author: levno-710

readme.md

@@ -55,5 +55,6 @@ lua ./cli.lua [options]
 - [pnlmon](https://github.com/pnlmon)
 - [britzl](https://github.com/britzl)
 - [SpinnySpiwal](https://github.com/SpinnySpiwal)
+- [oxince](https://github.com/oxince)
 ## License
 This Project is Licensed under the GNU General Public License v3.0. For more details, please refer to [LICENSE](https://github.com/levno-710/Prometheus/blob/master/LICENSE).

src/presets.lua

@@ -108,6 +108,12 @@ return {
 
                 }
             },
+            {
+                Name = "Watermark";
+                Settings = {
+                    Content = "This Script was obfuscated using the Medium Preset of Prometheus Obfuscator by Levno_710"
+                }
+            }
         }
     };
     ["Strong"] = {

src/prometheus/steps.lua

@@ -1,11 +1,12 @@
 return {
-	WrapInFunction = require("prometheus.steps.WrapInFunction");
-	SplitStrings   = require("prometheus.steps.SplitStrings");
-	Vmify          = require("prometheus.steps.Vmify");
-	ConstantArray  = require("prometheus.steps.ConstantArray");
-	ProxifyLocals  = require("prometheus.steps.ProxifyLocals");
-	AntiTamper  = require("prometheus.steps.AntiTamper");
-	EncryptStrings = require("prometheus.steps.EncryptStrings");
+	WrapInFunction       = require("prometheus.steps.WrapInFunction");
+	SplitStrings         = require("prometheus.steps.SplitStrings");
+	Vmify                = require("prometheus.steps.Vmify");
+	ConstantArray        = require("prometheus.steps.ConstantArray");
+	ProxifyLocals  			 = require("prometheus.steps.ProxifyLocals");
+	AntiTamper  				 = require("prometheus.steps.AntiTamper");
+	EncryptStrings 			 = require("prometheus.steps.EncryptStrings");
 	NumbersToExpressions = require("prometheus.steps.NumbersToExpressions");
-	AddVararg 	= require("prometheus.steps.AddVararg");
+	AddVararg 					 = require("prometheus.steps.AddVararg");
+	Watermark 	         = require("prometheus.steps.Watermark");
 }

src/prometheus/steps/Watermark.lua

@@ -0,0 +1,36 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- Watermark.lua
+--
+-- This Script provides a Step that will add a watermark to the script
+
+local Step = require("prometheus.step");
+local Ast = require("prometheus.ast");
+local Scope = require("prometheus.scope");
+
+local Watermark = Step:extend();
+Watermark.Description = "This Step will add a watermark to the script";
+Watermark.Name = "Watermark";
+
+Watermark.SettingsDescriptor = {
+  Content = {
+    name = "Content",
+    description = "The Content of the Watermark",
+    type = "string",
+    default = "This Script is Part of the Prometheus Obfuscator by Levno_710",
+  },
+}
+
+function Watermark:init(settings)
+	
+end
+
+function Watermark:apply(ast)
+  local body = ast.body;
+  if string.len(self.Content) > 0 then
+    local watermark = body.scope:addVariable();
+    table.insert(body.statements, 1, Ast.LocalVariableDeclaration(body.scope, {watermark}, {Ast.StringExpression(self.Content)}));
+  end
+end
+
+return Watermark;