From 203936fa84d69da6e8c2fea594b98a362c9ec770 Mon Sep 17 00:00:00 2001
From: Leo Hemsted <leo.hemsted@digital.cabinet-office.gov.uk>
Date: Tue, 20 Sep 2016 17:35:15 +0100
Subject: [PATCH] add GET/POST rest endpoints for whitelist

GET /<service_id>/whitelist
    returns all whitelisted contacts for a service, separated into two lists

POST /<service_id>/whitelist
    removes all existing whitelisted contacts, and replaces them with the
    provided new entries

(todo: dao work + tests)
---
 app/dao/service_whitelist_dao.py | 11 +++++++++++
 app/service/rest.py              | 29 +++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+)
 create mode 100644 app/dao/service_whitelist_dao.py

diff --git a/app/dao/service_whitelist_dao.py b/app/dao/service_whitelist_dao.py
new file mode 100644
index 000000000..359ad0bdc
--- /dev/null
+++ b/app/dao/service_whitelist_dao.py
@@ -0,0 +1,11 @@
+from sqlalchemy import or_
+
+from app import db
+from app.models import ServiceWhitelist
+
+def dao_fetch_service_whitelist(service_id):
+    return ServiceWhitelist.query().filter(ServiceWhitelist.service_id == service_id).all()
+
+
+def dao_add_whitelisted_contact(obj):
+    db.session.add(obj)
diff --git a/app/service/rest.py b/app/service/rest.py
index a6adebdd2..7dce1e19a 100644
--- a/app/service/rest.py
+++ b/app/service/rest.py
@@ -26,6 +26,10 @@ from app.dao.services_dao import (
     dao_fetch_weekly_historical_stats_for_service,
     dao_fetch_todays_stats_for_all_services
 )
+from app.dao.service_whitelist_dao import (
+    dao_fetch_service_whitelist,
+    dao_add_whitelisted_contact
+)
 from app.dao import notifications_dao
 from app.dao.provider_statistics_dao import get_fragment_count
 from app.dao.users_dao import get_model_users
@@ -45,6 +49,7 @@ from app.errors import (
 )
 from app.service import statistics
 
+
 service_blueprint = Blueprint('service', __name__)
 register_errors(service_blueprint)
 
@@ -266,3 +271,27 @@ def get_detailed_services():
             service.statistics = statistics.create_zeroed_stats_dicts()
 
     return detailed_service_schema.dump(services.values(), many=True).data
+
+
+@service_blueprint.route('/<uuid:service_id>/whitelist', methods=['GET'])
+def get_whitelist(service_id):
+    whitelist = dao_fetch_service_whitelist(service_id)
+
+    return {
+        'emails': [
+            {'id': item.id, 'email_address': item.email_address}
+            for item in whitelist if item.email_address is not None
+        ],
+        'mobile_numbers': [
+            {'id': item.id, 'mobile_number': item.mobile_number}
+            for item in whitelist if item.mobile_number is not None
+        ]
+    }
+
+
+@service_blueprint.route('/<uuid:service_id>/whitelist', methods=['POST'])
+def update_whitelist(service_id):
+    # todo: make this transactional
+    dao_remove_service_whitelist(service_id)
+    for contact in request.get_json():
+        dao_add_whitelisted_contact(ServiceWhitelist.from_string(contact))