From 78d9b221ccf076b6ca613ee573cb47b33a8cbf43 Mon Sep 17 00:00:00 2001
From: geoffrey <geoffrey.bucchino@protonmail.com>
Date: Sat, 22 Jun 2024 15:13:39 +0200
Subject: [PATCH] Add project name variable

---
 config.py | 2 ++
 main.py   | 8 ++++----
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/config.py b/config.py
index 77679ff..a94cfd5 100644
--- a/config.py
+++ b/config.py
@@ -1,6 +1,8 @@
 #!/usr/bin/venv python
 # -*- coding: utf-8 -*-
 
+PROJECT_NAME = "baoSOC"
+
 VT_ATTRIBUTES_MAPPING = {
     'asn': 'str',
     'as_owner': 'int', 
diff --git a/main.py b/main.py
index 93ce930..7e9e699 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 
 from argparse import ArgumentParser
-from config import VT_ATTRIBUTES_MAPPING
+from config import VT_ATTRIBUTES_MAPPING, PROJECT_NAME
 from vt import VT
 from dnschecker import DNSChecker as DNS
 import ipaddress
@@ -12,7 +12,7 @@ from os.path import exists
 
 
 def checkArguments():
-    parser = ArgumentParser(description="baoSOC")
+    parser = ArgumentParser(description=PROJECT_NAME)
     parser.add_argument('-c', '--config', help='Config file')
     parser.add_argument('--dns', help='Get domain name information', action="store_true")
     # For dns command
@@ -33,7 +33,7 @@ def checkArguments():
 
 def usage():
     print("------------------------------")
-    print("|           baoSOC           |")
+    print(f"|           {PROJECT_NAME}           |")
     print("------------------------------\n")
     print("A tool for SOC analyst\n")
     print("Usage: main.py [COMMAND]")
@@ -56,7 +56,7 @@ def usage():
     print("\t --sha512 FILE\t\tGet the SHA512 of the file")
 
 def mainMenu():
-    print("\n           baoSOC           ")
+    print(f"\n           {PROJECT_NAME}           ")
     print(" What would you like to do? ")
     print("\n OPTION 1: Sanitise URL For emails ")
     print(" OPTION 2: Decoders (PP, URL, SafeLinks) ")