#! /usr/bin/env python
# -*- coding: utf-8 -*-
#
# pkipplib : IPP and CUPS support for Python
#
# (c) 2003-2009 Jerome Alet <alet@librelogiciel.com>
# 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 3 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, see <http://www.gnu.org/licenses/>.
#
# $Id$

"""pksubscribe is a command line tool which can be used to create
or delete IPP subscriptions."""

import sys
import locale
import gettext
import optparse

from pkipplib import pkipplib

if __name__ == "__main__" :
    try :
        locale.setlocale(locale.LC_ALL, "")
    except (locale.Error, IOError) :
        sys.stderr.write("Problem while setting locale.\n")
    try :
        gettext.install("pksubscribe")
    except :
        gettext.NullTranslations().install()
    parser = optparse.OptionParser(usage="pksubscribe [options] [subscriptions ids]")
    parser.add_option("-v", "--version",
                            action="store_true",
                            dest="version",
                            help=_("show pksubscribe's version number and exit."))
    parser.add_option("-c", "--cups",
                            default="http://localhost:631",
                            dest="cups",
                            help=_("the CUPS server to connect to. Defaults to http://localhost:631"))
    parser.add_option("-d", "--debug",
                            action="store_true",
                            dest="debug",
                            help=_("activate debug mode."))
    parser.add_option("-X", "--delete",
                            action="store_true",
                            dest="delete",
                            help=_("deletes subscriptions."))
    parser.add_option("-p", "--printer",
                            dest="printer",
                            help=_("the printer's name for a printer subscription."))
    parser.add_option("-j", "--job",
                            type="int",
                            dest="job",
                            help=_("the job's id for a job subscripition."))
    parser.add_option("-r", "--recipient",
                            dest="recipient",
                            help=_("the recipient's uri."))
    parser.add_option("-C", "--charset",
                            dest="charset",
                            help=_("the charset to use in notifications sent for this subscription."))
    parser.add_option("-L", "--language",
                            dest="language",
                            help=_("the language to use in notifications sent for this subscription."))
    parser.add_option("-u", "--userdata",
                            dest="userdata",
                            help=_("the user's data to use in notifications for this subscription."))
    parser.add_option("-U", "--username",
                            dest="username",
                            help=_("the user's name to use when connecting to the CUPS server."))
    parser.add_option("-W", "--password",
                            dest="password",
                            help=_("the user's password to use when connecting to the CUPS server."))
    parser.add_option("-E", "--events",
                            dest="events",
                            help=_("a comma separated list of events to subscribe to."))
    parser.add_option("-P", "--pullmethod",
                            dest="pullmethod",
                            help=_("the optional pull method's name."))
    parser.add_option("-D", "--duration",
                            type="int",
                            dest="duration",
                            help=_("the duration of the subscription."))
    parser.add_option("-I", "--interval",
                            type="int",
                            dest="interval",
                            help=_("the time interval of the subscription."))

    (options, arguments) = parser.parse_args()
    if options.version :
        print "pksubscribe v%(__version__)s" % globals()
    else :
        if not options.events and not options.delete :
            sys.stderr.write(_("You MUST pass a list of events to subscribe to.\n"))
        elif not options.recipient and not options.delete :
            sys.stderr.write(_("You MUST pass a recipient for the subscription.\n"))
        elif options.delete and not arguments :
            sys.stderr.write(_("You MUST pass a subscriptions ids at the end of your command line.\n"))
        else :
            cups = pkipplib.CUPS(options.cups,
                                 options.username,
                                 options.password,
                                 debug=options.debug)
            baseurl = options.cups.replace("http://", "ipp://")
            if baseurl.endswith(":631") :
                baseurl = baseurl[:-4]
            if options.printer :
                url = "%s/printers/%s" % (baseurl, options.printer)
            elif options.job :
                url = "%s/jobs/%i" % (baseurl, options.job)
            else :
                url = baseurl
            if not options.delete :
                answer = cups.createSubscription(url,
                                                 [e.strip() for e in options.events.split(",")],
                                                 userdata=options.userdata,
                                                 recipient=options.recipient,
                                                 pullmethod=options.pullmethod,
                                                 charset=options.charset,
                                                 naturallanguage=options.language,
                                                 leaseduration=options.duration,
                                                 timeinterval=options.interval,
                                                 jobid=options.job)
                try :
                    subscriptionid = answer.subscription["notify-subscription-id"][0][1]
                except KeyError :
                    sys.stderr.write("%s\n" % answer.operation["status-message"][0][1])
                else :
                    print _("Subscription %i registered.") % subscriptionid
            else :
                for subid in [int(arg) for arg in arguments] :
                    answer = cups.cancelSubscription(url, subid, options.job)
                    if answer :
                        try :
                            error = answer.operation["status-message"][0][1]
                        except KeyError :
                            print _("Subscription %i cancelled.") % subid
                        else :
                            sys.stderr.write("%s\n" % error)
                    else :
                        sys.stderr.write(_("Incorrect answer (None)\n"))