#!/usr/bin/env python

# usage ./extract_tampermonkey_script.py "/home/<user>/.config/<browser>/Default/Local Extension Settings/<extension_id>"
# i.e.: ./extract_tampermonkey_script.py "/home/foo/.config/google-chrome-beta/Default/Local Extension Settings/gcalenpjmijncebpfijmoaglllgpjagf"
#
import leveldb
import sys
import re
import ast

pattern = re.compile("(.*)@source$")

db = leveldb.LevelDB(sys.argv[1:][0])

for k,v in db.RangeIter():
    m = pattern.match(k)
    if m:
        name = re.sub("[\W\d\b]", "_", m.groups()[0].strip())
        full_name = "%s.user.js" % name

        print "Writing to %s" % full_name

        text_file = open(full_name, "w")
        text_file.write(ast.literal_eval(v))
        text_file.close()