summaryrefslogtreecommitdiff
path: root/bin/rbot
blob: cee5974a05331920187a3dc4dff306a1d5dfeac5 (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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
#!/usr/bin/env ruby

=begin rdoc

= rbot main executable

Usage:

  % rbot [options] [config directory]

== Options

[-h, --help]
    display a help message and exit
[-v, --version]
    display version information and exit
[-d, --debug]
    enable debug messages
[-l, --loglevel _level_]
    sets the minimum log level verbosity
[-b, --background]
    background (daemonize) the bot
[-p, --pidfile _filename_]
    write the bot pid to _filename_

The default config directory is <tt>~/.rbot</tt>.

The default pidfile is <tt><i>botdir</i>/rbot.pid</tt>.

The logfile is located at <tt><i>botdir</i>/<i>botname</i>.log</tt>, and
the default loglevel is 1 (INFO messages). Possible values for the loglevel
are 0 (DEBUG), 1 (INFO), 2 (WARN), 3 (ERROR), 4 (FATAL).

Please note that the logfile doesn't contain IRC logs (which are located at
<tt><i>botdir</i>/logs/*</tt>, but only rbot diagnostic messages.

=end

# Copyright (C) 2002-2006 Tom Gilbert.
# Copyright (C) 2007-2008 Giuseppe Bilotta and the rbot development team
#
# This is free software, see COPYING for licensing details

$VERBOSE=true

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

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

if $version =~ /git/
  if defined?(SCM_DIR) and File.exists?(File.join(SCM_DIR, '.git'))
    begin
      git_out = `git status`
      git_out.match(/^# On branch (.*)\n/)
      if $1 # git 1.5.x
        branch = $1.dup || "unknown"
        changed = git_out.match(/^# Change(.*)\n/)
        rev = "revision "
        git_out = `git log -1 --pretty=format:"%h%n%b%n%ct"`.split("\n")
        rev << git_out.first
        $version_timestamp = git_out.last.to_i
        if git_out[1].match(/^git-svn-id: \S+@(\d+)/)
          rev << "(svn #{$1})"
        end
        rev << ", local changes" if changed
      else # older gits
        git_out = `git branch`
        git_out.match(/^\* (.*)\n/)
        branch = $1.dup rescue "unknown"
        rev = "revision " + `git rev-parse HEAD`[0,6]
      end
    rescue => e
      puts e.inspect
      branch = "unknown branch"
      rev = "unknown revision"
    end

    $version << " (#{branch} branch, #{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 if File.directory?(File.join(up, '.svn'))
    $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 = $DEBUG
$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 "  -l, --loglevel     sets the log level verbosity"
  puts "  -b, --background   background (daemonize) the bot"
  puts "  -p, --pidfile      write the bot pid to 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