From f2a2e4a90b41361fdf5b521fe3f727240eb1e183 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Fri, 11 Mar 2022 17:14:14 +0100
Subject: [PATCH] fix: Missing files

---
 pymispwarninglists/exceptions.py |  8 ++++++
 pymispwarninglists/tools.py      | 46 ++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 pymispwarninglists/exceptions.py
 create mode 100644 pymispwarninglists/tools.py

diff --git a/pymispwarninglists/exceptions.py b/pymispwarninglists/exceptions.py
new file mode 100644
index 0000000..0c5325d
--- /dev/null
+++ b/pymispwarninglists/exceptions.py
@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+class PyMISPWarningListsError(Exception):
+    def __init__(self, message: str):
+        super(PyMISPWarningListsError, self).__init__(message)
+        self.message = message
diff --git a/pymispwarninglists/tools.py b/pymispwarninglists/tools.py
new file mode 100644
index 0000000..1f8f2b1
--- /dev/null
+++ b/pymispwarninglists/tools.py
@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import os
+import shutil
+
+from io import BytesIO
+from pathlib import Path
+from zipfile import ZipFile
+
+from .exceptions import PyMISPWarningListsError
+
+try:
+    import requests
+    HAS_REQUESTS = True
+except ImportError:
+    HAS_REQUESTS = False
+
+
+def get_xdg_home_dir() -> Path:
+    if os.getenv('XDG_MISP_HOME'):
+        xdg_home_dir = Path(os.environ['XDG_MISP_HOME'])
+        if not xdg_home_dir.exists():
+            xdg_home_dir.mkdir(parents=True)
+        return xdg_home_dir / 'misp-warninglists'
+    if os.name != 'posix':
+        raise PyMISPWarningListsError('Cannot initialize XDG variable for OS {os.name}. Must be posix.')
+    shared_data_dir = Path(os.environ['XDG_DATA_HOME']) if os.environ.get('XDG_DATA_HOME') else Path.home() / '.local' / 'share'
+    xdg_home_dir = shared_data_dir / 'misp'
+    if not xdg_home_dir.exists():
+        xdg_home_dir.mkdir(parents=True)
+    os.putenv('XDG_MISP_HOME', str(xdg_home_dir))
+    return xdg_home_dir / 'misp-warninglists'
+
+
+def update_warninglists():
+    if not HAS_REQUESTS:
+        raise PyMISPWarningListsError('Cannot update local warning lists, please install pymispwarninglists this way: pip install -E fetch_lists pymispwarninglists ')
+    storage_dir = get_xdg_home_dir()
+    if storage_dir.exists():
+        shutil.rmtree(storage_dir)
+    r = requests.get('https://github.com/MISP/misp-warninglists/archive/refs/heads/main.zip')
+    r.raise_for_status()
+    with ZipFile(BytesIO(r.content)) as zipfile:
+        zipfile.extractall(storage_dir.parent)
+    os.rename(storage_dir.parent / 'misp-warninglists-main', storage_dir)