#  Copyright (c) 2019 Canonical Ltd.
#
#  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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
#  USA

from aptsources.distro import get_distro
from aptsources.sourceslist import SourceEntry

from softwareproperties.sourceslist import SourcesListShortcutHandler


class URIShortcutHandler(SourcesListShortcutHandler):
    def __init__(self, shortcut, **kwargs):
        (uri, _, line_comps) = shortcut.strip().partition(' ')

        line_comps = set(line_comps.split() if line_comps else [])
        param_comps = set(kwargs.get('components', []))
        comps = list(line_comps | param_comps)

        if not comps:
            # if no comps provided, we default to 'main'
            comps = ['main']

        kwargs['components'] = comps

        suite = kwargs.get('codename')
        pocket = kwargs.get('pocket')

        s = SourceEntry('')
        s.invalid = False
        s.uri = uri
        s.type = get_distro().binary_type
        s.dist = suite or get_distro().codename
        if pocket is not None:
            s.dist = '{}-{}'.format(s.dist, pocket)

        super(URIShortcutHandler, self).__init__(str(s), **kwargs)


# vi: ts=4 expandtab