first sync

iris-web/source/app/blueprints/reports/__init__.py

@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+#
+# This file is part of IRIS Project
+# MIT Licence
+# Author : PAM
+# File creation date : 4/27/20
+
+# IMPORTS ------------------------------------------------
+
+# VARS ---------------------------------------------------
+
+# CONTENT ------------------------------------------------

iris-web/source/app/blueprints/reports/reports_route.py

@@ -0,0 +1,138 @@
+#!/usr/bin/env python3
+#
+#  IRIS Source Code
+#  Copyright (C) 2021 - Airbus CyberSecurity (SAS)
+#  ir@cyberactionlab.net
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU Lesser General Public
+#  License as published by the Free Software Foundation; either
+#  version 3 of the License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#  Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+# IMPORTS ------------------------------------------------
+
+# VARS ---------------------------------------------------
+
+import os
+# CONTENT ------------------------------------------------
+import tempfile
+
+from flask import Blueprint
+from flask import redirect
+from flask import request
+from flask import send_file
+from flask import url_for
+from flask_login import current_user
+
+from app.iris_engine.module_handler.module_handler import call_modules_hook
+from app.iris_engine.reporter.reporter import IrisMakeDocReport, IrisMakeMdReport
+from app.iris_engine.utils.tracker import track_activity
+from app.models import CaseTemplateReport
+from app.util import FileRemover, ac_api_requires
+from app.util import not_authenticated_redirection_url
+from app.util import response_error
+
+reports_blueprint = Blueprint('reports',
+                              __name__,
+                              template_folder='templates')
+
+file_remover = FileRemover()
+
+
+@reports_blueprint.route('/case/report/generate-activities/<int:report_id>', methods=['GET'])
+@ac_api_requires()
+def download_case_activity(report_id, caseid):
+
+    call_modules_hook('on_preload_activities_report_create', data=report_id, caseid=caseid)
+    if report_id:
+        report = CaseTemplateReport.query.filter(CaseTemplateReport.id == report_id).first()
+        if report:
+            tmp_dir = tempfile.mkdtemp()
+
+            safe_mode = False
+
+            if request.args.get('safe-mode') == 'true':
+                safe_mode = True
+
+            # Get file extension
+            _, report_format = os.path.splitext(report.internal_reference)
+            
+            # Depending on the template format, the generation process is different
+            if report_format == ".docx":
+                mreport = IrisMakeDocReport(tmp_dir, report_id, caseid, safe_mode)
+                fpath, logs = mreport.generate_doc_report(doc_type="Activities")
+
+            elif report_format == ".md" or report_format == ".html" :
+                mreport = IrisMakeMdReport(tmp_dir, report_id, caseid, safe_mode)
+                fpath, logs = mreport.generate_md_report(doc_type="Activities")
+
+            else:
+                return response_error("Report error", "Unknown report format.")
+
+            if fpath is None:
+                track_activity("failed to generate a report")
+                return response_error(msg="Failed to generate the report", data=logs)
+
+            call_modules_hook('on_postload_activities_report_create', data=report_id, caseid=caseid)
+            resp = send_file(fpath, as_attachment=True)
+            file_remover.cleanup_once_done(resp, tmp_dir)
+
+            track_activity("generated a report")
+
+            return resp
+
+    return response_error("Unknown report", status=404)
+
+
+@reports_blueprint.route("/case/report/generate-investigation/<int:report_id>", methods=['GET'])
+@ac_api_requires()
+def _gen_report(report_id, caseid):
+
+    safe_mode = False
+
+    call_modules_hook('on_preload_report_create', data=report_id, caseid=caseid)
+    if report_id:
+        report = CaseTemplateReport.query.filter(CaseTemplateReport.id == report_id).first()
+        if report:
+            tmp_dir = tempfile.mkdtemp()
+
+            if request.args.get('safe-mode') == 'true':
+                safe_mode = True
+
+            _, report_format = os.path.splitext(report.internal_reference)
+            
+            if report_format == ".md" or report_format == ".html":
+                mreport = IrisMakeMdReport(tmp_dir, report_id, caseid, safe_mode)
+                fpath, logs = mreport.generate_md_report(doc_type="Investigation")
+
+            elif report_format == ".docx":
+                mreport = IrisMakeDocReport(tmp_dir, report_id, caseid, safe_mode)
+                fpath, logs = mreport.generate_doc_report(doc_type="Investigation")
+
+            else:
+                return response_error("Report error", "Unknown report format.")
+
+            if fpath is None:
+                track_activity("failed to generate the report")
+                return response_error(msg="Failed to generate the report", data=logs)
+
+            call_modules_hook('on_postload_report_create', data=fpath, caseid=caseid)
+
+            resp = send_file(fpath, as_attachment=True)
+            file_remover.cleanup_once_done(resp, tmp_dir)
+
+            track_activity("generated a report")
+
+            return resp
+
+    return response_error("Unknown report", status=404)
+