Added support for conf file and multiplataform paths

setup.py

@@ -1,16 +1,22 @@
 #!/usr/bin/env python3
 
-"""The setup script."""
-
+from appdirs import *
 from setuptools import setup, find_packages
+from colorama import init, Fore, Back, Style
+import os
+
+init(autoreset=True)
+
+# Identify OS config default path
+os_dirs = AppDirs("IntegrityGuard", "IntegrityGuard")
 
-with open('README.md') as readme_file:
+with open('README.rst') as readme_file:
     readme = readme_file.read()
 
-with open('HISTORY.md') as history_file:
+with open('HISTORY.rst') as history_file:
     history = history_file.read()
 
-requirements = ['Click>=7.0', ]
+requirements = [ 'appdirs>=1.4.4','watchdog>=1.0.2', 'colorama>=0.4.4' ]
 
 test_requirements = [ ]
 
@@ -19,7 +25,7 @@
     author_email='[email protected]',
     python_requires='>=3.6',
     classifiers=[
-        'Development Status :: 2 - Pre-Alpha',
+        'Development Status :: 3 - Alpha',
         'Intended Audience :: Developers',
         'License :: OSI Approved :: MIT License',
         'Natural Language :: English',
@@ -34,6 +40,9 @@
             'integrityguard=integrityguard.cli:main',
         ],
     },
+    data_files=[
+        (os_dirs.user_config_dir, ['src/helpers/integrityguard.conf'])
+    ],
     install_requires=requirements,
     license="MIT license",
     long_description=readme + '\n\n' + history,
@@ -47,3 +56,8 @@
     version='0.1.1',
     zip_safe=False,
 )
+
+# Print basic instructions for the user
+print(Fore.GREEN + "Success!! See important information below:")
+print(Fore.YELLOW + "Default config file path: " + os.path.join(os_dirs.user_config_dir, "integrityguard.conf") )
+print(Fore.YELLOW + "Default hashes store path: " + os.path.join(os_dirs.user_data_dir, "hashes.json") )

src/hashreport.py

@@ -1,24 +1,28 @@
 #!/usr/bin/env python3
 
 import os
-from helpers.loadconfig import load_config
-from data.hash import hash_file
 import json
 import pathlib
+from helpers.loadconfig import load_config
+from data.hash import hash_file
+from appdirs import *
 
 # Load configuration
 config = load_config()
 
 # Get root path to scan
-path = config['targetPath']
+path = config['monitor']['target_path']
 
 # Get hash type
-hash_type =  config['hash_type'].lower()
+hash_type = config['hash']['hash_type'].lower()
 
 # Check if the directory is empty
 if len( os.listdir(path) ) == 0:
     raise ValueError("The directory is empty.")
 
+# Identify OS config default path
+os_dirs = AppDirs("IntegrityGuard", "IntegrityGuard")
+
 # Initiate hashes variable
 hashes = []
 
@@ -30,13 +34,14 @@
         try:
             getHash = hash_file(file_fullpath, hash_type)
             hashes.append({ "path": os.path.abspath(file_fullpath), "hash": getHash })
-            print(file_fullpath + ";" + getHash)
+            print(file_fullpath + " > " + getHash)
         except ValueError as e:
             print("Something went wrong hashing the files. " + e)
 
 # Store hashes
-hash_file_path = pathlib.Path(__file__).parent.resolve()
-hash_file_path = str(hash_file_path) + "/data/hashes"
+hash_file_path = os.path.join(os_dirs.user_data_dir, "hashes.json")
 f = open(hash_file_path, "w")
 f.write(json.dumps(hashes))
-f.close()
+f.close()
+
+print("Hashes stored at " + hash_file_path)

src/helpers/integrityguard.conf

@@ -0,0 +1,24 @@
+# Hash algorithm type (MD5, SHA1, SHA224, SHA256, SHA384, and SHA512).
+[hash]
+hash_type = MD5
+
+# Directory monitoring settings
+# Use full path e.g. /var/www/myproject.com/
+[monitor]
+target_path = ./sample
+to_ignore =
+  ./*.log
+  ./ignoreTest/*
+log_path =
+report_type = JSON
+
+# Notification settings
+# Supported methods: API or EMAIL
+# Leave empty if a setting is not applicable for your notification option
+[notify]
+notify_method = API
+api_endpoint =
+smtp_host =
+smtp_user =
+smtp_pass =
+smtp_port =

src/helpers/loadconfig.py

@@ -1,11 +1,24 @@
-import yaml
+import configparser
+from appdirs import *
+import os
 
-def load_config(configPath=None):
+def load_config(config_path=None):
 
-    configPath = "./config.yml" if configPath is None else configPath
+    # Identify OS config default path
+    dirs = AppDirs("IntegrityGuard", "IntegrityGuard")
 
-    with open(configPath, "r") as stream:
-        try:
-            return yaml.safe_load(stream)
-        except yaml.YAMLError as exc:
-            raise ValueError('Something went wrong loading the config.yml file. ' + exc)
+    # Define default config file path
+    config_file = os.path.join(dirs.user_config_dir, "integrityguard.conf")
+
+    # Check if the user provided a config path
+    if config_path != None:
+        config_file = os.path.abspath(config_path)
+
+    # Check if the config file exist
+    if os.path.exists(config_file) == False:
+        raise ValueError("The configuration file *" + config_file + "* doesn't exist.")
+
+    config = configparser.ConfigParser()
+    config.read(config_file)
+
+    return config