summaryrefslogtreecommitdiff
path: root/src/util.py
blob: 91d175517c3666a3564b0132e8932c01c39a6c96 (plain)
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
#!/usr/bin/env python

import subprocess
import atexit
import os
import sys
import signal
import dbus

from threading import Thread
from gi.repository import GLib, Gio
from gi.repository import TelepathyGLib as Tp


def _got_line (d, r, u):
    (prefix, quiet) = u
    line = d.read_line_finish (r)
    if not quiet:
        print prefix + ": " + str(line)
    d.read_line_async (0, None, _got_line, u)

def _process_input(f, prefix, quiet):
    i = Gio.UnixInputStream.new (f.fileno(), True)
    d = Gio.DataInputStream.new (i)
    d.read_line_async (0, None, _got_line, (prefix, quiet))


def spawnbus(quiet = False):
    command = [ "dbus-daemon", "--session", "--nofork", "--print-address" ]

    process = subprocess.Popen (command,
        stdout = subprocess.PIPE,
        stderr = subprocess.PIPE,
    )
    atexit.register (lambda p: os.kill (p.pid, signal.SIGKILL), process)

    address = process.stdout.readline().rstrip()
    os.environ ["DBUS_SESSION_BUS_ADDRESS"] = address
    if not quiet:
        print "Temporary Session bus: %s" % address

    # Process stdout
    _process_input (process.stdout, "STDOUT", quiet)
    _process_input (process.stderr, "STDERR", quiet)
    return process

# TODO: it might be more useful to read the input stream synchronously
# for this
def show_messages(quiet):
    command = ['dbus-monitor', '--session']

    process = subprocess.Popen(command,
                               stdout = subprocess.PIPE,
                               stderr = subprocess.PIPE,
                               )
    atexit.register(lambda p: os.kill(p.pid, signal.SIGKILL), process)

    _process_input(process.stdout, "MONITOR", quiet)
    _process_input(process.stderr, "MONITOR", quiet)

def override_env (variable, var, quiet = False):
    os.environ[variable] = var
    if not quiet:
        print "Setting %s to %s" % (variable, var)

def prepend_env_path (variable, prefix, quiet = False):
    current = os.getenv(variable)
    if current != None:
        p = prefix + ":" + current
    else:
        p = prefix
    override_env (variable, p, quiet)

def setup_data_dir (path, quiet = False):
    # Define default if not set as per XDG spec
    if os.getenv("XDG_DATA_DIRS") == None:
        os.environ["XDG_DATA_DIRS"] = "/usr/local/share/:/usr/share/"
    prepend_env_path ("XDG_DATA_DIRS", path, quiet)

def setup_run_dir (path, quiet= False):
    # Setup all the various XDG paths
    override_env ("XDG_CONFIG_HOME", os.path.join (path, ".config"), quiet)
    override_env ("XDG_CACHE_HOME", os.path.join (path, ".cache"), quiet)
    override_env ("XDG_DATA_HOME", os.path.join (path, ".local"), quiet)
    override_env ("MC_ACCOUNT_DIR",
        os.path.join (path, ".config", "mission-control", "accounts"), quiet)

def drop_env_var(variable):
    if variable in os.environ.keys():
        os.environ.pop(variable)

def scrub_env():
    drop_env_var ("GNOME_KEYRING_CONTROL")

def got_account_cb (o, r, password, func):
    account = o.create_account_finish (r)
    # Put the password in our credentials store
    if password:
        try:
            os.makedirs (os.path.join (GLib.get_user_config_dir (), "phoenix"))
        except:
            pass
        authfile = os.path.join (GLib.get_user_config_dir (), "phoenix", "auth")
        f = open (authfile, "a+")
        f.write (account.get_path_suffix() + " " + password + "\n")

    func(account)

def create_account (am, cm, protocol, name, parameters, password, func):
    request = Tp.AccountRequest.new (am, cm, protocol, name)
    for k,v in parameters.items():
        request.set_parameter (k, v)

    request.create_account_async (
        lambda o, r, u: got_account_cb (o, r, password, func),
        None)