'.interpolate({ emote: emote, description: emote });
842 | }
843 | return result;
844 | },
845 |
846 | insertEmotes: function(text) {
847 | var result = '';
848 | $A(text.split(/(:[^ ]*)/)).each(function(segment) {
849 | if (segment && segment.match(/^:/)) {
850 | segment = EmoteHelper.emoteToImage(segment);
851 | }
852 | result += segment;
853 | });
854 | return result;
855 | }
856 | };
857 | JsChat.SignOnController = Class.create({
858 | initialize: function() {
859 | this.retries = 0;
860 | setTimeout(function() { $('name').activate(); }, 500);
861 | $('sign-on').observe('submit', this.submitEvent.bindAsEventListener(this));
862 | },
863 |
864 | submitEvent: function(e) {
865 | this.signOn();
866 | Event.stop(e);
867 | return false;
868 | },
869 |
870 | showError: function(message) {
871 | $('feedback').innerHTML = '#{message}
'.interpolate({ message: message });
872 | $('feedback').show();
873 | $('sign-on-submit').enable();
874 | },
875 |
876 | signOn: function() {
877 | $('loading').show();
878 | $('sign-on-submit').disable();
879 | this.retries += 1;
880 |
881 | new Ajax.Request('/identify', {
882 | parameters: $('sign-on').serialize(true),
883 | onSuccess: function(transport) {
884 | try {
885 | var json = transport.responseText.evalJSON(true);
886 | if (json['action'] == 'reload' && this.retries < 4) {
887 | setTimeout(function() { this.signOn() }.bind(this), 50);
888 | } else if (json['action'] == 'redirect') {
889 | if (window.location.toString().match(new RegExp(json['to'] + '$'))) {
890 | window.location.reload();
891 | } else {
892 | window.location = json['to'];
893 | }
894 | } else if (json['error']) {
895 | this.showError(json['error']['message']);
896 | $('loading').hide();
897 | } else {
898 | this.showError('Connection error');
899 | }
900 | } catch (exception) {
901 | this.showError('Connection error: #{error}'.interpolate({ error: exception }));
902 | }
903 | }.bind(this),
904 | onFailure: function() {
905 | this.showError('Connection error');
906 | }.bind(this),
907 | onComplete: function() {
908 | $('loading').hide();
909 | }
910 | });
911 | }
912 | });
913 | JsChat.ChatController = Class.create({
914 | initialize: function() {
915 | $('loading').show();
916 |
917 | this.resizeEvent();
918 | setTimeout(this.initDisplay.bind(this), 50);
919 | this.tabCompletion = new TabCompletion('message');
920 |
921 | Event.observe(window, 'focus', this.focusEvent.bindAsEventListener(this));
922 | Event.observe(window, 'blur', this.blurEvent.bindAsEventListener(this));
923 | Event.observe(window, 'resize', this.resizeEvent.bindAsEventListener(this));
924 |
925 | $('post_message').observe('submit', this.postMessageFormEvent.bindAsEventListener(this));
926 | $('messages').observe('scroll', this.messagesScrolled.bindAsEventListener(this));
927 | $$('#rooms li.join a').first().observe('click', this.joinRoomClicked.bindAsEventListener(this));
928 | Event.observe(document, 'click', this.roomTabClick.bindAsEventListener(this));
929 | this.allRecentMessages();
930 | },
931 |
932 | allRecentMessages: function() {
933 | new Ajax.Request('/room_update_times', {
934 | method: 'get',
935 | onComplete: function(request) {
936 | var times = request.responseText.evalJSON();
937 | $H(this.lastUpdateTimes).each(function(data) {
938 | var room = data[0],
939 | time = data[1];
940 | if (Date.parse(time) < Date.parse(times[room])) {
941 | this.roomTabAlert(room);
942 | }
943 | }.bind(this));
944 | this.lastUpdateTimes = times;
945 | }.bind(this)
946 | });
947 | },
948 |
949 | roomTabAlert: function(room) {
950 | if (room === PageHelper.currentRoom()) return;
951 |
952 | $$('ul#rooms li a').each(function(roomLink) {
953 | if (roomLink.innerHTML === room) {
954 | roomLink.addClassName('new');
955 | }
956 | });
957 | },
958 |
959 | clearRoomTabAlert: function(room) {
960 | $$('ul#rooms li a').each(function(roomLink) {
961 | if (roomLink.innerHTML === room) {
962 | roomLink.removeClassName('new');
963 | }
964 | });
965 | },
966 |
967 | joinRoomClicked: function(e) {
968 | this.addRoomPrompt(e);
969 | Event.stop(e);
970 | return false;
971 | },
972 |
973 | roomTabClick: function(e) {
974 | var element = Event.element(e);
975 |
976 | if (element.tagName == 'A' && element.up('#rooms') && !element.up('li').hasClassName('join')) {
977 | this.switchRoom(element.innerHTML);
978 | Event.stop(e);
979 | return false;
980 | }
981 | },
982 |
983 | messagesScrolled: function() {
984 | Display.scrolled = (($('messages').scrollHeight - $('messages').scrollTop) > $('messages').getHeight());
985 | },
986 |
987 | focusEvent: function() {
988 | Display.unread = 0;
989 | Display.show_unread = false;
990 | document.title = PageHelper.title();
991 | },
992 |
993 | blurEvent: function() {
994 | Display.show_unread = true;
995 | },
996 |
997 | resizeEvent: function() {
998 | var messageInset = PageHelper.isDevice('iphone') ? 390 : 290,
999 | heightInset = PageHelper.isDevice('iphone') ? 200 : 100,
1000 | windowSize = document.viewport.getDimensions();
1001 |
1002 | if (PageHelper.isDevice('ipad')) {
1003 | messageInset = 330;
1004 | heightInset = 130;
1005 | }
1006 |
1007 | $('messages').setStyle({ width: windowSize.width - 220 + 'px' });
1008 | $('messages').setStyle({ height: windowSize.height - heightInset + 'px' });
1009 | $('message').setStyle({ width: windowSize.width - messageInset + 'px' });
1010 | $('names').setStyle({ height: windowSize.height - 200 + 'px' });
1011 | Display.scrollMessagesToTop();
1012 | },
1013 |
1014 | postMessageFormEvent: function(e) {
1015 | try {
1016 | var element = Event.element(e);
1017 | var message = $('message').value;
1018 | $('message').value = '';
1019 |
1020 | this.tabCompletion.history.add(message);
1021 |
1022 | if (message.length > 0) {
1023 | var command_posted = $H(UserCommands).find(function(command) {
1024 | var name = command[0];
1025 | var matches = message.match(new RegExp('^' + name + '$'));
1026 | if (matches) {
1027 | command[1].bind(this)(matches);
1028 | return true;
1029 | }
1030 | }.bind(this));
1031 |
1032 | if (!command_posted) {
1033 | if (message.match(/^\/\s?\//)) {
1034 | this.postMessage(message.replace(/\//, '').strip());
1035 | } else if (message.match(/^\//)) {
1036 | Display.add_message('Error: Command not found. Use /help display commands.', 'error');
1037 | } else {
1038 | this.postMessage(message);
1039 | }
1040 | }
1041 | }
1042 | } catch (exception) {
1043 | console.log(exception);
1044 | }
1045 |
1046 | Event.stop(e);
1047 | return false;
1048 | },
1049 |
1050 | postMessage: function(message) {
1051 | Display.message({ 'message': message.escapeHTML(), 'user': JsChat.user.name }, new Date());
1052 | new Ajax.Request('/message', {
1053 | method: 'post',
1054 | parameters: { 'message': message, 'to': PageHelper.currentRoom() }
1055 | });
1056 | },
1057 |
1058 | sendTweet: function(message) {
1059 | new Ajax.Request('/tweet', {
1060 | method: 'post',
1061 | parameters: { 'tweet': message }
1062 | });
1063 | },
1064 |
1065 | initDisplay: function() {
1066 | Display.unread = 0;
1067 | Display.show_unread = false;
1068 | Display.ignore_notices = false;
1069 |
1070 | PageHelper.setCurrentRoomName(window.location.hash);
1071 | $('message').activate();
1072 | $$('.header .navigation li').invoke('hide');
1073 | $('quit-nav').show();
1074 | $('help-nav').show();
1075 |
1076 | $('help-link').observe('click', function(e) {
1077 | UserCommands['/help']();
1078 | $('message').activate();
1079 | Event.stop(e);
1080 | return false;
1081 | });
1082 |
1083 | this.createPollers();
1084 | this.getRoomList(this.addRoomAndCheckSelected);
1085 | this.joinRoom(PageHelper.currentRoom());
1086 | },
1087 |
1088 | getRoomList: function(callback) {
1089 | new Ajax.Request('/rooms', {
1090 | method: 'get',
1091 | parameters: { time: new Date().getTime() },
1092 | onComplete: function(response) {
1093 | response.responseText.evalJSON().sort().each(function(roomName) {
1094 | try {
1095 | callback.apply(this, [roomName]);
1096 | } catch (exception) {
1097 | console.log(exception);
1098 | }
1099 | }.bind(this));
1100 | }.bind(this)
1101 | });
1102 | },
1103 |
1104 | joinRoom: function(roomName) {
1105 | new Ajax.Request('/join', {
1106 | method: 'post',
1107 | parameters: { time: new Date().getTime(), room: roomName },
1108 | onFailure: function() {
1109 | Display.add_message("Error: Couldn't join channel", 'server');
1110 | $('loading').hide();
1111 | },
1112 | onComplete: function() {
1113 | // Make the server update the last polled time
1114 | JsChat.Request.get('/messages', function() {});
1115 | document.title = PageHelper.title();
1116 | UserCommands['/lastlog'].apply(this);
1117 | $('loading').hide();
1118 | $('rooms').show();
1119 | this.addRoomToNav(roomName, true);
1120 | }.bind(this)
1121 | });
1122 | },
1123 |
1124 | isValidRoom: function(roomName) {
1125 | if (PageHelper.allRoomNames().include(roomName)) {
1126 | return false;
1127 | }
1128 | return true;
1129 | },
1130 |
1131 | validateAndJoinRoom: function(roomName) {
1132 | if (roomName === null || roomName.length == 0) {
1133 | return;
1134 | }
1135 |
1136 | if (!roomName.match(/^#/)) {
1137 | roomName = '#' + roomName;
1138 | }
1139 |
1140 | if (this.isValidRoom(roomName)) {
1141 | this.joinRoomInTab(roomName);
1142 | }
1143 | },
1144 |
1145 | addRoomPrompt: function() {
1146 | var roomName = prompt('Enter a room name:');
1147 | this.validateAndJoinRoom(roomName);
1148 | },
1149 |
1150 | addRoomToNav: function(roomName, selected) {
1151 | if (PageHelper.allRoomNames().include(roomName)) return;
1152 |
1153 | var classAttribute = selected ? ' class="selected"' : '';
1154 | $('rooms').insert({ bottom: '#{message}
'.interpolate({ message: message });
16 | $('feedback').show();
17 | $('sign-on-submit').enable();
18 | },
19 |
20 | signOn: function() {
21 | $('loading').show();
22 | $('sign-on-submit').disable();
23 | this.retries += 1;
24 |
25 | new Ajax.Request('/identify', {
26 | parameters: $('sign-on').serialize(true),
27 | onSuccess: function(transport) {
28 | try {
29 | var json = transport.responseText.evalJSON(true);
30 | if (json['action'] == 'reload' && this.retries < 4) {
31 | setTimeout(function() { this.signOn() }.bind(this), 50);
32 | } else if (json['action'] == 'redirect') {
33 | if (window.location.toString().match(new RegExp(json['to'] + '$'))) {
34 | window.location.reload();
35 | } else {
36 | window.location = json['to'];
37 | }
38 | } else if (json['error']) {
39 | this.showError(json['error']['message']);
40 | $('loading').hide();
41 | } else {
42 | this.showError('Connection error');
43 | }
44 | } catch (exception) {
45 | this.showError('Connection error: #{error}'.interpolate({ error: exception }));
46 | }
47 | }.bind(this),
48 | onFailure: function() {
49 | this.showError('Connection error');
50 | }.bind(this),
51 | onComplete: function() {
52 | $('loading').hide();
53 | }
54 | });
55 | }
56 | });
57 |
--------------------------------------------------------------------------------
/lib/jschat/http/public/javascripts/app/helpers/emote_helper.js:
--------------------------------------------------------------------------------
1 | var EmoteHelper = {
2 | legalEmotes: ['angry', 'arr', 'blink', 'blush', 'brucelee', 'btw', 'chuckle', 'clap', 'cool', 'drool', 'drunk', 'dry', 'eek', 'flex', 'happy', 'holmes', 'huh', 'laugh', 'lol', 'mad', 'mellow', 'noclue', 'oh', 'ohmy', 'panic', 'ph34r', 'pimp', 'punch', 'realmad', 'rock', 'rofl', 'rolleyes', 'sad', 'scratch', 'shifty', 'shock', 'shrug', 'sleep', 'sleeping', 'smile', 'suicide', 'sweat', 'thumbs', 'tongue', 'unsure', 'w00t', 'wacko', 'whistling', 'wink', 'worship', 'yucky'],
3 |
4 | emoteToImage: function(emote) {
5 | var result = emote;
6 | emote = emote.replace(/^:/, '').toLowerCase();
7 | if (EmoteHelper.legalEmotes.find(function(v) { return v == emote })) {
8 | result = ''.interpolate({ emote: emote, description: emote });
9 | }
10 | return result;
11 | },
12 |
13 | insertEmotes: function(text) {
14 | var result = '';
15 | $A(text.split(/(:[^ ]*)/)).each(function(segment) {
16 | if (segment && segment.match(/^:/)) {
17 | segment = EmoteHelper.emoteToImage(segment);
18 | }
19 | result += segment;
20 | });
21 | return result;
22 | }
23 | };
24 |
--------------------------------------------------------------------------------
/lib/jschat/http/public/javascripts/app/helpers/form_helpers.js:
--------------------------------------------------------------------------------
1 | var FormHelpers = {
2 | getCaretPosition: function(element) {
3 | if (element.setSelectionRange) {
4 | return element.selectionStart;
5 | } else if (element.createTextRange) {
6 | try {
7 | // The current selection
8 | var range = document.selection.createRange();
9 | // We'll use this as a 'dummy'
10 | var stored_range = range.duplicate();
11 | // Select all text
12 | stored_range.moveToElementText(element);
13 | // Now move 'dummy' end point to end point of original range
14 | stored_range.setEndPoint('EndToEnd', range);
15 |
16 | return stored_range.text.length - range.text.length;
17 | } catch (exception) {
18 | // IE is being mental. TODO: Figure out what IE's issue is
19 | return 0;
20 | }
21 | }
22 | },
23 |
24 | setCaretPosition: function(element, pos) {
25 | if (element.setSelectionRange) {
26 | element.focus()
27 | element.setSelectionRange(pos, pos)
28 | } else if (element.createTextRange) {
29 | var range = element.createTextRange()
30 |
31 | range.collapse(true)
32 | range.moveEnd('character', pos)
33 | range.moveStart('character', pos)
34 | range.select()
35 | }
36 | }
37 | };
38 |
--------------------------------------------------------------------------------
/lib/jschat/http/public/javascripts/app/helpers/link_helper.js:
--------------------------------------------------------------------------------
1 | var LinkHelper = {
2 | url: function(url) {
3 | return url.match(/(https?:\/\/[^\s]*)/gi);
4 | },
5 |
6 | link: function(url) {
7 | return '\#{link_name}'.interpolate({ url: url, link_name: url});
8 | },
9 |
10 | image_url: function(url) {
11 | return url.match(/\.(jpe?g|png|gif)/i);
12 | },
13 |
14 | image: function(url) {
15 | return '
2 |
7 |
8 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/index.erb:
--------------------------------------------------------------------------------
1 |
2 |
22 |
25 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/ipad.erb:
--------------------------------------------------------------------------------
1 |
3 |
4 |
5 | About
3 |JsChat is an open source chat system that uses a simple protocol based on JSON.
4 |To download the code and an irssi-like console client, visit GitHub.
5 |Read more on the JsChat Blog.
6 |Try JsChat Now
7 | 8 | <%= erb :form, :layout => false %> 9 | <% if JsChat::Auth::Twitter.loaded? %> 10 | <%= erb JsChat::Auth::Twitter.template, :layout => false %> 11 | <% end %> 12 |Features
13 |-
14 |
- Simple protocol that makes it easy to implement clients and bots 15 |
- Console client designed to look and feel like IRC clients 16 |
- Web client auto links and displays images/YouTube videos inline 17 |
- Protocol designed to be close to executable code, so creating clients and bots is easy 18 |
- Optional mongodb logging 19 |
- Optional Twitter authentication 20 |
18 |
19 |
22 |
28 |
29 |
30 | 
31 | <%= yield %>
32 |
33 |
34 |
35 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/iphone.erb:
--------------------------------------------------------------------------------
1 |
3 |
4 |
5 |
19 |
20 |
23 |
29 |
30 |
31 |
32 | <%= yield %>
33 |
34 |
35 |
36 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/iphone_message_form.erb:
--------------------------------------------------------------------------------
1 | -
2 |
4 |
9 | Loading...
5 | 6 |-
7 |
10 |
14 |
15 |
16 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/layout.erb:
--------------------------------------------------------------------------------
1 |
3 |
4 |
5 |
15 |
16 |
19 |
25 |
26 |
27 |
28 | <%= yield %>
29 |
30 |
31 |
32 |
33 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/message_form.erb:
--------------------------------------------------------------------------------
1 | -
2 |
4 |
8 | Loading...
5 |-
6 |
9 |
13 |
14 |
15 |
--------------------------------------------------------------------------------
/lib/jschat/http/views/twitter.erb:
--------------------------------------------------------------------------------
1 | Login with Twitter
2 | 3 |JsChat will save your rooms and keep your presence online until you click Quit.
4 |This will persist even if you login on another computer.
5 | 6 | 7 | -------------------------------------------------------------------------------- /lib/jschat/init.rb: -------------------------------------------------------------------------------- 1 | module JsChat ; end 2 | 3 | require 'jschat/server_options' 4 | require 'jschat/storage/init' 5 | 6 | module JsChat 7 | STATELESS_TIMEOUT = 60 8 | LASTLOG_DEFAULT = 100 9 | 10 | def self.init_storage 11 | if JsChat::Storage::MongoDriver.available? 12 | JsChat::Storage.enabled = true 13 | JsChat::Storage.driver = JsChat::Storage::MongoDriver 14 | else 15 | JsChat::Storage.enabled = false 16 | JsChat::Storage.driver = JsChat::Storage::NullDriver 17 | end 18 | end 19 | end 20 | -------------------------------------------------------------------------------- /lib/jschat/server.rb: -------------------------------------------------------------------------------- 1 | require 'rubygems' 2 | require 'eventmachine' 3 | gem 'json', '>= 1.1.9' 4 | require 'json' 5 | require 'time' 6 | require 'socket' 7 | 8 | # JsChat libraries 9 | require 'jschat/init' 10 | require 'jschat/errors' 11 | require 'jschat/flood_protection' 12 | 13 | module JsChat 14 | module Server 15 | def self.pid_file_name 16 | File.join(ServerConfig['tmp_files'], 'jschat.pid') 17 | end 18 | 19 | def self.write_pid_file 20 | return unless ServerConfig['use_tmp_files'] 21 | File.open(pid_file_name, 'w') { |f| f << Process.pid } 22 | end 23 | 24 | def self.rm_pid_file 25 | FileUtils.rm pid_file_name 26 | end 27 | 28 | def self.stop 29 | rm_pid_file 30 | end 31 | 32 | def self.run! 33 | write_pid_file 34 | JsChat.init_storage 35 | 36 | at_exit do 37 | stop 38 | end 39 | 40 | EM.run do 41 | EM.start_server ServerConfig['ip'], ServerConfig['port'], JsChat 42 | end 43 | end 44 | end 45 | 46 | class User 47 | include JsChat::FloodProtection 48 | 49 | attr_accessor :name, :connection, :rooms, :last_activity, 50 | :identified, :ip, :last_poll, :session_length 51 | 52 | def initialize(connection) 53 | @name = nil 54 | @connection = connection 55 | @rooms = [] 56 | @last_activity = Time.now.utc 57 | @last_poll = Time.now.utc 58 | @identified = false 59 | @ip = '' 60 | @expires = nil 61 | @session_length = nil 62 | end 63 | 64 | def session_expired? 65 | return true if @expires.nil? 66 | Time.now.utc >= @expires 67 | end 68 | 69 | def update_session_expiration 70 | return if @session_length.nil? 71 | @expires = Time.now.utc + @session_length 72 | end 73 | 74 | def to_json(*a) 75 | { 'name' => @name, 'last_activity' => @last_activity }.to_json(*a) 76 | end 77 | 78 | def name=(name) 79 | if @connection and @connection.name_taken? name 80 | raise JsChat::Errors::InvalidName.new(:name_taken, 'Name taken') 81 | elsif User.valid_name?(name) 82 | @identified = true 83 | @name = name 84 | else 85 | raise JsChat::Errors::InvalidName.new(:invalid_name, 'Invalid name') 86 | end 87 | end 88 | 89 | def self.valid_name?(name) 90 | not name.match /[^[:alnum:]._\-\[\]^C]/ and name.size > 0 91 | end 92 | 93 | def private_message(message) 94 | response = { 'display' => 'message', 'message' => message } 95 | @connection.send_response response 96 | end 97 | 98 | def change(params) 99 | # Valid options for change 100 | ['name'].each do |field| 101 | if params[field] 102 | old_value = send(field) 103 | send "#{field}=", params[field] 104 | @rooms.each do |room| 105 | response = { 'change' => 'user', 106 | 'room' => room.name, 107 | 'user' => { field => { old_value => params[field] } } } 108 | room.change_notice self, response 109 | return [field, params[field]] 110 | end 111 | end 112 | end 113 | end 114 | end 115 | 116 | class Room 117 | attr_accessor :name, :users 118 | 119 | def initialize(name) 120 | @name = name 121 | @users = [] 122 | end 123 | 124 | def self.valid_name?(name) 125 | User.valid_name?(name[1..-1]) and name[0].chr == '#' 126 | end 127 | 128 | def self.find(item) 129 | @@rooms ||= [] 130 | 131 | if item.kind_of? String 132 | @@rooms.find { |room| room.name.downcase == item.downcase if room.name } 133 | elsif item.kind_of? User 134 | @@rooms.find_all { |room| room.users.include? item } 135 | end 136 | end 137 | 138 | def self.find_or_create(room_name) 139 | room = find room_name 140 | if room.nil? 141 | room = new(room_name) 142 | @@rooms << room 143 | end 144 | room 145 | end 146 | 147 | def self.rooms 148 | @@rooms 149 | end 150 | 151 | def lastlog(since = nil) 152 | { 'display' => 'messages', 'messages' => messages_since(since) } 153 | end 154 | 155 | def search(query, limit = 100) 156 | { 'display' => 'messages', 'messages' => message_search(query, limit) } 157 | end 158 | 159 | def last_update_time 160 | message = JsChat::Storage.driver.lastlog(LASTLOG_DEFAULT, name).last 161 | message['time'] if message 162 | end 163 | 164 | def messages_since(since) 165 | messages = JsChat::Storage.driver.lastlog(LASTLOG_DEFAULT, name) 166 | if since.nil? 167 | messages 168 | else 169 | messages.select { |m| m['time'] && m['time'] > since } 170 | end 171 | end 172 | 173 | def message_search(query, limit) 174 | JsChat::Storage.driver.search(query, name, limit) 175 | end 176 | 177 | def add_to_lastlog(message) 178 | if message 179 | message['time'] = Time.now.utc 180 | JsChat::Storage.driver.log message, name 181 | end 182 | end 183 | 184 | def join(user) 185 | if @users.include? user 186 | Error.new(:already_joined, 'Already in that room') 187 | else 188 | @users << user 189 | user.rooms << self 190 | join_notice user 191 | { 'display' => 'join', 'join' => { 'user' => user.name, 'room' => @name } } 192 | end 193 | end 194 | 195 | def part(user) 196 | if not @users.include?(user) 197 | Error.new(:not_in_room, 'Not in that room') 198 | else 199 | user.rooms.delete_if { |r| r == self } 200 | @users.delete_if { |u| u == user } 201 | part_notice user 202 | { 'display' => 'part', 'part' => { 'user' => user.name, 'room' => @name } } 203 | end 204 | end 205 | 206 | def send_message(message) 207 | message['room'] = name 208 | response = { 'display' => 'message', 'message' => message } 209 | 210 | add_to_lastlog response 211 | 212 | @users.each do |user| 213 | user.connection.send_response response 214 | end 215 | end 216 | 217 | def member_names 218 | @users.collect { |user| user.name } 219 | end 220 | 221 | def to_json(*a) 222 | { 'name' => @name, 'members' => member_names }.to_json(*a) 223 | end 224 | 225 | def notice(user, message, all = false) 226 | add_to_lastlog message 227 | 228 | @users.each do |u| 229 | if (u != user and !all) or all 230 | u.connection.send_response(message) 231 | end 232 | end 233 | end 234 | 235 | def change_notice(user, response) 236 | notice(user, response, true) 237 | end 238 | 239 | def join_notice(user) 240 | notice(user, { 'display' => 'join_notice', 'join_notice' => { 'user' => user.name, 'room' => @name } }) 241 | end 242 | 243 | def part_notice(user) 244 | notice(user, { 'display' => 'part_notice', 'part_notice' => { 'user' => user.name, 'room' => @name } }) 245 | end 246 | 247 | def quit_notice(user) 248 | notice(user, { 'display' => 'quit_notice', 'quit_notice' => { 'user' => user.name, 'room' => @name } }) 249 | @users.delete_if { |u| u == user } 250 | end 251 | end 252 | 253 | # User initially has a nil name 254 | def users_with_names 255 | @@users.find_all { |u| u.name } 256 | end 257 | 258 | def name_taken?(name) 259 | users_with_names.find { |user| user.name.downcase == name.downcase } 260 | end 261 | 262 | # {"identify":"alex"} 263 | def identify(name, ip, session_length, options = {}) 264 | if @user and @user.identified 265 | Error.new :already_identified, 'You have already identified' 266 | elsif name_taken? name 267 | Error.new :name_taken, 'Name already taken' 268 | else 269 | @user.name = name 270 | @user.ip = ip 271 | @user.session_length = session_length 272 | @user.update_session_expiration 273 | register_stateless_user if @stateless 274 | { 'display' => 'identified', 'identified' => @user } 275 | end 276 | rescue JsChat::Errors::InvalidName => exception 277 | exception 278 | end 279 | 280 | def lastlog(room, options = {}) 281 | room = Room.find room 282 | if room and room.users.include? @user 283 | room.lastlog 284 | else 285 | Error.new(:not_in_room, "Please join this room first") 286 | end 287 | end 288 | 289 | def search(query, options = {}) 290 | room = Room.find options['room'] 291 | if room and room.users.include? @user 292 | room.search query 293 | else 294 | Error.new(:not_in_room, "Please join this room first") 295 | end 296 | end 297 | 298 | def since(room, options = {}) 299 | room = Room.find room 300 | if room and room.users.include? @user 301 | response = room.lastlog(@user.last_poll) 302 | @user.last_poll = Time.now.utc 303 | response 304 | else 305 | Error.new(:not_in_room, "Please join this room first") 306 | end 307 | end 308 | 309 | def times(message, options = {}) 310 | times = {} 311 | @user.rooms.each do |room| 312 | times[room.name] = room.last_update_time 313 | end 314 | times 315 | end 316 | 317 | def ping(message, options = {}) 318 | if @user and @user.last_poll and Time.now.utc > @user.last_poll 319 | time = Time.now.utc 320 | @user.update_session_expiration 321 | { 'pong' => time } 322 | else 323 | # TODO: HANDLE PING OUTS 324 | Error.new(:ping_out, 'Your connection has been lost') 325 | end 326 | end 327 | 328 | def quit(message, options = {}) 329 | if @user 330 | disconnect_user @user 331 | end 332 | end 333 | 334 | def room_message(message, options) 335 | room = Room.find options['to'] 336 | if room and room.users.include? @user 337 | room.send_message({ 'message' => message, 'user' => @user.name }) 338 | else 339 | send_response Error.new(:not_in_room, "Please join this room first") 340 | end 341 | end 342 | 343 | def private_message(message, options) 344 | user = users_with_names.find { |u| u.name.downcase == options['to'].downcase } 345 | if user 346 | # Return the message to the user, and send it to the other person too 347 | now = Time.now.utc 348 | user.private_message({ 'message' => message, 'user' => @user.name }) 349 | @user.private_message({ 'message' => message, 'user' => @user.name }) 350 | else 351 | Error.new(:not_online, 'User not online') 352 | end 353 | end 354 | 355 | def send_message(message, options) 356 | if options['to'].nil? 357 | send_response Error.new(:to_required, 'Please specify who to send the message to or join a channel') 358 | elsif options['to'][0].chr == '#' 359 | room_message message, options 360 | else 361 | private_message message, options 362 | end 363 | end 364 | 365 | def join(room_name, options = {}) 366 | if Room.valid_name? room_name 367 | room = Room.find_or_create(room_name) 368 | room.join @user 369 | else 370 | Error.new(:invalid_room, 'Invalid room name') 371 | end 372 | end 373 | 374 | def part(room_name, options = {}) 375 | room = @user.rooms.find { |r| r.name == room_name } 376 | if room 377 | room.part @user 378 | else 379 | Error.new(:not_in_room, "You are not in that room") 380 | end 381 | end 382 | 383 | def names(room_name, options = {}) 384 | room = Room.find(room_name) 385 | if room 386 | { 'display' => 'names', 'names' => room.users, 'room' => room.name } 387 | else 388 | Error.new(:room_not_available, 'No such room') 389 | end 390 | end 391 | 392 | def new_cookie 393 | chars = ("a".."z").to_a + ("1".."9").to_a 394 | Array.new(8, '').collect { chars[rand(chars.size)] }.join 395 | end 396 | 397 | def register_stateless_client 398 | @stateless_cookie = new_cookie 399 | user = User.new(self) 400 | @@stateless_cookies << { :cookie => @stateless_cookie, :user => user } 401 | @@users << user 402 | { 'cookie' => @stateless_cookie } 403 | end 404 | 405 | def current_stateless_client 406 | @@stateless_cookies.find { |c| c[:cookie] == @stateless_cookie } 407 | end 408 | 409 | def register_stateless_user 410 | current_stateless_client[:user] = @user 411 | end 412 | 413 | def valid_stateless_user? 414 | current_stateless_client 415 | end 416 | 417 | def load_stateless_user 418 | if client = current_stateless_client 419 | @user = client[:user] 420 | @stateless = true 421 | else 422 | raise JsChat::Errors::InvalidCookie.new(:invalid_cookie, 'Invalid cookie') 423 | end 424 | end 425 | 426 | def disconnect_lagged_users 427 | @@stateless_cookies.delete_if do |cookie| 428 | if cookie[:user].session_expired? 429 | lagged?(cookie[:user].last_poll) ? disconnect_user(cookie[:user]) && true : false 430 | end 431 | end 432 | end 433 | 434 | def lagged?(time) 435 | Time.now.utc - time > STATELESS_TIMEOUT 436 | end 437 | 438 | def unbind 439 | return if @stateless 440 | disconnect_user(@user) 441 | @user = nil 442 | end 443 | 444 | def disconnect_user(user) 445 | log :info, "Removing a connection" 446 | Room.find(user).each do |room| 447 | room.quit_notice user 448 | end 449 | 450 | @@users.delete_if { |u| u == user } 451 | end 452 | 453 | def post_init 454 | @@users ||= [] 455 | @@stateless_cookies ||= [] 456 | @user = User.new(self) 457 | end 458 | 459 | def log(level, message) 460 | if Object.const_defined? :ServerConfig and ServerConfig['logger'] 461 | if @user 462 | message = "#{@user.name} (#{@user.ip}): #{message}" 463 | end 464 | ServerConfig['logger'].send level, message 465 | end 466 | end 467 | 468 | def change(change, options = {}) 469 | if change == 'user' 470 | field, value = @user.send :change, options[change] 471 | { 'display' => 'notice', 'notice' => "Your #{field} has been changed to: #{value}" } 472 | else 473 | Error.new(:invalid_request, 'Invalid change request') 474 | end 475 | rescue JsChat::Errors::InvalidName => exception 476 | exception 477 | end 478 | 479 | def list(list, options = {}) 480 | case list 481 | when 'rooms' 482 | @user.rooms.collect { |room| room.name } 483 | else 484 | Error.new(:invalid_request, 'Invalid list command') 485 | end 486 | end 487 | 488 | def send_response(data) 489 | response = '' 490 | case data 491 | when String 492 | response = data 493 | when Error 494 | response = data.to_json + "\n" 495 | log :error, data.message 496 | else 497 | # Other objects should be safe for to_json 498 | response = data.to_json + "\n" 499 | log :info, response.strip 500 | end 501 | 502 | send_data response 503 | end 504 | 505 | include EM::Protocols::LineText2 506 | 507 | def get_remote_ip 508 | Socket.unpack_sockaddr_in(get_peername)[1] 509 | end 510 | 511 | def receive_line(data) 512 | response = '' 513 | disconnect_lagged_users 514 | 515 | if data and data.size > ServerConfig['max_message_length'] 516 | raise JsChat::Errors::MessageTooLong.new(:message_too_long, 'Message too long') 517 | end 518 | 519 | data.chomp.split("\n").each do |line| 520 | # Receive the identify request 521 | input = JSON.parse line 522 | 523 | @user.seen! 524 | 525 | # Unbind when a stateless connection doesn't match the cookie 526 | if input.has_key?('cookie') 527 | @stateless_cookie = input['cookie'] 528 | load_stateless_user 529 | end 530 | 531 | if input.has_key? 'protocol' 532 | if input['protocol'] == 'stateless' 533 | @stateless = true 534 | response << send_response(register_stateless_client) 535 | end 536 | elsif input.has_key? 'identify' 537 | input['ip'] ||= get_remote_ip 538 | response << send_response(identify(input['identify'], input['ip'], input['session_length'])) 539 | else 540 | %w{search lastlog change send join names part since ping list quit times}.each do |command| 541 | if @user.name.nil? 542 | response << send_response(Error.new(:identity_required, 'Identify first')) 543 | return response 544 | end 545 | 546 | if input.has_key? command 547 | if command == 'send' 548 | @user.last_activity = Time.now.utc 549 | message_result = send('send_message', input[command], input) 550 | response << message_result if message_result.kind_of? String 551 | else 552 | result = send_response(send(command, input[command], input)) 553 | response << result if result.kind_of? String 554 | end 555 | end 556 | end 557 | end 558 | end 559 | 560 | response 561 | rescue JsChat::Errors::StillFlooding 562 | "" 563 | rescue JsChat::Errors::Flooding => exception 564 | send_response exception 565 | rescue JsChat::Errors::MessageTooLong => exception 566 | send_response exception 567 | rescue JsChat::Errors::InvalidCookie => exception 568 | send_response exception 569 | rescue Exception => exception 570 | puts "Data that raised exception: #{exception}" 571 | p data 572 | print_call_stack 573 | end 574 | 575 | def print_call_stack(from = 0, to = 10) 576 | puts "Stack:" 577 | (from..to).each do |index| 578 | puts "\t#{caller[index]}" 579 | end 580 | end 581 | end 582 | -------------------------------------------------------------------------------- /lib/jschat/server_options.rb: -------------------------------------------------------------------------------- 1 | require 'optparse' 2 | require 'tmpdir' 3 | 4 | logger = nil 5 | 6 | if Object.const_defined? :Logger 7 | logger = Logger.new(STDERR) 8 | logger = Logger.new(STDOUT) 9 | end 10 | 11 | ServerConfigDefaults = { 12 | 'port' => 6789, 13 | 'ip' => '0.0.0.0', 14 | 'logger' => logger, 15 | 'max_message_length' => 500, 16 | 'tmp_files' => File.join(Dir::tmpdir, 'jschat'), 17 | 'db_name' => 'jschat', 18 | 'db_host' => 'localhost', 19 | 'db_port' => 27017, 20 | #'db_username' => '', 21 | #'db_password' => '', 22 | # Register your instance of JsChat here: http://twitter.com/apps/create 23 | # 'twitter' => { 'key' => '', 'secret' => '' } 24 | } 25 | 26 | # Command line options will overrides these 27 | def load_options(path) 28 | path = File.expand_path path 29 | if File.exists? path 30 | JSON.parse(File.read path) 31 | else 32 | {} 33 | end 34 | end 35 | 36 | def make_tmp_files 37 | ServerConfig['use_tmp_files'] = false 38 | if File.exists? ServerConfig['tmp_files'] 39 | ServerConfig['use_tmp_files'] = true 40 | else 41 | if Dir.mkdir ServerConfig['tmp_files'] 42 | ServerConfig['use_tmp_files'] = true 43 | end 44 | end 45 | end 46 | 47 | options = {} 48 | default_config_file = '/etc/jschat/config.json' 49 | 50 | ARGV.clone.options do |opts| 51 | script_name = File.basename($0) 52 | opts.banner = "Usage: #{$0} [options]" 53 | 54 | opts.separator "" 55 | 56 | opts.on("-c", "--config=PATH", String, "Configuration file location (#{default_config_file}") { |o| options['config'] = o } 57 | opts.on("-p", "--port=PORT", String, "Port number") { |o| options['port'] = o } 58 | opts.on("-t", "--tmp_files=PATH", String, "Temporary files location (including pid file)") { |o| options['tmp_files'] = o } 59 | opts.on("--help", "-H", "This text") { puts opts; exit 0 } 60 | 61 | opts.parse! 62 | end 63 | 64 | options = load_options(options['config'] || default_config_file).merge options 65 | 66 | ServerConfig = ServerConfigDefaults.merge options 67 | make_tmp_files 68 | -------------------------------------------------------------------------------- /lib/jschat/storage/init.rb: -------------------------------------------------------------------------------- 1 | require File.join(File.dirname(__FILE__), 'mongo') 2 | require File.join(File.dirname(__FILE__), 'null') 3 | 4 | module JsChat::Storage 5 | def self.driver=(driver) 6 | @driver = driver 7 | end 8 | 9 | def self.driver ; @driver ; end 10 | 11 | def self.enabled=(enabled) 12 | @enabled = enabled 13 | end 14 | 15 | def self.enabled? 16 | @enabled 17 | end 18 | end 19 | 20 | -------------------------------------------------------------------------------- /lib/jschat/storage/mongo.rb: -------------------------------------------------------------------------------- 1 | begin 2 | require 'mongo' 3 | rescue LoadError 4 | end 5 | 6 | module JsChat::Storage 7 | module MongoDriver 8 | def self.connect! 9 | @db = Mongo::Connection.new(ServerConfig['db_host'], ServerConfig['db_port'], :slave_ok => true).db(ServerConfig['db_name']) 10 | if ServerConfig['db_username'] and ServerConfig['db_password'] 11 | if @db.authenticate(ServerConfig['db_username'], ServerConfig['db_password']) 12 | true 13 | else 14 | raise 'Bad Mongo username or password' 15 | end 16 | else 17 | true 18 | end 19 | end 20 | 21 | def self.log(message, room) 22 | message['room'] = room 23 | @db['events'].insert(message) 24 | end 25 | 26 | def self.lastlog(number, room) 27 | @db['events'].find({ :room => room }, { :limit => number, :sort => ['time', Mongo::DESCENDING] }).to_a.reverse 28 | end 29 | 30 | def self.search(query, room, limit) 31 | query = /\b#{query}\b/i 32 | @db['events'].find({ 'message.message' => query, 'room' => room }, 33 | { :limit => limit, :sort => ['time', Mongo::DESCENDING] } 34 | ).to_a.reverse 35 | end 36 | 37 | # TODO: use twitter oauth for the key 38 | def self.find_user(options) 39 | @db['users'].find_one(options) 40 | end 41 | 42 | def self.save_user(user) 43 | @db['users'].save user 44 | end 45 | 46 | def self.delete_user(user) 47 | @db['users'].remove user 48 | end 49 | 50 | def self.available? 51 | return unless Object.const_defined?(:Mongo) 52 | connect! 53 | rescue 54 | p $! 55 | puts 'Failed to connect to mongo' 56 | false 57 | end 58 | end 59 | end 60 | -------------------------------------------------------------------------------- /lib/jschat/storage/null.rb: -------------------------------------------------------------------------------- 1 | module JsChat::Storage 2 | MEMORY_MESSAGE_LIMIT = 100 3 | 4 | module NullDriver 5 | def self.log(message, room) 6 | @messages ||= [] 7 | message['room'] = room 8 | @messages.push message 9 | @messages = @messages[-MEMORY_MESSAGE_LIMIT..-1] if @messages.size > MEMORY_MESSAGE_LIMIT 10 | end 11 | 12 | def self.lastlog(number, room) 13 | @messages ||= [] 14 | @messages.select { |m| m['room'] == room }.reverse[0..number].reverse 15 | end 16 | 17 | def self.search(query, room, limit) 18 | @messages ||= [] 19 | @messages.select do |m| 20 | m['message'] and m['message']['message'].match(query) and m['room'] == room 21 | end.reverse[0..limit].reverse 22 | end 23 | 24 | def self.find_user(options) 25 | end 26 | 27 | def self.save_user(user) 28 | end 29 | 30 | def self.delete_user(user) 31 | end 32 | end 33 | end 34 | 35 | -------------------------------------------------------------------------------- /test/server_test.rb: -------------------------------------------------------------------------------- 1 | require 'test_helper' 2 | 3 | class ServerTest < Test::Unit::TestCase 4 | include JsChatHelpers 5 | 6 | def setup 7 | @jschat = JsChatMock.new 8 | @jschat.post_init 9 | end 10 | 11 | def teardown 12 | @jschat.reset 13 | end 14 | 15 | def test_identify 16 | response = JSON.parse @jschat.receive_line({ 'identify' => 'alex' }.to_json) 17 | assert_equal 'identified', response['display'] 18 | end 19 | 20 | def test_invalid_identify 21 | response = JSON.parse @jschat.receive_line({ 'identify' => '@lex' }.to_json) 22 | assert_equal JsChat::Errors::Codes.invert[:invalid_name], response['error']['code'] 23 | end 24 | 25 | def test_identify_twice_fails 26 | identify_as 'alex' 27 | result = JSON.parse identify_as('alex') 28 | assert_equal 106, result['error']['code'] 29 | end 30 | 31 | def test_ensure_nicks_are_longer_than_0 32 | result = JSON.parse identify_as('') 33 | assert result['error'] 34 | end 35 | 36 | def test_ensure_nicks_are_unique 37 | identify_as 'alex' 38 | 39 | # Obvious duplicate 40 | result = identify_as 'alex' 41 | assert result['error'] 42 | 43 | # Case 44 | result = identify_as 'Alex' 45 | assert result['error'] 46 | end 47 | 48 | def test_invalid_room_name 49 | identify_as 'bob' 50 | response = JSON.parse @jschat.receive_line({ 'join' => 'oublinet' }.to_json) 51 | assert_equal 'Invalid room name', response['error']['message'] 52 | end 53 | 54 | def test_join 55 | identify_as 'bob' 56 | expected = { 'display' => 'join', 'join' => { 'user' => 'bob', 'room' => '#oublinet' } }.to_json + "\n" 57 | assert_equal expected, @jschat.receive_line({ 'join' => '#oublinet' }.to_json) 58 | end 59 | 60 | def test_join_without_identifying 61 | response = JSON.parse @jschat.receive_line({ 'join' => '#oublinet' }.to_json) 62 | assert response['error'] 63 | end 64 | 65 | def test_join_more_than_once 66 | identify_as 'bob' 67 | 68 | expected = { 'display' => 'error', 'error' => { 'message' => 'Already in that room' } }.to_json + "\n" 69 | @jschat.receive_line({ 'join' => '#oublinet' }.to_json) 70 | response = JSON.parse @jschat.receive_line({ 'join' => '#oublinet' }.to_json) 71 | assert_equal JsChat::Errors::Codes.invert[:already_joined], response['error']['code'] 72 | end 73 | 74 | def test_names 75 | identify_as 'nick', '#oublinet' 76 | 77 | # Add a user 78 | @jschat.add_user 'alex', '#oublinet' 79 | 80 | response = JSON.parse(@jschat.receive_line({ 'names' => '#oublinet' }.to_json)) 81 | assert response['names'] 82 | end 83 | 84 | def test_valid_names 85 | user = JsChat::User.new nil 86 | ['alex*', "alex\n"].each do |name| 87 | assert_raises JsChat::Errors::InvalidName do 88 | user.name = name 89 | end 90 | end 91 | end 92 | 93 | def test_message_not_in_room 94 | identify_as 'nick', '#oublinet' 95 | @jschat.add_user 'alex', '#oublinet' 96 | response = JSON.parse @jschat.receive_line({ 'send' => 'hello', 'to' => '#merk' }.to_json) 97 | assert_equal 'Please join this room first', response['error']['message'] 98 | end 99 | 100 | def test_message 101 | identify_as 'nick', '#oublinet' 102 | @jschat.add_user 'alex', '#oublinet' 103 | assert @jschat.receive_line({ 'send' => 'hello', 'to' => '#oublinet' }.to_json) 104 | end 105 | 106 | def test_message_ignores_case 107 | identify_as 'nick', '#oublinet' 108 | @jschat.add_user 'alex', '#oublinet' 109 | response = @jschat.receive_line({ 'send' => 'hello', 'to' => '#Oublinet' }.to_json) 110 | assert response 111 | end 112 | 113 | def test_part 114 | identify_as 'nick', '#oublinet' 115 | @jschat.add_user 'alex', '#oublinet' 116 | response = JSON.parse @jschat.receive_line({ 'part' => '#oublinet'}.to_json) 117 | assert_equal '#oublinet', response['part']['room'] 118 | end 119 | 120 | def test_private_message 121 | identify_as 'nick' 122 | @jschat.add_user 'alex', '#oublinet' 123 | response = JSON.parse @jschat.receive_line({ 'send' => 'hello', 'to' => 'alex' }.to_json) 124 | assert_equal 'hello', response['message']['message'] 125 | end 126 | 127 | def test_private_message_ignores_case 128 | identify_as 'nick' 129 | @jschat.add_user 'alex', '#oublinet' 130 | response = JSON.parse @jschat.receive_line({ 'send' => 'hello', 'to' => 'Alex' }.to_json) 131 | assert_equal 'hello', response['message']['message'] 132 | end 133 | 134 | def test_log_request 135 | identify_as 'nick', '#oublinet' 136 | @jschat.receive_line({ 'send' => 'hello', 'to' => '#oublinet' }.to_json) 137 | response = JSON.parse @jschat.receive_line({ 'lastlog' => '#oublinet' }.to_json) 138 | assert_equal 'hello', response['messages'].last['message']['message'] 139 | end 140 | 141 | def test_name_change 142 | identify_as 'nick', '#oublinet' 143 | @jschat.add_user 'alex', '#oublinet' 144 | response = JSON.parse @jschat.receive_line({ 'change' => 'user', 'user' => { 'name' => 'bob' }}.to_json) 145 | assert_equal 'notice', response['display'] 146 | end 147 | 148 | def test_name_change_duplicate 149 | identify_as 'nick', '#oublinet' 150 | @jschat.add_user 'alex', '#oublinet' 151 | response = JSON.parse @jschat.receive_line({ 'change' => 'user', 'user' => { 'name' => 'alex' }}.to_json) 152 | assert_equal 'error', response['display'] 153 | end 154 | 155 | def test_max_message_length 156 | identify_as 'nick', '#oublinet' 157 | response = JSON.parse @jschat.receive_line({ 'send' => 'a' * 1000, 'to' => '#oublinet' }.to_json) 158 | assert response['error'] 159 | end 160 | 161 | def test_flood_protection 162 | identify_as 'nick', '#oublinet' 163 | response = '' 164 | # simulate a flood and extract the error response 165 | (1..50).detect do 166 | response = @jschat.receive_line({ 'send' => 'a' * 10, 'to' => '#oublinet' }.to_json) 167 | response.match /error/ 168 | end 169 | response = JSON.parse response 170 | assert response['error'] 171 | assert_match /wait a few seconds/i, response['error']['message'] 172 | end 173 | 174 | end 175 | 176 | -------------------------------------------------------------------------------- /test/stateless_test.rb: -------------------------------------------------------------------------------- 1 | require 'test_helper' 2 | 3 | class StatelessTest < Test::Unit::TestCase 4 | include JsChatHelpers 5 | 6 | def setup 7 | @jschat = JsChatMock.new 8 | @jschat.post_init 9 | @cookie = get_cookie 10 | end 11 | 12 | def test_identify 13 | response = send_to_jschat({ 'identify' => 'alex', 'cookie' => @cookie }) 14 | assert_equal 'identified', response['display'] 15 | end 16 | 17 | def test_join 18 | response = identify_as 'alex2', '#jschat' 19 | assert JSON.parse(response)['join'] 20 | end 21 | 22 | def test_message 23 | response = identify_as 'nick', '#jschat' 24 | assert send_to_jschat({ 'cookie' => @cookie, 'send' => 'hello', 'to' => '#jschat' }, false) 25 | end 26 | 27 | private 28 | 29 | def get_cookie 30 | JSON.parse(@jschat.receive_line({ 'protocol' => 'stateless' }.to_json))['cookie'] 31 | end 32 | 33 | end 34 | -------------------------------------------------------------------------------- /test/test_helper.rb: -------------------------------------------------------------------------------- 1 | require 'test/unit' 2 | require 'rubygems' 3 | require 'eventmachine' 4 | gem 'json', '>= 1.1.9' 5 | require 'json' 6 | $:.unshift File.join(File.dirname(__FILE__), '..', 'lib') 7 | require File.join(File.dirname(__FILE__), '..', 'lib', 'jschat', 'server.rb') 8 | 9 | ServerConfig['max_message_length'] = 500 10 | 11 | class JsChat::Room 12 | def self.reset 13 | @@rooms = nil 14 | end 15 | end 16 | 17 | module JsChatHelpers 18 | def identify_as(name, channel = nil) 19 | if @cookie 20 | result = @jschat.receive_line({ 'identify' => name, :cookie => @cookie }.to_json) 21 | result = @jschat.receive_line({ 'join' => channel, :cookie => @cookie }.to_json) if channel 22 | else 23 | result = @jschat.receive_line({ 'identify' => name }.to_json) 24 | result = @jschat.receive_line({ 'join' => channel }.to_json) if channel 25 | end 26 | result 27 | end 28 | 29 | def send_to_jschat(h, parse = true) 30 | response = @jschat.receive_line(h.to_json) 31 | parse ? JSON.parse(response) : response 32 | end 33 | end 34 | 35 | class JsChatMock 36 | include JsChat 37 | 38 | def get_remote_ip 39 | '' 40 | end 41 | 42 | def send_data(data) 43 | data 44 | end 45 | 46 | def reset 47 | @@users = nil 48 | @user = nil 49 | Room.reset 50 | end 51 | 52 | # Helper for testing 53 | def add_user(name, room_name) 54 | room = Room.find_or_create room_name 55 | user = User.new self 56 | user.name = name 57 | user.rooms << room 58 | @@users << user 59 | room.users << user 60 | end 61 | end 62 | 63 | JsChat::Storage.enabled = false 64 | JsChat::Storage.driver = JsChat::Storage::NullDriver 65 | 66 | --------------------------------------------------------------------------------