From cf927e22e0c2c05b6dac58e6363fb09a347ed21b Mon Sep 17 00:00:00 2001
From: Dobin <dobin.rutishauser@gmail.com>
Date: Sat, 23 Mar 2024 18:36:22 +0000
Subject: [PATCH] fix: save comment

---
 app/templates/project.html | 3 ++-
 app/views.py               | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/templates/project.html b/app/templates/project.html
index 13a1dcb..32b9c38 100644
--- a/app/templates/project.html
+++ b/app/templates/project.html
@@ -14,7 +14,8 @@
 
         <div class="row">
             <div class="col-3">
-                <input type="text" name="comment" class="hidden form-control" placeholder="Comment" value="{{project.comment}}"
+                <input type="text" name="comment" class="hidden form-control" 
+                       placeholder="Comment" value="{{project.comment}}"
                 placeholder="" aria-label="PROJECTNAME" aria-describedby="basic-addon1">
             
                 <select class="form-select" name="shellcode" aria-label="SHELLCODE">
diff --git a/app/views.py b/app/views.py
index a254354..139f062 100644
--- a/app/views.py
+++ b/app/views.py
@@ -78,6 +78,7 @@ def add_project():
         settings = Settings()
 
         project_name = request.form['project_name']
+        comment = request.form['comment']
 
         settings.payload_path = "app/upload/shellcode/" + request.form['shellcode']
         if request.form['shellcode'] == "createfile.bin":
@@ -106,6 +107,7 @@ def add_project():
             # overwrite project
             project = storage.get_project(project_name)
             project.settings = settings
+            project.comment = comment
         else:
             # add new project
             project = Project(project_name, settings)
@@ -113,6 +115,7 @@ def add_project():
             project.project_exe = request.form['exe'].replace(".exe", ".infected.exe")
             project.settings = settings
             settings.project_name = project_name
+            project.comment = comment
             storage.add_project(project)
         storage.save_data()
         return redirect("/project/{}".format(project_name), code=302)