#!/usr/bin/env python
# encoding: utf-8

import socket
import re

version = "0.1"

def get_name(line):
    p = re.compile(":[a-zA-Z0-9_-]*!")
    m = p.search(line)
    if m == None:
        return
    return str(m.group())[1:-1]

class irc:
    def __init__(self, host, port=6667, nick, realname="", name="irc"):
        self.nick = nick
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.connect((host, port))
        self.raw("NICK %s" % self.nick)
        self.raw("USER %s 0 0: %s" % name, realname)

    def raw(self, data):
        return self.socket.send(data)

    def join(self, channel):
        data = "JOIN %s\r\n" % channel
        return self.raw(data)

    def part(self, channel):
        data = "PART %s\r\n" % channel
        return self.raw(data)

    def send(self, to, message):
        data = "PRIVMSG %s :%s\r\n" % (to, message)
        return self.raw(data)

    def quit(self, message="Disconnect"):
        data = "QUIT %s\r\n" % message
        return self.raw(data)