~kelemeng/software-properties/bug945245

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
#!/usr/bin/python

import os
import sys
import gettext
import locale

from softwareproperties.SoftwareProperties import SoftwareProperties
from softwareproperties.ppa import DEFAULT_KEYSERVER, expand_ppa_line
from softwareproperties import lp_application_name
from aptsources.sourceslist import SourceEntry
from optparse import OptionParser
from gettext import gettext as _
from urllib2 import HTTPError, URLError

def utf8(s):
    """
    Takes a string or unicode object and returns a utf-8 encoded
    string, errors are ignored
    """
    if s is None:
        return None
    if isinstance(s, unicode):
        return s.encode("utf-8", "ignore")
    return unicode(s, "utf8", "ignore").encode("utf8")

def _maybe_suggest_ppa_name_based_on_user(user):
    try:
        from launchpadlib.launchpad import Launchpad
        lp = Launchpad.login_anonymously(lp_application_name, "production")
        try:
            user_inst = lp.people[user]
            entity_name = _("team") if user_inst.is_team else _("user")
            if len(user_inst.ppas) > 0:
                # Translators: %(entity)s is either "team" or "user"
                print _("The %(entity)s named '%(user)s' has no PPA named '%(ppa)s'") % {
                        'entity' : entity_name, 
                         'user' : user,
                         'ppa' : ppa_name}
                print _("Please choose from the following available PPAs:")
                for ppa in user_inst.ppas:
                    print _(" * '%(name)s':  %(displayname)s") % {
                            'name' : ppa.name,
                            'displayname' : ppa.displayname}
            else:
                # Translators: %(entity)s is either "team" or "user"
                print _("The %(entity)s named '%(user)s' does not have any PPA") % {
                        'entity' : entity_name, 
                         'user' : user}
        except KeyError:
            pass
    except ImportError:
        print _("Please check that the PPA name or format is correct.")



def utf8(s):
    """
    Takes a string or unicode object and returns a utf-8 encoded
    string, errors are ignored
    """
    if s is None:
        return None
    if isinstance(s, unicode):
        return s.encode("utf-8", "ignore")
    return unicode(s, "utf8", "ignore").encode("utf8")

if __name__ == "__main__":
    try:
        locale.setlocale(locale.LC_ALL, "")
    except:
        pass
    gettext.textdomain("software-properties")
    #FIXME: Workaround a bug in optparser which doesn't handle unicode/str
    # correctly, see http://bugs.python.org/issue4391
    # Shoudl be resolved by Python3
    enc = locale.getpreferredencoding()
    usage = """Usage: %prog <sourceline>

%prog is a script for adding apt sources.list entries. 
It can be used to add any repository and also provides a shorthand 
syntax for adding a Launchpad PPA (Personal Package Archive)
repository.

<sourceline> - The apt repository source line to add. This is one of:
  a complete apt line in quotes, 
  a repo url and areas in quotes (areas defaults to 'main')
  a PPA shortcut.

  Examples:
    apt-add-repository 'deb http://myserver/path/to/repo stable myrepo'
    apt-add-repository 'http://myserver/path/to/repo myrepo'
    apt-add-repository 'https://packages.medibuntu.org free non-free'
    apt-add-repository http://extras.ubuntu.com/ubuntu 
    apt-add-repository ppa:user/repository

If --remove is given the tool will remove the given sourceline from your
sources.list
"""
    parser = OptionParser(usage)
    # FIXME: provide a --sources-list-file= option that 
    #        puts the line into a specific file in sources.list.d
    parser.add_option ("-m", "--massive-debug", action="store_true",
        dest="massive_debug", default=False,
        help=_("Print a lot of debug information to the command line").decode(enc))
    parser.add_option("-r", "--remove", action="store_true",
        dest="remove", default=False,
        help=_("remove repository from sources.list.d directory").decode(enc))
    parser.add_option("-k", "--keyserver",
        dest="keyserver", default=DEFAULT_KEYSERVER,
        help=_("URL of keyserver. Default: %default").decode(enc))
    parser.add_option("-y", "--yes", action="store_true",
        dest="assume_yes", default=False,
        help=_("Assume yes to all queries").decode(enc))
    (options, args) = parser.parse_args()

    if os.geteuid() != 0:
        print _("Error: must run as root")
        sys.exit(1)

    if (len(args) != 1):
        print _("Error: need a repository as argument")
        sys.exit(1)

    # force new ppa file to be 644 (LP: #399709)
    os.umask(0022)

    # get the line
    line = args[0]

    # display PPA info (if needed)
    if line.startswith("ppa:") and not options.assume_yes:
        from softwareproperties.ppa import get_ppa_info_from_lp, LAUNCHPAD_PPA_API
        user, sep, ppa_name = line.split(":")[1].partition("/")
        ppa_name = ppa_name or "ppa"
        try:
            ppa_info = get_ppa_info_from_lp(user, ppa_name)
        except HTTPError:
            print _("Cannot add PPA: '%s'.") % line
            if user.startswith("~"):
                print _("Did you mean 'ppa:%s/%s' ?") %(user[1:], ppa_name)
                sys.exit(1) # Exit because the user cannot be correct
            # If the PPA does not exist, then try to find if the user/team 
            # exists. If it exists, list down the PPAs
            _maybe_suggest_ppa_name_based_on_user(user)
            sys.exit(1)
        except URLError:
            print _("Cannot access PPA (%s) to get PPA information, "
                    "please check your internet connection.") % \
                (LAUNCHPAD_PPA_API % (user, ppa_name))
            sys.exit(1)
        # private PPAs are not supported
        if "private" in ppa_info and ppa_info["private"]:
            print _("Adding private PPAs is not supported currently")
            sys.exit(1)
        
        if options.remove:
            print _("You are about to remove the following PPA from your system:")
        else:
            print _("You are about to add the following PPA to your system:")
        print " %s" % utf8(ppa_info["description"] or "")
        print _(" More info: %s") % ppa_info["web_link"]
        if (sys.stdin.isatty() and
            not "FORCE_ADD_APT_REPOSITORY" in os.environ):
            if options.remove:
                print _("Press [ENTER] to continue or ctrl-c to cancel removing it")
            else:
                print _("Press [ENTER] to continue or ctrl-c to cancel adding it")
            sys.stdin.readline()

    # add it
    sp = SoftwareProperties(options=options)
    if options.remove:
        (line, file) = expand_ppa_line(line.strip(), sp.distro.codename)
        deb_line = sp.expand_http_line(line)
        debsrc_line = 'deb-src' + deb_line[3:]
        deb_entry = SourceEntry(deb_line, file)
        debsrc_entry = SourceEntry(debsrc_line, file)
        try:
            sp.remove_source(deb_entry)
        except ValueError:
            print _("Error: '%s' doesn't exist in a sourcelist file") % deb_line
        try:
            sp.remove_source(debsrc_entry)
        except ValueError:
            print _("Error: '%s' doesn't exist in a sourcelist file") % debsrc_line

    else:
        if not sp.add_source_from_line(line):
            print _("Error: '%s' invalid") % line
            sys.exit(1)
        sp.sourceslist.save()