time_bomb.rb in ragent-0.0.4

- old
+ new

@@ -1,63 +1,33 @@
+# frozen_string_literal: true
+require_relative 'time_bomb/bomb'
+
 module Ragent
   module Plugin
     class TimeBomb
-      class TestBomb
-        include Celluloid
-        include Celluloid::Notifications
-        include Ragent::Logging
-
-        finalizer :stop
-
-        def initialize
-          async.start
-        end
-
-        def start
-          every(5) do
-            publish 'time-bomb-boom', expode_at: Time.now
-          end
-        end
-
-        def stop
-          info 'disarmed'
-        end
-      end
-
       include Ragent::Plugin
 
-      def configure
+      plugin_name 'time_bomb'
+      commands :tick, :list, :disarm
+
+      def configure(*_args)
         @next_time_bomb_id = 1
         @time_bombs = {}
-        command('tick', :command_tick)
-        command('list', :command_list)
-        command('disarm', :command_disarm)
         subscribe('time-bomb-boom', :boom_callback)
       end
 
-      def start
-      end
-
-      def command(sub, method)
-        cmd = Ragent::Command.new(main: 'time_bomb',
-                                  sub: sub,
-                                  recipient: self,
-                                  method: method)
-        @ragent.commands.add(cmd)
-      end
-
       def command_disarm(options)
         bomb_name = "time_bomb_#{options['bomb']}"
-        bomb = Actor[bomb_name]
+        bomb = agents(bomb_name)
         if bomb
           bomb.terminate
           @time_bombs.delete(bomb_name)
           "terminated #{bomb_name}"
         else
           "no such bomb #{bomb_name}"
         end
-    end
+      end
 
       def command_list(_options)
         @time_bombs.keys.join("\n")
       end
 
@@ -65,15 +35,12 @@
         # how to track the bombs to disarm them?
         as = "time_bomb_#{@next_time_bomb_id}"
         @time_bombs[as] = true
         @next_time_bomb_id += 1
 
-        @ragent.supervisor.supervise(
-          type: TimeBomb::TestBomb,
-          as: as
-        )
+        agent(type: TimeBomb::Bomb, as: as)
 
-        'starting timebomb'
+        "starting #{as}"
       end
 
       def boom_callback(_topic, params)
         debug "boom: #{params.inspect}"
       end