Merge branch 'master' into licenses-unstable

awx/api/management/commands/uses_mongo.py

@@ -0,0 +1,45 @@
+# Copyright (c) 2015 Ansible, Inc.
+# All Rights Reserved
+
+import sys
+
+from django.core.management.base import BaseCommand
+
+from awx.main.task_engine import TaskSerializer
+
+
+class Command(BaseCommand):
+    """Return a exit status of 0 if MongoDB should be active, and an
+    exit status of 1 otherwise.
+
+    This script is intended to be used by bash and init scripts to
+    conditionally start MongoDB, so its focus is on being bash-friendly.
+    """
+    def handle(self, **kwargs):
+        # Get the license data.
+        license_reader = TaskSerializer()
+        license_data = license_reader.from_file()
+
+        # Does the license have features, at all?
+        # If there is no license yet, then all features are clearly off.
+        if 'features' not in license_data:
+            return False
+
+        # Does the license contain the system tracking feature?
+        # If and only if it does, MongoDB should run.
+        system_tracking = license_data['features']['system_tracking']
+
+        # Okay, do we need MongoDB to be turned on?
+        # This is a silly variable assignment right now, but I expect the
+        # rules here will grow more complicated over time.
+        # FIXME: Most likely this should be False if HA is active
+        #        (not just enabled by license, but actually in use).
+        uses_mongo = system_tracking  # noqa
+
+        # If we do not need Mongo, return a non-zero exit status.
+        print('MongoDB NOT required')
+        sys.exit(1)
+
+        # We do need Mongo, return zero.
+        print('MongoDB required')
+        sys.exit(0)

awx/api/views.py

@@ -8,6 +8,7 @@ import datetime
 import dateutil
 import time
 import socket
+import subprocess
 import sys
 
 # Django
@@ -228,10 +229,21 @@ class ApiV1ConfigView(APIView):
         except Exception:
             # FIX: Log
             return Response({"error": "Invalid License"}, status=status.HTTP_400_BAD_REQUEST)
+
+        # If the license is valid, write it to disk.
         if license_data['valid_key']:
             fh = open(TASK_FILE, "w")
             fh.write(data_actual)
             fh.close()
+
+            # Spawn a task to ensure that MongoDB is started (or stopped)
+            # as appropriate, based on whether the license uses it.
+            if license_data['features']['system_tracking']:
+                subprocess.call('sudo service mongod start', shell=True)
+            else:
+                subprocess.call('sudo service mongod stop', shell=True)
+
+            # Done; return the response.
             return Response(license_data)
         return Response({"error": "Invalid license"}, status=status.HTTP_400_BAD_REQUEST)