No controllers yet

README.md

@@ -1,5 +1,13 @@
 # angel_websocket
-WebSocket plugin for Angel. Features JWT support.
+WebSocket plugin for Angel.
+
+This plugin broadcasts events from hooked services via WebSockets. 
+
+In addition,
+it adds itself to the app's IoC container as `AngelWebSocket`, so that it can be used
+in controllers as well.
+
+WebSocket contexts are add to `req.params` as `'socket'`.
 
 
 # Usage
@@ -51,6 +59,8 @@ class Car extends srv.Model {
 
 main() async {
   Angel app = new WebSocketClient("/ws");
+  // Wait for WebSocket connection...
+  await app.connect();
   var Cars = app.service("api/cars", type: Car);
 
   Cars.onCreated.listen((e) {

lib/browser.dart

@@ -12,6 +12,7 @@ class WebSocketClient extends Angel {
 
   WebSocketClient(String wsEndpoint) : super(wsEndpoint) {
     _socket = new WebSocket(wsEndpoint);
+    _connect();
   }
 
   onData(data) {
@@ -60,7 +61,7 @@ class WebSocketClient extends Angel {
     }
   }
 
-  Future connect() async {
+  void _connect() {
     _socket.onMessage.listen((MessageEvent event) {
       onData(event.data);
     });

lib/server.dart

@@ -17,9 +17,10 @@ class Realtime {
   const Realtime();
 }
 
-class AngelWebSocket {
+class AngelWebSocket extends AngelPlugin {
   Angel _app;
   List<WebSocket> _clients = [];
+  List<WebSocket> get clients => new List.from(_clients, growable: false);
   List<String> servicesAlreadyWired = [];
   String endpoint;
 
@@ -153,8 +154,12 @@ class AngelWebSocket {
     }
   }
 
+  @override
   Future call(Angel app) async {
-    this._app = app;
+    this._app = app..container.singleton(this);
+
+    if (runtimeType != AngelWebSocket)
+      app.container.singleton(this, as: AngelWebSocket);
 
     // Set up services
     wireAllServices(app);
@@ -177,6 +182,8 @@ class AngelWebSocket {
       var socket = new WebSocketContext(ws, req, res);
       await onConnect(socket);
 
+      req.params['socket'] = socket;
+
       ws.listen((data) {
         onData(socket, data);
       }, onDone: () {

test/common.dart

@@ -1,7 +1,7 @@
 import 'dart:async';
 import 'dart:io';
 import 'package:angel_framework/angel_framework.dart';
-import 'package:angel_framework/defs.dart';
+import 'package:angel_framework/src/defs.dart';
 
 class Todo extends MemoryModel {
   String text;