165 lines
4.4 KiB
Python
Executable File
165 lines
4.4 KiB
Python
Executable File
#! /usr/bin/env python
|
|
#
|
|
# Example program using irclib.py.
|
|
#
|
|
# Copyright (C) 1999-2002 Joel Rosdahl
|
|
#
|
|
# This library is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 2.1 of the License, or (at your option) any later version.
|
|
#
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this library; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
# Joel Rosdahl <joel@rosdahl.net>
|
|
#
|
|
# servermap connects to an IRC server and finds out what other IRC
|
|
# servers there are in the net and prints a tree-like map of their
|
|
# interconnections.
|
|
#
|
|
# Example:
|
|
#
|
|
# % ./servermap irc.dal.net somenickname
|
|
# Connecting to server...
|
|
# Getting links...
|
|
#
|
|
# 26 servers (18 leaves and 8 hubs)
|
|
#
|
|
# splitrock.tx.us.dal.net
|
|
# `-vader.ny.us.dal.net
|
|
# |-twisted.ma.us.dal.net
|
|
# |-sodre.nj.us.dal.net
|
|
# |-glass.oh.us.dal.net
|
|
# |-distant.ny.us.dal.net
|
|
# | |-algo.se.eu.dal.net
|
|
# | | |-borg.se.eu.dal.net
|
|
# | | | `-ced.se.eu.dal.net
|
|
# | | |-viking.no.eu.dal.net
|
|
# | | |-inco.fr.eu.dal.net
|
|
# | | |-paranoia.se.eu.dal.net
|
|
# | | |-gaston.se.eu.dal.net
|
|
# | | | `-powertech.no.eu.dal.net
|
|
# | | `-algo-u.se.eu.dal.net
|
|
# | |-philly.pa.us.dal.net
|
|
# | |-liberty.nj.us.dal.net
|
|
# | `-jade.va.us.dal.net
|
|
# `-journey.ca.us.dal.net
|
|
# |-ion.va.us.dal.net
|
|
# |-dragons.ca.us.dal.net
|
|
# |-toronto.on.ca.dal.net
|
|
# | `-netropolis-r.uk.eu.dal.net
|
|
# | |-traced.de.eu.dal.net
|
|
# | `-lineone.uk.eu.dal.net
|
|
# `-omega.ca.us.dal.net
|
|
|
|
import irclib
|
|
import sys
|
|
|
|
if len(sys.argv) != 3:
|
|
print "Usage: servermap <server[:port]> <nickname>"
|
|
sys.exit(1)
|
|
|
|
links = []
|
|
|
|
def on_connect(connection, event):
|
|
sys.stdout.write("\nGetting links...")
|
|
sys.stdout.flush()
|
|
connection.links()
|
|
|
|
def on_passwdmismatch(connection, event):
|
|
print "Password required."
|
|
sys.exit(1)
|
|
|
|
def on_links(connection, event):
|
|
global links
|
|
|
|
links.append((event.arguments()[0],
|
|
event.arguments()[1],
|
|
event.arguments()[2]))
|
|
|
|
def on_endoflinks(connection, event):
|
|
global links
|
|
|
|
print "\n"
|
|
|
|
m = {}
|
|
for (to_node, from_node, desc) in links:
|
|
if from_node != to_node:
|
|
m[from_node] = m.get(from_node, []) + [to_node]
|
|
|
|
if connection.get_server_name() in m:
|
|
if len(m[connection.get_server_name()]) == 1:
|
|
hubs = len(m) - 1
|
|
else:
|
|
hubs = len(m)
|
|
else:
|
|
hubs = 0
|
|
|
|
print "%d servers (%d leaves and %d hubs)\n" % (len(links), len(links)-hubs, hubs)
|
|
|
|
print_tree(0, [], connection.get_server_name(), m)
|
|
connection.quit("Using irclib.py")
|
|
|
|
def on_disconnect(connection, event):
|
|
sys.exit(0)
|
|
|
|
def indent_string(level, active_levels, last):
|
|
if level == 0:
|
|
return ""
|
|
s = ""
|
|
for i in range(level-1):
|
|
if i in active_levels:
|
|
s = s + "| "
|
|
else:
|
|
s = s + " "
|
|
if last:
|
|
s = s + "`-"
|
|
else:
|
|
s = s + "|-"
|
|
return s
|
|
|
|
def print_tree(level, active_levels, root, map, last=0):
|
|
sys.stdout.write(indent_string(level, active_levels, last)
|
|
+ root + "\n")
|
|
if root in map:
|
|
list = map[root]
|
|
for r in list[:-1]:
|
|
print_tree(level+1, active_levels[:]+[level], r, map)
|
|
print_tree(level+1, active_levels[:], list[-1], map, 1)
|
|
|
|
s = sys.argv[1].split(":", 1)
|
|
server = s[0]
|
|
if len(s) == 2:
|
|
try:
|
|
port = int(s[1])
|
|
except ValueError:
|
|
print "Error: Erroneous port."
|
|
sys.exit(1)
|
|
else:
|
|
port = 6667
|
|
nickname = sys.argv[2]
|
|
|
|
irc = irclib.IRC()
|
|
sys.stdout.write("Connecting to server...")
|
|
sys.stdout.flush()
|
|
try:
|
|
c = irc.server().connect(server, port, nickname)
|
|
except irclib.ServerConnectionError, x:
|
|
print x
|
|
sys.exit(1)
|
|
|
|
c.add_global_handler("welcome", on_connect)
|
|
c.add_global_handler("passwdmismatch", on_passwdmismatch)
|
|
c.add_global_handler("links", on_links)
|
|
c.add_global_handler("endoflinks", on_endoflinks)
|
|
c.add_global_handler("disconnect", on_disconnect)
|
|
|
|
irc.process_forever()
|