#!/usr/bin/env python
# Written by Stephane Graber <stgraber@stgraber.org>
#            Daniel Bartlett <dan@f-box.org>
# Last modification : Mon Jan 28 22:33:23 CET 2008

# 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

defaultPB = "http://pastebin.com" #Default pastebin
try:
    import lsb_release
    release = lsb_release.get_distro_information()['ID'].lower()
    if release == 'debian':
        defaultPB = "http://paste.debian.net"
    elif release == 'fedora':
        defaultPB = "http://fpaste.org"
    elif release == 'ubuntu':
        defaultPB = "http://paste.ubuntu.com"
except ImportError:
    pass

try:
    import urllib, os, sys, re, getopt, xml.dom.minidom, gettext
    try:
        import json # For python 2.6 and newer
    except ImportError:
        try:
            import simplejson as json
        except ImportError:
            json = None
    from gettext import gettext as _
    import configobj

    gettext.textdomain("pastebinit")

    version = "1.3" #Version number to show in the usage
    configfile = os.path.expanduser("~/.pastebinit.xml")

    # Custom urlopener to handle 401's
    class pasteURLopener(urllib.FancyURLopener):
        def http_error_401(self, url, fp, errcode, errmsg, headers, data=None):
            return None

    def preloadPastebins():
        # Check several places for config files:
        #  - global config in /etc/pastebin.d
        #  - for source checkout, config in the checkout
        #  - user's overrides in ~/.pastebin.d
        # Files found later override files found earlier.
        pastebind = {}
        for confdir in ['/usr/share/pastebin.d','/etc/pastebin.d','/usr/local/etc/pastebin.d',
                        os.path.expanduser('~/.pastebin.d'),
                        os.path.join(os.path.dirname(__file__), 'pastebin.d')]:
            try:
                confdirlist = os.listdir(confdir)
            except OSError:
                continue

            for fileitem in confdirlist:
                if fileitem.startswith('.'):
                    continue

                filename = os.path.join(confdir, fileitem)
                try:
                    bininstance = configobj.ConfigObj(filename)
                except configobj.ConfigObjError, e:
                    print >> sys.stderr, '%s: %s' % (filename, e)
                    continue

                try:
                    section = bininstance['pastebin']
                except KeyError:
                    print >> sys.stderr, _('%s: no section [pastebin]') % filename
                    continue

                try:
                    basename = section['basename']
                except KeyError:
                    print >> sys.stderr, _("%s: no 'basename' in [pastebin]") % filename
                    continue

                pastebind[basename] = bininstance
        return pastebind

    # pastey.net obfuscates parent ids for replies.  Rather than taking the
    # post ID given as the parent ID, we must handle this by going to that
    # post page and looking up what the invisible parent ID field will be
    # set to for children.
    def doParentFixup(website, paramname, parentid):
        if parentid == "":
            return ""
        url_opener = pasteURLopener()
        page = url_opener.open(website + '/' + parentid, None)
        matches = re.split('<input.*?name="' + paramname + '".*?value="(.*?)"', page.read())
        if len(matches) <= 1 or re.match(parentid, matches[1]) == None:
            # The obfuscated version didn't begin with the partial version,
            # or unable to find the obfuscated version for some reason!
            # Create a paste with no parent (should we throw, instead?)
            return ""
        return matches[1]

    #Return the parameters depending of the pastebin used
    def getParameters(website, pastebind, content, user, jabberid, version, format, parentpid, permatag, title, username, password):
        "Return the parameters array for the selected pastebin"
        params = {}
        for pastebin in pastebind:
            if re.search(pastebind[pastebin]['pastebin']['regexp'], website):
                if "sizelimit" in pastebind[pastebin]['pastebin']:
                    params['sizelimit'] = pastebind[pastebin]['pastebin']['sizelimit']

                for param in pastebind[pastebin]['format'].keys():
                    paramname = pastebind[pastebin]['format'][param]
                    if param == 'user':
                        params[paramname] = user
                    elif param == 'content':
                        params[paramname] = content
                    elif param == 'title':
                        params[paramname] = title
                    elif param == 'version':
                        params[paramname] = version
                    elif param == 'format':
                        params[paramname] = format
                    elif param == 'parentpid':
                        params[paramname] = doParentFixup(website, paramname, parentpid)
                    elif param == 'permatag':
                        params[paramname] = permatag
                    elif param == 'username':
                        params[paramname] = username
                    elif param == 'password':
                        params[paramname] = password
                    elif param == 'jabberid':
                        params[paramname] = jabberid
                    else:
                        params[paramname] = pastebind[pastebin]['defaults'][param]
        if params:
            return params
        else:
            sys.exit(_("Unknown website, please post a bugreport to request this pastebin to be added (%s)") % website)

    #XML Handling methods
    def getText(nodelist):
        rc = ""
        for node in nodelist:
            if node.nodeType == node.TEXT_NODE:
                rc = rc + node.data
        return rc.encode('utf-8')

    def getNodes(nodes, title):
        return nodes.getElementsByTagName(title)

    def getFirstNode(nodes, title):
        return getNodes(nodes, title)[0]

    def getFirstNodeText(nodes, title):
        return getText(getFirstNode(nodes, title).childNodes)

    # Display usage instructions
    def Usage():
        print "pastebinit v" + version
        print _("Reads on stdin for input or takes a filename as first parameter")
        print _("Optional arguments (not supported by all pastebins):")
        print _("\t-a <author:default is '%s'>") % user
        print _("\t-b <pastebin url:default is '%s'>") % website
        print _("\t-f <format of paste:default is '%s'>") % format
        print _("\t-h This help screen")
        print _("\t-i <input file>")
        print _("\t-l List all supported pastebins")
        print _("\t-j <jabberid for notifications:default is '%s'>") % jabberid
        print _("\t-m <permatag for all versions of a post:default is blank>")
        print _("\t-r <parent posts ID:defaults to none>")
        print _("\t-t <title of paste:default is blank>")
        print _("\t-u <username> -p <password>")

    # Set defaults
    website = defaultPB
    user = os.environ.get('USER')
    jabberid = ""
    title = ""
    permatag = ""
    format = "text"
    username = ""
    password = ""
    filename = ""
    content = ""
    parentpid = ""

    #Example configuration file string
    configexample = """\
<pastebinit>
<pastebin>http://paste.debian.net</pastebin>
<author>A pastebinit user</author>
<jabberid>nobody@nowhere.org</jabberid>
<format>text</format>
</pastebinit>
"""

    #Open configuration file if it exists
    try:
        f = open(configfile)
        configtext = f.read()
        f.close()
        gotconfigxml = True
    except KeyboardInterrupt:
        sys.exit(_("KeyboardInterrupt caught."))
    except:
        gotconfigxml = False

    #Parse configuration file
    if gotconfigxml:
        try:
            configxml = xml.dom.minidom.parseString(configtext)
            for variable,key in (('pastebin','website'),('author','user'),('format','format'),('jabberid','jabberid')):
                try:
                    value = getFirstNodeText(configxml, variable)
                    vars()[key]=value
                except:
                    pass
        except KeyboardInterrupt:
            sys.exit(_("KeyboardInterrupt caught."))
        except:
            print _("Error parsing configuration file!")
            print _("Please ensure that your configuration file looks similar to the following:")
            print configexample
            sys.exit(1)

    # Get options
    try:
        optlist, list = getopt.getopt(sys.argv[1:], 'hli:f:b:a:r:j:t:m:u:p:')
    except KeyboardInterrupt:
        sys.exit(_("KeyboardInterrupt caught."))
    except getopt.GetoptError:
        print _("Invalid arguments!\n")
        Usage()
        sys.exit(1)

    pastebind = preloadPastebins() #get the config from /etc/pastebin.d/
    # Iterate through options
    for opt in optlist:
        if opt[0] == "-h":
            Usage()
            sys.exit(1)
        if opt[0] == "-i":
            filename = opt[1]
        elif opt[0] == "-f":
            format = opt[1]
        elif opt[0] == "-b":
            website = opt[1]
        elif opt[0] == "-a":
            user = opt[1]
        elif opt[0] == "-r":
            parentpid = opt[1]
        elif opt[0] == "-j":
            jabberid = opt[1]
        elif opt[0] == "-l":
            print _("Supported pastebins:")
            for pastebin in sorted(pastebind):
                print "- %s" % pastebin
            sys.exit(0)
        elif opt[0] == "-t":
            title = opt[1]
        elif opt[0] == "-m":
            permatag = opt[1]
        elif opt[0] == "-u":
            username = opt[1]
        elif opt[0] == "-p":
            password = opt[1]

    if filename == "" and len(list) >= 1:
        filename = list[0]

    #If - is specified as a filename read from stdin, otherwise load the specified file.
    if (filename == "" or filename == "-") and content == "":
        content = sys.stdin.read()
    elif content == "":
        try:
            f = open(filename)
            content = f.read()
            f.close()
        except KeyboardInterrupt:
            sys.exit(_("KeyboardInterrupt caught."))
        except:
            sys.exit(_("Unable to read from: %s") % filename)
    if not content:
        sys.exit(_("You are trying to send an empty document, exiting."))

    params = getParameters(website, pastebind, content, user, jabberid, version, format, parentpid, permatag, title, username, password) #Get the parameters array

    if not website.endswith("/"):
        website += "/"

    if "sizelimit" in params:
        if len(content) > int(params['sizelimit']):
            sys.exit(_("The content you are trying to send exceeds the pastebin's size limit."))
        else:
            del params['sizelimit']

    reLink = None
    tmp_page = ""
    if "page" in params:
        website += params['page']
        tmp_page = params['page']
        del params["page"]
    if "regexp" in params:
        reLink = params['regexp']
        del params["regexp"]
    if "target_url" in params:
        target_url = params["target_url"]
        del params["target_url"]
    else:
        target_url = None
    if 'post_format' in params:
        post_format = params['post_format']
        del params['post_format']
    else:
        post_format = 'standard'

    url_opener = pasteURLopener()

    if post_format == 'json':
        if json:
            params = json.dumps(params)
            url_opener.addheader('Content-type: text/json')
        else:
            sys.exit(_("Could not find any json library."))
    else:
        params = urllib.urlencode(params) #Convert to a format usable with the HTML POST

    page = url_opener.open(website, params) #Send the informations and be redirected to the final page

    try:
        if reLink: #Check if we have to apply a regexp
            if target_url:
                website = target_url
            else:
                website = website.replace(tmp_page, "")

            if reLink == '(.*)':
                print page.read().strip()
            else:
                print website + re.split(reLink, page.read())[1] #Print the result of the Regexp
        else:
            print page.url #Get the final page and show the ur
    except KeyboardInterrupt:
        sys.exit(_("KeyboardInterrupt caught."))
    except:
        sys.exit(_("Unable to read or parse the result page, it could be a server timeout or a change server side, try with another pastebin."))

except KeyboardInterrupt:
    sys.exit(_("KeyboardInterrupt caught."))
