diamant 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5f4a7116de355c0b04280914d27758f287116298d2cde5a36e311382352342b1
-  data.tar.gz: 04a5cd38b4596383a220bd553b6e6d1a9360310810156d009f8cad569db9bd01
+  metadata.gz: 3aae272eff26676b737dc65ff08ea76b0b4ea1822ec4785dfa47b97f0647dbe8
+  data.tar.gz: dc803566505cbed2653c44ac2bc7d4d3e0767ceb4604f91f32c1523917045192
 SHA512:
-  metadata.gz: e948fce49088c1b5483a18dd4c4dfccfbaefb7b160d4a9ce0fc880a0bf4487c6ded0eda9a3b58de9ad74fe5a05f3922c790716d26789db0a18621381fe6acefe
-  data.tar.gz: 1113f9c27f505edcc445b116a3d9dbb900c681a5305b48655d72faee52714a1ca7c4b3b71a7123b80e8f93468ecfcfa88afd5f455e2fc4222adef6e73070fa61
+  metadata.gz: dc0d90505b8e7ef55f679c290aca6288dddd7c5b3208fbcab86dfa9501be7316ae8f425a3e5e4e5a3525d03c8765c397729231a590836bbe168783f162c3a852
+  data.tar.gz: b181654c75861b8a20308513b451999b63ce08b47c527ce2258de451daa16835aba9676c0f0ede70cb609eeb7b1a71f0a8f933b3c031ab39d505e4308396d532

data/bin/diamant

@@ -1,9 +1,9 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require 'diamant'
-require 'diamant/cert_generator'
 require 'optparse'
+require_relative '../lib/diamant'
+require_relative '../lib/diamant/cert_generator'
 
 options = { hostname: '127.0.0.1', port: 1965 }
 OptionParser.new do |parser| # rubocop:disable Metrics/BlockLength

data/lib/diamant/mimefile.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+module Diamant
+  class MimeError < StandardError; end
+
+  # MIME aware wrapper for file
+  class MimeFile
+    attr_reader :body, :category, :content_type, :extension
+
+    MIMETYPES = {
+      '.gemini' => 'text/gemini',
+      '.gmi' => 'text/gemini',
+      '.txt' => 'text/plain',
+      '.md' => 'text/markdown',
+      '.org' => 'text/org',
+      '.rss' => 'application/rss+xml',
+      '.atom' => 'application/atom+xml',
+      '.xml' => 'application/xml',
+      '.svg' => 'image/svg+xml',
+      '.bmp' => 'image/bmp',
+      '.png' => 'image/png',
+      '.jpg' => 'image/jpeg',
+      '.jpeg' => 'image/jpeg',
+      '.gif' => 'image/gif',
+      '.webp' => 'image/webp',
+      '.mp3' => 'audio/mpeg',
+      '.ogg' => 'audio/ogg'
+    }.freeze
+
+    def initialize(path)
+      @path = path
+      @body = File.read path
+      @extension, @content_type = extract_info
+      @category = classify
+      validate!
+      prepare_gem_file if @content_type == 'text/gemini'
+    end
+
+    # Disable metrics on purpose for big switch
+    # rubocop:disable Metrics/CyclomaticComplexity, Metrics/MethodLength
+    def validate
+      return true if @category == :text
+
+      case @content_type
+      when 'image/jpeg'
+        @body[0..1] == "\xFF\xD8"
+      when 'image/png'
+        @body[0..7] == "\x89PNG\r\n\u001A\n"
+      when 'image/gif'
+        @body[0..2] == 'GIF'
+      when 'image/webp'
+        @body[0..3] == 'RIFF' && @body[8..11] == 'WEBP'
+      when 'image/bmp'
+        @body[0..2] == 'BM'
+      when 'image/svg+xml', 'application/xml',
+           'application/rss+xml', 'application/atom+xml'
+        @body[0..5] == '<?xml '
+      when 'audio/mpeg'
+        @body[0..3] == 'ID3'
+      when 'audio/ogg'
+        @body[0..4] == 'OggS'
+      else
+        false
+      end
+    end
+    # rubocop:enable Metrics/CyclomaticComplexity, Metrics/MethodLength
+
+    def validate!
+      return if validate
+
+      raise MimeError, "#{@path} does not looks like a #{@content_type} file!"
+    end
+
+    private
+
+    def extract_info
+      extension = File.extname @path
+      mime = MIMETYPES[extension]
+      raise MimeError, "#{@path} format is not supported!" if mime == ''
+
+      # Any other supported extension
+      [extension, mime]
+    end
+
+    def classify
+      return unless @content_type
+
+      @content_type.split('/', 2).first.to_sym
+    end
+
+    def prepare_gem_file
+      # Ensure each lines finishes with \r\n
+      @body = @body.each_line(chomp: true).to_a.join "\r\n"
+    end
+  end
+end

data/lib/diamant/response.rb

@@ -3,9 +3,13 @@
 require 'net/gemini/request'
 require 'uri/gemini'
 
+require_relative 'mimefile'
+
 module Diamant
   # Methods to generate requests responses
   module Response
+    class NotFound < Net::Gemini::BadResponse; end
+
     private
 
     def reject_request?(sock, current_load)
@@ -25,34 +29,39 @@ module Diamant
       true
     end
 
-    def read_file(client)
+    def build_response(client)
       r = Net::Gemini::Request.read_new(client)
-      [r.uri, route(r.path)]
+      file_path = route r.path
+      [r.uri, read_mime_file(file_path)]
+    rescue NotFound => e
+      @logger.warn "51 - #{e}"
+      [r.uri, "51 Not found!\r\n"]
     rescue Net::Gemini::BadRequest, URI::InvalidURIError => e
       @logger.error "59 - #{e}"
-      [nil, ["59\r\n"]]
+      [nil, "59\r\n"]
     end
 
     def route(path)
       # In any case, remove the / prefix
-      route = File.expand_path path.delete_prefix('/'), Dir.pwd
+      route = File.expand_path path.delete_prefix('/')
       # We better should use some sort of chroot...
       unless route.start_with?(Dir.pwd)
-        @logger.warn "Bad attempt to get something out of public_dir: #{route}"
-        return ['51 Not found!']
+        raise NotFound, "Attempt to get something out of public_dir: #{route}"
       end
+
       route << '/index.gmi' if File.directory?(route)
-      return ['51 Not found!'] unless File.exist?(route)
+      return route if File.exist?(route)
 
-      build_response route
+      raise NotFound, "File #{route} does not exist"
     end
 
-    def build_response(route)
-      info = Diamant::MimeType.new(route)
-      answer = File.readlines route, chomp: true
-      answer.prepend "20 #{info.content_type}"
-    rescue Diamant::MimeError
-      ['50 Not a supported file!']
+    def read_mime_file(route)
+      info = Diamant::MimeFile.new route
+      header = "20 #{info.content_type}"
+      [header, info.body].join("\r\n")
+    rescue Diamant::MimeError => e
+      @logger.error "50 - #{e}"
+      "50 Not a supported file!\r\n"
     end
   end
 end

data/lib/diamant/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Diamant
-  VERSION = '0.0.9'
+  VERSION = '0.0.10'
 end

data/lib/diamant.rb

@@ -6,9 +6,7 @@ require 'English'
 require 'openssl'
 require 'fileutils'
 
-require 'diamant/version'
-require 'diamant/mimetype'
-require 'diamant/response'
+require_relative 'diamant/response'
 
 module Diamant
   # Runs the server request/answer loop.
@@ -62,16 +60,14 @@ module Diamant
     end
 
     def handle_client(client)
-      current_load = Thread.list.length - 1
+      current_load = Thread.list.length - 1 # Exclude main thread
       return if reject_request?(client, current_load)
 
-      uri, answer = read_file(client)
+      uri, answer = build_response(client)
       log_line = [current_load, client.peeraddr[3], answer[0]]
       log_line << uri if uri
       @logger.info log_line.join(' - ')
-      answer.each do |line|
-        client.puts "#{line}\r\n"
-      end
+      client.puts answer
     end
 
     def ssl_context

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: diamant
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Étienne Deparis
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-07-13 00:00:00.000000000 Z
+date: 2024-07-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-net-text
@@ -38,7 +38,7 @@ files:
 - bin/diamant
 - lib/diamant.rb
 - lib/diamant/cert_generator.rb
-- lib/diamant/mimetype.rb
+- lib/diamant/mimefile.rb
 - lib/diamant/response.rb
 - lib/diamant/version.rb
 homepage: https://git.umaneti.net/diamant/about/

data/lib/diamant/mimetype.rb

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-module Diamant
-  class MimeError < StandardError; end
-
-  # Helper to understand what mimetype has a given file
-  class MimeType
-    attr_reader :extension, :content_type
-
-    MIMETYPES = {
-      '.gemini' => 'text/gemini',
-      '.gmi' => 'text/gemini',
-      '.txt' => 'text/plain',
-      '.md' => 'text/markdown',
-      '.org' => 'text/org',
-      '.xml' => 'application/xml',
-      '.png' => 'image/png',
-      '.jpg' => 'image/jpeg',
-      '.jpeg' => 'image/jpeg',
-      '.gif' => 'image/gif'
-    }.freeze
-
-    def initialize(path)
-      @path = path
-      extract_info
-    end
-
-    def supported?
-      @extension != '' && MIMETYPES.has_key?(@extension)
-    end
-
-    private
-
-    def extract_info
-      @extension = File.extname @path
-      raise MimeError, "#{@path} format is not supported!" unless supported?
-
-      # Any other supported extension
-      @content_type = MIMETYPES[@extension]
-    end
-  end
-end