icloudtabs.py

#!/usr/bin/python
#
# Alfred2 workflow for listing iCloud tabs
#

import os
import subprocess
import shutil
import xml.etree.ElementTree as ET
import getpass
import sqlite3


local_user = getpass.getuser()
cloudtabs_db = '/Users/%s/Library/Safari/CloudTabs.db' % local_user

conn = sqlite3.connect(cloudtabs_db)
cursor = conn.cursor()

cursor.execute("select device_uuid, device_name from cloud_tab_devices")
cloud_tab_devices = cursor.fetchall()

# Structure of cloud_tab_devices:
#
# [
#     (device1_uuid, device1_name),
#     (device2_uuid, device2_name)
# ]

cloud_tab_devices_lookup = {}

for device in cloud_tab_devices:
        cloud_tab_devices_lookup[device[0]] = device[1]

cloud_tab_devices_uuid_list = []

for device in cloud_tab_devices:
        cloud_tab_devices_uuid_list.append(device[0])

cursor.execute("select device_uuid, title, url from cloud_tabs")
cloud_tabs = cursor.fetchall()

# Structure of cloud_tabs:
#
# [
#     (device_uuid, tab_title, tab_url),
#     (device_uuid, tab_title, tab_url),
# ]

conn.close()


# Structure needed for all_device_tabs:
#
# all_device_tabs = [
#     ['device1_name', [
#         {'Title': 'device1_tab1_title', 'URL': 'device1_tab1_url'},
#         {'Title': 'device1_tab1_title', 'URL': 'device1_tab1_url'}
#     ]
#     ]
#     ['device2_name', [
#         {'Title': 'device2_tab1_title', 'URL': 'device2_tab1_url'},
#         {'Title': 'device2_tab1_title', 'URL': 'device2_tab1_url'}
#     ]
#     ]
# ]

all_device_tabs = []

for device_uuid in cloud_tab_devices_uuid_list:
    lookup_name = cloud_tab_devices_lookup[device_uuid]
    device_tabs = []
    for tab in cloud_tabs:
        if tab[0] == device_uuid:
            tabinfo = {'Title': tab[1], 'URL': tab[2]}
            device_tabs.append(tabinfo)
    all_device_tabs.append([lookup_name, device_tabs])


# Get local machine's host and computer names to exclude both from the list

hostname_proc = subprocess.Popen(
    ['scutil --get LocalHostName'], stdout=subprocess.PIPE, shell=True)
(hostname_out, hostname_err) = hostname_proc.communicate()
hostname = hostname_out.strip()

computername_proc = subprocess.Popen(
    ['scutil --get ComputerName'], stdout=subprocess.PIPE, shell=True)
(computername_out, computername_err) = computername_proc.communicate()
computername = computername_out.strip()

# Generate Alfred's XML

root = ET.Element('items')

for device in all_device_tabs:

    device_name = device[0]

    if device_name not in [hostname, computername.decode("utf-8")]:

        for tab in device[1]:

            if '{query}'.lower() not in tab['Title'].lower():
                continue

            item = ET.SubElement(root, 'item')
            item.set('uid', tab['URL'])
            item.set('arg', tab['URL'])

            title = ET.SubElement(item, 'title')
            title.text = tab['Title']

            subtitle = ET.SubElement(item, 'subtitle')
            subtitle.text = 'iCloud Device: '+device_name

            icon = ET.SubElement(item, 'icon')
            icon.set('type', 'fileicon')
            icon.text = '/Applications/Safari.app'

print ET.tostring(root)