pass

#!/usr/bin/env python3

import argparse
import base64
import codecs
import os
import sqlite3
import tempfile
from urllib.parse import urlparse
import io
import percol
import percol.cli
import percol.action
import percol.actions
import pyperclip
from Crypto.Cipher import AES
import hashlib
import keyring
import collections
import getpass


HOME = os.path.expanduser('~')
CHROME = 'Library/Application Support/Google/Chrome'
PROFILE = 'Default'
LOGIN_DATA = 'Login Data'

CHROME_SERVICE = 'Chrome Safe Storage'
CHROME_USERNAME = 'Chrome'


def get_master_password():
  app_name = os.path.splitext(os.path.basename(__file__))[0]
  master = keyring.get_password(app_name, getpass.getuser())
  if master is None:
    master = str(keyring.get_password(CHROME_SERVICE, CHROME_USERNAME))
    keyring.set_password(app_name, getpass.getuser(), master)
  return master.encode('utf8')


def decrypt(password):
  master_password = get_master_password()
  key = hashlib.pbkdf2_hmac(hash_name='sha1', password=master_password,
                            salt=b'saltysalt', iterations=1003, dklen=16)
  cipher = AES.new(key, AES.MODE_CBC, IV=b' ' * 16)
  binary = cipher.decrypt(base64.b64decode(password))
  return ''.join([chr(i) for i in binary if i >= 32])


def main(query, profile, out):
  passwords = []
  password2id = []

  def candidates():
    with tempfile.NamedTemporaryFile() as tmp:
      with open(os.path.join(HOME, CHROME, profile, LOGIN_DATA), 'rb') as f:
        tmp.write(f.read())
        tmp.flush()
      cursor = sqlite3.connect(tmp.name).cursor()
      cursor.execute('''SELECT origin_url, username_value, password_value
                        FROM logins ORDER BY times_used desc''')
      accounts = []
      for i, (origin_url, account, password) in enumerate(cursor.fetchall()):
        password = base64.b64encode(password[3:]).decode('utf8')
        url = urlparse(origin_url)
        title = codecs.decode(url.netloc.encode('utf8'), 'idna')
        if title.lower().startswith('www.'):
          title = title[4:]
        if url.scheme == 'android':
          title = '%s://%s' % (url.scheme, title.split('@')[1])
        accounts.append((title, account))
        passwords.append(password)

    for key, _ in collections.Counter(passwords).most_common():
      password2id.append(key)

    for i, (title, account) in enumerate(accounts):
      pass_id = password2id.index(passwords[i])
      yield '{:>50} {:<40}   {:0>3d} '.format(title, account, pass_id)

  @percol.action.action()
  def copy(lines, _):
    pyperclip.copy(decrypt(password2id[int(lines[0].split()[-1])]))

  if out:
    # None-interactive, E.g. with --out
    descriptors = {'stdin': io.StringIO(), 'stdout': io.StringIO(), 'stderr': io.StringIO()}
    p = percol.Percol(actions=[], candidates=candidates(), query=query, descriptors=descriptors)
    p.model.do_search(p.model.query)
    results = p.model_candidate.get_selected_results_with_index()
    if len(results) == 1:
      print(decrypt(password2id[int(results[0][0].split()[-1])]), end='')
      exit_code = 0
    else:
      exit_code = 1
    exit(exit_code)
  else:
    # Interactive
    with percol.Percol(actions=[copy], candidates=candidates(), query=query) as p:
      percol.cli.load_rc(p)
      p.view.PROMPT  = "<bold><yellow>Search</yellow></bold> %q" # type: ignore
      p.view.CANDIDATES_LINE_SELECTED = ("underline", "on_blue", "black") # type: ignore
      exit_code = p.loop()
    exit(exit_code)


if __name__ == '__main__':
  parser = argparse.ArgumentParser()
  parser.add_argument('query', default='', nargs='?')
  parser.add_argument('--profile', default=PROFILE)
  parser.add_argument('--out', default=False, action='store_true', help='None-interactive; Output password to stdout')
  args = parser.parse_args()
  main(args.query, args.profile, args.out)