"""Handle nntp: and news: links"""
# Copyright (C) 2000,2001  Bastian Kleineidam
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

import re, time, sys, nntplib, urlparse, linkcheck
from UrlData import ExcList,UrlData
from debuglevels import *
debug = linkcheck.Config.debug

ExcList.extend([nntplib.error_reply,
               nntplib.error_temp,
               nntplib.error_perm,
               nntplib.error_proto,
               ])

class NntpUrlData(UrlData):
    "Url link with NNTP scheme"

    def buildUrl(self):
        # use nntp instead of news to comply with the unofficial internet
	# draft of Alfred Gilman which unifies (s)news and nntp URLs
        # note: we use this only internally (for parsing and caching)
        if self.urlName[:4].lower()=='news':
            self.url = 'nntp'+self.urlName[4:]
        else:
            self.url = self.urlName
        self.urlTuple = urlparse.urlparse(self.url)
        debug(BRING_IT_ON, self.urlTuple)


    def checkConnection(self, config):
        nntpserver = self.urlTuple[1] or config["nntpserver"]
        if not nntpserver:
            self.setWarning(linkcheck._("No NNTP server specified, skipping this URL"))
            return
        nntp = self._connectNntp(nntpserver)
        group = self.urlTuple[2]
        while group[:1]=='/':
            group = group[1:]
        if '@' in group:
            # request article
            resp,number,id = nntp.stat("<"+group+">")
            self.setInfo(linkcheck._('Articel number %s found' % number))
        else:
            # split off trailing articel span
            group = group.split('/',1)[0]
            if group:
                # request group info
                resp,count,first,last,name = nntp.group(group)
                self.setInfo(linkcheck._("Group %s has %s articles, range %s to %s") %\
                             (name, count, first, last))
            else:
                # group name is the empty string
                self.setWarning(linkcheck._("No newsgroup specified in NNTP URL"))


    def _connectNntp(self, nntpserver):
        """This is done only once per checking task."""
        timeout = 1
        while timeout:
            try:
                nntp=nntplib.NNTP(nntpserver)
                timeout = 0
            except nntplib.error_perm:
                value = sys.exc_info()[1]
                debug(BRING_IT_ON, "NNTP:", value)
                if re.compile("^505").search(str(value)):
                    import whrandom
                    time.sleep(whrandom.randint(10,20))
                else:
                    raise
        return nntp


    def getCacheKey(self):
        return self.url