From 79b42cf8584c66222d994bb5a3b2c1f6c428bdf7 Mon Sep 17 00:00:00 2001
From: Adam Robertson <adam@arcreative.net>
Date: Thu, 5 Nov 2020 15:11:59 -0800
Subject: [PATCH] Use symbols (but allow strings) for `foreman_init_system`
 setting

---
 README.md                       |  2 +-
 lib/capistrano/tasks/foreman.rb | 24 ++++++++++++------------
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/README.md b/README.md
index fb08cc0..3759efd 100644
--- a/README.md
+++ b/README.md
@@ -16,7 +16,7 @@ require 'capistrano/foreman'
 # Default settings
 set :foreman_use_sudo, false # Set to :rbenv for rbenv sudo, :rvm for rvmsudo or true for normal sudo
 set :foreman_roles, :all
-set :foreman_init_system, 'upstart'
+set :foreman_init_system, :upstart
 set :foreman_export_path, ->{ File.join(Dir.home, '.init') }
 set :foreman_app, -> { fetch(:application) }
 set :foreman_app_name_systemd, -> { "#{ fetch(:foreman_app) }.target" }
diff --git a/lib/capistrano/tasks/foreman.rb b/lib/capistrano/tasks/foreman.rb
index 796920c..e911133 100644
--- a/lib/capistrano/tasks/foreman.rb
+++ b/lib/capistrano/tasks/foreman.rb
@@ -23,7 +23,7 @@
           opts.map { |opt, value| "--#{opt}=\"#{value}\"" }.join(' ')
       end
 
-      if fetch(:foreman_init_system) == 'systemd'
+      if fetch(:foreman_init_system).to_sym == :systemd
         foreman_exec :systemctl, :'daemon-reload'
       end
     end
@@ -32,8 +32,8 @@
   desc 'Start the application services'
   task :start do
     on roles fetch(:foreman_roles) do
-      case fetch(:foreman_init_system)
-      when 'systemd'
+      case fetch(:foreman_init_system).to_sym
+      when :systemd
         foreman_exec :systemctl, :enable, fetch(:foreman_app_name_systemd)
         foreman_exec :systemctl, :start, fetch(:foreman_app_name_systemd)
       else
@@ -45,8 +45,8 @@
   desc 'Stop the application services'
   task :stop do
     on roles fetch(:foreman_roles) do
-      case fetch(:foreman_init_system)
-      when 'systemd'
+      case fetch(:foreman_init_system).to_sym
+      when :systemd
         foreman_exec :systemctl, :stop, fetch(:foreman_app_name_systemd)
         foreman_exec :systemctl, :disable, fetch(:foreman_app_name_systemd)
       else
@@ -58,8 +58,8 @@
   desc 'Restart the application services'
   task :restart do
     on roles fetch(:foreman_roles) do
-      case fetch(:foreman_init_system)
-      when 'systemd'
+      case fetch(:foreman_init_system).to_sym
+      when :systemd
         foreman_exec :systemctl, :restart, fetch(:foreman_app_name_systemd)
       else
         foreman_exec :restart, fetch(:foreman_app)
@@ -69,15 +69,15 @@
 
   def foreman_exec(*args)
     sudo_type = fetch(:foreman_use_sudo)
-    case sudo_type.to_s
-    when 'rbenv'
+    case sudo_type.to_sym
+    when :rbenv
       # this is required because 'rbenv sudo'
       # is not recognized by bundle_bins
       args.unshift(:bundle, :exec) if args[0].to_s == "foreman"
       execute(:rbenv, :sudo, *args)
-    when 'rvm'
+    when :rvm
       execute(:rvmsudo, *args)
-    when 'chruby'
+    when :chruby
       execute(:sudo, 'chruby-exec', fetch(:chruby_ruby), '--', *args)
     else
       sudo_type ? sudo(*args) : execute(*args)
@@ -89,7 +89,7 @@ def foreman_exec(*args)
   task :defaults do
     set :bundle_bins, fetch(:bundle_bins, []).push(:foreman)
     set :foreman_use_sudo, false
-    set :foreman_init_system, 'upstart'
+    set :foreman_init_system, :upstart
     set :foreman_export_path, '/etc/init/sites'
     set :foreman_roles, :all
     set :foreman_app, -> { fetch(:application) }