From d564aba27e932f414c1f576f5872ecda46e04a16 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Bompard?= <aurelien@bompard.org>
Date: Fri, 18 May 2018 14:56:52 +0000
Subject: [PATCH] SAR: add the FMN script

---
 .../notifs-web01.phx2.fedoraproject.org       |  5 ++
 inventory/inventory                           |  1 +
 roles/notifs/frontend/files/fmn-sar.py        | 68 +++++++++++++++++++
 roles/notifs/frontend/tasks/main.yml          | 11 +++
 4 files changed, 85 insertions(+)
 create mode 100644 roles/notifs/frontend/files/fmn-sar.py

diff --git a/inventory/host_vars/notifs-web01.phx2.fedoraproject.org b/inventory/host_vars/notifs-web01.phx2.fedoraproject.org
index c10d223bfd..c936d688c4 100644
--- a/inventory/host_vars/notifs-web01.phx2.fedoraproject.org
+++ b/inventory/host_vars/notifs-web01.phx2.fedoraproject.org
@@ -12,3 +12,8 @@ volgroup: /dev/vg_guests
 vmhost: virthost02.phx2.fedoraproject.org
 
 datacenter: phx2
+
+# GDPR SAR variables
+sar_script: /usr/local/bin/fmn-sar.py
+sar_script_user: apache
+sar_output_file: fmn.json
diff --git a/inventory/inventory b/inventory/inventory
index 34419014bc..5104b4afc6 100644
--- a/inventory/inventory
+++ b/inventory/inventory
@@ -52,6 +52,7 @@ pagure01.fedoraproject.org
 value01.phx2.fedoraproject.org
 fas03.phx2.fedoraproject.org
 pkgs02.phx2.fedoraproject.org
+notifs-web01.phx2.fedoraproject.org
 
 [certgetter]
 certgetter01.phx2.fedoraproject.org
diff --git a/roles/notifs/frontend/files/fmn-sar.py b/roles/notifs/frontend/files/fmn-sar.py
new file mode 100644
index 0000000000..5886459789
--- /dev/null
+++ b/roles/notifs/frontend/files/fmn-sar.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+"""
+GDPR SAR script for FMN.
+
+Extract all preferences from a selected username and prints them in JSON to the
+standard output.
+"""
+
+from __future__ import absolute_import, unicode_literals, print_function
+
+import argparse
+import json
+import logging
+import os
+import sys
+
+import requests
+from six.moves.urllib.parse import urljoin
+
+
+ENV_USERNAME = "SAR_USERNAME"
+FMN_INSTANCE = "http://localhost/notifications/"
+FMN_CONTEXTS = ["email", "irc"]
+
+log = logging.getLogger()
+
+
+def get_prefs(username, context):
+    url = urljoin(
+        FMN_INSTANCE,
+        "api/{username}.id.fedoraproject.org/{context}/".format(
+            username=username, context=context
+        )
+    )
+    response = requests.get(url)
+    if response.status_code >= 300:
+        log.error("Could not get URL %s: %d %s",
+                  url, response.status_code, response.reason)
+        return {}
+    result = response.json()
+    return result
+
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--debug", action="store_true")
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args()
+    try:
+        username = os.environ[ENV_USERNAME]
+    except KeyError as e:
+        print("Missing environment variable. {}".format(e), file=sys.stderr)
+        sys.exit(1)
+    logging.basicConfig(
+        level=logging.DEBUG if args.debug else logging.WARNING,
+        stream=sys.stderr,
+    )
+    result = {}
+    for context in FMN_CONTEXTS:
+        result[context] = get_prefs(username, context)
+    print(json.dumps(result, indent=2))
+
+
+if __name__ == "__main__":
+    main()
diff --git a/roles/notifs/frontend/tasks/main.yml b/roles/notifs/frontend/tasks/main.yml
index c266fb5cf7..6f97ef9f96 100644
--- a/roles/notifs/frontend/tasks/main.yml
+++ b/roles/notifs/frontend/tasks/main.yml
@@ -129,3 +129,14 @@
   - notifs
   - notifs/frontend
   - selinux
+
+- name: Install SAR script
+  copy:
+    src: fmn-sar.py
+    dest: /usr/local/bin/fmn-sar.py
+    mode: 0755
+    owner: apache
+    group: apache
+  when: inventory_hostname.startswith('notifs-web01')
+  tags:
+  - notifs