import sqlite3 as sqlite
import os
from glob import glob
from sys import platform as _platform

class FirefoxProfile:
    def __init__(self, dbfile):
        self.db = sqlite.connect(dbfile)
        self.cur = self.db.cursor()

    def __enter__(self):
        return FirefoxProfile.default()

    def __exit__(self, _, __, ___):
        self.close()

    @staticmethod
    def default():
        dbfile = None
        paths = ["~/.mozilla/firefox", "~/Library/Application Support/Firefox/Profiles"]
        while paths:
            path = os.path.expanduser(paths.pop())
            files = glob('{}/*.default/places.sqlite'.format(path))
            if files:
                dbfile = files[0]
                break
        if not dbfile:
            raise Exception("No places.sqlite file")
        return FirefoxProfile(dbfile)

    def bookmarks(self):
        self.cur.execute(
            'SELECT DISTINCT mb.title, mp.url ' +
            'FROM moz_bookmarks mb ' +
            'INNER JOIN moz_places mp ' +
            'ON mp.id = mb.fk ' +
            'WHERE mb.type = 1 AND mb.title IS NOT NULL '
            'ORDER BY mp.frecency DESC'
        )
        return self._process(self.cur.fetchall())

    def history(self):
        self.cur.execute(
            'SELECT DISTINCT title, url ' +
            'FROM moz_places ' +
            'WHERE title IS NOT NULL '
            'ORDER BY last_visit_date DESC'
        )
        return self._process(self.cur.fetchall())

    def _process(self, data):
        title_lookup = {}
        url_lookup = {}
        ret = []
        for row in data:
            title = row[0].encode('utf-8').decode()
            url = row[1].encode('utf-8').decode()
            if url.startswith("http") and title not in title_lookup and url not in url_lookup:
                title_lookup[title] = 1
                url_lookup[url] = 1
                ret.append("{}, {}".format(title, url))
        return ret

    def browse(self, url):
        if _platform.startswith("linux"):
            os.system('xdg-open "{}"'.format(url))
        elif _platform == "darwin":
            os.system('open "{}"'.format(url))
        else:
            raise Exception("What you doin' fool")

    def close(self):
        self.db.close()