lib/itamae/resource/service.rb in itamae-1.0.0.beta6 vs lib/itamae/resource/service.rb in itamae-1.0.0.beta7

@@ -4,9 +4,27 @@
   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