data.rb in lita-tweet-0.4.0

- old
+ new

@@ -7,20 +7,30 @@
       Data = Struct.new(:redis, :config, :robot) do
         def usernames
           redis.smembers("twitter_accounts")
         end
 
+        def default_account
+          account(redis.get("default_username"))
+        end
+
         def account(username)
+          return nil if username.nil?
+
           hash = redis.hgetall("twitter_accounts:#{username}")
+          return nil if hash.empty?
+
           data = hash.each_with_object({}){|(k,v),h| h[k.to_sym] = v }
           Account.new(**data.merge(config: config))
         end
 
-        def add_account(token, secret)
-          account = Account.new(token: token, secret: secret, config: config)
-          username = account.username
+        def add_account(token, secret, username = nil)
+          account = Account.new(token: token, secret: secret,
+                                config: config, username: username)
+          username ||= account.lookup_username!
 
+          redis.setnx("default_username", username)
           redis.sadd("twitter_accounts", username)
           redis.hmset("twitter_accounts:#{username}",
             "username", username,
             "token", token,
             "secret", secret
@@ -30,17 +40,53 @@
         end
 
         def remove_account(username)
           redis.del("twitter_accounts:#{username}")
           redis.srem("twitter_accounts", username)
+
+          if redis.get("default_username") == username
+            next_username = usernames.first
+            redis.set("default_username", next_username)
+          end
         end
 
         def set_last_tweet(username, tweet_id)
           redis.hset("twitter_accounts:#{username}", "last_tweet", tweet_id)
         end
 
         def get_last_tweet(username)
           redis.hget("twitter_accounts:#{username}", "last_tweet")
+        end
+
+        def channel_map
+          redis.smembers("channels").map do |name|
+            [name, get_channel_map(name)]
+          end.to_h
+        end
+
+        def get_channel_account(channel)
+          return nil unless channel
+          account(get_channel_map(channel))
+        end
+
+        def get_channel_map(channel)
+          redis.get("channels:#{channel}")
+        end
+
+        def set_channel_map(channel, username)
+          return false unless usernames.include?(username)
+          redis.sadd("channels", channel)
+          redis.set("channels:#{channel}", username)
+        end
+
+        def clear_channel_map(channel)
+          redis.del("channels:#{channel}")
+          redis.srem("channels", channel)
+        end
+
+        def set_default(username)
+          return false unless usernames.include?(username)
+          redis.set("default_username", username)
         end
 
         def create_request_token(callback_path)
           request_token = consumer.get_request_token(
             oauth_callback: bot_uri(callback_path).to_s)