core/aliaser.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

require_relative 'store'

module Dinobot
  module Core
    class Aliaser
      attr_accessor :data

      @@instance = nil
      @@mutex = Mutex.new

      def initialize
        @store = Dinobot::Core::Store.new('aliaser.db')
        @data = @store.data
      end

      def add(from, to, channel=:global)
        @data[channel] = Hash.new unless @data.key?(channel)
        @data[channel][from] = to

        save
      end

      def remove(from, channel=:global)
        return unless data.key?(channel)

        @data[channel].delete(from)
        @data.delete(channel) if @data[channel].empty?

        save
      end

      def aliases
        @data.dup
      end

      def save
        @store.save
      end

      class << self
        def instance
          return @@instance if @@instance

          @@mutex.synchronize do
            @@instance ||= new
          end
        end
      end

      private_class_method :allocate, :new
    end
  end
end