# frozen_string_literal: true
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            require "iodine"
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            require_relative "io/version"
          
        
      
        

          

            1
            

            
              
            

            require_relative "io/server"
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            module Websocket
          
        
      
        

          

            1
            

            
              
            

              module Echo
          
        
      
        

          

            1
            

            
              
            

                module Io
          
        
      
        

          

            1
            

            
              
            

                  class Error < StandardError; end
          
        
      
        

          

            
            

            
              
            

                end
          
        
      
        

          

            
            

            
              
            

              end
          
        
      
        

          

            
            

            
              
            

            end
          
        
      
    
  

    

      
        

          

            
            

            
              
            

            # frozen_string_literal: true
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            module Websocket
          
        
      
        

          

            1
            

            
              
            

              module Echo
          
        
      
        

          

            1
            

            
              
            

                module Io
          
        
      
        

          

            1
            

            
              
            

                  class Handler
          
        
      
        

          

            1
            

            
              
            

                    GOODBYE_LINE = "See you in a bit."
          
        
      
        

          

            1
            

            
              
            

                    SHUTDOWN_LINE = "Server going away\r\n"
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                    def on_open(client)
          
        
      
        

          

            
            

            
              
            

                      # Set a connection timeout
          
        
      
        

          

            1
            

            
              
            

                      client.timeout = 10
          
        
      
        

          

            
            

            
              
            

                      # Write a welcome message
          
        
      
        

          

            1
            

            
              
            

                      client.write "WebSocket-Echo-IO server powered by Iodine #{Iodine::VERSION}.\r\n"
          
        
      
        

          

            
            

            
              
            

                    end
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                    def on_message(client, data)
          
        
      
        

          

            2
            

            
              
            

                      client.write data
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            2
            

            
              
                
                  else: 1
                
              
                
                  then: 1
                
              
            

                      return unless /^Goodbye\b/.match?(data)
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                      client.write GOODBYE_LINE
          
        
      
        

          

            1
            

            
              
            

                      client.close
          
        
      
        

          

            
            

            
              
            

                    end
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                    def on_shutdown(client)
          
        
      
        

          

            1
            

            
              
            

                      client.write SHUTDOWN_LINE
          
        
      
        

          

            
            

            
              
            

                    end
          
        
      
        

          

            
            

            
              
            

                  end
          
        
      
        

          

            
            

            
              
            

                end
          
        
      
        

          

            
            

            
              
            

              end
          
        
      
        

          

            
            

            
              
            

            end
          
        
      
    
  

    

      
        

          

            
            

            
              
            

            # frozen_string_literal: true
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            require_relative "handler"
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            module Websocket
          
        
      
        

          

            1
            

            
              
            

              module Echo
          
        
      
        

          

            1
            

            
              
            

                module Io
          
        
      
        

          

            1
            

            
              
            

                  class Server
          
        
      
        

          

            
            

            
              
            

                    # support heroku custom port
          
        
      
        

          

            1
            

            
              
                
                  then: 0
                
              
                
                  else: 1
                
              
            

                    PORT = ENV["PORT"]&.to_i || 3003
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                    class << self
          
        
      
        

          

            1
            

            
              
            

                      def start
          
        
      
        

          

            1
            

            
              
            

                        handler = Handler.new
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                        app = proc do |env|
          
        
      
        

          

            
            skipped

            
              
            

                          # :nocov: for some reason clover can't detect coverage of these lines
          
        
      
        

          

            
            skipped

            
              
            

                          if env["rack.upgrade?"] == :websocket
          
        
      
        

          

            
            skipped

            
              
            

                            env["rack.upgrade"] = handler
          
        
      
        

          

            
            skipped

            
              
            

                            [0, {}, []] # It's possible to set cookies for the response.
          
        
      
        

          

            
            skipped

            
              
            

                          elsif env["rack.upgrade?"] == :sse
          
        
      
        

          

            
            skipped

            
              
            

                            puts "SSE connections can only receive data from the server, they can't write."
          
        
      
        

          

            
            skipped

            
              
            

                            env["rack.upgrade"] = handler
          
        
      
        

          

            
            skipped

            
              
            

                            [0, {}, []] # It's possible to set cookies for the response.
          
        
      
        

          

            
            skipped

            
              
            

                          else
          
        
      
        

          

            
            skipped

            
              
            

                            body = "Upgrade connection to Websocket to enter echo mode."
          
        
      
        

          

            
            skipped

            
              
            

                            [418, { "Content-Length" => body.bytesize, "Content-Type" => "text/plain" }, [body]]
          
        
      
        

          

            
            skipped

            
              
            

                          end
          
        
      
        

          

            
            skipped

            
              
            

                          # :nocov:
          
        
      
        

          

            
            

            
              
            

                        end
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

                        Iodine.listen service: :ws, address: "::", port: PORT, handler: app # max_msg: 128
          
        
      
        

          

            1
            

            
              
            

                        Iodine.start
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            
            

            
              
            

                        # in case long term running clogs the server, hot restart regularly
          
        
      
        

          

            
            

            
              
            

                        # see https://github.com/boazsegev/iodine#hot-restart
          
        
      
        

          

            
            

            
              
            

                        # Iodine.run_every(4 * 60 * 60 * 1000) do
          
        
      
        

          

            
            

            
              
            

                        #   Process.kill("SIGUSR1", Process.pid) unless Iodine.worker?
          
        
      
        

          

            
            

            
              
            

                        # end
          
        
      
        

          

            
            

            
              
            

                      end
          
        
      
        

          

            
            

            
              
            

                    end
          
        
      
        

          

            
            

            
              
            

                  end
          
        
      
        

          

            
            

            
              
            

                end
          
        
      
        

          

            
            

            
              
            

              end
          
        
      
        

          

            
            

            
              
            

            end
          
        
      
    
  

    

      
        

          

            
            

            
              
            

            # frozen_string_literal: true
          
        
      
        

          

            
            

            
              
            

            
          
        
      
        

          

            1
            

            
              
            

            module Websocket
          
        
      
        

          

            1
            

            
              
            

              module Echo
          
        
      
        

          

            1
            

            
              
            

                module Io
          
        
      
        

          

            1
            

            
              
            

                  VERSION = "0.1.0"
          
        
      
        

          

            
            

            
              
            

                end
          
        
      
        

          

            
            

            
              
            

              end
          
        
      
        

          

            
            

            
              
            

            end