#!/usr/bin/env python

# SPDX-License-Identifier: GPL-2.0+

#

# Copyright (C) 2009 Novell, Inc.

# Copyright (C) 2009 Red Hat, Inc.

#

# Run this script without any arguments to list the available connection uuids.

# The uuid of the connection to activate

CONNECTION_UUID = "ac6dc9b2-85ef-4311-83d8-add5d7db3f59"

# UID to use. Note that NM only allows the owner of the connection to activate it.

# UID=1000

UID = 0

import sys

import os

import dbus

from dbus.mainloop.glib import DBusGMainLoop

import gobject

DBusGMainLoop(set_as_default=True)

def get_connections():

    bus = dbus.SystemBus()

    proxy = bus.get_object(

        "org.freedesktop.NetworkManager", "/org/freedesktop/NetworkManager/Settings"

    )

    iface = dbus.Interface(

        proxy, dbus_interface="org.freedesktop.NetworkManager.Settings"

    )

    return iface.ListConnections()

def get_connection_by_uuid(uuid):

    bus = dbus.SystemBus()

    for c in get_connections():

        proxy = bus.get_object("org.freedesktop.NetworkManager", c)

        iface = dbus.Interface(

            proxy, dbus_interface="org.freedesktop.NetworkManager.Settings.Connection"

        )

        settings = iface.GetSettings()

        if settings["connection"]["uuid"] == uuid:

            return c

    return None

def list_uuids():

    bus = dbus.SystemBus()

    for c in get_connections():

        proxy = bus.get_object("org.freedesktop.NetworkManager", c)

        iface = dbus.Interface(

            proxy, dbus_interface="org.freedesktop.NetworkManager.Settings.Connection"

        )

        settings = iface.GetSettings()

        conn = settings["connection"]

        print("%s - %s (%s)" % (conn["uuid"], conn["id"], conn["type"]))

def get_active_connection_path(uuid):

    bus = dbus.SystemBus()

    proxy = bus.get_object(

        "org.freedesktop.NetworkManager", "/org/freedesktop/NetworkManager"

    )

    iface = dbus.Interface(proxy, dbus_interface="org.freedesktop.DBus.Properties")

    active_connections = iface.Get(

        "org.freedesktop.NetworkManager", "ActiveConnections"

    )

    for a in active_connections:

        proxy = bus.get_object("org.freedesktop.NetworkManager", a)

        iface = dbus.Interface(proxy, dbus_interface="org.freedesktop.DBus.Properties")

        path = iface.Get(

            "org.freedesktop.NetworkManager.Connection.Active", "Connection"

        )

        proxy = bus.get_object("org.freedesktop.NetworkManager", path)

        iface = dbus.Interface(

            proxy, dbus_interface="org.freedesktop.NetworkManager.Settings.Connection"

        )

        settings = iface.GetSettings()

        if settings["connection"]["uuid"] == uuid:

            return a

    return None

def get_wifi_device_path():

    bus = dbus.SystemBus()

    proxy = bus.get_object(

        "org.freedesktop.NetworkManager", "/org/freedesktop/NetworkManager"

    )

    iface = dbus.Interface(proxy, dbus_interface="org.freedesktop.NetworkManager")

    devices = iface.GetDevices()

    for d in devices:

        proxy = bus.get_object("org.freedesktop.NetworkManager", d)

        iface = dbus.Interface(proxy, dbus_interface="org.freedesktop.DBus.Properties")

        devtype = iface.Get("org.freedesktop.NetworkManager.Device", "DeviceType")

        if devtype == 2:

            return d

    return None

def activate_connection(connection_path, device_path):

    def reply_handler(opath):

        print("Success: device activating")

        sys.exit(0)

    def error_handler(*args):

        sys.stderr.write("Error activating device: %s\n" % args)

        sys.exit(1)

    bus = dbus.SystemBus()

    proxy = bus.get_object(

        "org.freedesktop.NetworkManager", "/org/freedesktop/NetworkManager"

    )

    iface = dbus.Interface(proxy, dbus_interface="org.freedesktop.NetworkManager")

    iface.ActivateConnection(

        "org.freedesktop.NetworkManager",

        connection_path,

        device_path,

        "/",

        reply_handler=reply_handler,

        error_handler=error_handler,

    )

# Change the UID first if required

if UID != 0:

    os.setuid(UID)

# Are we configured?

if not len(CONNECTION_UUID):

    print("missing connection UUID")

    sys.exit(0)

connection_path = get_connection_by_uuid(CONNECTION_UUID)

if not connection_path:

    # Configured VPN connection is not known to NM, check CONNECTION_UUID.

    print("couldn't find the connection")

    sys.exit(1)

device_path = get_wifi_device_path()

if not device_path:

    print("no Wi-Fi device found")

    sys.exit(1)

# Is it already activated?

if get_active_connection_path(CONNECTION_UUID):

    print("already connected")

    sys.exit(0)

print("Activating connection...")

activate_connection(connection_path, device_path)

loop = gobject.MainLoop()

loop.run()