Get the basic tutorial working completely, including reacting to server-side events

endpoints/index.py

@@ -5,9 +5,9 @@ import urlparse
 from flask import request, make_response, jsonify, session, Blueprint
 from functools import wraps
 
-from data import model
+from data import model, userevent
 from data.queue import webhook_queue
-from app import mixpanel
+from app import mixpanel, app
 from auth.auth import (process_auth, get_authenticated_user,
                        get_validated_token)
 from util.names import parse_repository_name
@@ -80,8 +80,16 @@ def create_user():
   if existing_user:
     verified = model.verify_user(username, password)
     if verified:
+      # Mark that the user was logged in.
+      event = app.config['USER_EVENTS'].get_event(username)
+      event.publish_event_data('docker-cli', {'action': 'login'})
+
       return make_response('Verified', 201)
     else:
+      # Mark that the login failed.
+      event = app.config['USER_EVENTS'].get_event(username)
+      event.publish_event_data('docker-cli', {'action': 'loginfailure'})
+
       abort(400, 'Invalid password.', issue='login-failure')
 
   else:
@@ -186,9 +194,21 @@ def create_repository(namespace, repository):
   }
 
   if get_authenticated_user():
-    mixpanel.track(get_authenticated_user().username, 'push_repo',
-                   extra_params)
-    metadata['username'] = get_authenticated_user().username
+    username = get_authenticated_user().username
+
+    mixpanel.track(username, 'push_repo', extra_params)
+    metadata['username'] = username
+
+    # Mark that the user has started pushing the repo.
+    user_data = {
+      'action': 'push_repo',
+      'repository': repository,
+      'namespace': namespace
+    }
+
+    event = app.config['USER_EVENTS'].get_event(username)
+    event.publish_event_data('docker-cli', user_data)
+
   else:
     mixpanel.track(get_validated_token().code, 'push_repo', extra_params)
     metadata['token'] = get_validated_token().friendly_name
@@ -222,6 +242,19 @@ def update_images(namespace, repository):
       updated_tags[image['Tag']] = image['id']
       model.set_image_checksum(image['id'], repo, image['checksum'])
 
+    if get_authenticated_user():
+      username = get_authenticated_user().username
+
+      # Mark that the user has pushed the repo.
+      user_data = {
+        'action': 'pushed_repo',
+        'repository': repository,
+        'namespace': namespace
+        }
+
+      event = app.config['USER_EVENTS'].get_event(username)
+      event.publish_event_data('docker-cli', user_data)
+
     # Generate a job for each webhook that has been added to this repo
     webhooks = model.list_webhooks(namespace, repository)
     for webhook in webhooks: