ruboty-ruby_persistence 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: de217b9be23db369bc8ff9a5aae2ade6b54bba64
-  data.tar.gz: 79a01481307d1a5648c8b9ac33a68616705fa664
+  metadata.gz: 14d35d6042c421a1d15051f95a3391052538fc6d
+  data.tar.gz: ce10f722debf3f1ed76765735b5f40b9bc3ac432
 SHA512:
-  metadata.gz: c72db9e113af16f96e94877026f50e71e7dd3fc973fc0a8a1de861cd69ac2620478d04edaff8804be66e3583b52f9a011d48b4fe7854f3a464a4ff8564ce6213
-  data.tar.gz: 0f5a2870e619ba67721b31647db8e574ceccc54ad72fa2dfae29fc9e8006def4363c6e4a3200e9607b7dcfb19b809227ddf71a0c26a7064437e848057a928e04
+  metadata.gz: 61a455cb07e949ab6286b806cc276496aa3f88408ee02d39fac1715080e73d6815a6db92c06bf8d711506e2f545746ae36f2f243702d52170bd9e7fed1afc9bd
+  data.tar.gz: 8ef775aabf4f7f7aafab8a8c5ecaf6e499f249168fbcc1c1a45584bf8708de96faf712a085aa441102cf8fea346d5d0ae43b3af21010c7b69801967a4be5d8b9

data/lib/ruboty/handlers/ruby_persistence.rb

@@ -17,15 +17,18 @@ module Ruboty
         name: 'unset',
       )
 
+      def initialize(*)
+        super
+        restore
+      end
+
       def set(message)
         name = message[:name]
 
-        if variables[name]
-          set_and_persist(message)
-        elsif (method = predefined_method(name))
+        if !variables.key?(name) && (method = predefined_method(name))
           message.reply("Error: `#{name}` is predefined by `#{method.owner}`")
         else
-          set_and_persist(message)
+          define_and_persist_variable(message)
         end
       end
 
@@ -33,6 +36,7 @@ module Ruboty
         name = message[:name]
 
         if variables[name]
+          undefine_variable(name)
           variables.delete(name)
           message.reply("`#{name}` is now unset")
         else
@@ -42,22 +46,36 @@ module Ruboty
 
       private
 
+      def restore
+        variables.each do |name, value|
+          define_variable(name, value)
+        end
+      end
+
       def predefined_method(name)
         context.method(name)
       rescue NameError
         nil
       end
 
-      def set_and_persist(message)
+      def define_and_persist_variable(message)
         name = message[:name]
-        result = context.instance_eval(message[:code])
-        context.__send__(:define_method, name) { result }
-        variables[name] = result
-        message.reply("`#{name}` is now `#{result.inspect}`")
+        value = context.instance_eval(message[:code])
+        define_variable(name, value)
+        variables[name] = value
+        message.reply("`#{name}` is now `#{value.inspect}`")
       rescue Exception => exception
         message.reply("#{exception.class}: #{exception}")
       end
 
+      def define_variable(name, value)
+        context.__send__(:define_method, name) { value }
+      end
+
+      def undefine_variable(name)
+        context.singleton_class.__send__(:undef_method, name)
+      end
+
       def variables
         robot.brain.data[NAMESPACE] ||= {}
       end

data/lib/ruboty/ruby_persistence/version.rb

@@ -1,5 +1,5 @@
 module Ruboty
   module RubyPersistence
-    VERSION = "0.1.3"
+    VERSION = "0.1.4"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruboty-ruby_persistence
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Yuji Nakayama