lib/puma/daemon.rb in puma-daemon-0.1.0 vs lib/puma/daemon.rb in puma-daemon-0.1.1

@@ -1,10 +1,27 @@
 # frozen_string_literal: true
 
-require_relative "daemon/version"
+require 'puma/runner'
+require 'puma/single'
+require 'puma/cluster'
+require 'puma/dsl'
+require 'puma/cli'
 
+require 'puma/daemon/version'
+require 'puma/daemon/runner'
+require 'puma/daemon/configuration'
+require 'puma/daemon/cli'
+require 'puma/daemon/dsl'
+
 module Puma
   module Daemon
-    class Error < StandardError; end
-    # Your code goes here...
+    def self.daemonize!
+      ::Puma::Single.include(::Puma::Daemon::Runner)
+      ::Puma::Cluster.include(::Puma::Daemon::Runner)
+      ::Puma::DSL.include(::Puma::Daemon::DSL)
+      ::Puma::Configuration.prepend(::Puma::Daemon::Configuration)
+      ::Puma::CLI.instance_eval { attr_reader :options }
+    end
   end
 end
+
+Puma::Daemon.daemonize!