Add --serve flag.

Author: davejacobs

lib/mint/commandline/parse.rb

@@ -26,6 +26,14 @@ def self.parse!(argv)
           commandline_options[:verbose] = true
         end
 
+        cli.on "--serve", "Publish files and serve them on a local HTTP server" do
+          commandline_options[:serve_mode] = true
+        end
+
+        cli.on "--serve-port PORT", Integer, "Port for the local server (default: 4000)" do |port|
+          commandline_options[:serve_port] = port
+        end
+
         cli.on "-t", "--template TEMPLATE", "Specify a template by name (default: default)" do |t|
           commandline_options[:layout_name] = t
           commandline_options[:style_name] = t

lib/mint/commandline/run.rb

@@ -1,6 +1,7 @@
 require "tempfile"
 
 require_relative "../commandline/parse"
+require_relative "../commandline/serve"
 require_relative "../config"
 require_relative "../workspace"
 
@@ -14,7 +15,11 @@ def self.run!(argv)
         exit 0
       else
         begin
-          publish!(files, config: config)
+          if config.serve_mode
+            serve!(files, config: config)
+          else
+            publish!(files, config: config)
+          end
         rescue ArgumentError => e
           $stderr.puts "Error: #{e.message}"
           exit 1

lib/mint/commandline/serve.rb

@@ -0,0 +1,64 @@
+require "tmpdir"
+require "webrick"
+require "fileutils"
+
+require_relative "../config"
+
+module Mint
+  module Commandline
+    def self.serve!(source_files, config:)
+      config = Config.ensure_config(config)
+      port = config.serve_port
+
+      tmp_dir = Dir.mktmpdir("mint-serve-")
+
+      begin
+        serve_config = Config.new(config.to_h.merge(
+          destination_directory: Pathname.new(tmp_dir),
+          style_mode: :inline
+        ))
+
+        publish!(source_files, config: serve_config)
+
+        log = WEBrick::Log.new(nil, WEBrick::BasicLog::ERROR)
+        server = WEBrick::HTTPServer.new(
+          Port: port,
+          Logger: log,
+          AccessLog: []
+        )
+
+        server.mount_proc "/" do |req, res|
+          path = req.path.gsub(/\.\./, "")
+          file_path = File.join(tmp_dir, path)
+
+          resolved =
+            if File.file?(file_path)
+              file_path
+            elsif File.file?("#{file_path}.html")
+              "#{file_path}.html"
+            elsif File.file?(File.join(file_path, "index.html"))
+              File.join(file_path, "index.html")
+            end
+
+          if resolved
+            res.status = 200
+            res.content_type = WEBrick::HTTPUtils.mime_type(resolved, WEBrick::HTTPUtils::DefaultMimeTypes)
+            res.body = File.binread(resolved)
+          else
+            res.status = 404
+            res.content_type = "text/plain"
+            res.body = "Not found: #{req.path}"
+          end
+        end
+
+        trap("INT") { server.shutdown }
+        trap("TERM") { server.shutdown }
+
+        puts "Serving on http://localhost:#{port} (press Ctrl+C to stop)"
+        server.start
+      ensure
+        FileUtils.rm_rf(tmp_dir)
+      end
+    end
+  end
+end

lib/mint/config.rb

@@ -5,6 +5,8 @@ class Config
     attr_accessor :files
     attr_accessor :stdin_mode
     attr_accessor :stdout_mode
+    attr_accessor :serve_mode
+    attr_accessor :serve_port
     attr_accessor :help
     attr_accessor :verbose
     attr_accessor :layout_name
@@ -17,9 +19,11 @@ class Config
     attr_accessor :preserve_structure
     attr_accessor :autodrop
     attr_accessor :options
-    
+
     DEFAULT_STDIN_MODE                  = false
     DEFAULT_STDOUT_MODE                 = false
+    DEFAULT_SERVE_MODE                  = false
+    DEFAULT_SERVE_PORT                  = 4000
     DEFAULT_VERBOSE                     = false
     DEFAULT_LAYOUT_NAME                 = "default"
     DEFAULT_STYLE_NAME                  = "default"
@@ -34,6 +38,8 @@ class Config
     def initialize(options = {})
       @stdin_mode = options[:stdin_mode] if options.key?(:stdin_mode)
       @stdout_mode = options[:stdout_mode] if options.key?(:stdout_mode)
+      @serve_mode = options[:serve_mode] if options.key?(:serve_mode)
+      @serve_port = options[:serve_port] if options.key?(:serve_port)
       @help = options[:help]
       @verbose = options[:verbose] if options.key?(:verbose)
       @layout_name = options[:layout_name] || options[:template_name]
@@ -52,6 +58,8 @@ def to_h
       {
         stdin_mode: @stdin_mode,
         stdout_mode: @stdout_mode,
+        serve_mode: @serve_mode,
+        serve_port: @serve_port,
         help: @help,
         verbose: @verbose,
         layout_name: @layout_name,
@@ -100,6 +108,8 @@ def self.defaults
       Config.new({
         stdin_mode: DEFAULT_STDIN_MODE,
         stdout_mode: DEFAULT_STDOUT_MODE,
+        serve_mode: DEFAULT_SERVE_MODE,
+        serve_port: DEFAULT_SERVE_PORT,
         verbose: DEFAULT_VERBOSE,
         layout_name: DEFAULT_LAYOUT_NAME,
         style_name: DEFAULT_STYLE_NAME,

mint.gemspec

@@ -17,6 +17,7 @@ Gem::Specification.new do |s|
   s.files = Dir["{bin,config,lib,man}/**/*"] + [ "README.md", "Gemfile", "LICENSE" ]
   s.test_files = Dir["{features,spec}/**/*"]
 
+  s.add_dependency "webrick", "~> 1.8"
   s.add_dependency "redcarpet", "~> 3.6", ">= 3.6.1"
   s.add_dependency "sass-embedded", "~> 1.89", ">= 1.89.2"
   s.add_dependency "toml", "~> 0.3"