#!/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 = "" else: grp_name = group or '' 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 = "" else: grp_name = group or '' 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)