/b-gtk/fix-viz

# Copyright (C) 2009 Jelmer Vernooij <jelmer@samba.org>

# 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 gi.repository import GObject

try:

    import gnomekeyring

except ImportError:

    gnomekeyring = None

from bzrlib.config import (

    CredentialStore,

    )

class GnomeKeyringCredentialStore(CredentialStore):

    def __init__(self):

        CredentialStore.__init__(self)

        if GObject.get_application_name() is None:

            # External applications that load bzrlib may already have set the

            # application name so we don't contradict them (when we can

            # determine it that is).

            GObject.set_application_name("bzr")

    def decode_password(self, credentials):

        if gnomekeyring is None:

            return None

        attrs = {}

        if "scheme" in credentials:

            attrs["protocol"] = credentials["scheme"].encode("utf-8")

        if "host" in credentials:

            attrs["server"] = credentials["host"].encode("utf-8")

        if "user" in credentials:

            attrs["user"] = credentials["user"].encode("utf-8")

        if credentials.get("port") is not None:

            attrs["port"] = credentials["port"]

        try:

            items = gnomekeyring.find_items_sync(

                gnomekeyring.ITEM_NETWORK_PASSWORD, attrs)

            return items[0].secret

        except (gnomekeyring.NoMatchError, gnomekeyring.DeniedError):

            return None

    def get_credentials(self, scheme, host, port=None, user=None, path=None, 

                        realm=None):

        if gnomekeyring is None:

            return None

        attrs = {

            "protocol": scheme.encode("utf-8"),

            "server": host.encode("utf-8"),

            }

        # TODO: realm ?

        if port is not None:

            attrs["port"] = port

        if user is not None:

            attrs["user"] = user.encode("utf-8")

        credentials = { "scheme": scheme, "host": host, "port": port, 

            "realm": realm, "user": user}

        try:

            items = gnomekeyring.find_items_sync(

                gnomekeyring.ITEM_NETWORK_PASSWORD, attrs)

            credentials["user"] = items[0].attributes["user"]

            credentials["password"] = items[0].secret

            return credentials

        except (gnomekeyring.NoMatchError, gnomekeyring.DeniedError, gnomekeyring.NoKeyringDaemonError,

                gnomekeyring.IOError), e:

            from bzrlib import trace

            trace.mutter('Unable to obtain credentials for %r from GNOME keyring: %r',

                         attrs, e)

            return None