require 'itamae'

module Itamae
  module Resource
    class Service < Base
      define_attribute :action, default: :nothing
      define_attribute :name, type: String, default_name: true

      def set_current_attributes
        @current_attributes[:running?] = run_specinfra(:check_service_is_running, name)
        @current_attributes[:enabled?] = run_specinfra(:check_service_is_enabled, name)

        actions = [action].flatten
        if actions.include?(:start) || actions.include?(:restart)
          @attributes[:running?] = true
        elsif actions.include?(:stop)
          @attributes[:running?] = false
        end

        if actions.include?(:enable)
          @attributes[:enabled?] = true
        elsif actions.include?(:disable)
          @attributes[:enabled?] = false
        end
      end

      def start_action
        run_specinfra(:start_service, name)
      end

      def stop_action
        run_specinfra(:stop_service, name)
      end

      def restart_action
        run_specinfra(:restart_service, name)
      end

      def reload_action
        run_specinfra(:reload_service, name)
      end

      def enable_action
        run_specinfra(:enable_service, name)
      end

      def disable_action
        run_specinfra(:disable_service, name)
      end
    end
  end
end