summaryrefslogtreecommitdiff
path: root/bin/rbot
blob: 946344586bfbac2c0bc3632219733cecccad5228 (plain)
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/usr/bin/env ruby

# Copyright (C) 2002 Tom Gilbert.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to
# deal in the Software without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies of the Software and its documentation and acknowledgment shall be
# given in the documentation and software packages that this Software was
# used.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
# THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

# Most of the string processing across rbot is done against IRC messages, which
# do not have a well-defined encoding. Although many clients are now using
# UTF-8, there is no guarantee that an arbitrary string received from IRC will
# be UTF-8 encoded. We have to force ASCII (byte-wise/charset agnostic)
# matching because otherwise some strings can give problems: in particular, for
# example, the bytesequence "\340\350\354\362\371" (that is the aeiou vowels,
# each with a grave accent) will cause the string to be considered up to the
# "\354" (i with grave accent) only: so either the rest of the message is
# ignored, or the matching fails.
$KCODE = 'a'

$VERBOSE=true

require 'etc'
require 'getoptlong'
require 'fileutils'

$version="0.9.11-svn"
$opts = Hash.new

if $version =~ /svn/
  if defined?(SVN_DIR) and File.exists?(File.join(SVN_DIR, '.git'))
    $version.sub!('svn', 'git')

    git_out = `git status`

    git_out.match(/^# On branch (.*)\n/)
    branch = $1
    changed = git_out.match(/^# Change(.*)\n/)
    rev = " @ #{`git rev-parse HEAD`[0..6]}"
    rev << ", local changes" if changed

    $version << "-#{branch}" unless branch.empty?
    $version << rev
  else
    up = File.dirname(__FILE__) + "/.."
    rev = " (unknown revision)"
    begin
      svn_out = `svn info #{up}`
      if svn_out =~ /Last Changed Rev: (\d+)/
        rev = " (revision #{$1}"
      end
      svn_st = `svn st #{up}`
      if svn_st =~ /^[MDA] /
        rev << ", local changes"
      end
      rev << ")"
    rescue => e
      puts e.inspect
    end
    $version += rev
  end
end

orig_opts = ARGV.dup

opts = GetoptLong.new(
  ["--background", "-b", GetoptLong::NO_ARGUMENT],
  ["--debug", "-d", GetoptLong::NO_ARGUMENT],
  ["--help",  "-h", GetoptLong::NO_ARGUMENT],
  ["--loglevel",  "-l", GetoptLong::REQUIRED_ARGUMENT],
  ["--trace",  "-t", GetoptLong::REQUIRED_ARGUMENT],
  ["--pidfile", "-p", GetoptLong::REQUIRED_ARGUMENT],
  ["--version", "-v", GetoptLong::NO_ARGUMENT]
)

$debug = false
$daemonize = false

opts.each {|opt, arg|
  $debug = true if(opt == "--debug")
  $daemonize = true if(opt == "--background")
  $opts[opt.sub(/^-+/, "")] = arg
}

$cl_loglevel = $opts["loglevel"].to_i if $opts["loglevel"]

if ($opts["trace"])
  set_trace_func proc { |event, file, line, id, binding, classname|
    if classname.to_s == $opts["trace"]
      printf "TRACE: %8s %s:%-2d %10s %8s\n", event, File.basename(file), line, id, classname
    end
  }
end

defaultlib = File.expand_path(File.dirname($0) + '/../lib')

if File.directory? "#{defaultlib}/rbot"
  unless $:.include? defaultlib
    $:.unshift defaultlib
  end
end
  
begin
  require 'rbot/ircbot'
rescue LoadError => e
  puts "Error: couldn't find the rbot/ircbot module (or one of its dependencies)\n"
  puts e
  exit 2
end

if ($opts["version"])
  puts "rbot #{$version}"
  exit 0
end

if ($opts["help"])
  puts "usage: rbot [options] [config directory]"
  puts "  -h, --help         this message"
  puts "  -v, --version      version information"
  puts "  -d, --debug        enable debug messages"
  puts "  -b, --background   background (daemonize) the bot"
  puts "  -p, --pidfile      write the bot pid to this file"
  puts "config directory defaults to ~/.rbot"
  exit 0
end

if(bot = Irc::Bot.new(ARGV.shift, :argv => orig_opts))
  # just run the bot
  bot.mainloop
end