utility-scripts/firefox_tabgroups_export.py

457 lines
17 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Quick and Dirty Tab Groups Dumper for Firefox
--snip--
Copyright (C) 2014 Stephan Sokolow
Permission is hereby granted, free of charge, to any person obtaining
a copy of this software and associated documentation files (the "Software"),
to deal in the Software without restriction, including without limitation
the rights to use, copy, modify, merge, publish, distribute, sublicense,
and/or sell copies of the Software, and to permit persons to whom the
Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included
in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
"""
__appname__ = "Quick And Dirty Tab Groups Dumper for Firefox"
__author__ = "Stephan Sokolow (deitarion/SSokolow)"
__version__ = "0.2"
__license__ = "MIT"
import logging
log = logging.getLogger(__name__)
import json, os, re
#{{{ Data Types for Schema
bad_anchor_char_re = re.compile('[^A-Za-z0-9-_:.]+')
url_re = re.compile("""
^data:image/png;base64,|
^(chrome|file|https?)://|
^about:(blank|home|newtab)$|
^javascript:
""", re.VERBOSE)
is_nonempty_string = lambda x: isinstance(x, unicode) and x
is_url = lambda x: isinstance(x, unicode) and url_re.match(x)
is_nullable_url = lambda x: x is None or is_url(x)
is_natural_int = lambda x: isinstance(x, int) and x >= 0
is_positive_int = lambda x: is_natural_int(x) and x > 0
is_bool_string = lambda x: isinstance(x, unicode) and x in ('true', 'false')
def is_int_string(x):
"""Return true if the input is a string containing an integer"""
if not is_nonempty_string(x):
return False
try:
int(x)
return True
except ValueError:
return False
is_int_ish = lambda x: is_int_string(x) or is_natural_int(x)
def is_coord_pair(x):
"""Return true if the input is two comma-separated integers"""
return (is_nonempty_string(x) and
len(x.split(',')) == 2 and
all(is_int_string(y) for y in x.split(',')))
#}}}
#{{{ Schema
def apply_subschema(prefix, subschema):
"""Generate a new schema dict with prefixed paths"""
return {prefix + x: y for x, y in subschema.items()}
TAB_ENTRY_SCHEMA = {
'ID/': is_natural_int,
'docIdentifier/': is_natural_int,
'docshellID/': is_natural_int,
'owner_b64/': is_nonempty_string,
'referrer/': is_url,
'scroll/': is_nonempty_string,
'subframe/': lambda x: isinstance(x, bool),
'title/': is_nonempty_string,
'url/': is_url,
}
TABS_SCHEMA = {
'tabs/': lambda x: isinstance(x, list) and x,
'tabs/list/': lambda x: isinstance(x, dict) and x,
'tabs/list/attributes/': lambda x: isinstance(x, dict),
'tabs/list/attributes/image/': is_url,
'tabs/list/entries/': lambda x: isinstance(x, list) and x,
'tabs/list/entries/list/': lambda x: isinstance(x, dict) and x,
'tabs/list/entries/list/ID/': is_natural_int,
'tabs/list/entries/list/children/': lambda x: isinstance(x, list),
'tabs/list/entries/list/children/list/': lambda x: isinstance(x, dict),
'tabs/list/entries/list/url/': is_url,
'tabs/list/extData/': lambda x: isinstance(x, dict) and x,
'tabs/list/extData/tabview-tab/': is_nonempty_string,
'tabs/list/extData/weaveLastUsed/': is_nonempty_string,
'tabs/list/hidden/': lambda x: isinstance(x, bool),
'tabs/list/image/': is_nullable_url,
'tabs/list/index/': is_positive_int,
'tabs/list/lastAccessed/': is_natural_int,
'tabs/list/pinned/': lambda x: isinstance(x, bool),
'tabs/list/scroll/': lambda x: isinstance(x, dict),
'tabs/list/scroll/scroll/': is_nonempty_string,
}
TABS_SCHEMA.update(apply_subschema(
'tabs/list/entries/list/', TAB_ENTRY_SCHEMA))
TABS_SCHEMA.update(apply_subschema(
'tabs/list/entries/list/children/list/', TAB_ENTRY_SCHEMA))
TABS_SCHEMA.update(apply_subschema(
'tabs/list/entries/list/children/list/children/list/', TAB_ENTRY_SCHEMA))
CLOSED_TABS_SCHEMA = {
'_closedTabs/': lambda x: isinstance(x, list),
'_closedTabs/list/': lambda x: isinstance(x, dict),
'_closedTabs/list/closedAt/': is_natural_int,
'_closedTabs/list/image/': is_url,
'_closedTabs/list/pos/': is_positive_int,
'_closedTabs/list/state/': lambda x: isinstance(x, dict),
'_closedTabs/list/state/attributes/': lambda x: isinstance(x, dict),
'_closedTabs/list/state/entries/': lambda x: isinstance(x, list),
'_closedTabs/list/state/entries/list/': lambda x: isinstance(x, dict),
'_closedTabs/list/state/extData/': lambda x: isinstance(x, dict),
'_closedTabs/list/state/extData/tabview-tab/':
lambda x: isinstance(x, unicode),
'_closedTabs/list/state/hidden/': lambda x: isinstance(x, bool),
'_closedTabs/list/state/image/': is_url,
'_closedTabs/list/state/index/': is_positive_int,
'_closedTabs/list/state/lastAccessed/': is_natural_int,
'_closedTabs/list/state/scroll/': lambda x: isinstance(x, dict),
'_closedTabs/list/state/scroll/scroll/': is_coord_pair,
'_closedTabs/list/title/': lambda x: isinstance(x, unicode),
}
CLOSED_TABS_SCHEMA.update(apply_subschema(
'_closedTabs/list/state/entries/list/', TAB_ENTRY_SCHEMA))
WINDOW_SCHEMA = {
'extData/': lambda x: isinstance(x, dict),
'extData/tabview-group/': is_nonempty_string,
'extData/tabview-groups/': is_nonempty_string,
'extData/tabview-ui/': is_nonempty_string,
'height/': is_int_ish,
'screenX/': is_int_ish,
'screenY/': is_int_ish,
'selected/': is_positive_int,
'sizemode/': lambda x: isinstance(x, unicode),
'title/': is_nonempty_string,
'width/': is_int_ish,
}
#TODO: Finish deduplicating this schema definition
SCHEMA = {
'': lambda x: isinstance(x, dict) and x,
'_closedWindows/': lambda x: isinstance(x, list),
'_closedWindows/list/': lambda x: isinstance(x, dict),
'_closedWindows/list/closedAt/': lambda x: isinstance(x, int),
'_closedWindows/list/extData/__SessionManagerWindowId/':
lambda x: isinstance(x, unicode),
'global/': lambda x: isinstance(x, dict),
'scratchpads/': lambda x: isinstance(x, list),
'selectedWindow/': is_natural_int,
'session/': lambda x: isinstance(x, dict),
'session/lastUpdate/': is_natural_int,
'session/recentCrashes/': is_natural_int,
'session/startTime/': is_natural_int,
'windows/': lambda x: isinstance(x, list) and x,
'windows/list/': lambda x: isinstance(x, dict) and x,
'windows/list/busy/': lambda x: isinstance(x, bool),
'windows/list/extData/': lambda x: isinstance(x, dict) and x,
'windows/list/extData/__SessionManagerWindowId/': is_nonempty_string,
'windows/list/extData/tabview-last-session-group-name/':
is_nonempty_string,
'windows/list/extData/tabview-visibility/': is_bool_string,
'windows/list/tabs/list/disallow/': is_nonempty_string,
'windows/list/tabs/list/entries/list/cacheKey/': is_positive_int,
'windows/list/tabs/list/entries/list/children/list/children/':
lambda x: isinstance(x, list),
'windows/list/tabs/list/entries/list/children/list/children/list/':
lambda x: isinstance(x, dict),
'windows/list/tabs/list/entries/list/structuredCloneState/':
is_nonempty_string,
'windows/list/tabs/list/entries/list/structuredCloneVersion/':
is_positive_int,
'windows/list/tabs/list/extData/': lambda x: isinstance(x, dict),
'windows/list/tabs/list/extData/tabview-tab/': is_nonempty_string,
'windows/list/tabs/list/extData/weaveLastUsed/': is_nonempty_string,
'windows/list/tabs/list/lastAccessed/': is_natural_int,
'windows/list/tabs/list/pageStyle/':
lambda x: isinstance(x, (unicode, dict)),
'windows/list/tabs/list/pageStyle/pageStyle/': is_nonempty_string,
'windows/list/tabs/list/userTypedClear/': is_natural_int,
'windows/list/tabs/list/userTypedValue/': is_nonempty_string,
}
SCHEMA.update(apply_subschema('_closedWindows/list/', CLOSED_TABS_SCHEMA))
SCHEMA.update(apply_subschema('_closedWindows/list/', TABS_SCHEMA))
SCHEMA.update(apply_subschema('_closedWindows/list/', WINDOW_SCHEMA))
SCHEMA.update(apply_subschema('windows/list/', CLOSED_TABS_SCHEMA))
SCHEMA.update(apply_subschema('windows/list/', TABS_SCHEMA))
SCHEMA.update(apply_subschema('windows/list/', WINDOW_SCHEMA))
#}}}
#{{{ Schema-related routines
def make_schema_line(path, data):
"""Generate a draft schema line to be copy-pasted"""
tmp = "'%s': lambda x: isinstance(x, %s)," % (path, type(data).__name__)
return '%-100s # %s' % (tmp, repr(data)[:80])
def check_schema(data, dom_path='', schema=None, make_schema=False):
"""Recursive exploration for JSON via schemas"""
schema = schema or {}
result = {}
def fail(msg):
"""Unified failure message"""
dump = result[dom_path] = make_schema_line(dom_path, data)
if isinstance(data, (list, dict)):
raise ValueError("%s: %s @ %s\n %s" % (
msg, type(data), dom_path, dump))
else:
raise ValueError("%s: %s(%s) @ %s\n %s" % (
msg, type(data), data, dom_path, dump))
if dom_path not in schema:
if make_schema:
result[dom_path] = make_schema_line(dom_path, data)
else:
fail("Unexpected element")
if make_schema or schema[dom_path](data):
if isinstance(data, list):
for x in data:
result.update(check_schema(x, dom_path + 'list/',
schema=schema, make_schema=make_schema))
elif isinstance(data, dict):
for x in data:
result.update(check_schema(data[x], '%s%s/' % (dom_path, x),
schema=schema, make_schema=make_schema))
elif not (data is None or isinstance(data, (int, float, unicode))):
fail("Unexpected data type")
else:
fail("Element failed schema")
return result
#}}}
def _collect_tab_metadata(tab):
"""Restructure the tab metadata into a single dict"""
grp_id = None
grp_data = tab.get('extData', {}).get('tabview-tab', {})
if grp_data:
grp_data = json.loads(grp_data)
if grp_data:
grp_id = grp_data.get('groupID', None)
del grp_data
empty = {
'title' : 'New tab',
'url' : 'about:newtab',
}
if len(tab['entries']) == 0:
content = empty
else:
content = tab['entries'][-1] # -1 is most recent
return {
'index': tab.get('index'),
'group': grp_id,
'title': content.get('title', content['url']),
'url': content['url'],
'favicon': tab.get('image', None),
'pinned': tab.get('pinned', False)
}
def dump_tab_groups(data):
"""Load a JSON file from a path and extract the relevant data structure."""
windows = []
for window in data['windows']:
grp_names = json.loads(window.get('extData', {}).get(
'tabview-group', '{}'))
grp_names = {int(x): grp_names[x]['title'] for x in grp_names.keys()}
tabs = [_collect_tab_metadata(tab) for tab in window['tabs']]
# Group the tabs by group ID and then replace the IDs with the
# group names without risking naming collisions
groups = {}
[groups.setdefault(tab['group'], []).append(tab) for tab in tabs]
groups = [(grp_names.get(k, None), v) for k, v in groups.items()]
#groups.sort() # TODO: Sort case-insensitively
windows.append(groups)
return windows
def dump_to_html(dump, for_tiddlywiki=False):
"""Convert `dump_tab_groups` output to HTML
Specifically, an outline represented in HTML via the XOXO microformat
so that it's both human- and machine-readable:
http://www.microformats.org/wiki/xoxo
@todo: Clean up this code and reuse anything possible to produce an XBEL
output option:
http://en.wikipedia.org/wiki/XBEL
"""
from lxml import etree
from lxml.builder import E
def attr_class(*args):
"""workaround for `class` being a reserved word"""
return {"class": ' '.join(args)}
h3_prefix = lambda: (E.span('!!!', attr_class('copy-only'))
if for_tiddlywiki else '')
noicon = PLACEHOLDER_FAVICON_URI
tab_count, windows = 0, []
for pos, window in enumerate(dump):
k_toc = []
e_groups = []
for group, tabs in window:
if not group and all(x.get('pinned') for x in tabs):
grp_name = "<Pinned Tabs>"
else:
grp_name = group or '<Unnamed Group>'
grp_key = (grp_name, bad_anchor_char_re.sub('_', grp_name).lower())
grp_key_actual, idx = grp_key, 0
while grp_key_actual in k_toc:
idx += 1
grp_key_actual = '%s%s' % (grp_key, idx)
k_toc.append(grp_key_actual)
tab_count += len(tabs)
if for_tiddlywiki:
lines = []
for x in tabs:
lines.extend(['* [[', E.b(x['title']), '|',
E.a(x['url'], href=x['url']),
']]\n'])
e_tabs = E.pre(*lines)
else:
e_tabs = E.ul(*[E.li(E.a(E.img(src=x['favicon'] or noicon,
alt='', width='16', height='16'),
x['title'], href=x['url'])) for x in tabs])
e_groups.append(E.li(
E.h3(h3_prefix(), grp_name, id=grp_key_actual[1]),
e_tabs
))
windows.append(E.li(E.h2("Window %d" % pos), *e_groups))
e_toc = E.ul(*[E.li(E.a(x[0], href='#%s' % x[1])) for x in k_toc])
title_str = '%s (%s tabs total)' % (HTML_EXPORT_TITLE, tab_count)
return etree.tostring(E.html(
E.head(
E.title(title_str),
E.style(HTML_EXPORT_STYLE)),
E.body(
E.h1(title_str), e_toc,
E.ul(attr_class('xoxo'), *windows))))
def dump_to_text(dump):
"""Convert `dump_tab_groups` output to text
"""
tab_count, windows = 0, []
text = ''
for pos, window in enumerate(dump):
text += "Window {0}\n".format(pos)
for group, tabs in window:
if not group and all(x.get('pinned') for x in tabs):
grp_name = "<Pinned Tabs>"
else:
grp_name = group or '<Unnamed Group>'
text += "{0}\n".format(grp_name)
tab_count += len(tabs)
for x in tabs:
text += "\t{1}\n{0}\n".format(x['url'], x['title'])
text += "\n"
text += '({0} tabs total)'.format(tab_count)
return text
# From https://gist.github.com/Tblue/62ff47bef7f894e92ed5
def decompress(file_obj):
import lz4.block
if file_obj.read(8) != b"mozLz40\0":
raise "Invalid magic number"
return lz4.block.decompress(file_obj.read())
if __name__ == '__main__':
from optparse import OptionParser
parser = OptionParser(version="%%prog v%s" % __version__,
usage="%prog [opts] [path]",
description=__doc__.replace('\r\n', '\n').split('\n--snip--\n')[0])
parser.add_option('-v', '--verbose', action="count", dest="verbose",
default=2, help="Increase the verbosity. Use twice for extra effect")
parser.add_option('-q', '--quiet', action="count", dest="quiet",
default=0, help="Decrease the verbosity. Use twice for extra effect")
parser.add_option('--check-schema', action="store_true",
dest="check_schema", default=False, help="Check JSON against the "
"embedded sessionstore schema")
parser.add_option('--make-schema', action="store_true", dest="make_schema",
default=False, help="Auto-generate a first draft schema from the JSON")
# Allow pre-formatted descriptions
parser.formatter.format_description = lambda description: description
opts, args = parser.parse_args()
# Set up clean logging to stderr
log_levels = [logging.CRITICAL, logging.ERROR, logging.WARNING,
logging.INFO, logging.DEBUG]
opts.verbose = min(opts.verbose - opts.quiet, len(log_levels) - 1)
opts.verbose = max(opts.verbose, 0)
logging.basicConfig(level=log_levels[opts.verbose],
format='%(levelname)s: %(message)s')
if not args:
import glob
args = glob.glob(os.path.expanduser(
'~/.mozilla/firefox/*/sessionstore-backups/recovery.jsonlz4'))
results = []
for arg in args:
with open(arg, 'rb') as fobj:
data = decompress(fobj)
data = json.loads(data)
if opts.check_schema:
result = check_schema(data, schema=SCHEMA)
elif opts.make_schema:
result = check_schema(data, schema=SCHEMA, make_schema=True)
print("---------------")
print('\n'.join(sorted(result.values())))
else:
results += dump_tab_groups(data)
text = dump_to_text(results)
print(text)