itsi-server 0.1.1 → 0.1.2

data/lib/itsi/server.rb

@@ -2,12 +2,78 @@
 
 require_relative "server/version"
 require_relative "server/itsi_server"
+require_relative "signals"
+require_relative "request"
+require_relative "stream_io"
+require_relative "server/rack/handler/itsi"
 
 module Itsi
   class Server
-    # Call our Rack app with our request ENV.
     def self.call(app, request)
-      app.call(request.to_env)
+      respond request, app.call(request.to_env)
+    end
+
+    def self.streaming_body?(body)
+      body.respond_to?(:call) && !body.respond_to?(:each)
+    end
+
+    def self.respond(request, (status, headers, body))
+      response = request.response
+
+      # Don't try and respond if we've been hijacked.
+      # The hijacker is now responsible for this.
+      return if request.hijacked
+
+      # 1. Set Status
+      response.status = status
+
+      # 2. Set Headers
+      headers.each do |key, value|
+        next response.add_header(key, value) unless value.is_a?(Array)
+
+        value.each do |v|
+          response.add_header(key, v)
+        end
+      end
+
+      # 3. Set Body
+      # As soon as we start setting the response
+      # the server will begin to stream it to the client.
+
+      # If we're partially hijacked or returned a streaming body,
+      # stream this response.
+
+      if (body_streamer = streaming_body?(body) ? body : headers.delete("rack.hijack"))
+        body_streamer.call(StreamIO.new(response))
+
+      # If we're enumerable with more than one chunk
+      # also stream, otherwise write in a single chunk
+      elsif body.respond_to?(:each) || body.respond_to?(:to_ary)
+        unless body.respond_to?(:each)
+          body = body.to_ary
+          raise "Body #to_ary didn't return an array" unless body.is_a?(Array)
+        end
+        # We offset this iteration intentionally,
+        # to optimize for the case where there's only one chunk.
+        buffer = nil
+        body.each do |part|
+          response.send_frame(buffer.to_s) if buffer
+          buffer = part
+        end
+
+        response.send_and_close(buffer.to_s)
+      else
+        response.send_and_close(body.to_s)
+      end
+    ensure
+      response.close_write
+      body.close if body.respond_to?(:close)
+    end
+
+    def self.start_scheduler_loop(scheduler_class, scheduler_task)
+      scheduler = scheduler_class.new
+      Fiber.set_scheduler(scheduler)
+      [scheduler, Fiber.schedule(&scheduler_task)]
     end
 
     # If scheduler is enabled

data/lib/itsi/signals.rb

@@ -0,0 +1,18 @@
+module Itsi
+  module Signals
+    DEFAULT_SIGNALS = ["DEFAULT", ""].freeze
+    module TrapInterceptor
+      def trap(signal, command = nil, &block)
+        return super unless DEFAULT_SIGNALS.include?(command.to_s) && block.nil?
+        Itsi::Server.reset_signal_handlers
+      end
+    end
+    [Kernel, Signal].each do |receiver|
+      receiver.singleton_class.prepend(TrapInterceptor)
+    end
+
+    [Object].each do |receiver|
+      receiver.include(TrapInterceptor)
+    end
+  end
+end

data/lib/itsi/stream_io.rb

@@ -0,0 +1,38 @@
+module Itsi
+  class StreamIO
+    attr :response
+    def initialize(response)
+      @response = response
+    end
+
+    def read
+      response.recv_frame
+    end
+
+    def write(string)
+      response.send_frame(string)
+    end
+
+    def <<(string)
+      response.send_frame(string)
+    end
+
+    def flush
+      # No-op
+    end
+
+    def close
+      close_read
+      close_write
+    end
+
+    def close_read
+      response.close_read
+    end
+
+    def close_write
+      response.close_write
+    end
+
+  end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: itsi-server
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Wouter Coppieters
 bindir: exe
 cert_chain: []
-date: 2025-03-01 00:00:00.000000000 Z
+date: 2025-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: libclang
@@ -24,33 +24,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '14.0'
 - !ruby/object:Gem::Dependency
-  name: rb_sys
+  name: rack
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.9.91
+        version: '1.6'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.9.91
+        version: '1.6'
 - !ruby/object:Gem::Dependency
-  name: rack
+  name: rb_sys
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 0.9.91
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 0.9.91
 description: Itsi Server - A light-weight Rack Server implementation for Ruby
 email:
 - wc@pico.net.nz
@@ -69,30 +69,57 @@ files:
 - exe/itsi
 - ext/itsi_error/Cargo.lock
 - ext/itsi_error/Cargo.toml
+- ext/itsi_error/src/from.rs
 - ext/itsi_error/src/lib.rs
+- ext/itsi_instrument_entry/Cargo.toml
+- ext/itsi_instrument_entry/src/lib.rs
 - ext/itsi_rb_helpers/Cargo.lock
 - ext/itsi_rb_helpers/Cargo.toml
+- ext/itsi_rb_helpers/src/heap_value.rs
 - ext/itsi_rb_helpers/src/lib.rs
+- ext/itsi_scheduler/Cargo.toml
+- ext/itsi_scheduler/extconf.rb
+- ext/itsi_scheduler/src/itsi_scheduler.rs
+- ext/itsi_scheduler/src/itsi_scheduler/io_helpers.rs
+- ext/itsi_scheduler/src/itsi_scheduler/io_waiter.rs
+- ext/itsi_scheduler/src/itsi_scheduler/timer.rs
+- ext/itsi_scheduler/src/lib.rs
 - ext/itsi_server/Cargo.toml
 - ext/itsi_server/extconf.rb
+- ext/itsi_server/src/body_proxy/big_bytes.rs
+- ext/itsi_server/src/body_proxy/itsi_body_proxy.rs
+- ext/itsi_server/src/body_proxy/mod.rs
 - ext/itsi_server/src/lib.rs
 - ext/itsi_server/src/request/itsi_request.rs
 - ext/itsi_server/src/request/mod.rs
+- ext/itsi_server/src/response/itsi_response.rs
+- ext/itsi_server/src/response/mod.rs
 - ext/itsi_server/src/server/bind.rs
+- ext/itsi_server/src/server/bind_protocol.rs
+- ext/itsi_server/src/server/io_stream.rs
 - ext/itsi_server/src/server/itsi_ca/itsi_ca.crt
 - ext/itsi_server/src/server/itsi_ca/itsi_ca.key
 - ext/itsi_server/src/server/itsi_server.rs
+- ext/itsi_server/src/server/lifecycle_event.rs
 - ext/itsi_server/src/server/listener.rs
 - ext/itsi_server/src/server/mod.rs
+- ext/itsi_server/src/server/process_worker.rs
+- ext/itsi_server/src/server/serve_strategy/cluster_mode.rs
+- ext/itsi_server/src/server/serve_strategy/mod.rs
+- ext/itsi_server/src/server/serve_strategy/single_mode.rs
+- ext/itsi_server/src/server/signal.rs
+- ext/itsi_server/src/server/thread_worker.rs
 - ext/itsi_server/src/server/tls.rs
-- ext/itsi_server/src/server/transfer_protocol.rs
-- ext/itsi_server/src/stream_writer/mod.rs
 - ext/itsi_tracing/Cargo.lock
 - ext/itsi_tracing/Cargo.toml
 - ext/itsi_tracing/src/lib.rs
 - lib/itsi/request.rb
 - lib/itsi/server.rb
+- lib/itsi/server/rack/handler/itsi.rb
+- lib/itsi/server/scheduler_mode.rb
 - lib/itsi/server/version.rb
+- lib/itsi/signals.rb
+- lib/itsi/stream_io.rb
 - sig/itsi_server.rbs
 homepage: https://itsi.fyi
 licenses:
@@ -108,12 +135,12 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.1.0
+      version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.3.11
+      version: '3.1'
 requirements: []
 rubygems_version: 3.6.2
 specification_version: 4

data/ext/itsi_server/src/server/transfer_protocol.rs

@@ -1,23 +0,0 @@
-use itsi_error::ItsiError;
-use std::str::FromStr;
-
-#[derive(Debug, Default, Clone)]
-pub enum TransferProtocol {
-    #[default]
-    Https,
-    Http,
-    Unix,
-}
-
-impl FromStr for TransferProtocol {
-    type Err = ItsiError;
-
-    fn from_str(s: &str) -> Result<Self, Self::Err> {
-        match s {
-            "http" => Ok(TransferProtocol::Http),
-            "https" => Ok(TransferProtocol::Https),
-            "unix" => Ok(TransferProtocol::Unix),
-            _ => Err(ItsiError::UnsupportedProtocol(s.to_string())),
-        }
-    }
-}

data/ext/itsi_server/src/stream_writer/mod.rs

@@ -1,21 +0,0 @@
-use bytes::Bytes;
-use magnus::{error::Result as MagnusResult, exception::io_error, Error};
-use tokio::sync::mpsc::Sender;
-
-#[magnus::wrap(class = "Itsi::StreamWriter", free_immediately, size)]
-pub struct StreamWriter {
-    sender: Sender<Bytes>,
-}
-
-impl StreamWriter {
-    pub fn new(sender: Sender<Bytes>) -> Self {
-        StreamWriter { sender }
-    }
-
-    pub fn write(&self, bytes: Bytes) -> MagnusResult<()> {
-        self.sender
-            .blocking_send(bytes)
-            .map_err(|e| Error::new(io_error(), format!("{:?}", e)))?;
-        Ok(())
-    }
-}