lib/websocket/examples/websocket_server.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2014 Lucas Bajolet <r4pass@hotmail.com>
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #     http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 # Sample module for a minimal chat server using Websockets on port 8088
  18 module websocket_server
  19
  20 import websocket
  21
  22 var sock = new WebSocketListener(8088, 1)
  23
  24 var msg: String
  25
  26 if sock.listener.closed then
  27         print sys.errno.strerror
  28 end
  29
  30 var cli: TCPStream
  31
  32 while not sock.closed do
  33         cli = sock.accept
  34         while cli.connected do
  35                 if sys.stdin.poll_in then
  36                         msg = gets
  37                         printn "Received message : {msg}"
  38                         if msg == "disconnect" then cli.close
  39                         cli.write(msg)
  40                 end
  41                 if cli.can_read(10) then
  42                         msg = ""
  43                         while cli.can_read(0) do msg += cli.read(100)
  44                         if msg != "" then print msg
  45                 end
  46         end
  47 end