FEAT: add a user authentication token for websocket connections

Author: Christian Kruse

app/models/cf_user.rb

@@ -48,6 +48,12 @@ def self.find_first_by_auth_conditions(conditions = {})
     end
   end
 
+  def after_database_authentication
+    if websocket_token.blank?
+      update_attributes(websocket_token: SecureRandom.uuid)
+    end
+  end
+
   def has_badge?(type)
     badges.each do |b|
       return true if b.badge_type == type

db/migrate/0056_add_websocket_token.rb

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+
+class AddWebsocketToken < ActiveRecord::Migration
+  def up
+    execute <<-SQL
+ALTER TABLE users ADD COLUMN websocket_token CHARACTER VARYING(250) UNIQUE;
+    SQL
+  end
+
+  def down
+    execute <<-SQL
+ALTER TABLE users DROP COLUMN websocket_token;
+    SQL
+  end
+end
+
+# eof

db/structure.sql

@@ -1324,7 +1324,8 @@ CREATE TABLE users (
     avatar_file_name character varying(255),
     avatar_content_type character varying(255),
     avatar_file_size integer,
-    avatar_updated_at timestamp without time zone
+    avatar_updated_at timestamp without time zone,
+    websocket_token character varying(250)
 );
 
 
@@ -1793,6 +1794,14 @@ ALTER TABLE ONLY users
     ADD CONSTRAINT users_pkey PRIMARY KEY (user_id);
 
 
+--
+-- Name: users_websocket_token_key; Type: CONSTRAINT; Schema: cforum; Owner: -; Tablespace: 
+--
+
+ALTER TABLE ONLY users
+    ADD CONSTRAINT users_websocket_token_key UNIQUE (websocket_token);
+
+
 --
 -- Name: votes_pkey; Type: CONSTRAINT; Schema: cforum; Owner: -; Tablespace: 
 --
@@ -2617,6 +2626,8 @@ INSERT INTO schema_migrations (version) VALUES ('54');
 
 INSERT INTO schema_migrations (version) VALUES ('55');
 
+INSERT INTO schema_migrations (version) VALUES ('56');
+
 INSERT INTO schema_migrations (version) VALUES ('6');
 
 INSERT INTO schema_migrations (version) VALUES ('7');